o Replacing open_database with with_database to make sure it is closed after use.

1 files changed, 16 insertions, 15 deletions

diff --git a/trygvis/eda/cli/db_stats.py b/trygvis/eda/cli/db_stats.py
index 4780c20..5ee99ac 100755
--- a/trygvis/eda/cli/db_stats.py
+++ b/trygvis/eda/cli/db_stats.py
@@ -2,9 +2,8 @@ from trygvis.eda import cli
 
 
 def run(db_path):
-    g = cli.open_database(db_path)
-
-    res = cli.sparql(g, """
+    def db_stats(g):
+        res = cli.sparql(g, """
 SELECT ?schematic ?label
 WHERE {
     ?schematic a kicad-type:schematic
@@ -13,16 +12,16 @@ WHERE {
     }
 }
 """)
-    cli.info("Found %d schematics in database" % len(res))
-    for row in res:
-        name = row.label if row.label is not None else "<unnamed>"
-        url = row.schematic
+        cli.info("Found %d schematics in database" % len(res))
+        for row in res:
+            name = row.label if row.label is not None else "<unnamed>"
+            url = row.schematic
 
-        cli.info("%s:" % name)
-        cli.info("  URL: %s" % url)
-    cli.info()
+            cli.info("%s:" % name)
+            cli.info("  URL: %s" % url)
+        cli.info()
 
-    res = cli.sparql(g, """
+        res = cli.sparql(g, """
 SELECT
     ?dk_part ?dk_part_number ?label
 WHERE {
@@ -33,7 +32,9 @@ WHERE {
     }
 }
 ORDER BY ?dk_part_number
-    """)
-    cli.info("Found %d Digi-Key parts:" % len(res))
-    for row in res:
-        cli.info("  %-30s: %s" % (row.dk_part_number, row.label))
+""")
+        cli.info("Found %d Digi-Key parts:" % len(res))
+        for row in res:
+            cli.info("  %-30s: %s" % (row.dk_part_number, row.label))
+
+    cli.with_database(db_path, db_stats)