endorsement/test.py

43 lines
1,010 B
Python
Raw Normal View History

import random
import string
from app import db
from models import Achievement, User
db.drop_all()
db.create_all()
persoonA = User(id="AAA")
persoonB = User(id="BBB")
persoonC = User(id="CCC")
achievementA = Achievement(name="AA", achieved_by=persoonA.id)
achievementB = Achievement(name="BB", achieved_by=persoonB.id)
achievementA.endorsed_by.append(persoonC)
# Long generated lists
for i in range(30):
achievement = Achievement(name=''.join(
random.choices(string.ascii_uppercase + string.digits, k=6)),
achieved_by=persoonA.id)
db.session.add(achievement)
people = [
User(id=''.join(random.choices(string.ascii_uppercase +
string.digits, k=6))) for i in range(30)
]
for person in people:
db.session.add(person)
achievementA.endorsed_by.append(person)
db.session.add(persoonA)
db.session.add(persoonB)
db.session.add(persoonC)
db.session.add(achievementA)
db.session.add(achievementB)
db.session.commit()