-
Notifications
You must be signed in to change notification settings - Fork 0
/
routes.py
109 lines (88 loc) · 3.25 KB
/
routes.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
# coding=utf-8
from __init__ import *
from views import *
import sys
from flask import Flask, request, session, g, redirect, url_for, abort,\
render_template, flash
from flask_admin.model import filters
import flask_admin as admin
from flask.ext.security import SQLAlchemyUserDatastore
@app.errorhandler(404)
def pageNotFound(error):
app.logger.error(error)
return app.config['ERROR_404']
@app.errorhandler(500)
def internal_error(error):
app.logger.error(error)
return app.config['ERROR_500']
@app.errorhandler(403)
def internal_error(error):
return app.config['ERROR_403']
@app.errorhandler(Exception)
def unhandled_exception(e):
app.logger.error('Unhandled Exception: %s', (e))
return app.config['ERROR_500']
# Initialize flask-login
def init_login():
login_manager = LoginManager()
login_manager.init_app(app)
# login_manager.login_view = 'login'
# Create user loader function
@login_manager.user_loader
def load_user(user_id):
return db.session.query(User).get(int(user_id))
@login_manager.unauthorized_handler
def unauthorized_handler():
return 'Unauthorized'
@app.route('/')
def index():
return render_template('login.html')
# return redirect(url_for('login'))
@app.route('/login', methods=['GET', 'POST'])
def login():
error = None
if request.method == 'GET':
return render_template('login.html')
username = request.form.get('username')
password = request.form.get('password')
registered_user = User.query.filter_by(username=username,
password=password).first()
if registered_user is not None:
login_user(registered_user)
flash(u'登陆成功!')
return redirect(request.args.get('next') or url_for('admin.index'))
else:
error = u'不好意思,密码错误'
return render_template('login.html', error=error)
return redirect(url_for('.index'))
# login.login_user(registered_user)
init_login()
# Setup Flask-Security
user_datastore = SQLAlchemyUserDatastore(db, User, Role)
reload(sys)
sys.setdefaultencoding('utf8')
formatter = logging.Formatter(
"[%(asctime)s] {%(pathname)s:%(lineno)d} %(levelname)s - %(message)s")
handler = RotatingFileHandler('kuaidi.log', maxBytes=10000000, backupCount=5)
handler.setLevel(logging.ERROR)
handler.setFormatter(formatter)
app.logger.addHandler(handler)
admin = admin.Admin(
app,
name=u'阳成科技',
index_view=MyAdminIndexView(),
base_template='my_master.html'
)
# admin.locale_selector(get_locale)
admin.add_view(ReceiptView(Receipt, db.session, u'收快递'))
admin.add_view(PostView(Post, db.session, u'发快递'))
# for admin
admin.add_view(CommunityView(Community, db.session, u'小区'))
admin.add_view(UserView(User, db.session, u'管理员'))
admin.add_view(ReceiptReportView(ReceiptReport, db.session, u'收件单报表'))
admin.add_view(PostReportView(PostReport, db.session, u'发件单报表'))
admin.add_view(NotReceiveReportView(NotReceiveReport, db.session, u'未收快递报表'))
if __name__ == '__main__':
# admin.add_view(MyView(name='Hello'))
app.run()
# app.run(host='0.0.0.0',port=8080)