Merge pull request #55 from rediscovery-io/set_remo_home

Set remo home

Author: vmanilo

remo/__init__.py

@@ -5,7 +5,7 @@
 _sdk = None
 
 
-def connect(server: str = '', email: str = '', password: str = '', viewer: str = 'browser'):
+def connect(server: str = '', email: str = '', password: str = '', viewer: str = 'browser', remo_home: str = None):
     """
     Connect to a remo server.
     If no parameters are passed, it connects to a local running remo server. To connect to a remote remo, specify connection details.
@@ -15,7 +15,13 @@ def connect(server: str = '', email: str = '', password: str = '', viewer: str =
         email:  email address used for authentication
         password: password used for authentication
         (optional) viewer: viewer to use, one between 'browser', 'electron' and 'jupyter'
+        (optional) remo_home: location of remo home
     """
+
+    if remo_home:
+        from .config import set_REMO_HOME
+        set_REMO_HOME(remo_home)
+
     if not (server and email and password):
         from .config import Config
         config = Config.load()

remo/config.py

@@ -2,7 +2,17 @@
 import os
 from pathlib import Path
 
-REMO_HOME = os.getenv('REMO_HOME', str(Path.home().joinpath('.remo')))
+REMO_HOME_ENV = 'REMO_HOME'
+
+
+def REMO_HOME():
+    return os.getenv(REMO_HOME_ENV, str(Path.home().joinpath('.remo')))
+
+
+def set_REMO_HOME(path: str):
+    if not os.path.exists(path):
+        os.makedirs(path)
+    os.environ[REMO_HOME_ENV] = path
 
 
 class Config:
@@ -28,7 +38,7 @@ def server_url(self):
         return '{}:{}'.format(self.server, self.port)
 
     @staticmethod
-    def load(config_path: str = str(os.path.join(REMO_HOME, 'remo.json'))):
+    def load(config_path: str = str(os.path.join(REMO_HOME(), 'remo.json'))):
         if not os.path.exists(config_path):
             return None
 

remo/viewer.py

@@ -3,10 +3,11 @@
 import platform
 import subprocess
 from abc import abstractmethod, ABCMeta
-from pathlib import Path
 import webbrowser
 import uuid
 
+from remo.config import REMO_HOME
+
 
 class AbstractViewer(metaclass=ABCMeta):
     @abstractmethod
@@ -67,7 +68,6 @@ def browse(self, url: str):
 
 class ElectronViewer(AbstractViewer):
     exe_path = {'Linux': 'app/remo', 'Darwin': 'app/remo.app/Contents/MacOS/remo', 'Windows': 'app/remo.exe'}
-    REMO_HOME = os.getenv('REMO_HOME', str(Path.home().joinpath('.remo')))
     url_rxp = re.compile(r'(http[s]?://[.\w-]+)(:([0-9]+))?/?(.+)?')
 
     def browse(self, url):
@@ -86,7 +86,7 @@ def build_cmd(executable, **kwargs):
         return '{} {}'.format(executable, ' '.join('--{}={}'.format(k, v) for k, v in kwargs.items() if v))
 
     def get_remo_path(self):
-        return str(os.path.join(self.REMO_HOME, self.exe_path.get(platform.system())))
+        return str(os.path.join(REMO_HOME(), self.exe_path.get(platform.system())))
 
 
 def factory(name: str):