大作业-修改了主页

2024-06-12 00:02:55 +08:00
parent b59880f301
commit e48e752b91
20 changed files with 415 additions and 190 deletions
--- a/Project/func/signup.py
+++ b/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'))