feat:项目完成

db/fz_data.py

@@ -0,0 +1,83 @@
+import pymysql, json
+import app_conf
+
+conf = app_conf.conf()
+
+
+def getConn():
+    return pymysql.Connect(
+        host=conf.get("database", "host"),
+        user=conf.get("database", "user"),
+        port=conf.getint("database", "port"),
+        database=conf.get("database", "database"),
+        password=conf.get("database", "password"),
+    )
+
+
+def saveFzData(data) -> bool:
+    "将数据存入数据库"
+    sql = """
+    INSERT INTO table_zhapian 
+    VALUES (
+    %(uuid)s,%(callednm)s,%(zptype)s,%(zpfeatures)s,%(early_warning_time)s,
+    %(warninglevel)s,%(zpcallernm)s,%(talktime)s,%(auth_code)s,%(student_number)s,
+    %(transfer_amout)s,%(url_or_app_name)s,%(zp_account)s
+    );
+    """
+    conn = getConn()
+    cursor = conn.cursor()
+    try:
+        for message in data:
+            message.setdefault("uuid",None)
+            message.setdefault("callednm",None)
+            message.setdefault("zptype",None)
+            message.setdefault("zpfeatures",None)
+            message.setdefault("early_warning_time",None)
+            message.setdefault("warninglevel",None)
+            message.setdefault("zpcallernm",None)
+            message.setdefault("talktime",None)
+            message.setdefault("auth_code",None)
+            message.setdefault("student_number",None)
+            message.setdefault("transfer_amout",None)
+            message.setdefault("url_or_app_name",None)
+            message.setdefault("zp_account",None)
+            cursor.execute(sql,message)
+        conn.commit()
+    except:
+        return False
+    conn.close()
+    return True
+
+def LoadFzData():
+    "从数据库读取数据"
+    sql = """
+    SELECT * FROM `table_zhapian` ORDER BY `early_warning_time` DESC LIMIT 0,1000;
+    """
+    conn = getConn()
+    cursor = conn.cursor()
+    try:
+        cursor.execute(sql)
+        results = cursor.fetchall()
+    except:
+        conn.close()
+        return None
+    conn.close()
+
+    return results
+
+def total():
+    "返回数据库总条目数量"
+    sql = """
+    SELECT COUNT(*) AS total_count FROM table_zhapian;
+    """
+    conn = getConn()
+    cursor = conn.cursor()
+    try:
+        cursor.execute(sql)
+        results = cursor.fetchone()
+    except:
+        conn.close()
+        return None
+    conn.close()
+
+    return results[0]

db/user.py

@@ -0,0 +1,116 @@
+import hashlib
+from db.fz_data import getConn
+
+
+def Check(username: str, password: str) -> bool:
+    password_hash = hashlib.md5(password.encode()).hexdigest()
+    "检查用户密码"
+    sql = """
+    SELECT * FROM `table_user` WHERE username = %s and password = %s;
+    """
+    conn = getConn()
+    cursor = conn.cursor()
+    try:
+        cursor.execute(sql, (username, password_hash))
+        results = cursor.fetchone()
+        conn.close()
+    except:
+        conn.close()
+        return False
+    if results != None:
+        return True
+    return False
+
+
+def UpdatePassword(username: str, password: str) -> bool:
+    password_hash = hashlib.md5(password.encode()).hexdigest()
+    "更新用户密码"
+    sql = """
+    UPDATE `table_user` SET `password` = %s WHERE `username` = %s;
+    """
+    conn = getConn()
+    cursor = conn.cursor()
+    try:
+        c_row = cursor.execute(sql, (password_hash, username))
+        conn.commit()
+        conn.close()
+    except:
+        conn.close()
+        return False
+    if c_row > 0:
+        return True
+    return False
+
+
+def New(username: str, password: str) -> bool:
+    password_hash = hashlib.md5(password.encode()).hexdigest()
+    "更新用户密码"
+    sql = """
+    INSERT INTO `table_user` (`username`, `password`) VALUES (%s, %s);
+    """
+    conn = getConn()
+    cursor = conn.cursor()
+    try:
+        c_row = cursor.execute(sql, (username,password_hash))
+        conn.commit()
+        conn.close()
+    except:
+        conn.close()
+        return False
+    if c_row > 0:
+        return True
+    return False
+
+
+def total() -> int:
+    "返回数据库总条目数量"
+    sql = """
+    SELECT COUNT(*) AS total_count FROM table_user;
+    """
+    conn = getConn()
+    cursor = conn.cursor()
+    try:
+        cursor.execute(sql)
+        results = cursor.fetchone()
+    except:
+        conn.close()
+        return None
+    conn.close()
+
+    return results[0]
+
+
+def list():
+    "返回数据库所有条目"
+    sql = """
+    SELECT * FROM table_user;
+    """
+    conn = getConn()
+    cursor = conn.cursor()
+    try:
+        cursor.execute(sql)
+        results = cursor.fetchall()
+    except:
+        conn.close()
+        return None
+    conn.close()
+    return results
+
+
+def Del(username: str) -> bool:
+    "删除用户"
+    sql = """
+    DELETE FROM `table_user` WHERE `username` = %s;
+    """
+    conn = getConn()
+    cursor = conn.cursor()
+    try:
+        c_row = cursor.execute(sql, (username))
+        conn.commit()
+        conn.close()
+    except:
+        conn.close()
+        return False
+    if c_row > 0:
+        return True
+    return False