Return YAML in command get

Author: danlooo

pathtraits/cli.py

@@ -14,7 +14,7 @@ def main():
 
 
 @main.command(help="Update database once, searches for all directories recursively.")
-@click.argument("path", required=True)
+@click.argument("path", required=True, type=click.Path(exists=True))
 @click.option("-v", "--verbose", "verbose", flag_value=True, default=False)
 @click.option(
     "--include-files",
@@ -42,7 +42,7 @@ def batch(path, verbose, include_files):
 
 
 @main.command(help="Update database continiously, watches for new or changed files.")
-@click.argument("path", required=True)
+@click.argument("path", required=True, type=click.Path(exists=True))
 @click.option("-v", "--verbose", "verbose", flag_value=True, default=False)
 def watch(path, verbose):
     if verbose:
@@ -65,5 +65,34 @@ def watch(path, verbose):
             db.add_pathpair(pair)
 
 
+@main.command(help="Get traits of a given path")
+@click.argument("path", required=True, type=click.Path(exists=True))
+@click.option("-v", "--verbose", "verbose", flag_value=True, default=False)
+def get(path, verbose):
+    if verbose:
+        logging.basicConfig(level=logging.DEBUG)
+
+    abs_path = os.path.abspath(path)
+    leaf_dir = os.path.dirname(abs_path) if os.path.isfile(abs_path) else abs_path
+    dirs = leaf_dir.split("/")
+    for i in reversed(range(0, len(dirs))):
+        if i == 0:
+            db_dir = "/"
+        else:
+            db_dir = "/".join(dirs[0 : i + 1])
+
+        db_path = db_dir + "/.pathtraits.db"
+        if not os.path.exists(db_path):
+            continue
+
+        # TODO: recursive inheritance of pathtraits
+        db = TraitsDB(db_dir)
+        data = db.get("data", path=abs_path)
+        print(yaml.safe_dump(data))
+        return
+
+    KeyError(f"No pathtraits database found in {abs_path} and its parents.")
+
+
 if __name__ == "__main__":
     main()

pathtraits/traitsdb.py

@@ -43,7 +43,14 @@ def get(self, table, cols="*", condition=None, **kwargs):
             }
             condition = " AND ".join([f"{k}={v}" for (k, v) in escaped_kwargs.items()])
         get_row_query = f"SELECT {cols} FROM {table} WHERE {condition} LIMIT 1;"
-        res = self.execute(get_row_query).fetchone()
+        response = self.execute(get_row_query)
+        values = response.fetchone()
+
+        if values is None:
+            return None
+
+        keys = map(lambda x: x[0], response.description)
+        res = {k: v for k, v in zip(keys, values)}
         return res
 
     def put_path_id(self, path):
@@ -54,7 +61,7 @@ def put_path_id(self, path):
         else:
             # create
             self.put("path", path=path)
-            path_id = self.get("path", path=path, cols="id")[0]
+            path_id = self.get("path", path=path, cols="id")["id"]
             return path_id
 
     def escape(value):