dbgraph 0.2.9__tar.gz → 0.2.10__tar.gz

{dbgraph-0.2.9 → dbgraph-0.2.10}/.idea/workspace.xml

@@ -6,10 +6,7 @@
   <component name="ChangeListManager">
     <list default="true" id="263a96da-9a82-40fd-89b2-ba71dd2c194d" name="Changes" comment="update sql builder and tests">
       <change beforePath="$PROJECT_DIR$/.idea/dbgraph.iml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/dbgraph.iml" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/dbgraph/__init__.py" beforeDir="false" afterPath="$PROJECT_DIR$/dbgraph/__init__.py" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/dbgraph/entity/aspect.py" beforeDir="false" afterPath="$PROJECT_DIR$/dbgraph/entity/aspect.py" afterDir="false" />
       <change beforePath="$PROJECT_DIR$/dbgraph/persistent/orm_graph_persistent.py" beforeDir="false" afterPath="$PROJECT_DIR$/dbgraph/persistent/orm_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/units/test_orm_graph_persistent.py" beforeDir="false" afterPath="$PROJECT_DIR$/tests/units/test_orm_graph_persistent.py" afterDir="false" />
       <change beforePath="$PROJECT_DIR$/uv.lock" beforeDir="false" afterPath="$PROJECT_DIR$/uv.lock" afterDir="false" />
     </list>
@@ -71,67 +68,68 @@
     <option name="hideEmptyMiddlePackages" value="true" />
     <option name="showLibraryContents" value="true" />
   </component>
