dbgraph 0.2.5__tar.gz → 0.2.7__tar.gz

{dbgraph-0.2.5 → dbgraph-0.2.7}/.idea/workspace.xml

@@ -71,32 +71,32 @@
     <option name="hideEmptyMiddlePackages" value="true" />
     <option name="showLibraryContents" value="true" />
   </component>
-  <component name="PropertiesComponent"><![CDATA[{
-  "keyToString": {
-    "ASKED_SHARE_PROJECT_CONFIGURATION_FILES": "true",
-    "ModuleVcsDetector.initialDetectionPerformed": "true",
-    "Python tests.Python tests for tests.test_dbgraph.TestDBGraph.executor": "Run",
-    "Python tests.Python tests for tests.test_json_graph_loader.TestJSONGraphLoader.test_get_neighbors_sub_graph.executor": "Run",
-    "Python tests.Python tests for tests.test_load_graph_sql.TestLoadGraphSQL.test_load.executor": "Run",
-    "Python tests.Python tests for tests.test_sql_graph_builder.TestSQLGraphBuilder.executor": "Run",
-    "Python tests.Python tests for tests.test_sql_graph_builder.TestSQLGraphBuilder.test__build_links.executor": "Run",
-    "Python tests.Python tests for tests.test_sql_graph_builder.TestSQLGraphBuilder.test_build_graph.executor": "Run",
-    "Python tests.Python tests for tests.test_sql_graph_persistent.TestSQLGraphPersistent.executor": "Run",
-    "Python tests.Python tests for tests.test_sql_graph_persistent.TestSQLGraphPersistent.test_delete_graph.executor": "Run",
-    "Python tests.Python tests for tests.test_sql_graph_persistent.TestSQLGraphPersistent.test_load_graph.executor": "Run",
-    "Python tests.Python tests for tests.test_sql_graph_persistent.TestSQLGraphPersistent.test_save_graph.executor": "Run",
-    "Python tests.Python tests in test_load_graph_sql.py.executor": "Run",
-    "Python tests.Python tests in tests.executor": "Run",
-    "RunOnceActivity.ShowReadmeOnStart": "true",
-    "RunOnceActivity.git.unshallow": "true",
-    "RunOnceActivity.typescript.service.memoryLimit.init": "true",
-    "codeWithMe.voiceChat.enabledByDefault": "false",
-    "git-widget-placeholder": "main",
-    "ignore.virus.scanning.warn.message": "true",
-    "nodejs_package_manager_path": "npm",
-    "settings.editor.selected.configurable": "editor.preferences.folding"
+  <component name="PropertiesComponent">{
+  &quot;keyToString&quot;: {
+    &quot;ASKED_SHARE_PROJECT_CONFIGURATION_FILES&quot;: &quot;true&quot;,
+    &quot;ModuleVcsDetector.initialDetectionPerformed&quot;: &quot;true&quot;,
+    &quot;Python tests.Python tests for tests.test_dbgraph.TestDBGraph.executor&quot;: &quot;Run&quot;,
+    &quot;Python tests.Python tests for tests.test_json_graph_loader.TestJSONGraphLoader.test_get_neighbors_sub_graph.executor&quot;: &quot;Run&quot;,
+    &quot;Python tests.Python tests for tests.test_load_graph_sql.TestLoadGraphSQL.test_load.executor&quot;: &quot;Run&quot;,
+    &quot;Python tests.Python tests for tests.test_sql_graph_builder.TestSQLGraphBuilder.executor&quot;: &quot;Run&quot;,
+    &quot;Python tests.Python tests for tests.test_sql_graph_builder.TestSQLGraphBuilder.test__build_links.executor&quot;: &quot;Run&quot;,
+    &quot;Python tests.Python tests for tests.test_sql_graph_builder.TestSQLGraphBuilder.test_build_graph.executor&quot;: &quot;Run&quot;,
+    &quot;Python tests.Python tests for tests.test_sql_graph_persistent.TestSQLGraphPersistent.executor&quot;: &quot;Run&quot;,
+    &quot;Python tests.Python tests for tests.test_sql_graph_persistent.TestSQLGraphPersistent.test_delete_graph.executor&quot;: &quot;Run&quot;,
+    &quot;Python tests.Python tests for tests.test_sql_graph_persistent.TestSQLGraphPersistent.test_load_graph.executor&quot;: &quot;Run&quot;,
+    &quot;Python tests.Python tests for tests.test_sql_graph_persistent.TestSQLGraphPersistent.test_save_graph.executor&quot;: &quot;Run&quot;,
+    &quot;Python tests.Python tests in test_load_graph_sql.py.executor&quot;: &quot;Run&quot;,
+    &quot;Python tests.Python tests in tests.executor&quot;: &quot;Run&quot;,
+    &quot;RunOnceActivity.ShowReadmeOnStart&quot;: &quot;true&quot;,
+    &quot;RunOnceActivity.git.unshallow&quot;: &quot;true&quot;,
+    &quot;RunOnceActivity.typescript.service.memoryLimit.init&quot;: &quot;true&quot;,
+    &quot;codeWithMe.voiceChat.enabledByDefault&quot;: &quot;false&quot;,
+    &quot;git-widget-placeholder&quot;: &quot;main&quot;,
+    &quot;ignore.virus.scanning.warn.message&quot;: &quot;true&quot;,
+    &quot;nodejs_package_manager_path&quot;: &quot;npm&quot;,
+    &quot;settings.editor.selected.configurable&quot;: &quot;editor.preferences.folding&quot;
   }
-}]]></component>
+}</component>
   <component name="RunManager" selected="Python tests.Python tests for tests.test_sql_graph_persistent.TestSQLGraphPersistent.test_delete_graph">
     <configuration name="Python tests for tests.test_json_graph_loader.TestJSONGraphLoader.test_get_neighbors_sub_graph" type="tests" factoryName="Autodetect" temporary="true" nameIsGenerated="true">
       <module name="dbgraph" />
@@ -318,7 +318,7 @@
       <breakpoints>
         <line-breakpoint enabled="true" suspend="THREAD" type="python-line">
           <url>file://$PROJECT_DIR$/dbgraph/persistent/sql_graph_persistent.py</url>
-          <line>353</line>
+          <line>360</line>
           <option name="timeStamp" value="5" />
         </line-breakpoint>
       </breakpoints>

{dbgraph-0.2.5 → dbgraph-0.2.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: dbgraph
-Version: 0.2.5
+Version: 0.2.7
 Summary: dbgraph is a database profiling tool which model the database as a graph
 Requires-Python: >=3.11
 Requires-Dist: bm25s>=0.3.9

{dbgraph-0.2.5 → dbgraph-0.2.7}/dbgraph/persistent/sql_graph_persistent.py

@@ -1,7 +1,7 @@
 import json
 from dataclasses import asdict, dataclass
 
-from sqlalchemy import create_engine, delete, select, update
+from sqlalchemy import create_engine, delete, select, update, event
 from sqlalchemy.orm import Session
 
 from dbgraph.entity.aspect import (
@@ -41,8 +41,17 @@ class SQLGraphPersistent(GraphPersistent):
 
     def __post_init__(self):
         self.engine = create_engine(self.db_uri)
+
+        @event.listens_for(self.engine, "connect")
+        def set_sqlite_pragma(dbapi_connection, connection_record):
+            if dbapi_connection.__class__.__module__.startswith("sqlite3"):
+                cursor = dbapi_connection.cursor()
+                cursor.execute("PRAGMA foreign_keys=ON")
+                cursor.close()
+
         Base.metadata.create_all(bind=self.engine)
 
+
     def create_graph(self, graph_id: str, graph_name: str):
         graph_model = GraphModel(graph_id=graph_id, name=graph_name, completed=False)
         with Session(self.engine) as session:
@@ -52,8 +61,6 @@ class SQLGraphPersistent(GraphPersistent):
     def delete_graph(self, graph_id: str):
         with Session(self.engine) as session:
             session.execute(delete(GraphModel).where(GraphModel.graph_id == graph_id))
-            session.execute(delete(AssetModel).where(AssetModel.graph_id == graph_id))
-            session.execute(delete(LinkModel).where(LinkModel.graph_id == graph_id))
             session.commit()
 
     def insert_assets(self, assets: list[Asset], graph_id: str):
@@ -370,7 +377,7 @@ class SQLGraphPersistent(GraphPersistent):
 
     def save_graph(self, graph: DatabaseGraph, graph_id: str):
         """Save the graph into database"""
-
+        self.create_graph(graph_id, graph_id)
         self.insert_assets(graph.assets, graph_id)
         self.insert_links(graph.links, graph_id)
 

{dbgraph-0.2.5 → dbgraph-0.2.7}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "dbgraph"
-version = "0.2.5"
+version = "0.2.7"
 description = "dbgraph is a database profiling tool which model the database as a graph"
 readme = "README.md"
 requires-python = ">=3.11"

{dbgraph-0.2.5 → dbgraph-0.2.7}/uv.lock

@@ -284,7 +284,7 @@ wheels = [
 
 [[package]]
 name = "dbgraph"
-version = "0.2.4"
+version = "0.2.6"
 source = { editable = "." }
 dependencies = [
     { name = "bm25s" },