宠物的id在表格中去除,统一使用宠物名字,给出修改后的完整代码和对应的html
时间: 2024-11-27 16:13:39 浏览: 20
为了将宠物ID从表单中移除并统一使用宠物名字进行操作,我们需要对现有的Flask应用进行一些调整。以下是修改后的完整代码:
### 修改后的Python代码 (`app.py`)
```python
from flask import Flask, render_template, request, redirect, url_for, flash, session
import mysql.connector
import bcrypt
import logging
app = Flask(__name__)
app.secret_key = 'your_secret_key'
# 配置日志
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
# 数据库连接
def connect_db():
conn = mysql.connector.connect(
host='localhost',
port=3306,
user='root',
password='123456',
database='pet_weight_management'
)
return conn
# 初始化数据库
def init_db():
conn = connect_db()
cursor = conn.cursor()
cursor.execute('''
CREATE TABLE IF NOT EXISTS users (
id INT AUTO_INCREMENT PRIMARY KEY,
username VARCHAR(255) UNIQUE NOT NULL,
password VARCHAR(255) NOT NULL
)
''')
cursor.execute('''
CREATE TABLE IF NOT EXISTS pets (
id INT AUTO_INCREMENT PRIMARY KEY,
user_id INT NOT NULL,
pet_name VARCHAR(255) NOT NULL,
FOREIGN KEY (user_id) REFERENCES users (id) ON DELETE CASCADE
)
''')
cursor.execute('''
CREATE TABLE IF NOT EXISTS pet_weights (
id INT AUTO_INCREMENT PRIMARY KEY,
pet_name VARCHAR(255) NOT NULL,
weight DECIMAL(10, 2) NOT NULL,
recorded_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
FOREIGN KEY (pet_name) REFERENCES pets (pet_name) ON DELETE CASCADE
)
''')
conn.commit()
conn.close()
@app.route('/')
def index():
return render_template('index.html')
@app.route('/register', methods=['GET', 'POST'])
def register():
if request.method == 'POST':
username = request.form['username']
password = request.form['password']
conn = connect_db()
cursor = conn.cursor()
hashed_password = bcrypt.hashpw(password.encode('utf-8'), bcrypt.gensalt())
try:
cursor.execute('INSERT INTO users (username, password) VALUES (%s, %s)', (username, hashed_password))
conn.commit()
flash("注册成功!", 'success')
logging.info("用户 %s 注册成功!", username)
return redirect(url_for('login'))
except mysql.connector.IntegrityError:
flash("用户名已存在!", 'warning')
logging.warning("用户名 %s 已存在!", username)
finally:
conn.close()
return render_template('register.html')
@app.route('/login', methods=['GET', 'POST'])
def login():
if request.method == 'POST':
username = request.form['username']
password = request.form['password']
conn = connect_db()
cursor = conn.cursor()
cursor.execute('SELECT id, password FROM users WHERE username = %s', (username,))
result = cursor.fetchone()
if result and bcrypt.checkpw(password.encode('utf-8'), result[1].encode('utf-8')):
session['user_id'] = result[0]
flash("登录成功!", 'success')
logging.info("用户 %s 登录成功!", username)
return redirect(url_for('dashboard'))
else:
flash("用户名或密码错误!", 'danger')
logging.warning("用户名或密码错误!")
conn.close()
return render_template('login.html')
@app.route('/logout')
def logout():
session.pop('user_id', None)
flash("已退出登录。", 'info')
logging.info("已退出登录。")
return redirect(url_for('index'))
@app.route('/dashboard')
def dashboard():
if 'user_id' not in session:
return redirect(url_for('login'))
user_id = session['user_id']
conn = connect_db()
cursor = conn.cursor()
cursor.execute('SELECT pet_name FROM pets WHERE user_id = %s', (user_id,))
pets = cursor.fetchall()
conn.close()
return render_template('dashboard.html', pets=pets)
@app.route('/add_pet', methods=['GET', 'POST'])
def add_pet():
if 'user_id' not in session:
return redirect(url_for('login'))
if request.method == 'POST':
user_id = session['user_id']
pet_name = request.form['pet_name']
weight = float(request.form['weight'])
conn = connect_db()
cursor = conn.cursor()
cursor.execute('INSERT INTO pets (user_id, pet_name) VALUES (%s, %s)', (user_id, pet_name))
cursor.execute('INSERT INTO pet_weights (pet_name, weight) VALUES (%s, %s)', (pet_name, weight))
conn.commit()
flash("宠物添加成功,初始体重为 %.2f kg" % weight, 'success')
logging.info("宠物 %s 添加成功,初始体重为 %.2f kg", pet_name, weight)
conn.close()
return redirect(url_for('dashboard'))
return render_template('add_pet.html')
@app.route('/view_pet/<string:pet_name>')
def view_pet(pet_name):
if 'user_id' not in session:
return redirect(url_for('login'))
conn = connect_db()
cursor = conn.cursor()
cursor.execute('SELECT * FROM pets WHERE pet_name = %s', (pet_name,))
pet = cursor.fetchone()
if not pet:
flash("宠物不存在!", 'warning')
logging.warning("宠物不存在!")
return redirect(url_for('dashboard'))
cursor.execute('SELECT weight, recorded_at FROM pet_weights WHERE pet_name = %s ORDER BY recorded_at', (pet_name,))
weights = cursor.fetchall()
conn.close()
return render_template('view_pet.html', pet=pet, weights=weights)
@app.route('/update_pet_weight/<string:pet_name>', methods=['GET', 'POST'])
def update_pet_weight(pet_name):
if 'user_id' not in session:
return redirect(url_for('login'))
conn = connect_db()
cursor = conn.cursor()
# 获取当前宠物的所有体重记录
cursor.execute('SELECT weight, recorded_at FROM pet_weights WHERE pet_name = %s ORDER BY recorded_at DESC', (pet_name,))
weights = cursor.fetchall()
if request.method == 'POST':
weight = float(request.form['weight'])
cursor.execute('INSERT INTO pet_weights (pet_name, weight) VALUES (%s, %s)', (pet_name, weight))
conn.commit()
flash("宠物体重更新成功!", 'success')
logging.info("宠物体重更新成功!")
conn.close()
return redirect(url_for('view_pet', pet_name=pet_name))
conn.close()
return render_template('update_pet_weight.html', pet_name=pet_name, weights=weights)
@app.route('/delete_account', methods=['GET', 'POST'])
def delete_account():
if 'user_id' not in session:
return redirect(url_for('login'))
if request.method == 'POST':
user_id = session['user_id']
conn = connect_db()
cursor = conn.cursor()
try:
# 删除宠物的体重记录
cursor.execute('DELETE FROM pet_weights WHERE pet_name IN (SELECT pet_name FROM pets WHERE user_id = %s)', (user_id,))
# 删除宠物信息
cursor.execute('DELETE FROM pets WHERE user_id = %s', (user_id,))
# 删除用户信息
cursor.execute('DELETE FROM users WHERE id = %s', (user_id,))
conn.commit()
session.pop('user_id', None)
flash("账号已注销。", 'success')
logging.info("用户 %s 账号已注销。", user_id)
except Exception as e:
flash("注销账号失败: %s" % str(e), 'danger')
logging.error("注销账号失败: %s", str(e))
finally:
conn.close()
return redirect(url_for('index'))
return render_template('delete_account.html')
if __name__ == '__main__':
init_db()
app.run(debug=True)
```
### 修改后的HTML模板
#### `templates/index.html`
```html
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<title>首页</title>
</head>
<body>
<h1>欢迎来到宠物体重管理平台</h1>
<a href="{{ url_for('login') }}">登录</a>
<a href="{{ url_for('register') }}">注册</a>
</body>
</html>
```
#### `templates/register.html`
```html
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<title>注册</title>
</head>
<body>
<h1>注册</h1>
{% for message in get_flashed_messages(with_categories=true) %}
<div class="alert {{ message[0] }}">{{ message[1] }}</div>
{% endfor %}
<form method="post">
<label for="username">用户名:</label>
<input type="text" name="username" required><br>
<label for="password">密码:</label>
<input type="password" name="password" required><br>
<button type="submit">注册</button>
</form>
<a href="{{ url_for('login') }}">已有账号?去登录</a>
</body>
</html>
```
#### `templates/login.html`
```html
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<title>登录</title>
</head>
<body>
<h1>登录</h1>
{% for message in get_flashed_messages(with_categories=true) %}
<div class="alert {{ message[0] }}">{{ message[1] }}</div>
{% endfor %}
<form method="post">
<label for="username">用户名:</label>
<input type="text" name="username" required><br>
<label for="password">密码:</label>
<input type="password" name="password" required><br>
<button type="submit">登录</button>
</form>
<a href="{{ url_for('register') }}">没有账号?去注册</a>
</body>
</html>
```
#### `templates/dashboard.html`
```html
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<title>仪表盘</title>
</head>
<body>
<h1>我的宠物</h1>
{% for message in get_flashed_messages(with_categories=true) %}
<div class="alert {{ message[0] }}">{{ message[1] }}</div>
{% endfor %}
<a href="{{ url_for('add_pet') }}">添加新宠物</a>
<ul>
{% for pet in pets %}
<li><a href="{{ url_for('view_pet', pet_name=pet[0]) }}">{{ pet[0] }}</a></li>
{% endfor %}
</ul>
<a href="{{ url_for('logout') }}">退出登录</a>
</body>
</html>
```
#### `templates/add_pet.html`
```html
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<title>添加宠物</title>
</head>
<body>
<h1>添加新宠物</h1>
{% for message in get_flashed_messages(with_categories=true) %}
<div class="alert {{ message[0] }}">{{ message[1] }}</div>
{% endfor %}
<form method="post">
<label for="pet_name">宠物名称:</label>
<input type="text" name="pet_name" required><br>
<label for="weight">初始体重 (kg):</label>
<input type="number" step="0.01" name="weight" required><br>
<button type="submit">添加宠物</button>
</form>
<a href="{{ url_for('dashboard') }}">返回仪表盘</a>
</body>
</html>
```
#### `templates/view_pet.html`
```html
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<title>{{ pet[0] }} 的体重记录</title>
</head>
<body>
<h1>{{ pet[0] }} 的体重记录</h1>
{% for message in get_flashed_messages(with_categories=true) %}
<div class="alert {{ message[0] }}">{{ message[1] }}</div>
{% endfor %}
<table border="1">
<tr>
<th>体重 (kg)</th>
<th>记录时间</th>
</tr>
{% for weight in weights %}
<tr>
<td>{{ weight[0] }}</td>
<td>{{ weight[1] }}</td>
</tr>
{% endfor %}
</table>
<a href="{{ url_for('update_pet_weight', pet_name=pet[0]) }}">更新体重</a>
<a href="{{ url_for('dashboard') }}">返回仪表盘</a>
</body>
</html>
```
#### `templates/update_pet_weight.html`
```html
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<title>更新 {{ pet_name }} 的体重</title>
</head>
<body>
<h1>更新 {{ pet_name }} 的体重</h1>
{% for message in get_flashed_messages(with_categories=true) %}
<div class="alert {{ message[0] }}">{{ message[1] }}</div>
{% endfor %}
<form method="post">
<label for="weight">新的体重 (kg):</label>
<input type="number" step="0.01" name="weight" required><br>
<button type="submit">更新体重</button>
</form>
<a href="{{ url_for('view_pet', pet_name=pet_name) }}">返回体重记录</a>
</body>
</html>
```
#### `templates/delete_account.html`
```html
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<title>注销账号</title>
</head>
<body>
<h1>确认注销账号</h1>
{% for message in get_flashed_messages(with_categories=true) %}
<div class="alert {{ message[0] }}">{{ message[1] }}</div>
{% endfor %}
<p>您确定要注销您的账号吗?这将删除所有与该账号相关的数据。</p>
<form method="post">
<button type="submit">确认注销</button>
</form>
<a href="{{ url_for('dashboard') }}">取消</a>
</body>
</html>
```
这些修改后的代码和模板文件将允许用户通过宠物名字而不是宠物ID来管理和查看宠物的体重记录。希望这些改动能满足您的需求。
阅读全文