def setUp(self):
#调用app
app.config.update(
#处于测试环境
TESTING=True,
#用缓存数据库 memory 内存数据库
SQLALCHEMY_DATABASE_URI="sqlite:///:memory:")
# 往库中
db.create_all()
user = User(name="Test", username="******")
user.set_password("123456")
movie = Movie(title="Test Movie Title", year="2020")
#add一条上传 add_all 多条列表上传
db.session.add_all([user, movie])
db.session.commit()
# 创建一个测试用的客户端模拟 浏览器
self.client = app.test_client()
#创建测试命令运行器
self.runner = app.test_cli_runner()
def sign_up():
"""SIGN-UP page which allows users to register."""
if current_user.is_authenticated:
flash("You have alredy logged in.")
return redirect(url_for("app.home"))
else:
form = SignUpForm()
if form.validate_on_submit():
user = User.query.filter_by(username=form.username.data).first()
if user is None:
user = User(username=form.username.data)
user.set_password(form.password.data)
db.session.add(user)
db.session.commit()
login_user(user)
flash("Sucessfully registered.")
flash("Sucessfully logged in.")
return redirect(url_for("app.home"))
flash("A user already exists with that email address.")
return render_template(
"sign-up.html",
form=form,
page_title="SIGN-UP",
page_color="black",
)
def register():
if current_user.is_authenticated:
return redirect(url_for('index'))
form = RegistrationForm()
if form.validate_on_submit():
user = User(username=form.username.data, email=form.email.data)
user.set_password(form.password.data)
db.session.add(user)
db.session.commit()
flash('Congratulations, you are now a registered user!')
return redirect(url_for('login'))
return render_template('register.html', title='Register', form=form)
def register():
if current_user.is_authenticated:
return redirect(url_for("main.home"))
form = RegistrationForm()
if form.validate_on_submit():
if form.validate_on_submit():
user = User(username=form.username.data, email=form.email.data)
user.set_password(form.password.data)
user.save()
flash("User registered! Please Log in", "success")
return redirect(url_for("users.login"))
return render_template("users/register.html", title="Register", form=form)
def fake_admin():
admin = User(name='Difan Xue',
username='******',
email='*****@*****.**',
bio=fake.sentence(),
location=fake.city(),
member_since=fake.date_this_decade(),
website='http://xuedifan.pythonanywhere.com',
)
admin.set_password('helloflask')
db.session.add(admin)
db.session.commit()
def register():
if request.method == 'POST':
email = request.form['email']
password = request.form['password']
new_user = User(email)
new_user.set_password(password)
db.session.add(new_user)
db.session.commit()
return redirect(url_for('users.login'))
return render_template('users/register.html')
def admin(username, password):
db.create_all()
user = User.query.first()
if user is not None:
click.echo('更新用户')
user.username = username
user.set_password(password)
else:
click.echo('创建用户')
user = User(username=username, name="ZY")
user.set_password(password)
db.session.add(user)
db.session.commit()
click.echo('创建管理员账号完成')
def register():
"""User Registration page view function"""
if current_user.is_authenticated:
return redirect(url_for('main.index'))
form = RegistrationForm()
if form.validate_on_submit():
user = User(username=form.username.data, email=form.email.data)
user.set_password(form.password.data)
db.session.add(user)
db.session.commit()
flash(_('Congratulations, you are now a registered user'))
return redirect(url_for('auth.login'))
return render_template(
'auth/register.html', title=_('Register'), form=form)
def register(request):
if ('email' or 'pwd') not in request.GET:
lf = LoginForm()
return render_to_response('register.html', {'lf': lf})
lf = LoginForm(request.GET)
email = lf.data['email']
pwd = lf.data['pwd']
user = User()
user.username = uuid.uuid1()
user.email = email
user.set_password(pwd)
user.desc = 'lazy'
user.save()
user.user_permissions = [Permission.objects.get(name='Can see news'), Permission.objects.get(name='Can edit news')]
return HttpResponseRedirect("/blog/login/")
def setUp(self):
# 更新配置
app.config.update(TESTING=True,
SQLALCHEMY_DATABASE_URI='sqlite:///:memory:')
db.create_all()
user = User(name='Test', username='******')
user.set_password('123456')
movies = Movies(title='test movie title', year=2020)
db.session.add_all([user, movies])
db.session.commit()
self.client = app.test_client() # 测试客户端
self.runner = app.test_cli_runner() # 创建测试
def post(self, request, format=None):
serializer = RegisterSerializer(data=request.data)
if not serializer.is_valid():
return Response(serializer.errors,
status=status.HTTP_400_BAD_REQUEST)
user = User.objects.get(username=serializer.data.get('username'))
if user is not None:
return Response(status=status.HTTP_400_BAD_REQUEST)
if serializer.data.get('password') != serializer.data.get('password1'):
return Response(status=status.HTTP_400_BAD_REQUEST)
user = User()
user.username = serializer.data.get('username')
user.set_password(serializer.data.get('password'))
user.save()
return Response(serializer.data, status=status.HTTP_201_CREATED)
def setUp(self):
app.config.update(TESTING=True,
SQLALCHEMY_DATABASE_URI='sqlite:///:memory:')
db.create_all()
user = User(name='Test', username='******')
user.set_password('123456')
movie = Movie(title='测试电影名称', year='2020')
db.session.add_all([user, movie])
db.session.commit()
self.client = app.test_client()
self.runner = app.test_cli_runner()
def fake_admin():
admin_role = Role.query.filter_by(name='Administrator').first()
admin = User(name='Xiao hao',
username='******',
email='*****@*****.**',
bio=fake.sentence(),
website='',
role_id=admin_role.id,
confirmed=True)
admin.set_password('xiaohaosang')
notification = Notification(message='Hello, welcome to Blog.',
receiver=admin)
db.session.add(notification)
db.session.add(admin)
db.session.commit()
def fake_user(count=10):
for i in range(count):
user = User(name=fake.name(),
username=fake.user_name(),
bio=fake.sentence(),
location=fake.city(),
website=fake.url(),
member_since=fake.date_this_decade(),
email=fake.email())
user.set_password('123456')
db.session.add(user)
try:
db.session.commit()
except IntegrityError:
db.session.rollback()
def setUp(self):
app = create_app('testing')
self.context = app.test_request_context()
self.context.push()
self.client = app.test_client()
self.runner = app.test_cli_runner()
db.create_all()
Role.init_role()
admin_user = User(email='*****@*****.**',
name='Admin',
username='******')
admin_user.set_password('123')
admin_user.set_role()
normal_user = User(email='*****@*****.**',
name='Normal User',
username='******')
normal_user.set_password('123')
unconfirmed_user = User(
email='*****@*****.**',
name='Unconfirmed',
username='******',
)
unconfirmed_user.set_password('123')
locked_user = User(email='*****@*****.**',
name='Locked User',
username='******',
locked=True)
locked_user.set_password('123')
locked_user.lock()
blocked_user = User(email='*****@*****.**',
name='Blocked User',
username='******',
active=False)
blocked_user.set_password('123')
category = Category(name='test category')
post = Post(title='test post', body='Test post', category=category)
comment = Comment(body='test comment body',
post=post,
author=normal_user)
db.session.add_all([
admin_user, normal_user, unconfirmed_user, locked_user,
blocked_user
])
db.session.commit()
def init_db():
db.create_all()
user1 = User(username="******")
user1.set_password("test")
user2 = User(username="******")
user2.set_password("password2")
post1 = Post(title="Post by user 1", contents="Contens from user 1", user=user1)
post2 = Post(title="Post by user 2", contents="Contens from user 2", user=user2)
db.session.add(user1, user2)
db.session.add(post1, post2)
db.session.commit()
def admin(username, password):
"""Create user."""
db.create_all()
user = User.query.first()
if user is not None:
click.echo('Updating user...')
user.username = username
user.set_password(password)
else:
click.echo('Creating user...')
user = User(username=username, name='Admin')
user.set_password(password)
db.session.add(user)
db.session.commit()
click.echo('Done.')
def setUp(self):
# 更新配置 在开发和测试时候通常配置是不一样
app.config.update(
TESTING=True, # 开启测试模式 出错时候不会输出多余的信息
SQLALCHEMY_DATABASE_URI=
'sqlite:///:memory:' # SQLite内存型数据库，不会干扰开发时使用的数据库文件
)
# 创建数据库和表
db.create_all()
# 创建测试数据，一个用户，一个电影信息
user = User(name='Test', username='******')
user.set_password('123456')
movie = Movie(title='测试电影名称', year='2020')
db.session.add_all([user, movie])
db.session.commit()
self.client = app.test_client() # 创建测试客户端（模拟客户端请求）
self.runner = app.test_cli_runner() # 创建测试命令运行器（触发自定义命令）
def register():
if current_user.is_authenticated:
return redirect(url_for('main.index'))
form = RegisterForm()
if form.validate_on_submit():
name = form.name.data
email = form.email.data.lower()
username = form.username.data
password = form.password.data
user = User(name=name, email=email, username=username)
user.set_password(password)
db.session.add(user)
db.session.commit()
token = generate_token(user=user, operation='confirm')
send_confirm_email(user=user, token=token)
flash('Confirm email sent, check your inbox.', 'info')
return redirect(url_for('.login'))
return render_template('auth/register.html', form=form)
def setUp(self):
# 更新配置
app.config.update(TESTING=True,
SQLALCHEMY_DATABASE_URI='sqlite:///:memory:')
# 创建数据库和表
db.create_all()
# 创建测试数据
user = User(name='Test1', username='******')
user.set_password('123456')
movie = Movie(title='Test Movie Title', year='2019')
# 使用add_all()方法一次添加多个模型类实例，传入列表
db.session.add_all([user, movie])
db.session.commit()
# 创建测试客户端
self.client = app.test_client()
# 创建测试命令运行器
self.runner = app.test_cli_runner()
def register(request):
if request.method == 'POST':
usr = User()
usr.username = request.POST.get('username')
# usr.password
usr.set_password(request.POST.get('password'))
usr.email = request.POST.get('email')
usr.phone = request.POST.get('phone')
usr.bio = request.POST.get('bio')
usr.save()
image_file = request.FILES.get('image')
if image_file:
img = ProfileImage()
img.file = image_file
img.alt_text = request.POST.get('alt_text', 'none')
img.description = request.POST.get('img_desc', 'none')
img.user = usr
img.save()
return redirect('login')
return render(request, 'register.html')
def test_password_hashing(self):
u = User(username='******')
u.set_password('cat')
self.assertFalse(u.check_password('dog'))
self.assertTrue(u.check_password('cat'))
