1 files changed, 10 insertions, 3 deletions
diff --git a/trygvis/eda/cli/digikey_download_metadata.py b/trygvis/eda/cli/digikey_download_metadata.py
index 60880ab..59c8099 100755
--- a/trygvis/eda/cli/digikey_download_metadata.py
+++ b/trygvis/eda/cli/digikey_download_metadata.py
@@ -1,4 +1,6 @@
-from trygvis.eda import cli
+from rdflib.term import URIRef
+
+from trygvis.eda import cli, EDA
 from trygvis.eda.digikey import *
 
 
@@ -19,11 +21,15 @@ def do_categories(db: DigikeyDatabase, client: DigikeyClient, output: str):
 
 
 def do_category(client: DigikeyClient, sc: DigikeyProductCategory, output: str):
-    cli.info('Downloading categories for %s' % sc.label)
+    cli.info('Downloading attributes for %s' % sc.label)
 
     def make_graph():
         attributes = download_attribute_types_from_category(sc, client)
-        g = cli.create_graph()
+        g = cli.create_graph(digikey=True)
+
+        g.add((sc.url(), RDF.type, OWL.Ontology))
+        g.add((sc.url(), OWL.imports, URIRef(rdf.DIGIKEY)))
+
         for a in attributes:
             [g.add(node) for node in a.to_nodes()]
 
@@ -31,6 +37,7 @@ def do_category(client: DigikeyClient, sc: DigikeyProductCategory, output: str):
 
     cli.write_graph(make_graph, output, force_write=True)
 
+
 def run(args):
     db = DigikeyDatabase()
     client = DigikeyClient()