[Fix]

1 files changed, 39 insertions, 32 deletions

diff --git a/ATRI/plugins/upload_sqlite.py b/ATRI/plugins/upload_sqlite.py
index a7c5b75..b399281 100644
--- a/ATRI/plugins/upload_sqlite.py
+++ b/ATRI/plugins/upload_sqlite.py
@@ -139,35 +139,42 @@ async def _(session: CommandSession):
 
 @on_command('data_list', aliases = ['查看上传数据', '数据总量'], only_to_me = False)
 async def _(session: CommandSession):
-    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()
-
-    await session.send(
-        f"""目前螃蟹™数据库收录了：
-涩图：
-    normal: {data_normal}
-    nearR18: {data_nearR18}
-    r18：{data_r18}
-网抑云语录：{data_cloudmusic}"""
-    )
\ No newline at end of file
+    user = session.event.user_id
+    with open('ATRI/plugins/noobList/noobList.json', 'r') as f:
+        data = json.load(f)
+
+    if data[f"{user}"] == str(user):
+        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()
+
+        await session.send(
+            f"""目前螃蟹™数据库收录了：
+    涩图：
+        normal: {data_normal}
+        nearR18: {data_nearR18}
+        r18：{data_r18}
+    网抑云语录：{data_cloudmusic}"""
+        )
\ No newline at end of file