closurizer 0.7.1__py3-none-any.whl → 0.7.3__py3-none-any.whl

closurizer/closurizer.py

@@ -156,18 +156,35 @@ def add_closure(kg_archive: str,
     """
     print(nodes_query)
 
+
     if not dry_run:
-        db.query(edges_query)
-        db.query(f"""
+
+        db.sql(edges_query)
+
+        edge_closure_replacements = [
+            f"""
+            list_aggregate({field}_closure, 'string_agg', '|') as {field}_closure,
+            list_aggregate({field}_closure_label, 'string_agg', '|') as {field}_closure_label
+            """
+            for field in edge_fields
+        ]
+
+        edge_closure_replacements = "REPLACE (\n" + ",\n".join(edge_closure_replacements) + ")\n"
+
+        edges_export_query = f"""
         -- write denormalized_edges as tsv
-        copy (select * from denormalized_edges) to '{edges_output_file}' (header, delimiter '\t')
-        """)
-        db.query(nodes_query)
-        db.query(f"""
+        copy (select * {edge_closure_replacements} from denormalized_edges) to '{edges_output_file}' (header, delimiter '\t')
+        """
+        print(edges_export_query)
+        db.sql(edges_export_query)
+
+        db.sql(nodes_query)
+        nodes_export_query = f"""
         -- write denormalized_nodes as tsv
         copy (select * from denormalized_nodes) to '{nodes_output_file}' (header, delimiter '\t')
-        """)
-
+        """
+        print(nodes_export_query)
+        db.sql(nodes_export_query)
 
         # Clean up extracted node & edge files
         if os.path.exists(f"{node_file}"):

{closurizer-0.7.1.dist-info → closurizer-0.7.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: closurizer
-Version: 0.7.1
+Version: 0.7.3
 Summary: Add closure expansion fields to kgx files following the Golr pattern
 Author: Kevin Schaper
 Author-email: kevin@tislab.org

closurizer-0.7.3.dist-info/RECORD

@@ -0,0 +1,6 @@
+closurizer/cli.py,sha256=xTFscsxGDnaKoTNhn1FefRPPeldI5tZvvp3DygNai7Y,2069
+closurizer/closurizer.py,sha256=rzxML6GSzVAeKdK1hq51ehvfIQ9zLYAa8YPH-IdRlJQ,7582
+closurizer-0.7.3.dist-info/METADATA,sha256=VSX_LhgP-_enQvhA4e2wsxLsOEoQfQc8lZr9eDUwaOY,661
+closurizer-0.7.3.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
+closurizer-0.7.3.dist-info/entry_points.txt,sha256=MnAVu1lgP6DqDb3BZGNzVs2AnDMsp4sThi3ccWbONFo,50
+closurizer-0.7.3.dist-info/RECORD,,

closurizer-0.7.1.dist-info/RECORD

@@ -1,6 +0,0 @@
-closurizer/cli.py,sha256=xTFscsxGDnaKoTNhn1FefRPPeldI5tZvvp3DygNai7Y,2069
-closurizer/closurizer.py,sha256=DY3NnSaCOf3XYF_NFqS_TB3o9WZVnBviXlvI0saXlxg,7000
-closurizer-0.7.1.dist-info/METADATA,sha256=S5ItMFYNZQZWBfF31M_fP9pS9LgNCFl_1QGbHU5cLKk,661
-closurizer-0.7.1.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-closurizer-0.7.1.dist-info/entry_points.txt,sha256=MnAVu1lgP6DqDb3BZGNzVs2AnDMsp4sThi3ccWbONFo,50
-closurizer-0.7.1.dist-info/RECORD,,