write data check

UI/main.py

@@ -9,6 +9,7 @@ from dotenv import load_dotenv
 import time
 import socket
 import utils
+from datacheck import *
 
 # Load environment variables from .env file
 load_dotenv()
@@ -66,9 +67,10 @@ def about():
 
 @app.route('/login', methods=['GET', 'POST'])
 def login():
-    if request.method == 'POST':
-        username = request.form['username']
-        password = request.form['password']
+    form = LoginForm()
+    if form.validate_on_submit():
+        username = form.username.data
+        password = form.password.data
         
         # 调用 ExecCommand 执行登录命令
         login_response = ExecCommand(f'login -u {username} -p {password}')
@@ -81,9 +83,10 @@ def login():
         else:
             error = 'An unknown error occurred.'
         
-        return render_template('login.html', error=error)
+        flash(error, 'error')
+        return redirect(url_for('login'))
     
-    return render_template('login.html')
+    return render_template('login.html', form=form)
 
 @app.before_request
 def before_request():