setupTool.py

import os
import json

gSetup =  None
SNAP_ROUTE_SRC = '/snaproute/src/'
EXTERNAL_SRC = '/external/src/'
GENERATED_SRC = '/generated/src/'

class setupGenie (object) :
    def __init__ (self, setupInfo, anchorDir , gitUsrName, role) :
        self.setupInfo = setupInfo
        self.info = None 
        self.homeDir = os.path.expanduser('~')
        self.anchor = self.homeDir + '/' + anchorDir
        self.usrName = gitUsrName
        if role in ['y', 'yes', 'Y', 'Yes', 'YES']:
            self.internalUser = True
            self.org = self.info['SnapRouteOrg']
        else:
            self.internalUser = False
            self.org = 'OpenSnapRoute'

        with open(self.setupInfo) as dataFile:
            self.info = json.load(dataFile)                                                                                 
        self.generateSrcInfo()

    def getExternalInstalls(self, comp = None) :
        if comp:
            return { comp+'Deps' : self.info['Externals'] [comp+'Deps']}
        else:
            return self.info['Externals']

    def generateSrcInfo (self) :
        srcInfoFile = 'srcInfo.json'
        with open(srcInfoFile, 'w') as fd :
            rpList =  self.getSRRepos()
            srcInfo = {'repos': { 'snaproute': rpList},
                       'srcLocation':SNAP_ROUTE_SRC }
            json.dump(srcInfo,fd, indent=4)

    def getGoDeps (self, comp = None) :
        return self.info['GoDeps']

    def getGoDepDirFor (self, comp ) :
        for dep in self.info['GoDeps']:
            if dep['repo'] == comp:
                if dep.has_key('renamedst') :
                    return self.anchor + EXTERNAL_SRC + dep['renamedst'] + dep['repo']
                else:
                    return self.anchor + EXTERNAL_SRC + dep['repo']

    def getSRRepos(self, comp = None) :
        if self.internalUser:
            return self.info['PrivateRepos']
        else:
            return self.info['PublicRepos']

    def getExtSrcDir (self ) :
        return self.anchor + EXTERNAL_SRC

    def getGenSrcDir (self ) :
        return self.anchor + GENERATED_SRC 

    def getSRSrcDir (self ) :
        return self.anchor + SNAP_ROUTE_SRC 

    def getAllSrcDir (self ) :
        return [ self.anchor + SNAP_ROUTE_SRC, self.anchor + EXTERNAL_SRC, self.anchor + GENERATED_SRC ]

    def getOrg (self) :
        return self.org
        
    def getUsrName(self) :
        return self.usrName
        
    def getUsrRole(self) :
        return self.internalUser
        
    def getAnchorDir(self ) :
        return self.anchor 

                
def getSetupHdl (setupInfo='setupInfo.json', anchorDir='git', gitUsrName='', role='n') :
    global gSetup
    if not gSetup:
        gSetup = setupGenie(setupInfo, anchorDir, gitUsrName, role)
    return gSetup