From efd76ee2285301e519d49cfaf0a30bcbd2785b9c Mon Sep 17 00:00:00 2001
From: Andrei Stoica <andrei.stoica.365@gmail.com>
Date: Wed, 3 Aug 2022 17:02:49 -0400
Subject: [PATCH] refactored table creation into seperate function

---
 .gitignore |  3 ++-
 src/db.py  | 16 ++++++++++++++--
 2 files changed, 16 insertions(+), 3 deletions(-)

diff --git a/.gitignore b/.gitignore
index deb14e9..64b7803 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,4 +1,5 @@
 data/
 *__pycache__
 *env
-*.code-workspace
\ No newline at end of file
+*.code-workspace
+sandbox/
\ No newline at end of file
diff --git a/src/db.py b/src/db.py
index df9b4f3..81c90b1 100644
--- a/src/db.py
+++ b/src/db.py
@@ -52,6 +52,14 @@ class RecipeIngredientParts(Base):
     ingredient = Column(String)
     supplement = Column(String)
 
+class IngredientConnection(Base):
+    __tablename__ = 'IngredientConnection'
+    
+    ingredient_a = Column(String, primary_key = True)
+    ingredient_b = Column(String, primary_key = True)
+    recipe_count = Column(Integer)
+    UniqueConstraint(ingredient_a, ingredient_b)
+
     
 
 def get_engine(use_dotenv = True, **kargs):
@@ -70,7 +78,11 @@ def get_engine(use_dotenv = True, **kargs):
     return create_engine(eng_url)
 
 
-if __name__ == "__main__":
-    eng = get_engine()
+def create_tables(eng):
     logging.info(f"Createing DB Tables: {eng.url}")
     Base.metadata.create_all(eng, checkfirst=True)
+
+
+if __name__ == "__main__":
+    eng = get_engine()
+    create_tables(eng)