Adding verbosity in loadLibrary

wrapper/nixWrapper.py

@@ -0,0 +1,247 @@
+import sys
+import os
+import urllib3
+import shutil
+import zipfile
+import pathlib
+import json
+import getpass
+import chardet
+import base64
+
+#this is meant to be used as a startup script
+#all commands will be executed when python -m script -i is used
+
+
+#generic function to load library from gitlab
+
+def getSuitePath():
+    installDir=os.path.join(os.path.expanduser('~'),'.labkey','software','src')
+    if not os.path.isdir(installDir):
+        os.makedirs(installDir)
+    return installDir
+
+def buildGITURL(server,project,path,branch='master'):
+    if server.find('wiscigt')>-1:
+        projectURL='%2f'.join(project)
+        pathURL='%2f'.join(path)
+        return server+'/api/v4/projects/'+projectURL+'/repository/files/'+pathURL+'?ref='+branch
+    if server.find('fmf.uni-lj.si')>-1:
+        projectURL='/'.join(project)#studen/nixSuite
+        pathURL='/'.join(path)
+        return '/'.join([server,projectURL,'raw',branch,pathURL])
+
+
+
+def getResources():
+    server='http://wiscigt.powertheword.com'
+    project=['labkey','nixsuite']
+    path=['remoteResources','resources.json']
+
+    server='https://git0.fmf.uni-lj.si'
+    project=['studen','nixSuite']
+    
+    remoteSourcesURL=buildGITURL(server,project,path)
+    print('remoteSourcesURL {}'.format(remoteSourcesURL))
+    http = urllib3.PoolManager()
+    r = http.request('GET', remoteSourcesURL)
+    #returns a JSON
+    encoding=chardet.detect(r.data)['encoding']
+    dataDecoded=r.data.decode(encoding)
+    jsonData=json.loads(dataDecoded)
+    print(jsonData)
+
+    #since gogs drops raw data, we already did the hard part 
+    if server.find('fmf.uni-lj.si')>-1:
+        return jsonData
+
+    #on gitlab a further decoding step is required
+    #we are interested in content, do looped way of decoding it 
+    b64_bytes=jsonData['content'].encode('ascii')
+    m_bytes=base64.b64decode(b64_bytes)
+    m=m_bytes.decode('ascii')
+    return json.loads(m)
+
+def loadModule(slicer,qt,name,moduleName):
+    loadLibrary(name)
+    modulePath=os.path.join(getSuitePath(),name,'slicerModules',moduleName+'.py')
+    factoryManager = slicer.app.moduleManager().factoryManager()
+
+    factoryManager.registerModule(qt.QFileInfo(modulePath))
+
+    factoryManager.loadModules([moduleName,])
+    slicer.util.selectModule(moduleName)
+
+
+def loadLibrary(name,doReload=True):
+
+    print('loadLibrary')
+    installDir=getSuitePath()
+    finalName=os.path.join(installDir,name)
+    if os.path.isdir(finalName):
+        if not doReload:
+            #1 keep existing copy, return
+            sys.path.append(finalName)
+            return
+        else:
+            #1 remove existing copy
+            shutil.rmtree(finalName)
+    
+    #load library from git, store it at a default location and 
+    #add path to the python sys
+    remoteSources=getResources()
+
+    #two steps:
+    #1 Download
+
+    tempDir=os.path.join(os.path.expanduser('~'),'temp')
+    if not os.path.isdir(tempDir):
+        os.mkdir(tempDir)
+    tempFile=os.path.join(tempDir,name+'.zip')
+    
+    http = urllib3.PoolManager()
+
+    rsource=remoteSources[name]
+
+    print('Rsource')
+    print(rsource)
+    
+    r = http.request('GET', rsource['url'], preload_content=False)
+    chunk_size=65536 
+    with open(tempFile, 'wb') as out:
+        while True:
+            data = r.read(chunk_size)
+            if not data:
+                break
+            out.write(data)
+
+    r.release_conn()
+    
+    #2 Unzip
+    with zipfile.ZipFile(tempFile,'r') as zip_ref:
+        zip_ref.extractall(installDir)
+    
+    #cleanup
+    os.remove(tempFile)
+
+    #rename
+    #this is the best guess of the relation between zip directory and name
+    try:
+        zipName=name.lower()+'-'+rsource['branch']
+        os.rename(os.path.join(installDir,zipName),finalName)
+    except FileNotFoundError:
+        try:
+            zipName=name+'-'+rsource['branch']
+            os.rename(os.path.join(installDir,zipName),finalName)
+        except FileNotFoundError:
+            #git0/gogs
+            zipName=name.lower()
+            os.rename(os.path.join(installDir,zipName),finalName)
+
+    
+    sys.path.append(finalName)
+    updateSetup(name,finalName)
+
+def updateSetup(name,path):
+
+    #update setup.json
+    #set paths[name]=finalName
+    #for use in slicer, use .labkey/slicer/setup.json
+    setupDir=os.path.join(os.path.expanduser('~'),'.labkey','slicer')
+    if not os.path.isdir(setupDir):
+        os.makedirs(setupDir)
+    setupFile=os.path.join(setupDir,'setup.json')
+    
+    #if f is not there, create an empty json
+    try:
+        with open(setupFile,'r') as f:
+            setup=json.load(f)
+    except OSError:
+        setup={}
+
+    try:
+        setup['paths'][name]=path
+    except KeyError:
+        setup['paths']={}
+        setup['paths'][name]=path
+
+    with open(setupFile,'w') as f:
+        json.dump(setup,f)
+    
+def setConfig(labkeyUser,labkeyServer='https://merlin.fmf.uni-lj.si',\
+        certificateZip=None):
+    connectionConfig={}
+    if certificateZip!=None:
+        zpath=pathlib.Path(certificateZip)
+        user=zpath.stem
+        userDir=os.path.join(os.path.expanduser('~'),'.labkey',user)
+        if not os.path.isdir(userDir):
+            os.makedirs(userDir)
+        caName=None
+        with zipfile.ZipFile(certificateZip,'r') as zipObj:
+            zipObj.extract(user+'.crt',userDir)
+            zipObj.extract(user+'.key',userDir)
+            zipList=zipObj.namelist()
+            for f in zipList:
+                if f.find('CA')>-1:
+                    caName=f
+                    zipObj.extract(f,userDir)
+    
+        #setup SSL
+        sslSetup={}
+        sslSetup['user']=os.path.join(userDir,user+'.crt')
+        sslSetup['key']=os.path.join(userDir,user+'.key')
+        sslSetup['ca']=os.path.join(userDir,caName)
+        sslSetup['keyPwd']='notUsed'
+        connectionConfig['SSL']=sslSetup
+
+    connectionConfig["host"]=labkeyServer
+    connectionConfig["context"]="labkey"
+    labkeySetup={}
+    labkeySetup['user']=labkeyUser
+    labkeyPwd='guest'
+    if not labkeyUser=='guest':
+        labkeyPwd=getpass.getpass(prompt='Enter labkey password:')
+    labkeySetup['password']=labkeyPwd 
+    connectionConfig['labkey']=labkeySetup
+
+    orthancSetup={}
+    if connectionConfig['host'].find('merlin')>-1:
+        orthancSetup['server']='https://orthanc.fmf.uni-lj.si'
+    if connectionConfig['host'].find('onko-nix')>-1:
+        orthancSetup['server']='http://onko-nix.onko-i.si:8042'
+    orthancSetup['user']='ask'
+    orthancSetup['password']='askPassword'
+    connectionConfig['orthanc']=orthancSetup
+    net=labkeyInterface.labkeyInterface()
+    net.connectionConfig=connectionConfig
+    net.initRemote()
+    print(net.getUserId())
+    return net
+
+def testCertificate(certificateZip):
+    return setConfig('guest',certificateZip=certificateZip)
+
+def testConnection(labkeyUser,labkeyServer='https://merlin.fmf.uni-lj.si',\
+        certificateZip=None):
+    return setConfig(labkeyUser,labkeyServer,certificateZip)
+
+def getDefaultConfig(configFile):
+    #if configFile is None, this will set it to default and leave unchanged otherwise
+    if configFile==None:
+        return os.path.join(os.path.expanduser('~'),'.labkey','network.json')
+    return configFile
+
+def storeConfig(net,configFile=None):
+    #if configFile is None, this will set it to default and leave unchanged otherwise
+    with open(getDefaultConfig(configFile),'w') as f:
+        json.dump(net.connectionConfig,f,indent='\t')
+
+def getInterface(configFile=None):
+    net=labkeyInterface.labkeyInterface()
+    #if configFile is None, this will set it to default and leave unchanged otherwise
+    net.init(getDefaultConfig(configFile))
+    return net
+
+#loadLibrary('labkeyInterface')
+