ComiPy/db/file.py
2024-04-15 15:19:56 +08:00

60 lines
1.2 KiB
Python

import shortuuid
import db.util as util
# 查找文件信息
def searchByid(id: str):
conn = util.getConn()
c = conn.cursor()
cursor = c.execute("SELECT * FROM Metadata WHERE id = ?", (id,))
out = []
for row in cursor:
out.append(row)
conn.close()
return out
# 查找文件信息
def searchByname(filename: str):
conn = util.getConn()
c = conn.cursor()
cursor = c.execute("SELECT * FROM Metadata WHERE filename = ?", (filename,))
out = []
for row in cursor:
out.append(row)
conn.close()
return out
# 在数据库中添加一个新的文件记录
def new(filename: str):
suuid = shortuuid.random(8)
conn = util.getConn()
c = conn.cursor()
c.execute(
"""
INSERT INTO Metadata
(id, filename)
VALUES
(?, ?, ?);
""",
(suuid, filename),
)
conn.commit()
conn.close()
return suuid
# 获取文件元数据
def getMetadata(form: int, num: int):
conn = util.getConn()
c = conn.cursor()
cursor = c.execute(
"SELECT * FROM Metadata ORDER BY num desc LIMIT ?, ?", (form, num)
)
out = []
for row in cursor:
out.append(list(row))
conn.close()
return out