Add frontend restart to service file, add check for custom db uri

Author: kizniche

install/mycodo.service

@@ -6,7 +6,7 @@ StartLimitBurst=5
 
 [Service]
 Restart=on-failure
-RestartSec=20s
+RestartSec=20
 TimeoutSec=0
 ExecStart=/opt/Mycodo/env/bin/python /opt/Mycodo/mycodo/mycodo_daemon.py
 ExecStop=/opt/Mycodo/env/bin/python /opt/Mycodo/mycodo/mycodo_client.py -t

install/mycodoflask.service

@@ -1,8 +1,12 @@
 [Unit]
 Description=Mycodo web user interface
 After=network.target
+StartLimitIntervalSec=300
+StartLimitBurst=5
 
 [Service]
+Restart=on-failure
+RestartSec=20
 User=root
 Group=mycodo
 WorkingDirectory=/opt/Mycodo/mycodo

mycodo/databases/utils.py

@@ -16,7 +16,15 @@ def session_scope(db_uri):
     Session = sessionmaker()
 
     # later, we create the engine
-    engine = create_engine(f"{db_uri}?check_same_thread=False")
+    try:
+        # Custom URI
+        from mycodo.config_override import SQLALCHEMY_DATABASE_URI
+        engine_url = SQLALCHEMY_DATABASE_URI
+    except:
+        # SQLite3
+        engine_url = f"{db_uri}?check_same_thread=False"
+
+    engine = create_engine(engine_url)
 
     # associate it with our custom Session class
     Session.configure(bind=engine)