- 编写要求登录的装饰器
from functools import wraps
def loginFirst(func): #参数是函数
@wraps(func)
def wrapper(*args, ** kwargs): #定义个函数将其返回
#要求登录
return func(*args, ** kwargs)
return wrapper #返回一个函数
- 应用装饰器,要求在发布前进行登录,登录后可发布。
@app.route('/question/',methods=['GET','POST'])
@loginFirst
def question():
- 建立发布内容的对象关系映射。
class Question(db.Model):
- 完成发布函数。
保存到数据库。
重定向到首页。
qusetion.html
{% extends'index.html' %} {% block questiontitle %}提问页面{% endblock %} {% block questionhead %} <link rel="stylesheet" type="text/css" href="../static/css/myweb.css"/> <script src="../static/js/question.js"></script> {% endblock %} {% block questiontbody %} <form action="{{ url_for('question') }}" method="post"> <div class="form-group"> <!--<label for="name">选择列表</label>--> <!--<select class="form-control">--> <!--<option>1</option>--> <!--<option>2</option>--> <!--<option>3</option>--> <!--<option>4</option>--> <!--<option>5</option>--> <!--</select>--> <label for="questionTitle">标题</label> <textarea class="form-control" rows="1" id="questionTitle"></textarea> <label for="questionDatail">详情</label> <textarea class="form-control" rows="5" id="questionDatail"></textarea> <input type="checkbox" name="c1" id="c1" value="">记住我 <br> <div id=""><br></div> <button class="btn btn-default" style="color:black" type="submit" οnclick="return fnQuestion()">发布</button> </div> </form> {% endblock %}
question.js
function fnQuestion() { var title =document.getElementById("title") var detail = document.getElementById("detail") var creat_time=document.getElementById("creat_time") var author_id =document.getElementById("author_id") var author=document.getElementById("author") var isError = true; if (title.value.length > 100 || oUpass.value.length < 0) { oError.innerHTML = "问题请输入100位以内字符" isError = false; } return isError; return isError; }
mis.py
from flask import Flask, render_template,redirect,url_for,request,session from flask_sqlalchemy import SQLAlchemy from functools import wraps from datetime import datetime 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=True, autoincrement=True) username = db.Column(db.String(20), nullable=False) password = db.Column(db.String(20), nullable=False) nickname = db.Column(db.String(20)) class Question(db.Model): __tablename__='question' id = db.Column(db.Integer,primary_key=True,autoincrement=True) title = db.Column(db.String(100),nullable=False) detail = db.Column(db.Text,nullable=False) create_time = db.Column(db.DateTime,default=datetime.now()) author_id = db.Column(db.Integer,db.ForeignKey('user.id')) author = db.relationship('User',backref=db.backref('question')) db.create_all() @app.route('/') def index(): return render_template('index.html') @app.route('/login/',methods=['GET','POST']) def login(): if request.method == "GET": return render_template("login.html") else: username = request.form.get("username") password = request.form.get("password") user = User.query.filter(User.username == username).first() if user: if user.password == password: session['user'] = username return redirect(url_for('login')) else: return '密码错误' else: return '用户名不存在' @app.context_processor def mycontext(): username = session.get('user') if username: return{'username':username} else: return{} @app.route('/logout/') def logout(): session.clear() return redirect(url_for('base')) @app.route('/regist/',methods=['GET','POST']) def regist(): if request.method == 'GET': return render_template("regist.html") else: username = request.form.get("username") password = request.form.get("password") user = User.query.filter(User.username == username).first() if user: return 'username existed.' else: user = User(username=username, password=password) db.session.add(user) #数据库,添加操作 db.session.commit() return redirect(url_for('login')) #发布前需要登录,定义装饰器 def loginFirst(func): @wraps(func) def wrapper(*args,**kwargs): if session.get('user'): return func(*args,**kwargs) else: return redirect(url_for('login')) return wrapper @app.route('/question/',methods=['GET','POST']) @loginFirst def question(): if request.methof=='GET': return render_template('question.html') else: title=request.form.get('title') detail=request.form.get('detail') author_id=User.query.filter(User.username==session.get('user')).first().id question=Question(title=title,detail=detail,author_id=author_id) db.session.ass(question) db.session.commit() return redirect(url_for('index')) if __name__ == '__main__': app.run(debug=True)
运行结果如下: