From ebcb6cb4d5043759e634dcc1bd1a28573d650813 Mon Sep 17 00:00:00 2001
From: Kyomotoi <1172294279@qq.com>
Date: Sun, 25 Oct 2020 16:33:33 +0800
Subject: [Update]
---
ATRI/plugins/plugin_sqlite/__init__.py | 110 +++++++++++++++++++++++++++++++++
1 file changed, 110 insertions(+)
create mode 100644 ATRI/plugins/plugin_sqlite/__init__.py
(limited to 'ATRI/plugins/plugin_sqlite')
diff --git a/ATRI/plugins/plugin_sqlite/__init__.py b/ATRI/plugins/plugin_sqlite/__init__.py
new file mode 100644
index 0000000..42ea60c
--- /dev/null
+++ b/ATRI/plugins/plugin_sqlite/__init__.py
@@ -0,0 +1,110 @@
+#!/usr/bin/env python3
+# -*- encoding: utf-8 -*-
+'''
+@File : __init__.py
+@Time : 2020/10/25 15:01:29
+@Author : Kyomotoi
+@Contact : kyomotoiowo@gmail.com
+@Github : https://github.com/Kyomotoi
+@License : Copyright © 2018-2020 Kyomotoi, All Rights Reserved.
+'''
+__author__ = 'kyomotoi'
+
+import os
+import json
+import sqlite3
+from pathlib import Path
+from nonebot.plugin import on_command
+from nonebot.permission import SUPERUSER
+from nonebot.adapters.cqhttp import Bot, Event
+
+from utils.utils_error import errorRepo
+from utils.utils_request import aio_get_bytes
+
+
+UploadSetu = on_command('setu', permission=SUPERUSER)
+
+@UploadSetu.handle() # type: ignore
+async def _(bot: Bot, event: Event, state: dict) -> None:
+ msg = str(event.message).strip().split(' ')
+
+ s_type = msg[0]
+ pid = msg[1]
+
+ URL = f'https://api.imjad.cn/pixiv/v1/?type=illust&id={pid}'
+ info = {}
+
+ try:
+ info = json.loads(await aio_get_bytes(URL))
+ except:
+ await UploadSetu.finish(errorRepo("网络请求出错"))
+
+ info = info["response"][0]
+ title = info["title"]
+ tags = info["tags"]
+ account = info["account"]
+ name = info["user"]["name"]
+ u_id = info["user"]["id"]
+ user_link = f'https://www.pixiv.net/users/' + f'{u_id}'
+ img = f'https://pixiv.cat/{pid}.jpg'
+
+ data_setu = (f'{pid}', f'{title}', f'{tags}', f'{account}', f'{name}', f'{u_id}', f'{user_link}', f'{img}')
+
+ if s_type == "normal":
+ if os.path.exists('ATRI/data/data_Sqlite/setu/normal.db'):
+ print('数据文件存在！')
+ else:
+ await bot.send(event, '数据库不存在，将开始构建...')
+ await bot.send(event, '开始构建数据库！')
+ con = sqlite3.connect(Path('.') / 'ATRI' / 'data' / 'data_Sqlite' / 'setu' / 'normal.db')
+ cur = con.cursor()
+ cur.execute('CREATE TABLE normal(pid PID, title TITLE, tags TAGS, account ACCOUNT, name NAME, u_id UID, user_link USERLINK, img IMG, UNIQUE(pid, title, tags, account, name, u_id, user_link, img))')
+ con.commit()
+ cur.close()
+ await bot.send(event, '完成')
+
+ con = sqlite3.connect(Path('.') / 'ATRI' / 'data' / 'data_Sqlite' / 'setu' / 'normal.db')
+ cur = con.cursor()
+ cur.execute('INSERT INTO normal(pid, title, tags, account, name, u_id, user_link, img) VALUES(?, ?, ?, ?, ?, ?, ?, ?)', data_setu)
+ con.commit()
+ con.close()
+
+ elif s_type == "nearr18":
+ if os.path.exists('ATRI/data/data_Sqlite/setu/nearR18.db'):
+ print('数据文件存在！')
+ else:
+ await bot.send(event, '数据库不存在，将开始构建...')
+ await bot.send(event, '开始构建数据库！')
+ con = sqlite3.connect(Path('.') / 'ATRI' / 'data' / 'data_Sqlite' / 'setu' / 'nearR18.db')
+ cur = con.cursor()
+ cur.execute('CREATE TABLE nearR18(pid PID, title TITLE, tags TAGS, account ACCOUNT, name NAME, u_id UID, user_link USERLINK, img IMG, UNIQUE(pid, title, tags, account, name, u_id, user_link, img))')
+ con.commit()
+ cur.close()
+ await bot.send(event, '完成')
+
+ con = sqlite3.connect(Path('.') / 'ATRI' / 'data' / 'data_Sqlite' / 'setu' / 'nearR18.db')
+ cur = con.cursor()
+ cur.execute('INSERT INTO nearR18(pid, title, tags, account, name, u_id, user_link, img) VALUES(?, ?, ?, ?, ?, ?, ?, ?)', data_setu)
+ con.commit()
+ con.close()
+
+ elif s_type == "r18":
+ if os.path.exists('ATRI/data/data_Sqlite/setu/r18.db'):
+ print('数据文件存在！')
+ else:
+ await bot.send(event, '数据库不存在，将开始构建...')
+ await bot.send(event, '开始构建数据库！')
+ con = sqlite3.connect(Path('.') / 'ATRI' / 'data' / 'data_Sqlite' / 'setu' / 'r18.db')
+ cur = con.cursor()
+ cur.execute('CREATE TABLE r18(pid PID, title TITLE, tags TAGS, account ACCOUNT, name NAME, u_id UID, user_link USERLINK, img IMG, UNIQUE(pid, title, tags, account, name, u_id, user_link, img))')
+ con.commit()
+ cur.close()
+ await bot.send(event, '完成')
+
+ con = sqlite3.connect(Path('.') / 'ATRI' / 'data' / 'data_Sqlite' / 'setu' / 'r18.db')
+ cur = con.cursor()
+ cur.execute('INSERT INTO r18(pid, title, tags, account, name, u_id, user_link, img) VALUES(?, ?, ?, ?, ?, ?, ?, ?)', data_setu)
+ con.commit()
+ con.close()
+
+ await UploadSetu.finish(f"数据上传完成~！\n涩图库[{s_type}]涩图 +1")
--
cgit v1.2.3