summaryrefslogtreecommitdiff
path: root/ATRI/plugins/checkData.py
blob: 0fd7217fd2136da260a70c0c91deaa6cd3cbc102 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
import json
import sqlite3
from pathlib import Path
from nonebot import on_command, CommandSession



@on_command('data_list', aliases = ['数据总量'], only_to_me = False)
async def _(session: CommandSession):
    user = session.event.user_id
    with open('ATRI/plugins/noobList/noobList.json', 'r') as f:
        data = json.load(f)

    if str(user) in data.keys():
        pass
    else:
        con = sqlite3.connect(Path('.') / 'ATRI' / 'data' / 'sqlite' / 'setu' / 'normal.db') # setu-normal
        cur = con.cursor()
        cur.execute("select * from normal")
        data_normal = len(cur.fetchall())
        con.close()

        con = sqlite3.connect(Path('.') / 'ATRI' / 'data' / 'sqlite' / 'setu' / 'nearR18.db') # setu-nearR18
        cur = con.cursor()
        cur.execute("select * from nearR18")
        data_nearR18 = len(cur.fetchall())
        con.close()

        con = sqlite3.connect(Path('.') / 'ATRI' / 'data' / 'sqlite' / 'setu' / 'r18.db') # setu-r18
        cur = con.cursor()
        cur.execute("select * from r18")
        data_r18 = len(cur.fetchall())
        con.close()

        con = sqlite3.connect(Path('.') / 'ATRI' / 'data' / 'sqlite' / 'cloudmusic' / 'cloudmusic.db') # cloudmusic
        cur = con.cursor()
        cur.execute("select * from cloudmusic")
        data_cloudmusic = len(cur.fetchall())
        con.close()

        with open('ATRI/plugins/wordcloud/wordcloud.json', 'r') as f:
            data = json.load(f)
        data_repo = len(data)

        await session.send(
            f"""目前螃蟹™数据库收录了:
涩图:
    normal: {data_normal}
    nearR18: {data_nearR18}
    r18:{data_r18}
网抑云语录:{data_cloudmusic}
词汇量:{data_repo}"""
        )