Calise calise

#!/usr/bin/env python2
# -*- coding: utf-8 -*-
#
#    This program is free software: you can redistribute it and/or modify
#    it under the terms of the GNU General Public License as published by
#    the Free Software Foundation, either version 3 of the License, or
#    any later version.

import os
import sys
import re
from DistUtilsExtra.auto import setup
from distutils.core import Extension


''' Reads file "infos.py" in that package and obtains from there the

This "fake" class has been implemented because import priority between
root and workingDir modules varies from Distro to Distro.

'''
class replacement():
    def __init__ (self, path ):
        self.__CapitalName__ = 'Calise'
        self.__LowerName__ = self.__CapitalName__.lower()
        with open( os.path.join(path, self.__LowerName__, 'infos.py') ) as fp:
            fc = fp.read().splitlines()
            for line in fc:
                if line[:11] == '__version__':
                    self.__version__ = (
                        '='.join(line.split('=')[1:]).replace(' ', '')[1:-1])


# Obtain Setup Directory (even if called from outside)
def getsd():
    baseName = sys.argv[0].\
        replace('./', '').\
        replace(os.path.basename(sys.argv[0]), '')
    path = os.path.join(os.getcwd(), baseName)
    return path


''' Developement versions track

Returns either:
  - SVN revision in the form SVN-XXXX, where XXXX is the revision number
  - GIT shorthash in the form GIT-xxxxxxx, where xxxxxxx are first 7 chars of
    sha1 hash

Returns calise.infos.__version__ in no git nor svn exist

'''
def get_svn_revision(path=None):
    if path is None:
        path = getsd()
    infos = replacement(path)
    # if there's git directory process as git
    if os.path.isdir(os.path.join(path, '.git')):
        gitRev = os.popen('git rev-parse --short HEAD')
        gitRev = gitRev.read()[:-1]
        versionString = u'GIT-%s' % gitRev
    # else if there's svn directory process as subversion
    elif os.path.isdir(os.path.join(path, '.svn')):
        db_path = os.path.join(path, '.svn', 'wc.db') % path
        with open(db_path) as fp:
            content = fp.read().replace('svn/ver/', '\nsvn/ver/').split('\n')
        prog = re.compile('svn/ver/([0-9]+)/')
        revisions = []
        for line in content:
            z = prog.match(line)
            if z is not None:
                revisions.append(int(z.group(1)))
        rev = max(revisions)
        versionString = u'SVN-%s' % rev
    else:
        versionString = None
    if versionString:
        with open(os.path.join(path, infos.__LowerName__, 'infos.py')) as fp:
            fc = fp.read()
            fc = fc.replace(
                '__version__ = \'%s\'' % infos.__version__,
                '__version__ = \'%s\'' % versionString, 1 )
        with open(os.path.join(path,infos.__LowerName__,'infos.py'),'w') as fp:
            fp.write(fc)
    else:
        versionString = infos.__version__
    return versionString


# remove '.py' suffix from scripts in 'bin' directory
def clearScriptExtensions(path=None):
    if path is None:
        path = getsd()
        binPath = os.path.join(path, 'bin')
    if os.path.isdir(binPath):
        retList = []
        for script in os.listdir(binPath):
            os.rename(
                os.path.join(binPath, script),
                os.path.join(binPath, script.replace('.py', '')))
            retList.append(os.path.join('bin', script.replace('.py', '')))


# restore '.py' suffix to scripts in 'bin' directory
def restoreScriptExtensions(path=None):
    if path is None:
        path = getsd()
        binPath = os.path.join(path, 'bin')
    if os.path.isdir(binPath):
        retList = []
        for script in os.listdir(binPath):
            os.rename(
                os.path.join(binPath, script),
                os.path.join(binPath, script + '.py'))
            retList.append(os.path.join('bin', script + '.py'))


clearScriptExtensions()
# instruct setup to build c modules with their included libraries
addModule1 = Extension(
    'calise.screenBrightness',
    sources = ['src/modules/screenBrightness.c'],
    libraries = ['X11'])
addModule2 = Extension(
    'calise.camera',
    sources = ['src/modules/camera.c'],)
# actual setup
setup(
    name='calise',
    version=get_svn_revision(),
    description="automatically adjust backlight trough a camera",
    author='Nicolò Barbon',
    author_email='smilzoboboz@gmail.com',
    url='http://calise.sourceforge.net/',
    license='GNU GPL v3',
    ext_modules=[addModule1, addModule2],
)
restoreScriptExtensions()