250 lines
6.1 KiB
Python
250 lines
6.1 KiB
Python
from flask import Flask, redirect, url_for, render_template, request, flash
|
||
from models import db, Contact, User
|
||
from forms import ContactForm
|
||
from flask_login import LoginManager, current_user, login_user, login_required
|
||
|
||
|
||
# Flask
|
||
app = Flask(__name__)
|
||
app.config['SECRET_KEY'] = 'my secret'
|
||
app.config['DEBUG'] = False
|
||
|
||
# Database
|
||
# app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///book.sqlite'
|
||
app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql+pymysql://root:123456@localhost/hp'
|
||
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
|
||
db.init_app(app)
|
||
|
||
# Login
|
||
login_manager = LoginManager()
|
||
login_manager.init_app(app)
|
||
password = '123456'
|
||
|
||
|
||
@login_manager.user_loader
|
||
#使用user_loader装饰器的回调函数非常重要,他将决定 user 对象是否在登录状态
|
||
def user_loader(id):
|
||
#这个id参数的值是在 login_user(user)中传入的 user 的 id 属性
|
||
user = User()
|
||
return user
|
||
|
||
|
||
@app.route('/login', methods=['GET', 'POST'])
|
||
def login():
|
||
if request.method == 'GET':
|
||
return '''
|
||
<form action="#" method="POST">
|
||
<span>请输入账号</span>
|
||
<input type="text" name="name" id="name" placeholder="name">
|
||
<span>请输入密码</span>
|
||
<input type="password" name="pw" id="pw" placeholder="password">
|
||
<input type="submit" name="submit">
|
||
</form>
|
||
'''
|
||
name = request.form.get('name')
|
||
if request.form.get('pw') == password:
|
||
user = User()
|
||
login_user(user)
|
||
return redirect(url_for('index'))
|
||
return redirect(url_for('login'))
|
||
|
||
@app.route("/")
|
||
@login_required
|
||
def index():
|
||
'''
|
||
Home page
|
||
'''
|
||
return redirect(url_for('contacts'))
|
||
|
||
|
||
@app.route("/new_contact", methods=('GET', 'POST'))
|
||
@login_required
|
||
def new_contact():
|
||
'''
|
||
Create new contact
|
||
'''
|
||
form = ContactForm()
|
||
if form.validate_on_submit():
|
||
my_contact = Contact()
|
||
form.populate_obj(my_contact)
|
||
db.session.add(my_contact)
|
||
try:
|
||
db.session.commit()
|
||
# User info
|
||
flash('Contact created correctly', 'success')
|
||
return redirect(url_for('contacts'))
|
||
except Exception as ex:
|
||
db.session.rollback()
|
||
flash('Error generating contact. {}'.format(ex), 'danger')
|
||
|
||
return render_template('web/new_contact.html', form=form)
|
||
|
||
|
||
@app.route("/edit_contact/<id>", methods=('GET', 'POST'))
|
||
@login_required
|
||
def edit_contact(id):
|
||
'''
|
||
Edit contact
|
||
|
||
:param id: Id from contact
|
||
'''
|
||
my_contact = Contact.query.filter_by(id=id).first()
|
||
form = ContactForm(obj=my_contact)
|
||
if form.validate_on_submit():
|
||
try:
|
||
# Update contact
|
||
form.populate_obj(my_contact)
|
||
db.session.add(my_contact)
|
||
db.session.commit()
|
||
# User info
|
||
flash('Saved successfully', 'success')
|
||
except:
|
||
db.session.rollback()
|
||
flash('Error update contact.', 'danger')
|
||
return render_template(
|
||
'web/edit_contact.html',
|
||
form=form)
|
||
|
||
|
||
@app.route("/contacts")
|
||
@login_required
|
||
def contacts():
|
||
'''
|
||
Show alls contacts
|
||
'''
|
||
contacts = Contact.query.order_by(Contact.name).all()
|
||
return render_template('web/contacts.html', contacts=contacts)
|
||
|
||
|
||
@app.route("/search")
|
||
@login_required
|
||
def search():
|
||
'''
|
||
Search
|
||
'''
|
||
name_search = request.args.get('name')
|
||
all_contacts = Contact.query.filter(
|
||
Contact.name.contains(name_search)
|
||
).order_by(Contact.name).all()
|
||
return render_template('web/contacts.html', contacts=all_contacts)
|
||
|
||
|
||
@app.route("/contacts/delete", methods=('POST',))
|
||
@login_required
|
||
def contacts_delete():
|
||
'''
|
||
Delete contact
|
||
'''
|
||
try:
|
||
mi_contacto = Contact.query.filter_by(id=request.form['id']).first()
|
||
db.session.delete(mi_contacto)
|
||
db.session.commit()
|
||
flash('Delete successfully.', 'danger')
|
||
except:
|
||
db.session.rollback()
|
||
flash('Error delete contact.', 'danger')
|
||
|
||
return redirect(url_for('contacts'))
|
||
|
||
@app.route("/test1", methods=('GET', 'POST'))
|
||
@login_required
|
||
def test1():
|
||
c1 = dict()
|
||
c1['account'] = '15120201012'
|
||
c1['ip'] = '23.14.234.12'
|
||
c1['location'] = '河南郑州'
|
||
c1['isp'] = '中国联通'
|
||
c1['cnt1'] = '3'
|
||
c1['cnt2'] = '2'
|
||
|
||
c2 = dict()
|
||
c2['account'] = '15120210128'
|
||
c2['ip'] = '12.221.21.2'
|
||
c2['location'] = '广州深圳'
|
||
c2['isp'] = '中国联通'
|
||
c2['cnt1'] = '1'
|
||
c2['cnt2'] = '1'
|
||
|
||
c3 = dict()
|
||
c3['account'] = '15220201214'
|
||
c3['ip'] = '122.23.11.25'
|
||
c3['location'] = '北京'
|
||
c3['isp'] = '中国电信'
|
||
c3['cnt1'] = '8'
|
||
c3['cnt2'] = '4'
|
||
|
||
data = []
|
||
data.append(c1)
|
||
data.append(c2)
|
||
data.append(c3)
|
||
|
||
return render_template('web/test1.html', contacts=data)
|
||
|
||
@app.route("/test2", methods=('GET', 'POST'))
|
||
@login_required
|
||
def test2():
|
||
|
||
t1 = dict()
|
||
t1['account'] = '15220201027'
|
||
t1['login'] = '2021-02-28 12:16:23'
|
||
t1['duration'] = '41'
|
||
t1['total'] = '953'
|
||
|
||
t2 = dict()
|
||
t2['account'] = '15220201214'
|
||
t2['login'] = '2021-02-22 09:23:51'
|
||
t2['duration'] = '342'
|
||
t2['total'] = '1234'
|
||
|
||
t3 = dict()
|
||
t3['account'] = '15220210129'
|
||
t3['login'] = '2020-12-31 12:34:22'
|
||
t3['duration'] = '12'
|
||
t3['total'] = '31'
|
||
|
||
t4 = dict()
|
||
t4['account'] = '15320201030'
|
||
t4['login'] = '2021-01-03 20:12:34'
|
||
t4['duration'] = '434'
|
||
t4['total'] = '1356'
|
||
|
||
data = []
|
||
data.append(t1)
|
||
data.append(t2)
|
||
data.append(t3)
|
||
data.append(t4)
|
||
|
||
return render_template('web/test2.html', contacts=data)
|
||
|
||
@app.route("/test3", methods=('GET', 'POST'))
|
||
@login_required
|
||
def test3():
|
||
|
||
d1 = dict()
|
||
d1['account'] = '15220210129'
|
||
d1['count'] = 9
|
||
|
||
d2 = dict()
|
||
d2['account'] = '15320201013'
|
||
d2['count'] = 22
|
||
|
||
d3 = dict()
|
||
d3['account'] = '15320210302'
|
||
d3['count'] = 78
|
||
|
||
d4 = dict()
|
||
d4['account'] = '15220210129'
|
||
d4['count'] = 12
|
||
|
||
data = []
|
||
data.append(d1)
|
||
data.append(d2)
|
||
data.append(d3)
|
||
data.append(d4)
|
||
|
||
return render_template('web/test3.html', contacts=data)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
app.run(host="0.0.0.0", port=5000, debug=True)
|