From 165ca069b017bf2f483de933ea5bd106c17d66b8 Mon Sep 17 00:00:00 2001
From: Kyomotoi <0w0@imki.moe>
Date: Sun, 18 Sep 2022 15:27:03 +0800
Subject: =?UTF-8?q?=E2=9C=A8=20=E6=96=B0=E5=A2=9E=E6=8F=92=E4=BB=B6:=20Rss?=
 =?UTF-8?q?=E8=AE=A2=E9=98=85?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ATRI/database/db.py     | 14 ++++++++++++++
 ATRI/database/models.py | 25 +++++++++++++++++++++++++
 2 files changed, 39 insertions(+)

(limited to 'ATRI/database')

diff --git a/ATRI/database/db.py b/ATRI/database/db.py
index a5a04c8..0917440 100644
--- a/ATRI/database/db.py
+++ b/ATRI/database/db.py
@@ -35,6 +35,18 @@ async def run():
                         "file_path": f"{DB_DIR}/thesaurusauditlist.sqlite3"
                     },
                 },
+                "rrs": {
+                    "engine": "tortoise.backends.sqlite",
+                    "credentials": {
+                        "file_path": f"{DB_DIR}/rssrsshubsubscription.sqlite3"
+                    },
+                },
+                "rms": {
+                    "engine": "tortoise.backends.sqlite",
+                    "credentials": {
+                        "file_path": f"{DB_DIR}/rssmikananisubscription.sqlite3"
+                    },
+                },
             },
             "apps": {
                 "bilibili": {
@@ -47,6 +59,8 @@ async def run():
                 },
                 "ts": {"models": [locals()["models"]], "default_connection": "ts"},
                 "tal": {"models": [locals()["models"]], "default_connection": "tal"},
+                "rrs": {"models": [locals()["models"]], "default_connection": "rrs"},
+                "rms": {"models": [locals()["models"]], "default_connection": "rms"},
             },
         }
     )
diff --git a/ATRI/database/models.py b/ATRI/database/models.py
index d326151..9b009e4 100644
--- a/ATRI/database/models.py
+++ b/ATRI/database/models.py
@@ -57,3 +57,28 @@ class ThesaurusAuditList(Model):
 
     class Meta:
         app = "tal"
+
+
+class RssRsshubSubcription(Model):
+    _id = fields.TextField()
+    group_id = fields.IntField(null=True)
+    title = fields.TextField(null=True)
+    raw_link = fields.TextField(null=True)
+    rss_link = fields.TextField(null=True)
+    discription = fields.TextField(null=True)
+    update_time = fields.DatetimeField(default=datetime.fromordinal(1))
+
+    class Meta:
+        app = "rrs"
+
+
+class RssMikananiSubcription(Model):
+    _id = fields.TextField()
+    group_id = fields.IntField(null=True)
+    title = fields.TextField(null=True)
+    rss_link = fields.TextField(null=True)
+    discription = fields.TextField(null=True)
+    update_time = fields.DatetimeField(default=datetime.fromordinal(1))
+
+    class Meta:
+        app = "rms"
-- 
cgit v1.2.3