博客园  :: 首页  :: 新随笔  :: 联系 :: 订阅 订阅  :: 管理

登录之后更新导航

Posted on 2017-11-24 12:18  占鹏  阅读(113)  评论(0编辑  收藏  举报

用上下文处理器app_context_processor定义函数

获取session中保存的值

返回字典

在父模板中更新导航,插入登录状态判断代码。、

注意用{% ... %}表示指令。

{{ }}表示变量

完成注销功能。

清除session

跳转

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>
        首页
        {% block logintitle %}{% endblock %}
        {% block registertitle %}{% endblock %}
        {% block wendatitle %}{% endblock %}
    </title>
    <script>
        function mySwitch() {
            var oBody= document.getElementById("myBody");
            var oOnoff=document.getElementById("myOnOff");
            if(oOnoff.src.match("bulbon")){
                oOnoff.src="http://www.runoob.com/images/pic_bulboff.gif";
                oBody.style.background="yellow";
                oBody.style.color="red";
            }else{oOnoff.src="http://www.runoob.com/images/pic_bulbon.gif";
                oBody.style.background="lightcyan";
                oBody.style.color="black";
            }
        }

    </script>
   <link rel="stylesheet" type="text/css" href="../static/css/switch.css">
{% block loginhead %}{% endblock %}
{% block registerhead %}{% endblock %}
{% block wendahead %}{% endblock %}

</head>

<body id="myBody" style="background-color: lightcyan">
<nav>

<img src="{{ url_for('static',filename='image/gzsxy.png') }}"alt="" width="20px">
    <script>
    document.write(Date())
</script>
<br>

    <a href="{{ url_for('switch') }}">首页</a>

    {% if username %}
    <a href="#">{{ username }}</a>
    <a href="{{ url_for('logout') }}">注销</a>
    {% else %}
 <a href="{{ url_for('login') }}">登录</a>
    <a href="{{ url_for('register') }}">注册</a>
    {% endif %}

    <button type="submit">搜索</button>







<p align="right"><img src="{{ url_for('static',filename='image/gzt.jpg') }}"alt="" width="20px"></p>
    <p align="right"> <img style="width: 20px" id="myOnOff" onclick="mySwitch()" src="
    http://www.runoob.com/images/pic_bulbon.gif"></p>
   </nav>
<p>{{username  }}context</p>




{% block loginbody %}{% endblock %}
{% block registerbody %}{% endblock %}
{% block wendabody %}{% endblock %}
</body>
</html>
from flask import Flask, render_template, request, redirect, url_for, session
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)
    nickname = db.Column(db.String(20), nullable=True)


 # db.create_all()    #是否链接成功


# 添加
# user=User(username='ma',password='pp')
# db.session.add(user)
# db.session.commit()
#
#
# 更新
#
# user = User.query.filter(User.username=='zz').first()
# user.password='gg'
# db.session.commit()
#
#
# 查询
# user = User.query.filter(User.username=='lz').first()
# print(user.username,user.password)
#
# 删除
# user = User.query.filter(User.username=='lz').first()
# db.session.delete(user)
# db.session.commit()


# session会话连接
# filter()过滤器
# route制定路径和函数之间的关系
# def定义一个变量

@app.route('/')
def moban():
    return render_template('lizi.html')


@app.route('/login/', methods=['GET', 'POST']) 
def login():
    if request.method == 'GET':
        return render_template('login.html')
    else:
        username = request.form.get('user')
        password = request.form.get('psw')
        user = User.query.filter(User.username == username).first()  
        if user:
           if user.password == password:
               session['user']=username
               session.permanent = True
               return redirect(url_for('lizi'))
        else:
            return '用户名不存在'


@app.context_processor
def mycontext():
    username=session.get('user')
    if username:
        return {'username':username}
    else:
        return {}

@app.route('/register/', methods=['GET', 'POST']) 
def register():
    if request.method == 'GET':
        return render_template('register.html')
    else:

        username = request.form.get('user')
        password = request.form.get('psw')
        nickname = request.form.get('nickname')
        user = User.query.filter(User.username == username).first()  
        if user:
            return u'该用户已存在'
        else:
            user = User(username=username, password=password, nickname=nickname)
            db.session.add(user)
            db.session.commit()
            return redirect(url_for('login'))

@app.route('/logout/')
def logout():
    session.clear()
    return render_template('lizi.html')

@app.route('/wenda/')
def wenda():
    return render_template('wenda.html')


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