-  <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.integrates.test_json_graph_loader.TestJSONGraphLoader.executor&quot;: &quot;Run&quot;,
-    &quot;Python tests.Python tests for tests.integrates.test_sql_graph_builder.TestSQLGraphBuilder.executor&quot;: &quot;Run&quot;,
-    &quot;Python tests.Python tests for tests.integrates.test_sql_graph_builder.TestSQLGraphBuilder.test_build_graph.executor&quot;: &quot;Run&quot;,
-    &quot;Python tests.Python tests for tests.integrates.test_trino_builder.TestTrinoBuilder.test__get_cat_stats_aspect.executor&quot;: &quot;Run&quot;,
-    &quot;Python tests.Python tests for tests.integrates.test_trino_builder.TestTrinoBuilder.test__get_columns.executor&quot;: &quot;Run&quot;,
-    &quot;Python tests.Python tests for tests.integrates.test_trino_builder.TestTrinoBuilder.test__get_columns_orm.executor&quot;: &quot;Run&quot;,
-    &quot;Python tests.Python tests for tests.integrates.test_trino_builder.TestTrinoBuilder.test__get_null_info.executor&quot;: &quot;Run&quot;,
-    &quot;Python tests.Python tests for tests.integrates.test_trino_builder.TestTrinoBuilder.test__get_numerical_stats_aspect.executor&quot;: &quot;Run&quot;,
-    &quot;Python tests.Python tests for tests.integrates.test_trino_builder.TestTrinoBuilder.test__get_table_orm.executor&quot;: &quot;Run&quot;,
-    &quot;Python tests.Python tests for tests.integrates.test_trino_builder.TestTrinoBuilder.test__get_table_row_count.executor&quot;: &quot;Run&quot;,
-    &quot;Python tests.Python tests for tests.integrates.test_trino_builder.TestTrinoBuilder.test__get_temp_aspect.executor&quot;: &quot;Run&quot;,
-    &quot;Python tests.Python tests for tests.integrates.test_trino_builder.TestTrinoBuilder.test__is_column_textual.executor&quot;: &quot;Run&quot;,
-    &quot;Python tests.Python tests for tests.integrates.test_trino_builder.TestTrinoBuilder.test_build_graph.executor&quot;: &quot;Run&quot;,
-    &quot;Python tests.Python tests for tests.integrates.test_trino_builder.TestTrinoBuilder.test_table_names.executor&quot;: &quot;Run&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.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_markdown_renderer.TestMarkdownRenderer.executor&quot;: &quot;Run&quot;,
-    &quot;Python tests.Python tests for tests.test_orm_graph_persistent.TestORMGraphPersistent.executor&quot;: &quot;Run&quot;,
-    &quot;Python tests.Python tests for tests.test_orm_graph_persistent.TestORMGraphPersistent.test_cr_asset_aspect.executor&quot;: &quot;Run&quot;,
-    &quot;Python tests.Python tests for tests.test_orm_graph_persistent.TestORMGraphPersistent.test_cr_link_aspect.executor&quot;: &quot;Run&quot;,
-    &quot;Python tests.Python tests for tests.test_orm_graph_persistent.TestORMGraphPersistent.test_crud_assets.executor&quot;: &quot;Run&quot;,
-    &quot;Python tests.Python tests for tests.test_orm_graph_persistent.TestORMGraphPersistent.test_crud_links.executor&quot;: &quot;Run&quot;,
-    &quot;Python tests.Python tests for tests.test_orm_graph_persistent.TestORMGraphPersistent.test_insert_assets.executor&quot;: &quot;Run&quot;,
-    &quot;Python tests.Python tests for tests.test_orm_graph_persistent.TestORMGraphPersistent.test_insert_graph.executor&quot;: &quot;Run&quot;,
-    &quot;Python tests.Python tests for tests.test_orm_graph_persistent.TestORMGraphPersistent.test_insert_links.executor&quot;: &quot;Run&quot;,
-    &quot;Python tests.Python tests for tests.test_orm_graph_persistent.TestORMGraphPersistent.test_save_load_graph.executor&quot;: &quot;Run&quot;,
-    &quot;Python tests.Python tests for tests.test_rdbgraph.TestRDBGraph.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_asset.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_link.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 test_trino_builder.py.executor&quot;: &quot;Run&quot;,
-    &quot;Python tests.Python tests in tests.executor&quot;: &quot;Run&quot;,
-    &quot;Python tests.Python tests in units.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;optimize-queries&quot;,
-    &quot;ignore.virus.scanning.warn.message&quot;: &quot;true&quot;,
-    &quot;node.js.detected.package.eslint&quot;: &quot;true&quot;,
-    &quot;node.js.detected.package.jshint&quot;: &quot;true&quot;,
-    &quot;node.js.selected.package.eslint&quot;: &quot;(autodetect)&quot;,
-    &quot;node.js.selected.package.jshint&quot;: &quot;&quot;,
-    &quot;nodejs_package_manager_path&quot;: &quot;npm&quot;,
-    &quot;run.code.analysis.last.selected.profile&quot;: &quot;aDefault&quot;,
-    &quot;settings.editor.selected.configurable&quot;: &quot;preferences.PingConfigurable&quot;
+  <component name="PropertiesComponent"><![CDATA[{
+  "keyToString": {
+    "ASKED_SHARE_PROJECT_CONFIGURATION_FILES": "true",
+    "ModuleVcsDetector.initialDetectionPerformed": "true",
+    "Python tests.Python tests for tests.integrates.test_json_graph_loader.TestJSONGraphLoader.executor": "Run",
+    "Python tests.Python tests for tests.integrates.test_sql_graph_builder.TestSQLGraphBuilder.executor": "Run",
+    "Python tests.Python tests for tests.integrates.test_sql_graph_builder.TestSQLGraphBuilder.test_build_graph.executor": "Run",
+    "Python tests.Python tests for tests.integrates.test_trino_builder.TestTrinoBuilder.test__get_cat_stats_aspect.executor": "Run",
+    "Python tests.Python tests for tests.integrates.test_trino_builder.TestTrinoBuilder.test__get_columns.executor": "Run",
+    "Python tests.Python tests for tests.integrates.test_trino_builder.TestTrinoBuilder.test__get_columns_orm.executor": "Run",
+    "Python tests.Python tests for tests.integrates.test_trino_builder.TestTrinoBuilder.test__get_null_info.executor": "Run",
+    "Python tests.Python tests for tests.integrates.test_trino_builder.TestTrinoBuilder.test__get_numerical_stats_aspect.executor": "Run",
+    "Python tests.Python tests for tests.integrates.test_trino_builder.TestTrinoBuilder.test__get_table_orm.executor": "Run",
+    "Python tests.Python tests for tests.integrates.test_trino_builder.TestTrinoBuilder.test__get_table_row_count.executor": "Run",
+    "Python tests.Python tests for tests.integrates.test_trino_builder.TestTrinoBuilder.test__get_temp_aspect.executor": "Run",
+    "Python tests.Python tests for tests.integrates.test_trino_builder.TestTrinoBuilder.test__is_column_textual.executor": "Run",
+    "Python tests.Python tests for tests.integrates.test_trino_builder.TestTrinoBuilder.test_build_graph.executor": "Run",
+    "Python tests.Python tests for tests.integrates.test_trino_builder.TestTrinoBuilder.test_table_names.executor": "Run",
+    "Python tests.Python tests for tests.test_dbgraph.TestDBGraph.executor": "Run",
+    "Python tests.Python tests for tests.test_json_graph_loader.TestJSONGraphLoader.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_markdown_renderer.TestMarkdownRenderer.executor": "Run",
+    "Python tests.Python tests for tests.test_orm_graph_persistent.TestORMGraphPersistent.executor": "Run",
+    "Python tests.Python tests for tests.test_orm_graph_persistent.TestORMGraphPersistent.test_cr_asset_aspect.executor": "Run",
+    "Python tests.Python tests for tests.test_orm_graph_persistent.TestORMGraphPersistent.test_cr_link_aspect.executor": "Run",
+    "Python tests.Python tests for tests.test_orm_graph_persistent.TestORMGraphPersistent.test_crud_assets.executor": "Run",
+    "Python tests.Python tests for tests.test_orm_graph_persistent.TestORMGraphPersistent.test_crud_links.executor": "Run",
+    "Python tests.Python tests for tests.test_orm_graph_persistent.TestORMGraphPersistent.test_insert_assets.executor": "Run",
+    "Python tests.Python tests for tests.test_orm_graph_persistent.TestORMGraphPersistent.test_insert_graph.executor": "Run",
+    "Python tests.Python tests for tests.test_orm_graph_persistent.TestORMGraphPersistent.test_insert_links.executor": "Run",
+    "Python tests.Python tests for tests.test_orm_graph_persistent.TestORMGraphPersistent.test_save_load_graph.executor": "Run",
+    "Python tests.Python tests for tests.test_rdbgraph.TestRDBGraph.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_asset.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_link.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 for tests.units.test_orm_graph_persistent.TestORMGraphPersistent.test_save_load_graph.executor": "Run",
+    "Python tests.Python tests in test_load_graph_sql.py.executor": "Run",
+    "Python tests.Python tests in test_trino_builder.py.executor": "Run",
+    "Python tests.Python tests in tests.executor": "Run",
+    "Python tests.Python tests in units.executor": "Run",
+    "RunOnceActivity.ShowReadmeOnStart": "true",
+    "RunOnceActivity.git.unshallow": "true",
+    "RunOnceActivity.typescript.service.memoryLimit.init": "true",
+    "codeWithMe.voiceChat.enabledByDefault": "false",
+    "git-widget-placeholder": "optimize-queries",
+    "ignore.virus.scanning.warn.message": "true",
+    "node.js.detected.package.eslint": "true",
+    "node.js.detected.package.jshint": "true",
+    "node.js.selected.package.eslint": "(autodetect)",
+    "node.js.selected.package.jshint": "",
+    "nodejs_package_manager_path": "npm",
+    "run.code.analysis.last.selected.profile": "aDefault",
+    "settings.editor.selected.configurable": "preferences.PingConfigurable"
   }
-}</component>
+}]]></component>
   <component name="PyConsoleOptionsProvider">
     <option name="myPythonConsoleState">
       <console-settings module-name="dbgraph" is-module-sdk="true">
