from flask import Flask, render_template, flash, redirect, url_for
from flask_sqlalchemy import SQLAlchemy
from flask_bootstrap import Bootstrap
from flask_wtf import FlaskForm
from wtforms import StringField, SubmitField, EmailField, TextAreaField, PasswordField
from wtforms.validators import DataRequired, Email, Length

app = Flask(__name__)
# 配置数据库URI
app.config['SECRET_KEY'] = 'abcdefghijklmnopqrstuvwxyz'
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///users.db'
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False

# 初始化扩展
db = SQLAlchemy(app)
bootstrap = Bootstrap(app)

# 定义数据模型
class User(db.Model):
    __tablename__ = 'users'
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(64), unique=True, index=True)
    email = db.Column(db.String(120), unique=True, index=True)
    password = db.Column(db.String(128))
    about_me = db.Column(db.Text)
    
    def __repr__(self):
        return f'<User {self.name}>'

# 定义表单
class UserForm(FlaskForm):
    name = StringField('用户名', validators=[DataRequired(), Length(min=2, max=20)])
    email = EmailField('邮箱', validators=[DataRequired(), Email()])
    password = PasswordField('密码', 
                           validators=[DataRequired(), 
                                     Length(min=6, message='密码至少需要6个字符')])
    about_me = TextAreaField('个人简介')
    submit = SubmitField('提交')

# 路由
@app.route('/', methods=['GET', 'POST'])
def index():
    form = UserForm()
    if form.validate_on_submit():
        # 检查用户名是否已存在
        user = User.query.filter_by(name=form.name.data).first()
        if user:
            flash('用户名已存在！', 'danger')
            return redirect(url_for('index'))
        
        # 检查邮箱是否已存在
        user = User.query.filter_by(email=form.email.data).first()
        if user:
            flash('邮箱已被注册！', 'danger')
            return redirect(url_for('index'))
        
        # 创建新用户
        user = User(
            name=form.name.data,
            email=form.email.data,
            password=form.password.data,  # 实际应用中应该加密存储
            about_me=form.about_me.data
        )
        
        # 添加到数据库
        db.session.add(user)
        db.session.commit()
        
        flash('注册成功！', 'success')
        return redirect(url_for('index'))
    
    return render_template('sqldb/users.html', form=form)

@app.route('/users')
def user_list():
    users = User.query.all()
    return render_template('sqldb/user_list.html', users=users)

# 创建数据库表
def init_db():
    with app.app_context():
        db.create_all()

if __name__ == '__main__':
    init_db()  # 创建数据库表
    app.run(host='0.0.0.0', port=9999, debug=True)