[Update] new function

1 files changed, 128 insertions, 0 deletions

diff --git a/ATRI/plugins/upload_setu.py b/ATRI/plugins/upload_setu.py
new file mode 100644
index 0000000..78158c3
--- /dev/null
+++ b/ATRI/plugins/upload_setu.py
@@ -0,0 +1,128 @@
+# -*- coding:utf-8 -*-
+import time
+import json
+import sqlite3
+import nonebot
+
+from nonebot import on_command, CommandSession
+from nonebot import session
+
+from ATRI.modules import response # type: ignore
+
+
+bot = nonebot.get_bot()
+master = bot.config.SUPERUSERS # type: ignore
+path_setu = f'ATRI\\splite\\setu\\'
+path_cloudmusic = f'ATRI\\splite\\cloudmusic\\'
+url = f'https://api.imjad.cn/pixiv/v1/?type=illust&id='
+
+
+@on_command('upload_setu', aliases = ['上传涩图'], only_to_me = False)
+async def _(session: CommandSession):
+    if session.event.user_id in master:
+        msg = session.event.raw_message.split(' ', 2)
+        print(msg)
+        i_tpye = msg[1]
+        pid = msg[2]
+
+        URL = url + pid
+
+        dc = json.loads(response.request_api(URL))
+        if not dc:
+            session.finish('ATRI在尝试解析数据时出问题...等会再试试吧...')
+        title = dc["response"][0]["title"]
+        tags = dc["response"][0]["tags"]
+        account = dc["response"][0]["user"]["account"]
+        name = dc["response"][0]["user"]["name"]
+        u_id = dc["response"][0]["user"]["id"]
+        user_link = f'https://www.pixiv.net/users/' + f'{u_id}'
+        img = f'https://pixiv.cat/{pid}.jpg'
+
+        if i_tpye == '正常':
+            con = sqlite3.connect(path_setu + 'normal.db')
+            cur = con.cursor()
+            
+            cur.execute('INSERT INTO normal VALUES ("%s", "%s", "%s", "%s", "%s", "%s", "%s", "%s")'%(pid, title, tags, account, name, u_id, user_link, img))
+            con.commit()
+
+            print('success!')
+            time.sleep(0.5)
+            con.close()
+
+        elif i_tpye == '擦边球':
+            con = sqlite3.connect(path_setu + 'nearR18.db')
+            cur = con.cursor()
+            
+            cur.execute('INSERT INTO nearR18 VALUES ("%s", "%s", "%s", "%s", "%s", "%s", "%s", "%s")'%(pid, title, tags, account, name, u_id, user_link, img))
+            con.commit()
+
+            print('success!')
+            time.sleep(0.5)
+            con.close()
+
+        elif i_tpye == 'r18':
+            con = sqlite3.connect(path_setu + 'r18.db')
+            cur = con.cursor()
+            
+            cur.execute('INSERT INTO r18 VALUES ("%s", "%s", "%s", "%s", "%s", "%s", "%s", "%s")'%(pid, title, tags, account, name, u_id, user_link, img))
+            con.commit()
+
+            print('success!')
+            time.sleep(0.5)
+            con.close()
+        
+        await session.send('数据上传完成！')
+
+@on_command('upload_cloudmusic', aliases = ['上传网抑语', '网抑云', '网易云'], only_to_me = False)
+async def _(session: CommandSession):
+    if session.event.user_id in master:
+        msg = session.event.raw_message.split(' ', 1)
+        msg = msg[1]
+
+        con = sqlite3.connect(path_cloudmusic + 'cloudmusic.db')
+        cur = con.cursor()
+
+        cur.execute('INSERT INTO cloudmusic VALUES ("%s")'%(msg))
+        con.commit()
+
+        print('success!')
+        time.sleep(0.5)
+        con.close()
+
+        await session.send('数据上传完成！')
+
+
+@on_command('data_list', aliases = ['查看上传数据', '数据总量'], only_to_me = False)
+async def _(session: CommandSession):
+    con = sqlite3.connect(path_setu + 'normal.db') # setu-normal
+    cur = con.cursor()
+    cur.execute("select * from normal")
+    data_normal = len(cur.fetchall())
+    con.close()
+
+    con = sqlite3.connect(path_setu + 'nearR18.db') # setu-nearR18
+    cur = con.cursor()
+    cur.execute("select * from nearR18")
+    data_nearR18 = len(cur.fetchall())
+    con.close()
+
+    con = sqlite3.connect(path_setu + 'r18.db') # setu-r18
+    cur = con.cursor()
+    cur.execute("select * from r18")
+    data_r18 = len(cur.fetchall())
+    con.close()
+
+    con = sqlite3.connect(path_cloudmusic + 'cloudmusic.db') # cloudmusic
+    cur = con.cursor()
+    cur.execute("select * from cloudmusic")
+    data_cloudmusic = len(cur.fetchall())
+    con.close()
+
+    await session.send(
+        f"""目前ATRI数据库收录了：
+    涩图：
+        normal: {data_normal}
+        nearR18: {data_nearR18}
+        r18：{data_r18}
+    网抑云语录：{data_cloudmusic}"""
+    )
\ No newline at end of file