luyou.py

# -*- coding: utf-8 -*
from flask import Blueprint,render_template,redirect,url_for,make_response,session
from werkzeug.routing import BaseConverter
from flask import request
#from mongo import *
from sql import *
import time
from datetime import timedelta
import sys

reload(sys)
sys.setdefaultencoding('utf-8')

#db=Mongo(host='localhost',db_name='sk')
dbsql=Sqlite("/root/chenzyhtml/db/db.db")

# 自定义正则转换器
class RegexConverter(BaseConverter):
    def __init__(self, url_map, *args):
        super(RegexConverter, self).__init__(url_map)
        # 将接受的第1个参数当作匹配规则进行保存
        self.regex = args[0]

b_shouye=Blueprint('b_luyou',__name__,url_prefix='/')
@b_shouye.route('/<re("(index.html){0,1}"):empty>',methods=['GET','POST'])
def index(empty):
    lyb=dbsql.find_all("lyb") 
    if request.method=="POST":
        g_name=request.form['guest_name']
        g_message=request.form['guest_message']
        g_time=time.asctime( time.localtime(time.time()) )
        dbsql.insert_one("lyb","NULL,'{}','{}','{}'".format(g_name,g_message,g_time))
        return redirect('/')
    else:
        return render_template('index.html',lyb=lyb)

# @b_shouye.route('/login',methods=['GET','POST'])
# def login():
#     if request.method=="POST":
#         f_username=request.form['username']
#         f_password=request.form['password']
#         check_data=db.find_one('users',{'name':f_username})
#         print(check_data)
#         if check_data:
#             if check_data['password'] == f_password:
#                 return redirect('/')
#             else:
#                 return render_template('login.html',login_flag=1)
#         else:
#             return render_template('login.html',login_flag=1)
#
#     return render_template('login.html',login_flag=0)
# @b_shouye.route('/register',methods=["GET","POST"])
# def register():
#     if request.method=="POST":
#         if not db.find_one('users',{'name':request.form['username']}):
#             f_username = request.form['username']
#             f_password = request.form['password']
#             db.insert_one('users',{'name':f_username,'password':f_password})
#             return render_template('login.html',login_flag=0)
#         else:
#             return render_template('register.html',register_flag=1)
#     return render_template('register.html',register_flag=0)
#
# @b_shouye.route('/test',methods=["GET","POST"])
# def test():
#     test=[1,2,3,4,5,6,7,8]
#     dtest={'a':'1','b':'2'}
#     return render_template('test.html',test=test,dtest=dtest)

@b_shouye.route('/register',methods=['GET','POST'])
def register():
    if request.method=="POST":
        salt=None
        r_time=time.asctime( time.localtime(time.time()) )
        r_name=request.form['register_name']
        r_password=request.form['register_password']
        last_login=r_time
        check=dbsql.find_one("users","user_name",r_name)
        if check is not None:
            return render_template('register.html',errormessage="该用户已存在，请尝试另外的用户名！")
        else:
            dbsql.insert_one("users",'NULL,"{}","{}","{}","{}"'.format(r_name,r_password,r_time,last_login))
            session.permanent = True
            session['user_name']=r_name
            return redirect('/index.html')
    else:
    	return render_template('register.html')

        # dbsql.insert_one("users","NULL,'{}','{}','{}','{}'".format(r_name,r_password,r_time,last_login))
    #     res=make_response(redirect('/index.html'))
    #     res.set_cookie('user_name',r_name,max_age=20)
    #     return res
    # else:
    #     return render_template('register.html')

@b_shouye.route('/login',methods=['GET','POST'])
def login():
    if request.method=="POST":
        l_name=request.form['login_name']
        l_password=request.form['login_password']
        check=dbsql.find_one("users","user_name",l_name)
        if check is None:
        	return render_template('login.html',errormessage="用户名或密码输入错误，请检查！")
        if check[1]==l_name and check[2]==l_password:
            dbsql.update_one("users","last_login",time.ctime(),"user_name",l_name)
            session.permanent = True
            session['user_name']=l_name
            return redirect('/index.html')
        else:
            return render_template("login.html",errormessage="用户名或密码输入错误，请检查！")
    else:
    	return render_template('login.html')

    #     res=make_response(redirect('/index.html'))
    #     res.set_cookie('user_name',l_name,max_age=20)
    #     return res
    # else:
    #     return render_template('login.html')

@b_shouye.route('/logout',methods=['GET','POST'])
def logout():
    try:
        session.pop('user_name')
        return redirect(request.args.get("backurl"))
    except:
        return redirect(request.args.get("backurl"))

if __name__=="__main__":
    print(url_for('index'))