1 files changed, 6 insertions, 6 deletions
diff --git a/trygvis/eda/cli/digikey_download_for_schematic.py b/trygvis/eda/cli/digikey_download_for_schematic.py
index 99f5266..ecfbc9e 100755
--- a/trygvis/eda/cli/digikey_download_for_schematic.py
+++ b/trygvis/eda/cli/digikey_download_for_schematic.py
@@ -1,5 +1,3 @@
-from os.path import isfile
-
 from rdflib.plugins.sparql import prepareQuery
 import rdflib.term
 
@@ -20,10 +18,7 @@ initNs = {
     "kicad-type": kicad_rdf.KICAD_TYPE}
 
 
-def run(schematic_url, db_path, args):
-    cli.info("Schematic: %s" % schematic_url)
-    g = cli.open_database(db_path)
-
+def work(schematic_url, g):
     client = DigikeyClient()
     db = DigikeyDatabase()
     download_category_tree(db, client)
@@ -131,3 +126,8 @@ ORDER BY ?digikey_pn
     #         for a in attributes:
     #             [tmpG.add(node) for node in a.toNodes()]
     #         writeGraph(tmpG, 'ttl/' + filename)
+
+
+def run(schematic_url, db_path, args):
+    cli.info("Schematic: %s" % schematic_url)
+    cli.with_database(db_path, lambda g: work(schematic_url, g))