add environment variables when not using Docker, add an ending message for the installation

Alexis Koralewski
1 parent 12aa6dce
Showing 1 changed file with 29 additions and 2 deletions Show diff stats
pyros.py
@@ -133,6 +133,33 @@ if type(WITH_DOCKER) is str and re.match(&quot;^y$|^Y$|^yes$|^Yes$&quot;,WITH_DOCKER.rstri
 else : 
     WITH_DOCKER = False
  
+ENV_PATH = "docker/.env"
+ENV_SAMPLE_PATH = "docker/.env-sample"
+is_environment_variables_not_defined = os.environ.get("MYSQL_ROOT_PASSWORD") == None
+if(is_environment_variables_not_defined):
+    try:
+        with open(ENV_PATH,"r") as env_file:
+            print("Reading env file")
+            for line in env_file:
+                if(line.startswith("#") and not env_sample_line.strip()):
+                    continue
+                else:
+                    key,value = line.split("=")
+                    # setting variables as environment variables
+                    os.environ[key] = value
+    except:
+        print(f".env not found at {ENV_PATH}, creating a file at this path from the .env-sample file stored at {ENV_SAMPLE_PATH}\n \
+            values from .env-sample will be used as environment variables")
+        with open(ENV_SAMPLE_PATH,'r') as env_sample_file:
+            with open(ENV_PATH,"w") as env_file:
+                for env_sample_line in env_sample_file:
+                    if(env_sample_line.startswith("#") or not env_sample_line.strip()):
+                        continue
+                    key,value = env_sample_line.split("=")
+                    os.environ[key] = value
+                    env_file.write(env_sample_line)
+    is_environment_variables_not_defined = os.environ.get("MYSQL_ROOT_PASSWORD") == None
+
 END_OF_LINE = '\n\n'
 ##VENV_BIN = '/bin/'
 # --------------------------------------------
@@ -437,7 +464,7 @@ def install_or_update(UPDATE:bool=False, packages_only:bool=False, database_only
             _updatedb() or die()
         else:
             install_database(VENV)
-
+    print("You can connect to PyROS as 'pyros' user with the password 'DjangoPyros' ")
     return True
  
  
@@ -1046,7 +1073,7 @@ def install_database(venv):
  
     # if the mysql_root_password isn't defined, it will ask to enter the password
     try:
-        # If we're using Docker, a root password should be defined as an environment variable
+        # a root password should be defined as an environment variable
         mysql_root_password = os.environ['MYSQL_ROOT_PASSWORD']
     except:
         # need a space as string so the command ask to the user to write his password.