@@ -146,7 +144,7 @@
       <recent name="C:\Users\minhden\Projects\dbgraph\tests\units" />
     </key>
   </component>
-  <component name="RunManager" selected="Python tests.Python tests in units">
+  <component name="RunManager" selected="Python tests.Python tests for tests.units.test_orm_graph_persistent.TestORMGraphPersistent.test_save_load_graph">
     <configuration name="Python tests for tests.integrates.test_sql_graph_builder.TestSQLGraphBuilder.test_build_graph" type="tests" factoryName="Autodetect" temporary="true" nameIsGenerated="true">
       <module name="dbgraph" />
       <option name="ENV_FILES" value="" />
@@ -198,7 +196,7 @@
       <option name="_new_targetType" value="&quot;PYTHON&quot;" />
       <method v="2" />
     </configuration>
-    <configuration name="Python tests for tests.integrates.test_trino_builder.TestTrinoBuilder.test_build_graph" type="tests" factoryName="Autodetect" temporary="true" nameIsGenerated="true">
+    <configuration name="Python tests for tests.units.test_orm_graph_persistent.TestORMGraphPersistent.test_save_load_graph" type="tests" factoryName="Autodetect" temporary="true" nameIsGenerated="true">
       <module name="dbgraph" />
       <option name="ENV_FILES" value="" />
       <option name="INTERPRETER_OPTIONS" value="" />
