使用flask实现一个模型注册网页,用于记录模型的版本、训练者、训练时间、所属部门等信息,并存储到mysql
时间: 2023-12-24 14:02:30 浏览: 23
好的,下面是一个使用Flask和MySQL实现模型注册网页的示例代码。在这个示例中,我们使用了一个名为`models`的表来存储模型的信息。如果您还没有安装MySQL,可以先安装MySQL并创建一个名为`models`的数据库和一个名为`models`的表。
```python
from flask import Flask, render_template, request
import datetime
import mysql.connector
app = Flask(__name__)
db = mysql.connector.connect(
host="localhost",
user="yourusername",
password="yourpassword",
database="models"
)
cursor = db.cursor()
class Model:
def __init__(self, name, version, trainer, department):
self.name = name
self.version = version
self.trainer = trainer
self.department = department
self.training_time = datetime.datetime.now()
@app.route('/', methods=['GET', 'POST'])
def index():
if request.method == 'POST':
name = request.form['name']
version = request.form['version']
trainer = request.form['trainer']
department = request.form['department']
model = Model(name, version, trainer, department)
sql = "INSERT INTO models (name, version, trainer, department, training_time) VALUES (%s, %s, %s, %s, %s)"
val = (model.name, model.version, model.trainer, model.department, model.training_time)
cursor.execute(sql, val)
db.commit()
return 'Model registered successfully!'
return render_template('index.html')
if __name__ == '__main__':
app.run(debug=True)
```
在这个应用程序中,我们连接到了MySQL数据库,并创建了一个名为`models`的表。我们定义了一个`Model`类,用于表示一个模型的信息,包括名称、版本、训练者、所属部门和训练时间。当用户提交模型注册表单时,我们从请求中获取模型的信息,并创建一个新的`Model`实例。然后,我们使用INSERT语句将模型的信息插入到MySQL表中。最后返回一个提示信息,表示模型注册成功。
这个应用程序的模型注册页面可以使用HTML和CSS来美化。你可以根据你的需求进行修改,并将其与Flask应用程序集成起来。如果出现任何问题,请确保你已经正确地安装了MySQL并创建了一个名为`models`的数据库和一个名为`models`的表。