Merge pull request #656 from frennkie/switch-project-sqlite-only

Allow switching projects only on SQLite

viper/core/database.py

@@ -193,6 +193,8 @@ class Database:
 
     def __init__(self):
 
+        self.supports_projects = False  # currently only sqlite support projects
+
         if cfg.database and cfg.database.connection:
             self._connect_database(cfg.database.connection)
         else:
@@ -218,6 +220,7 @@ def _connect_database(self, connection):
         elif connection.startswith("postgresql"):
             self.engine = create_engine(connection, connect_args={"sslmode": "disable"})
         else:
+            self.supports_projects = True
             db_path = os.path.join(__project__.get_path(), 'viper.db')
             self.engine = create_engine('sqlite:///{0}'.format(db_path), poolclass=NullPool)
 

viper/core/ui/cmd/projects.py

@@ -63,6 +63,11 @@ def run(self, *args):
 
             self.log("table", dict(header=["Project Name", "Creation Time", "Current"], rows=rows))
         elif args.switch:
+            db = Database()
+            if not db.supports_projects:
+                self.log('info', "The database type you are using does not support projects")
+                return
+
             if __sessions__.is_set():
                 __sessions__.close()
                 self.log("info", "Closed opened session")