@@ -209,9 +207,9 @@
       <option name="ADD_CONTENT_ROOTS" value="true" />
       <option name="ADD_SOURCE_ROOTS" value="true" />
       <option name="DEBUG_JUST_MY_CODE" value="false" />
-      <option name="RUN_TOOL" value="true" />
+      <option name="RUN_TOOL" value="" />
       <option name="_new_additionalArguments" value="&quot;&quot;" />
-      <option name="_new_target" value="&quot;tests.integrates.test_trino_builder.TestTrinoBuilder.test_build_graph&quot;" />
+      <option name="_new_target" value="&quot;tests.units.test_orm_graph_persistent.TestORMGraphPersistent.test_save_load_graph&quot;" />
       <option name="_new_targetType" value="&quot;PYTHON&quot;" />
       <method v="2" />
     </configuration>
@@ -234,11 +232,11 @@
     </configuration>
     <recent_temporary>
       <list>
+        <item itemvalue="Python tests.Python tests for tests.units.test_orm_graph_persistent.TestORMGraphPersistent.test_save_load_graph" />
         <item itemvalue="Python tests.Python tests in units" />
         <item itemvalue="Python tests.Python tests for tests.integrates.test_trino_builder.TestTrinoBuilder.test__get_table_orm" />
         <item itemvalue="Python tests.Python tests for tests.integrates.test_sql_graph_builder.TestSQLGraphBuilder.test_build_graph" />
         <item itemvalue="Python tests.Python tests for tests.integrates.test_trino_builder.TestTrinoBuilder.test__get_table_row_count" />
-        <item itemvalue="Python tests.Python tests for tests.integrates.test_trino_builder.TestTrinoBuilder.test_build_graph" />
       </list>
     </recent_temporary>
   </component>
@@ -345,7 +343,15 @@
       <option name="project" value="LOCAL" />
       <updated>1790214726709</updated>
     </task>
-    <option name="localTasksCounter" value="12" />
+    <task id="LOCAL-00012" summary="update sql builder and tests">
+      <option name="closed" value="true" />
+      <created>1790219300115</created>
+      <option name="number" value="00012" />
+      <option name="presentableId" value="LOCAL-00012" />
+      <option name="project" value="LOCAL" />
+      <updated>1790219300115</updated>
+    </task>
+    <option name="localTasksCounter" value="13" />
     <servers />
   </component>
   <component name="TypeScriptGeneratedFilesManager">

{dbgraph-0.2.9 → dbgraph-0.2.10}/PKG-INFO

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

{dbgraph-0.2.9 → dbgraph-0.2.10}/dbgraph/persistent/orm_graph_persistent.py

@@ -2,7 +2,7 @@ import json
 from dataclasses import dataclass, asdict
 from uuid import UUID
 
-from sqlalchemy import create_engine, select, delete, and_
+from sqlalchemy import create_engine, select, delete, and_, event
 from sqlalchemy.orm import Session
 
 from dbgraph import Asset, Link, DatabaseGraph, LinkType, AssetType
@@ -35,6 +35,13 @@ class ORMGraphPersistent(GraphPersistent):
 
     def __post_init__(self):
         self.engine = create_engine(self.persistent_uri)
+
+        @event.listens_for(self.engine, "connect")
+        def set_sqlite_pragma(dbapi_connection, connection_record):
+            cursor = dbapi_connection.cursor()
+            cursor.execute("PRAGMA foreign_keys=ON")
+            cursor.close()
+
         Base.metadata.create_all(self.engine)
 
     def insert_graph(self, graph_id: UUID, graph_name: str):

{dbgraph-0.2.9 → dbgraph-0.2.10}/pyproject.toml

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

{dbgraph-0.2.9 → dbgraph-0.2.10}/tests/units/test_orm_graph_persistent.py

@@ -365,6 +365,9 @@ class TestORMGraphPersistent(unittest.TestCase):
         self.persistent.save_graph(target_id, "test", self.graph)
         graph = self.persistent.load_graph(target_id)
         self.assertEqual(self.graph, graph)
+        self.persistent.delete_graph(target_id)
+        assets = self.persistent.get_assets(target_id)
+        self.assertEqual(0, len(assets))
 
 
 if __name__ == "__main__":

{dbgraph-0.2.9 → dbgraph-0.2.10}/uv.lock

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