一站式网上办事大厅

from flask import Flask, render_template, request, redirect, url_for

from flask_sqlalchemy import SQLAlchemy

app = Flask(__name__)

app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///users.db'

db = SQLAlchemy(app)

class User(db.Model):

id = db.Column(db.Integer, primary_key=True)

username = db.Column(db.String(80), unique=True, nullable=False)

password = db.Column(db.String(120), nullable=False)

@app.route('/register', methods=['GET', 'POST'])

def register():

if request.method == 'POST':

new_user = User(username=request.form['username'], password=request.form['password'])

db.session.add(new_user)

db.session.commit()

return redirect(url_for('login'))

return render_template('register.html')

@app.route('/login', methods=['GET', 'POST'])

def login():

if request.method == 'POST':

user = User.query.filter_by(username=request.form['username']).first()

if user and user.password == request.form['password']:

return "Login Successful"

else:

return "Invalid Credentials"

return render_template('login.html')

]]>