1 files changed, 69 insertions, 33 deletions
diff --git a/ATRI/plugins/UploadSqlite.py b/ATRI/plugins/UploadSqlite.py
index ffd2ecc..1e511ba 100644
--- a/ATRI/plugins/UploadSqlite.py
+++ b/ATRI/plugins/UploadSqlite.py
@@ -8,11 +8,12 @@ import nonebot
 from pathlib import Path
 from nonebot import on_command, CommandSession
 
-from ATRI.modules import response # type: ignore
+import config
+from ATRI.modules.response import request_api
 
 
 bot = nonebot.get_bot()
-master = bot.config.SUPERUSERS # type: ignore
+master = config.MASTER()
 url = f'https://api.imjad.cn/pixiv/v1/?type=illust&id='
 
 
@@ -44,7 +45,7 @@ async def _(session: CommandSession):
 
         URL = url + pid
 
-        dc = json.loads(response.request_api(URL))
+        dc = json.loads(request_api(URL))
         if not dc:
             session.finish('ATRI在尝试解析数据时出问题...等会再试试吧...')
         title = dc["response"][0]["title"]
@@ -157,39 +158,74 @@ async def _(session: CommandSession):
         await session.send(f'数据上传完成！\n耗时: {round(end - start, 3)}s')
 
 
-# @on_command('del_setu', aliases = ['删除涩图'], only_to_me = False)
-# async def _(session: CommandSession):
-#     user = session.event.user_id
-#     if user in master or user in sepi:
-#         start = time.perf_counter()
-#         msg = session.event.raw_message.split(' ', 2)
-#         print(msg)
-#         i_tpye = msg[1]
-#         pid = msg[2]
-
-#         if i_tpye == '正常':
-#             if os.path.exists(Path('.') / 'ATRI' / 'data' / 'sqlite' / 'setu' / 'normal.db'):
-#                 print('数据文件存在！')
-#             else:
-#                 session.finish('ERROR: 恁都没库删锤子')
-#             con = sqlite3.connect(Path('.') / 'ATRI' / 'data' / 'sqlite' / 'setu' / 'normal.db')
-#             cur = con.cursor()
-#             cur.execute(f'DELETE FROM COMPANY WHERE ID = {pid}')
-#             con.commit()
-#             con.close()
+@on_command('del_setu', aliases = ['删除涩图'], only_to_me = False)
+async def _(session: CommandSession):
+    user = session.event.user_id
+    if user in master or user in sepi:
+        start = time.perf_counter()
+        msg = session.event.raw_message.split(' ', 2)
+        i_tpye = msg[1]
+        pid = msg[2]
+
+        if i_tpye == '正常':
+            if os.path.exists(Path('.') / 'ATRI' / 'data' / 'sqlite' / 'setu' / 'normal.db'):
+                print('数据文件存在！')
+            else:
+                session.finish('ERROR: 恁都没库删锤子')
+            con = sqlite3.connect(Path('.') / 'ATRI' / 'data' / 'sqlite' / 'setu' / 'normal.db')
+            cur = con.cursor()
+            cur.execute(f'DELETE FROM COMPANY WHERE ID = {pid}')
+            con.commit()
+            con.close()
         
-#         elif i_tpye == '擦边球':
-#             if os.path.exists(Path('.') / 'ATRI' / 'data' / 'sqlite' / 'setu' / 'normal.db'):
-#                 print('数据文件存在！')
-#             else:
-#                 session.finish('ERROR: 恁都没库删锤子')
+        elif i_tpye == '擦边球':
+            if os.path.exists(Path('.') / 'ATRI' / 'data' / 'sqlite' / 'setu' / 'nearR18.db'):
+                print('数据文件存在！')
+            else:
+                session.finish('ERROR: 恁都没库删锤子')
+            con = sqlite3.connect(Path('.') / 'ATRI' / 'data' / 'sqlite' / 'setu' / 'nearR18.db')
+            cur = con.cursor()
+            cur.execute(f'DELETE FROM COMPANY WHERE ID = {pid}')
+            con.commit()
+            con.close()
         
 
-#         elif i_tpye == 'r18':
-#             if os.path.exists(Path('.') / 'ATRI' / 'data' / 'sqlite' / 'setu' / 'normal.db'):
-#                 print('数据文件存在！')
-#             else:
-#                 session.finish('ERROR: 恁都没库删锤子')
+        elif i_tpye == 'r18':
+            if os.path.exists(Path('.') / 'ATRI' / 'data' / 'sqlite' / 'setu' / 'r18.db'):
+                print('数据文件存在！')
+            else:
+                session.finish('ERROR: 恁都没库删锤子')
+            con = sqlite3.connect(Path('.') / 'ATRI' / 'data' / 'sqlite' / 'setu' / 'r18.db')
+            cur = con.cursor()
+            cur.execute(f'DELETE FROM COMPANY WHERE ID = {pid}')
+            con.commit()
+            con.close()
+        
+        end = time.perf_counter()
+        
+        await session.send(f'数据删除完成！\n耗时: {round(end - start, 3)}s')
+
+@on_command('del_cloudmusic', aliases = ['删除网易云'], only_to_me = False)
+async def _(session: CommandSession):
+    user = session.event.user_id
+    if user in master or user in cloudmusic:
+        start = time.perf_counter()
+        msg = session.event.raw_message.split(' ', 1)
+        msg = msg[1]
+
+        if os.path.exists('ATRI/data/sqlite/cloudmusic/cloudmusic.db'):
+            print('数据文件存在！')
+        else:
+            session.finish('ERROR: 恁都没库删锤子')
+        con = sqlite3.connect(Path('.') / 'ATRI' / 'data' / 'sqlite' / 'cloudmusic' / 'cloudmusic.db')
+        cur = con.cursor()
+        cur.execute('INSERT INTO cloudmusic(msg) VALUES (?)', msg)
+        con.commit()
+        con.close()
+
+        end = time.perf_counter()
+
+        await session.send(f'数据删除完成！\n耗时: {round(end - start, 3)}s')
 
 
 @on_command('add_check_sepi', aliases=['添加涩批', '移除涩批'], only_to_me = False)