Use sqlalchemy relationship

category is parent and label is child. easier to add categories and its labels

Use sqlalchemy relationship
category is parent and label is child. easier to add categories and its labels
hitier
1 parent d6b9daca
Showing 2 changed files with 6 additions and 3 deletions Show diff stats
app/commands/commands.py
app/models.py
@@ -200,10 +200,10 @@ def feed_from_irap(csv_file_name):
         print("Adding category ", c)
         n_c = Category(name=c)
         db.session.add(n_c)
-        db.session.commit()
+        # db.session.commit()
         for label in l:
-            print("Adding label {} for id {}".format(label, n_c.id))
-            n_l = CategoryLabel(name=label, category_id=n_c.id)
+            # print("Adding label {} for id {}".format(label, n_c.id))
+            n_l = CategoryLabel(name=label, category=n_c)
             db.session.add(n_l)
     db.session.commit()
  
 from flask_sqlalchemy import SQLAlchemy
+from sqlalchemy.orm import relationship
  
 db = SQLAlchemy()
  
@@ -6,12 +7,14 @@ db = SQLAlchemy()
 class Category(db.Model):
     id = db.Column(db.Integer, primary_key=True)
     name = db.Column(db.String(32))
+    labels = relationship("CategoryLabel", back_populates='category')
  
  
 class CategoryLabel(db.Model):
     id = db.Column(db.Integer, primary_key=True)
     name = db.Column(db.String(32))
     category_id = db.Column(db.Integer, db.ForeignKey('category.id'))
+    category = relationship("Category", back_populates='labels')
  
  
 class AgentBap(db.Model):