数据库接口重构

db/file.py

@@ -0,0 +1,59 @@
+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