大作业-修改了主页

Project/func/get_db.py

@@ -0,0 +1,7 @@
+import pymysql
+
+def get_db():
+    return pymysql.connect(
+        host='localhost', user='kejingfan', 
+        password='KJF2811879', database='TESTDB'
+    )

Project/func/index.py

@@ -0,0 +1,4 @@
+from flask import render_template
+
+def index():
+    return render_template("index.html")

Project/func/modify.py

@@ -0,0 +1,79 @@
+from flask import render_template, request, flash, redirect, url_for
+from typing import Dict
+
+from .get_db import get_db
+from .verify_user import verify_user
+
+
+class ModifyInfo:
+    def __init__(self, form:Dict[str, str]):
+        self.id = form['cardCode']
+        modifyType = form['modifyType']
+        self.new_password = form['encryptedNewPassword']
+        self.phone_number = form['mobileNo']
+        modifyType2command = {
+            '1':'delete account',
+            '2':'modify Password',
+            '3':'modify Phone_Number'
+        }
+        self.sql_dict = {
+            'delete account': 'DELETE FROM passengers WHERE ID = %s;',
+            'modify Password': 'UPDATE passengers SET `Password` = %s WHERE ID = %s;',
+            'modify Phone_Number': 'UPDATE passengers SET Phone_number = %s WHERE ID = %s;'
+        }
+        self.sql_args_dict = {
+            'delete account': (self.id,),
+            'modify Password': (self.new_password, self.id),
+            'modify Phone_Number': (self.phone_number, self.id)
+        }
+        self.ok_message_dict = {
+            'delete account': "删除账户成功",
+            'modify Password': "修改密码成功",
+            'modify Phone_Number': "修改手机号成功"
+        }
+        self.fail_message_dict = {
+            'delete account': "数据库异常，删除账户失败",
+            'modify Password': "数据库异常，修改密码失败",
+            'modify Phone_Number': "数据库异常，修改手机号失败"
+        }
+        self.command = modifyType2command[modifyType]
+    def get_sql(self):
+        return self.sql_dict[self.command]
+    def get_args(self):
+        return self.sql_args_dict[self.command]
+    def get_ok_message(self):
+        return self.ok_message_dict[self.command]
+    def get_fail_message(self):
+        return self.fail_message_dict[self.command]
+
+def modify():
+    if request.method == 'GET':
+        return render_template('modify.html')
+    
+    if request.method == 'POST':
+        id = request.form['cardCode']
+        password = request.form['encryptedPassword']
+        db = get_db()
+        cursor = db.cursor()
+
+        verify_info = verify_user(cursor, id, password)
+        if (verify_info == "NO_USER"):
+            flash("您未注册过，无法修改账号")
+            db.close()
+            return redirect(url_for('signup'))
+        elif (verify_info == "WRONG_PASSWORD"):
+            flash("密码错误")
+            db.close()
+            return redirect(url_for('modify'))
+
+        modifyInfo = ModifyInfo(request.form)
+        try:
+            cursor.execute(modifyInfo.get_sql(), modifyInfo.get_args())
+            db.commit()
+            flash(modifyInfo.get_ok_message())
+        except Exception as e:
+            db.rollback()
+            print(e)
+            flash(modifyInfo.get_fail_message())
+        db.close()
+        return redirect(url_for('index'))

Project/func/signup.py

@@ -0,0 +1,48 @@
+from flask import render_template, request, flash, redirect, url_for
+from .get_db import get_db
+
+def signup():
+    if request.method == 'GET':
+        return render_template('signup.html')
+    
+    if request.method == 'POST':
+        id = request.form['cardCode']
+        name = request.form['name']
+        phone_number = request.form['mobileNo']
+        password = request.form['encryptedPassword']
+
+        db = get_db()
+        cursor = db.cursor()
+
+        # 检查已有用户
+        sql = """
+        SELECT COUNT(*) FROM Users \
+        WHERE ID = %s;
+        """
+        try:
+            cursor.execute(sql, (id,))
+            id_exist = cursor.fetchall()[0][0]
+        except Exception as e:
+            flash("数据库异常，查询失败")
+            print(e)
+            return redirect(url_for('signup'))
+        if (id_exist != 0):
+            flash("您已注册过，请勿重复注册")
+            db.close()
+            return redirect(url_for('index'))
+
+        # 插入
+        sql = '''
+        INSERT INTO passengers (ID, `Name`, Phone_number, `Password`) \
+        VALUES (%s, %s, %s, %s); \
+        '''
+        try:
+            cursor.execute(sql, (id, name, phone_number, password))
+            db.commit()
+            flash("注册成功")
+        except Exception as e:
+            db.rollback()
+            print(e)
+            flash("数据库异常，注册失败")
+        db.close()
+        return redirect(url_for('index'))

Project/func/verify_user.py

@@ -0,0 +1,35 @@
+from flask import flash, redirect, url_for
+from pymysql.cursors import Cursor
+
+def verify_user(cursor:Cursor, id:str, password:str) -> str:
+    # 检查已有用户
+        sql = """
+        SELECT COUNT(*) FROM passengers \
+        WHERE ID = %s;
+        """
+        try:
+            cursor.execute(sql, (id,))
+            id_exist = cursor.fetchall()[0][0]
+        except Exception as e:
+            flash("数据库异常，查询失败")
+            print(e)
+            return redirect(url_for('signup'))
+        if (id_exist == 0):
+            return "NO_USER"
+        
+        # 检查密码
+        sql = """
+        SELECT `Password` FROM passengers \
+        WHERE ID = %s;
+        """
+        try:
+            cursor.execute(sql, (id,))
+            record_password = cursor.fetchall()[0][0]
+        except Exception as e:
+            flash("数据库异常，查询失败")
+            print(e)
+            return redirect(url_for('modify'))
+        if (record_password != password):
+            return "WRONG_PASSWORD"
+        
+        return "USER_VERIFIED"