完成注册功能

from flask import Flask, request,redirect,url_for
from flask import render_template
from flask_sqlalchemy import SQLAlchemy
import config

app=Flask(__name__)
app.config.from_object(config)
db=SQLAlchemy(app)

class User(db.Model):
    __tablename__='user'
    id=db.Column(db.Integer,primary_key=TabError,autoincrement=True)
    username=db.Column(db.String(20),nullable=False)
    password=db.Column(db.String(20),nullable=False)

'''
db.create_all()


# 数据添加
user=User(username='hh',password='gg')
db.session.add(user)
db.session.commit()

# 数据更改
user = User.query.filter(User.username=='hh').first()
user.password='0000'
db.session.commit()

# 数据删除
user = User.query.filter(User.username=='hh').first()
db.session.delete(user)
db.session.commit()

# 数据查询
user = User.query.filter(User.username=='hh').first()
print(user.username,user.password)
'''

app = Flask(__name__)
@app.route('/')
def switch():
    return render_template('moban.html')
@app.route('/login/')
def login():
    return render_template('login.html')
@app.route('/register/',methods=['GET','POST'])
def register():
    if request.method=='GET':
        return render_template('register.html')
    else:
        username=request.form.get('username')
        password = request.form.get('password')
        nickname = request.form.get('nickname')
        user=User.query.filter(User.username==username).first()
        if user:
            return 'username existed'
        else:
            user=User(username='hh',password='gg')
            db.session.add(user)
            db.session.commit()
            return  redirect(url_for('login'))

if __name__ == '__main__':
    app.run(debug=True)
if __name__ == '__main__':
      app.run()

 

posted on 2017-11-18 20:47  06黄良龙  阅读(105)  评论(0)    收藏  举报

导航