import os
from flask import Flask, render_template, request, redirect, url_for, flash, session
from functools import wraps
from flask_bootstrap import Bootstrap
from flask_mail import Message, Mail
from forms import RegisterForm, MailForm
users = [
{
‘username’: ‘root’,
‘password’: ‘root’
},
{
‘username’: ‘hello’,
‘password’: ‘root’
},
]
实现用户注册， 用户登录， 用户注销， 用户查看;
app = Flask(name)
app.config[‘SECRET_KEY’] = ‘westos’
bootstrap = Bootstrap(app)
配置发送邮件的相关信息;
指定邮件服务器的域名或者IP
app.config[‘MAIL_SERVER’] = ‘smtp.qq.com’
指定端口， 默认25， 但qq邮箱默认为 端口号465或587；
app.config[‘MAIL_PORT’] = 465
app.config[‘MAIL_USE_SSL’] = True
app.config[‘MAIL_USERNAME’] = ‘2287236639’
app.config[‘MAIL_PASSWORD’] = “epddcimhxnfndjcj”
def is_login(f):
“”“用来判断用户是否登录成功”""
@wraps(f)
def wrapper(*args, **kwargs):
# 判断session对象中是否有seesion['user'],
# 如果包含信息， 则登录成功， 可以访问主页；
# 如果不包含信息， 则未登录成功， 跳转到登录界面;；
if session.get('user', None):
return f(*args, **kwargs)
else:
flash("用户必须登录才能访问%s" % (f.__name__))
return redirect(url_for('login'))
return wrapper
def is_admin(f):
“”“用来判断用户是否登录成功”""
@wraps(f)
def wrapper(*args, **kwargs):
# 判断session对象中是否有seesion['user']等于root,
# 如果包含信息， 则登录成功， 可以访问主页；
# 如果不包含信息， 则未登录成功， 跳转到登录界面;；
if session.get('user', None) == 'root':
return f(*args, **kwargs)
else:
flash("只有管理员root才能访问%s" % (f.__name__))
return redirect(url_for('login'))
return wrapper
@app.route(’/’)
@is_login
def index():
return render_template(‘index.html’)
@app.route(’/register/’, methods=[‘GET’, ‘POST’])
def register():
if request.method == ‘POST’:
username = request.form.get(“username”, None)
password = request.form.get(‘password’, None)
# 当所有的信息遍历结束， 都没有发现注册的用户存在， 则将注册的用户添加到服务器， 并跳转登录界面;
for user in users:
if user['username'] == username:
return render_template('register.html', message="用户%s已经存在" % (username))
else:
users.append(dict(username=username, password=password))
# return redirect('/login/')
# 出现一个闪现信息;
flash("用户%s已经注册成功， 请登录....." % (username), category='info')
return redirect(url_for('login'))
return render_template('register.html')
@app.route(’/login/’, methods=[‘GET’, ‘POST’])
def login():
if request.method == ‘POST’:
username = request.form.get(“username”, None)
password = request.form.get(‘password’, None)
# 当所有的信息遍历结束， 都没有发的用户存在， 则将注册的用户添加到服务器， 并跳转登录界面;
for user in users:
if user['username'] == username and user['password'] == password:
# 将用户登录的信息存储到session中;
session['user'] = username
return redirect(url_for('index'))
else:
# 出现一个闪现信息;
flash("用户%s密码错误， 请重新登录....." % (username))
return redirect(url_for('login'))
return render_template('login.html')
@app.route(’/logout/’)
def logout():
# 将用户存储到session中的信息删除;
session.pop(‘user’)
flash(“注销成功…”)
return redirect(url_for(‘login’))
@app.route(’/delete/string:username/’)
def delete(username):
for user in users:
# 用户存在， 则删除;
if username == user[‘username’]:
users.remove(user)
flash(“删除用户%s成功” % (username))
# else:
# flash(“用户%s不存在” % (username))
# 删除成功， 跳转到/list/路由中.....
return redirect(url_for('list'))
@app.route(’/list/’)
@is_login
@is_admin
def list():
return render_template(‘list.html’,
users=users)
@app.route(’/upload/’, methods=[‘POST’, ‘GET’])
def upload():
if request.method == ‘POST’:
# 获取用户上传的文件对象
f = request.files[‘faceImg’]
# 获取上传文件的文件名
print(f.filename)
# 获取当前项目的目录位置;
basepath = os.path.dirname(__file__)
# print(__file__) # /root/PycharmProjects/day34/app.py
# print(basepath) # /root/PycharmProjects/day34
# /root/PycharmProjects/day34/static/img/face/xxx.png
# 拼接路径， 保存到本地的位置;
filepath = os.path.join(basepath, 'static', 'img', 'face', f.filename)
# 保存文件
f.save(filepath)
flash("上传文件%s成功" % (f.filename))
return redirect(url_for('upload'))
else:
return render_template('upload.html')
测试表单的应用案例
@app.route(’/wtf/’, methods=[‘GET’, ‘POST’])
def wtf():
from forms import RegisterForm
# 1. 实例化表单， 将来传递到前端， 进行生成对应的html;
form = RegisterForm()
# 2. 判断HTTP请求方式， 返回不同的内容
# 第一个判断: 是否为post请求? 判断提交的数据是否符合form表单定义的验证?
# if request.method == ‘POST’:
# print(request.form[‘name’])
# return ‘post’
if form.validate_on_submit():
username = request.form['name']
password = request.form['password']
gender = request.form['gender']
tech = request.form['tech']
# 当所有的信息遍历结束， 都没有发现注册的用户存在， 则将注册的用户添加到服务器， 并跳转登录界面;
for user in users:
if user['username'] == username:
flash(message="用户%s已经存在" % (username))
return redirect(url_for('wtf'))
else:
users.append(dict(username=username, password=password))
# return redirect('/login/')
# 出现一个闪现信息;
flash("用户%s已经注册成功， 请登录....." % (username))
return redirect(url_for('login'))
return render_template("wtf.html",
form=form)
def send_mail(to, subject, template, **kwargs):
mail = Mail(app)
# app.app_context(): 将之前Flask创建的app作为参数传入AppContext类中,
# 用于存储当前app的相关信息;
with app.app_context():
msg = Message(subject=subject,
sender=‘2287236639@qq.com’,
recipients=to,
# body=render_template(template + ‘.txt’, **kwargs),
html=render_template(template + ‘.html’, **kwargs)
)
mail.send(msg)
@app.route(’/mail/’, methods=[‘GET’, ‘POST’])
def mail():
# 实例化表单
form = MailForm()
if form.validate_on_submit():
# 获取用户提交的数据;
toEmails = form.toEmails.data.split(’,’)
toFilename = form.toFilename.data
try:
send_mail(toEmails, "西部开源邮件测试", toFilename)
except Exception as e:
flash("邮件发送失败, 失败原因: %s!" %(e))
else:
flash("邮件发送成功!")
return redirect(url_for('mail'))
return render_template('send_mail.html',
form = form)
if name == ‘main’:
app.run()
login.html:
{% extends 'base.html' %}
{% block title %} 登录页面{% endblock %}
{% block content %}
<!--<button type="button" class="btn btn-danger">失败</button>-->
<!--实现登录界面-->
<div class="container">
<div class="col-lg-4 col-lg-offset-4" style="margin-top: 50px">
<h1>
登录
<small>
没有账号？
{# <a href="/register/">注册</a>#}
<a href="{{ url_for('register') }}">注册</a>
</small>
</h1>
<form action="{{ url_for('login') }}" method="POST">
<div class="form-group">
<label for="exampleInputEmail1">用户名</label>
<input name='username' type="text" class="form-control" id="exampleInputEmail1" placeholder="Email">
</div>
<div class="form-group">
<label for="exampleInputPassword1">密码</label>
<input name='password' type="password" class="form-control" id="exampleInputPassword1"
placeholder="Password">
</div>
<div class="checkbox">
<label>
<input type="checkbox"> 记住密码
</label>
</div>
<button type="submit" class="btn btn-success btn-block">登录</button>
</form>
</div>
{% endblock %}
index.html
{% extends 'base.html' %}
{% block title %} 主页 {% endblock %}
{% block content %}
<h1>这是主页</h1>
{% endblock %}
register.html
{% extends 'base.html' %}
{% block title %} 注册页面{% endblock %}
{% block content %}
<!--<button type="button" class="btn btn-danger">失败</button>-->
<!--实现登录界面-->
<div class="container">
<div class="col-lg-4 col-lg-offset-4" style="margin-top: 50px">
<h1>
注册
<small>
已有帐号
{# <a href="/login/">登录</a>#}
<a href="{{ url_for('login') }}">登录</a>
</small>
</h1>
{# action : 将表单填写的数据提交到那个URL里面进行处理, #}
{# url_for('register') : 根据视图函数寻找对应的url地址,/register/ #}
{# method =['post', 'get'] , 这种是错误的， 提交数据只能选择一种HTTP请求方法; #}
<form action="{{ url_for('register') }}" method="post">
<div class="form-group">
<label for="exampleInputEmail1">用户名</label>
{# name="username" ====> 将name里面的值username作为key值， 表单里面填写的内容作为value值， 提交到request.form #}
{# type="text" : 要求填写的是一个字符串; #}
<input name="username" type="text" class="form-control" id="exampleInputEmail1" placeholder="Email">
</div>
<div class="form-group">
<label for="exampleInputPassword1">密码</label>
<input name="password" type="password" class="form-control" id="exampleInputPassword1"
placeholder="Password">
</div>
<div class="checkbox">
<label>
<input type="checkbox"> 记住密码
</label>
</div>
<button type="submit" class="btn btn-success btn-block">注册</button>
</form>
{% if message %}
<h5 style="color: red">{{ message }}</h5>
{% endif %}
</div>
{% endblock %}
list.html
{% extends 'base.html' %}
{% block title %} 用户显示{% endblock %}
{% block content %}
<div class="container">
<div class="col-lg-8 col-lg-offset-2">
<h1>用户信息显示</h1>
<br/>
<hr/>
<br/>
<table class="table table-hover">
<tr>
<td>用户名</td>
<td>密码</td>
<td>删除</td>
</tr>
{% for user in users %}
<tr>
{# user是一个字典， 获取字典的key值， user.keyname #}
<td>{{ user.username }}</td>
<td>{{ user.password }}</td>
{# 寻找删除的按钮 #}
<td>
{# <a href="/delete/{{ user.username }}/" class="btn btn-default btn-danger " role="button">删除</a>#}
<a href="{{ url_for('delete', username=user.username) }}"
class="btn btn-default btn-danger " role="button">删除</a>
</td>
</tr>
{% endfor %}
</table>
</div>
</div>
{% endblock %}
upload.html
{% extends 'base.html' %}
{% block title %} 上传文件 {% endblock %}
{% block content %}
<div class="container">
<div class="col-lg-8 col-lg-offset-2">
<h1>上传头像</h1>
<hr/>
{# enctype="multipart/form-data"用于接收特殊的数据， 包括视频， 图片， 音乐.....，
默认只能接收text/plain #}
<form action="{{ url_for('upload') }}" method="post" enctype="multipart/form-data">
<input name="faceImg" type="file"><br/>
<input type="submit" value="上传">
</form>
</div>
</div>
{% endblock %}
wtf.html
{% extends 'base.html' %}
{#导入模块， 并重命名为wtf#}
{% import "bootstrap/wtf.html" as wtf %}
{% block title %} wtf测试页面{% endblock %}
{% block content %}
<div class="container" style="margin-top: 50px">
<div class="col-lg-8 col-lg-offset-2">
<h1>注册-wtf测试</h1>
{# 根据后台传递过来的form对象生成对应的html代码; #}
{{ wtf.quick_form(form) }}
</div>
</div>
{% endblock %}
send_mail.html
{% extends ‘base.html’ %}
{% import “bootstrap/wtf.html” as wtf %}
{% block title %} 发送邮件 {% endblock %}
{% block content %}
<div class="container">
<div class="col-lg-8 col-lg-offset-2">
<h1>发送邮件测试案例</h1>
<hr/>
{{ wtf.quick_form(form) }}
</div>
</div>
{% endblock %}
630
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
