Schema proposal for Dgraph

thoth-station · Apr 24, 2019 · 0f4fa8e · 0f4fa8e
1 parent 1166a84
commit 0f4fa8e
Show file tree

Hide file tree

Showing 14 changed files with 1,746 additions and 870 deletions.
diff --git a/Pipfile b/Pipfile
@@ -10,6 +10,7 @@ thoth-common = "*"
 amun = "*"
 pydgraph = "*"
 python-dateutil = "*"
+thoth-python = "*"
 
 [dev-packages]
 pytest = "*"

diff --git a/Pipfile.lock b/Pipfile.lock
diff --git a/create_schema.py b/create_schema.py
@@ -51,7 +51,7 @@ def create_schema(models: frozenset):
             raise ValueError(f"Unknown entity - not derived from vertex nor edge: {model}")
 
         for attribute in model.__attrs_attrs__:
-            if attribute.name in ("source", "target", "_uid"):
+            if attribute.name in ("source", "target", "uid"):
                 # Edge attributes for storing source and target vertex or uid which is automatically generated.
                 continue
 

diff --git a/docs/schema.dia b/docs/schema.dia
diff --git a/docs/schema.png b/docs/schema.png
diff --git a/requirements.txt b/requirements.txt
@@ -4,3 +4,4 @@ thoth-common
 amun
 pydgraph
 python-dateutil
+thoth-python
diff --git a/setup.py b/setup.py
@@ -60,6 +60,11 @@ def run_tests(self):
         'thoth.storages',
         'thoth.storages.graph'
     ],
+    package_data={
+        'thoth.storages': [
+            os.path.join('graph', 'schema.rdf')
+        ]
+    },
     zip_safe=False,
     install_requires=get_install_requires(),
     tests_require=get_test_requires(),

diff --git a/thoth/storages/dependency_monkey_reports.py b/thoth/storages/dependency_monkey_reports.py
@@ -25,7 +25,7 @@ class DependencyMonkeyReportsStore(ResultStorageBase):
 
     RESULT_TYPE = "dependency-monkey-reports"
 
-    def iterate_inspection_ids() -> str:
+    def iterate_inspection_ids(self) -> str:
         """Iterate over all inspection ids that were run."""
         for _, report in self.iterate_results():
             # Yield inspections.

diff --git a/thoth/storages/exceptions.py b/thoth/storages/exceptions.py
@@ -36,3 +36,11 @@ class CacheMiss(ThothStorageException):
 
 class NotConnected(ThothStorageException):
     """Raised if there was no connection established when communicating with a storage."""
+
+
+class MultipleFoundError(ThothStorageException):
+    """Raised if there are multiple entities in the graph database when querying for a single one."""
+
+
+class PythonIndexNotRegistered(ThothStorageException):
+    """Raised if an attempt to insert a Python package with an index not being registered to the system."""