dbgraph 0.2.4__tar.gz → 0.2.6__tar.gz

{dbgraph-0.2.4 → dbgraph-0.2.6}/.idea/workspace.xml

@@ -10,6 +10,11 @@
       <change beforePath="$PROJECT_DIR$/data/northwind-index/indices.csc.index.npy" beforeDir="false" afterPath="$PROJECT_DIR$/data/northwind-index/indices.csc.index.npy" afterDir="false" />
       <change beforePath="$PROJECT_DIR$/data/northwind-index/indptr.csc.index.npy" beforeDir="false" afterPath="$PROJECT_DIR$/data/northwind-index/indptr.csc.index.npy" afterDir="false" />
       <change beforePath="$PROJECT_DIR$/data/northwind-index/vocab.index.json" beforeDir="false" afterPath="$PROJECT_DIR$/data/northwind-index/vocab.index.json" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/dbgraph/persistent/models.py" beforeDir="false" afterPath="$PROJECT_DIR$/dbgraph/persistent/models.py" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/dbgraph/persistent/sql_graph_persistent.py" beforeDir="false" afterPath="$PROJECT_DIR$/dbgraph/persistent/sql_graph_persistent.py" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/pyproject.toml" beforeDir="false" afterPath="$PROJECT_DIR$/pyproject.toml" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/tests/test_sql_graph_persistent.py" beforeDir="false" afterPath="$PROJECT_DIR$/tests/test_sql_graph_persistent.py" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/uv.lock" beforeDir="false" afterPath="$PROJECT_DIR$/uv.lock" afterDir="false" />
     </list>
     <option name="SHOW_DIALOG" value="false" />
     <option name="HIGHLIGHT_CONFLICTS" value="true" />
@@ -77,6 +82,7 @@
     &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;,
@@ -91,7 +97,7 @@
     &quot;settings.editor.selected.configurable&quot;: &quot;editor.preferences.folding&quot;
   }
 }</component>
-  <component name="RunManager" selected="Python tests.Python tests for tests.test_json_graph_loader.TestJSONGraphLoader.test_get_neighbors_sub_graph">
+  <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" />
       <option name="ENV_FILES" value="" />
@@ -143,7 +149,7 @@
       <option name="_new_targetType" value="&quot;PYTHON&quot;" />
       <method v="2" />
     </configuration>
-    <configuration name="Python tests for tests.test_sql_graph_builder.TestSQLGraphBuilder.test_build_graph" type="tests" factoryName="Autodetect" temporary="true" nameIsGenerated="true">
+    <configuration name="Python tests for tests.test_sql_graph_persistent.TestSQLGraphPersistent.test_delete_graph" type="tests" factoryName="Autodetect" temporary="true" nameIsGenerated="true">
       <module name="dbgraph" />
       <option name="ENV_FILES" value="" />
       <option name="INTERPRETER_OPTIONS" value="" />
@@ -156,7 +162,7 @@
       <option name="DEBUG_JUST_MY_CODE" value="false" />
       <option name="RUN_TOOL" value="" />
       <option name="_new_additionalArguments" value="&quot;&quot;" />
-      <option name="_new_target" value="&quot;tests.test_sql_graph_builder.TestSQLGraphBuilder.test_build_graph&quot;" />
+      <option name="_new_target" value="&quot;tests.test_sql_graph_persistent.TestSQLGraphPersistent.test_delete_graph&quot;" />
       <option name="_new_targetType" value="&quot;PYTHON&quot;" />
       <method v="2" />
     </configuration>
@@ -179,11 +185,11 @@
     </configuration>
     <recent_temporary>
       <list>
+        <item itemvalue="Python tests.Python tests for tests.test_sql_graph_persistent.TestSQLGraphPersistent.test_delete_graph" />
         <item itemvalue="Python tests.Python tests for tests.test_json_graph_loader.TestJSONGraphLoader.test_get_neighbors_sub_graph" />
         <item itemvalue="Python tests.Python tests in tests" />
         <item itemvalue="Python tests.Python tests for tests.test_sql_graph_builder.TestSQLGraphBuilder" />
         <item itemvalue="Python tests.Python tests for tests.test_sql_graph_builder.TestSQLGraphBuilder.test__build_links" />
-        <item itemvalue="Python tests.Python tests for tests.test_sql_graph_builder.TestSQLGraphBuilder.test_build_graph" />
       </list>
     </recent_temporary>
   </component>
@@ -312,12 +318,7 @@
       <breakpoints>
         <line-breakpoint enabled="true" suspend="THREAD" type="python-line">
           <url>file://$PROJECT_DIR$/dbgraph/persistent/sql_graph_persistent.py</url>
-          <line>330</line>
-          <option name="timeStamp" value="4" />
-        </line-breakpoint>
-        <line-breakpoint enabled="true" suspend="THREAD" type="python-line">
-          <url>file://$PROJECT_DIR$/dbgraph/persistent/sql_graph_persistent.py</url>
-          <line>351</line>
+          <line>360</line>
           <option name="timeStamp" value="5" />
         </line-breakpoint>
       </breakpoints>

{dbgraph-0.2.4 → dbgraph-0.2.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: dbgraph
-Version: 0.2.4
+Version: 0.2.6
 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.4 → dbgraph-0.2.6}/dbgraph/persistent/models.py

@@ -66,6 +66,7 @@ class AssetModel(Base):
         ForeignKey(
             "graphs.graph_id",
             ondelete="CASCADE",
+            onupdate="CASCADE"
         ),
         nullable=False,
         index=True,
@@ -116,6 +117,7 @@ class LinkModel(Base):
         ForeignKey(
             "assets.asset_id",
             ondelete="CASCADE",
+            onupdate="CASCADE"
         ),
         nullable=False,
         index=True,
@@ -125,6 +127,7 @@ class LinkModel(Base):
         ForeignKey(
             "assets.asset_id",
             ondelete="CASCADE",
+            onupdate="CASCADE"
         ),
         nullable=False,
         index=True,
@@ -134,6 +137,7 @@ class LinkModel(Base):
         ForeignKey(
             "graphs.graph_id",
             ondelete="CASCADE",
+            onupdate="CASCADE"
         ),
         nullable=False,
         index=True,
@@ -167,6 +171,7 @@ class AssetAspectModel(Base):
         ForeignKey(
             "assets.asset_id",
             ondelete="CASCADE",
+            onupdate="CASCADE"
         ),
         primary_key=True,
     )
@@ -198,6 +203,7 @@ class LinkAspectModel(Base):
         ForeignKey(
             "links.link_id",
             ondelete="CASCADE",
+            onupdate="CASCADE"
         ),
         primary_key=True,
     )

{dbgraph-0.2.4 → dbgraph-0.2.6}/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:

{dbgraph-0.2.4 → dbgraph-0.2.6}/pyproject.toml

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

{dbgraph-0.2.4 → dbgraph-0.2.6}/tests/test_sql_graph_persistent.py

@@ -441,6 +441,15 @@ class TestSQLGraphPersistent(unittest.TestCase):
         graph = self.persistent.load_graph(self.graph_id)
         self.assertEqual(self.graph, graph)
 
+    def test_delete_graph(self):
+        self.persistent.save_graph(self.graph, self.graph_id)
+        graph = self.persistent.load_graph(self.graph_id)
+        self.assertEqual(self.graph, graph)
+        self.persistent.delete_graph(self.graph_id)
+        graph = self.persistent.load_graph(self.graph_id)
+        self.assertEqual(0, len(graph.assets))
+        self.assertEqual(0, len(graph.links))
+
     def test_get_graph_info(self):
         graph_info = self.persistent.get_graph_info("1")
         self.assertEqual(graph_info[1], "graph-1")

{dbgraph-0.2.4 → dbgraph-0.2.6}/uv.lock

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