Add test_example

Author: danlooo

pathtraits/access.py

@@ -10,31 +10,8 @@ def get(path, db_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("/")
-
     db = TraitsDB(db_path)
-
-    # get traits from path and its parents
-    dirs_data = []
-    data = db.get("data", path=abs_path)
-    if data:
-        dirs_data.append(data)
-    for i in reversed(range(0, len(dirs))):
-        cur_path = "/".join(dirs[0 : i + 1])
-        data = db.get("data", path=cur_path)
-        if data:
-            dirs_data.append(data)
-
-    # inherit traits: children overwrite parent path traits
-    res = {}
-    for cur_data in reversed(dirs_data):
-        for k, v in cur_data.items():
-            if v and k != "path":
-                res[k] = v
-
-    # output
+    res = db.get_dict(path)
     if len(res) > 0:
         print(yaml.safe_dump(res))
     else:

pathtraits/db.py

@@ -55,6 +55,30 @@ def get(self, table, cols="*", condition=None, **kwargs):
         res = {k: v for k, v in zip(keys, values)}
         return res
 
+    def get_dict(self, path):
+        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("/")
+
+        # get traits from path and its parents
+        dirs_data = []
+        data = self.get("data", path=abs_path)
+        if data:
+            dirs_data.append(data)
+        for i in reversed(range(0, len(dirs))):
+            cur_path = "/".join(dirs[0 : i + 1])
+            data = self.get("data", path=cur_path)
+            if data:
+                dirs_data.append(data)
+
+        # inherit traits: children overwrite parent path traits
+        res = {}
+        for cur_data in reversed(dirs_data):
+            for k, v in cur_data.items():
+                if v and k != "path":
+                    res[k] = v
+        return res
+
     def put_path_id(self, path):
         get_row_query = f"SELECT id FROM path WHERE path = '{path}' LIMIT 1;"
         res = self.execute(get_row_query).fetchone()

test/test.py

@@ -10,14 +10,24 @@
 import pathtraits.scan
 
 
-class TestCLI(unittest.TestCase):
-    def test(self):
+class TestMain(unittest.TestCase):
+    def test_example(self):
         db_path = tempfile.mkstemp()[1]
         pathtraits.scan.batch("test/example", db_path, False)
 
         db = pathtraits.db.TraitsDB(db_path)
         self.assertTrue(db is not None)
 
+        source = db.get_dict("test/example/Europe/de.txt")
+        target = {
+            "description_TEXT": "Germany data",
+            "has_sidecar_meta_file_BOOL": 1,
+            "is_example_BOOL": 1,
+            "score_TEXT": "zero",
+        }
+        for k, v in target.items():
+            self.assertEqual(source[k], v)
+
 
 if __name__ == "__main__":
     unittest.main()