Merge pull request #5 from MitchellWeg/add_relationships

Add relationships

examples/demo_monetdb.py

@@ -1,6 +1,5 @@
 import pymonetdb
 
-import pyschemer
 from pyschemer.schemer import Database
 
 class Monet(Database):
@@ -32,7 +31,10 @@ def describe_tables(self):
         self.tables = all_tables
 
     def get_relationships(self, q=None):
-        return []
+        q = """
+        SELECT fk, fk_t, fk_c, pk_t, pk_c FROM describe_foreign_keys;
+        """
+        return super().get_relationships(q)
 
 def main():
     conn = pymonetdb.connect(

pyschemer/relationships.py

@@ -14,5 +14,12 @@ def __init__(
         self.referenced_table_name = ref_table
         self.referenced_column_name = ref_column
 
+    def __repr__(self) -> str:
+        return "".join([
+            f"{self.table_name}:{self.column_name}",
+             " 1--* ",
+             f"{self.referenced_table_name}:{self.referenced_column_name}"
+        ])
+
     def to_json(self):
         return self.__repr__()

pyschemer/schemer.py

@@ -30,7 +30,7 @@ def get_dot(self):
 
     def draw(self, dot_code, filename='out'):
         G = pgv.AGraph(string=dot_code)
-        schema_drawing = G.draw(path=None, format='jpeg', prog='dot')
+        schema_drawing = G.draw(path=None, format='jpeg', prog='circo')
 
         with open(f'{filename}.jpg', 'wb') as outfile:
             outfile.write(schema_drawing)

setup.py

@@ -5,7 +5,7 @@
 
 setup(
     name='pyschemer',
-    version='1.0.2',    
+    version='1.1.0',    
     description='A database visualization tool',
     url='https://github.com/MitchellWeg/schemer',
     author='Mitchell Weggemans',
@@ -20,7 +20,7 @@
                       ],
 
     classifiers=[
-        'Development Status :: 1 - Planning',
+        'Development Status :: 4 - Beta',
         'Intended Audience :: Science/Research',
         'License :: OSI Approved :: BSD License',  
         'Operating System :: POSIX :: Linux',