From 1459f8990c07467d50fec4217e22ee3dfdb6b19a Mon Sep 17 00:00:00 2001 From: ridethepig Date: Sun, 11 Dec 2022 23:57:14 +0800 Subject: [PATCH] file op record --- src/user.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/src/user.py b/src/user.py index 275ffd6..c7927fc 100644 --- a/src/user.py +++ b/src/user.py @@ -318,6 +318,10 @@ def doc_upload(id): (filename, filepath, filesz, filetype, id, g.user['user_id']) ) db.commit() + cur.execute("insert into record (`record_type`,`doc_url`,`user_id`) " + "values('upload', %s, %s)", + (filepath, g.user['user_id'])) + db.commit() except pymysql.Error as _e: if 'ck_usedspace' in str(_e): error = "空间不够用了，请向管理员申请扩容" @@ -376,6 +380,10 @@ def doc_delete(docid): try: cur.execute("delete from document where `doc_id`=%s", (docid)) db.commit() + cur.execute("insert into record (`record_type`,`doc_url`,`user_id`) " + "values('remove', %s, %s)", + (document['doc_url'], g.user['user_id'])) + db.commit() except pymysql.Error as _e: error = "严重错误：%s。\n请联系管理员解决问题" % (_e) cur.close()