Add pathtraits info

Author: danlooo

pathtraits/cli.py

@@ -5,7 +5,7 @@
 import logging
 import os
 import click
-from pathtraits import scan, access, create as _create
+from pathtraits import scan, access, db, create as _create
 
 logger = logging.getLogger(__name__)
 
@@ -111,5 +111,18 @@ def create(path, needed_until, overwrite, verbose):
     _create.generate_metadata(path, needed_until, overwrite, verbose)
 
 
+@main.command()
+@click.option(
+    "--db-path",
+    default=DB_PATH,
+    type=click.Path(file_okay=True, dir_okay=False),
+)
+def info(db_path):
+    """
+    Displays info about the database
+    """
+    db.TraitsDB(db_path).info()
+
+
 if __name__ == "__main__":
     main()

pathtraits/db.py

@@ -18,6 +18,7 @@ class TraitsDB:
     Database of pathtrait in 3NF with view of all joined trait tables
     """
 
+    db_path = None
     cursor = None
     traits = []
 
@@ -96,8 +97,8 @@ def flatten_dict(dictionary: dict, root_key: str = "", separator: str = "/"):
         return dict(items)
 
     def __init__(self, db_path):
-        db_path = os.path.join(db_path)
-        self.cursor = sqlite3.connect(db_path, autocommit=True).cursor()
+        self.db_path = os.path.join(db_path)
+        self.cursor = sqlite3.connect(self.db_path, autocommit=True).cursor()
         self.cursor.row_factory = TraitsDB.row_factory
 
         init_path_table_query = """
@@ -257,6 +258,20 @@ def get_paths(self, query_str):
         res = [x["path"] for x in res]
         return res
 
+    def info(self):
+        """
+        Prints info about this database
+
+        :param self: this database
+        """
+        res = {}
+        res["db_path"] = self.db_path
+        res["traits"] = self.traits
+        res["paths"] = len(self.get_paths("TRUE"))
+
+        if len(res) > 0:
+            print(yaml.safe_dump(res))
+
     def put_path_id(self, path):
         """
         Docstring for put_path_id