1 files changed, 44 insertions, 0 deletions
diff --git a/newspipe/models/tag.py b/newspipe/models/tag.py
new file mode 100644
index 00000000..b853c344
--- /dev/null
+++ b/newspipe/models/tag.py
@@ -0,0 +1,44 @@
+#! /usr/bin/env python
+# -*- coding: utf-8 -*-
+
+from newspipe.bootstrap import db
+
+
+class ArticleTag(db.Model):
+    text = db.Column(db.String, primary_key=True, unique=False)
+
+    # foreign keys
+    article_id = db.Column(
+        db.Integer, db.ForeignKey("article.id", ondelete="CASCADE"), primary_key=True
+    )
+
+    # relationships
+    article = db.relationship(
+        "Article", back_populates="tag_objs", foreign_keys=[article_id]
+    )
+
+    def __init__(self, text):
+        self.text = text
+
+
+class BookmarkTag(db.Model):
+    id = db.Column(db.Integer, primary_key=True)
+    text = db.Column(db.String, unique=False)
+
+    # foreign keys
+    user_id = db.Column(db.Integer, db.ForeignKey("user.id", ondelete="CASCADE"))
+    bookmark_id = db.Column(
+        db.Integer, db.ForeignKey("bookmark.id", ondelete="CASCADE")
+    )
+
+    # relationships
+    bookmark = db.relationship(
+        "Bookmark",
+        back_populates="tags",
+        cascade="all,delete",
+        foreign_keys=[bookmark_id],
+    )
+
+    # def __init__(self, text, user_id):
+    #     self.text = text
+    #     self.user_id = user_id