def parse_get_group_name_list_mask2(self, result):
gmarklist = result["gmarklist"]
gmasklist = result["gmasklist"]
gnamelist = result["gnamelist"]
ret_groups = {}
# parse gmarklist 似乎没有没使用
# parse gmasklist [群屏蔽] 暂时没写，我的qq获取没数据
# parse gnamelist
for i in range(0, len(gnamelist)):
group_obj = Group()
group_obj.flag = gnamelist[i]["flag"]
group_obj.name = gnamelist[i]["name"]
group_obj.gid = gnamelist[i]["gid"]
group_obj.code = gnamelist[i]["code"]
ret_groups[group_obj.gid] = group_obj
return ret_groups
def seed_db():
from models.Profile import Profile
from models.User import User
from models.Group import Group
from models.Content import Content
from models.Group_members import GroupMembers
from models.Admin import Admin
from main import bcrypt
import random
from faker import Faker
faker = Faker()
users = []
contents = []
profile_ids = list(range(1, 11))
random.shuffle(profile_ids)
admins = []
for i in range(1, 6):
user = User()
user.email = f"test{i}@test.com"
user.password = bcrypt.generate_password_hash("123456").decode("utf-8")
user.subscription_status = random.choice([0, 1])
db.session.add(user)
users.append(user)
db.session.commit()
print("User table seeded")
for i in range(30):
content = Content()
content.title = faker.sentence()
content.genre = faker.word()
content.year = faker.year()
db.session.add(content)
contents.append(content)
db.session.commit()
print("Content table seeded")
for i in range(10):
content = random.sample(contents, k=2)
restrictions = ("G", "PG", "M", "MA15+", "R18+")
profile = Profile()
profile.name = faker.first_name_nonbinary()
profile.restrictions = random.choice(restrictions)
profile.user_id = random.choice(users).user_id
profile.unrecommend.extend(content)
db.session.add(profile)
db.session.commit()
print("Profile table seeded")
for i in range(10):
content = random.sample(contents, k=3)
group = Group()
group.name = faker.word()
group.description = faker.text()
group.content.extend(content)
admin = GroupMembers()
admin.groups = group
admin.profile_id = profile_ids.pop(0)
admin.admin = True
member_ids = [i for i in range(1, 11) if i != admin.profile_id]
random.shuffle(member_ids)
for i in range(2):
member = GroupMembers()
member.groups = group
member.profile_id = member_ids.pop()
member.admin = False
db.session.add(group)
db.session.commit()
print("Group table seeded")
for i in range(1, 3):
admin = Admin()
admin.username = f"Admin{i}"
admin.password = bcrypt.generate_password_hash("654321").decode(
"utf-8")
db.session.add(admin)
admins.append(admin)
db.session.commit()
print("Admin table seeded")