-
Notifications
You must be signed in to change notification settings - Fork 42
Expand file tree
/
Copy pathcommands.py
More file actions
92 lines (72 loc) · 3.13 KB
/
commands.py
File metadata and controls
92 lines (72 loc) · 3.13 KB
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
from models.user import PermissionModel,RoleModel,PermissionEnum,UserModel
from models.post import BoardModel,PostModel
import click
from exts import db
from faker import Faker
import random
def create_permission():
for permission_name in dir(PermissionEnum):
if permission_name.startswith("__"):
continue
permission = PermissionModel(name=getattr(PermissionEnum,permission_name))
db.session.add(permission)
db.session.commit()
click.echo("权限添加成功！")
def create_role():
# 稽查员
inspector = RoleModel(name="稽查",desc="负责审核帖子和评论是否合法合规！")
inspector.permissions = PermissionModel.query.filter(PermissionModel.name.in_([PermissionEnum.POST,PermissionEnum.COMMENT])).all()
# 运营
operator = RoleModel(name="运营",desc="负责网站持续正常运营！")
operator.permissions = PermissionModel.query.filter(PermissionModel.name.in_([
PermissionEnum.POST,
PermissionEnum.COMMENT,
PermissionEnum.BOARD,
PermissionEnum.FRONT_USER
])).all()
# 管理员
administrator = RoleModel(name="管理员",desc="负责整个网站所有工作！")
administrator.permissions = PermissionModel.query.all()
db.session.add_all([inspector,operator,administrator])
db.session.commit()
click.echo("角色添加成功！")
def create_test_user():
admin_role = RoleModel.query.filter_by(name="管理员").first()
zhangsan = UserModel(username="张三",email="[email protected]",password="111111",is_staff=True,role=admin_role)
operator_role = RoleModel.query.filter_by(name="运营").first()
lisi = UserModel(username="李四",email="[email protected]",password="111111",is_staff=True,role=operator_role)
inspector_role = RoleModel.query.filter_by(name="稽查").first()
wangwu = UserModel(username="王五",email="[email protected]",password="111111",is_staff=True,role=inspector_role)
db.session.add_all([zhangsan,lisi,wangwu])
db.session.commit()
click.echo("测试用户添加成功！")
@click.option("--username",'-u')
@click.option("--email",'-e')
@click.option("--password",'-p')
def create_admin(username,email,password):
admin_role = RoleModel.query.filter_by(name="管理员").first()
admin_user = UserModel(username=username, email=email, password=password, is_staff=True, role=admin_role)
db.session.add(admin_user)
db.session.commit()
click.echo("管理员创建成功！")
def create_board():
board_names = ['Python语法', 'web开发', '数据分析', '测试开发', '运维开发']
for board_name in board_names:
board = BoardModel(name=board_name)
db.session.add(board)
db.session.commit()
click.echo("板块添加成功！")
def create_test_post():
fake = Faker(locale="zh_CN")
author = UserModel.query.first()
boards = BoardModel.query.all()
click.echo("开始生成测试帖子...")
for x in range(98):
title = fake.sentence()
content = fake.paragraph(nb_sentences=10)
random_index = random.randint(0,4)
board = boards[random_index]
post = PostModel(title=title, content=content, board=board, author=author)
db.session.add(post)
db.session.commit()
click.echo("测试帖子生成成功！")