fresh start

Pixiv/downloader.py

@@ -0,0 +1,37 @@
+from __future__ import annotations
+import traceback
+from typing import TYPE_CHECKING
+
+import nextcord
+
+from Discord import discordHelper
+from config import Global_Config
+if TYPE_CHECKING:
+    from runtimeBotData import RuntimeBotData
+
+async def download_loop(botData: RuntimeBotData):
+    try:
+        db = botData.db
+        guild = botData.client.guilds[0]
+
+        new_posts = await botData.pixivApi.get_new_posts(botData)
+        new_posts.sort(key= lambda x: x.id)
+        
+        for post in new_posts:
+            media = (await botData.pixivApi.download_illust(post, Global_Config["pixiv_download_path"]))[:4]
+            if post.x_restrict == 0:
+                channel = nextcord.utils.get(guild.channels, name="pixiv-sfw-feed")
+            elif post.x_restrict == 1:
+                channel = nextcord.utils.get(guild.channels, name="pixiv-r18-feed")
+            else:
+                channel = nextcord.utils.get(guild.channels, name="pixiv-r18g-feed")
+            await discordHelper.send_pixiv_post(post, media, channel)
+            
+            if db.pixiv_get_user(post.user.id) == None:
+                db.pixiv_insert_user(post.user)
+            db.pixiv_insert_post(post)
+
+    except Exception as ex:
+        print(ex)
+        await discordHelper.send_error(traceback.format_exc()[0:256], botData)
+    print("Pixiv done")