def fill_tasks(entry_number=10):
for _ in range(entry_number):
name = f'TEST{randint(10000, 99999999)}'
offices = []
# First task will be uncommon task
number_of_offices = 1 if _ == 0 else choice(range(1, 5))
while number_of_offices > len(offices):
office = choice(Office.query.all())
if office not in offices:
offices.append(office)
task = Task(name)
db.session.add(task)
db.session.commit()
task.offices = offices
db.session.commit()
# TODO: get refactor the helper to fit the testing process
# def fill_tickets(entery_number=10, s_task=None):
# for i in range(entery_number):
# forin = Task.query if s_task is None else Task.query.filter_by(
# id=s_task)
# for task in forin:
# num = Serial.query.order_by(Serial.timestamp.desc()).first()
# num = num.number if num is not None else None
# name = choice(names)
# t_id = task.id
# f_id = choice(task.offices).id
# # if i >= 11: WTF ?!
# db.session.add(Serial(number=num + 1,
# office_id=f_id,
# task_id=t_id, name=name, n=True))
# for a in range(Waiting.query.count(), 11):
# for b in Serial.query.filter_by(p=False).order_by(Serial.timestamp):
# if Waiting.query.filter_by(office_id=b.office_id,
# number=b.number,
# task_id=b.task_id).first() is None:
# db.session.add(Waiting(b.number, b.office_id,
# b.task_id, b.name, b.n))
# db.session.commit()
def fill_tasks(entry_number=ENTRY_NUMBER):
for _ in range(entry_number):
name = f'TEST{randint(10000, 99999999)}'
offices = []
# First task will be uncommon task and the second is common
number_of_offices = 1 if _ == 0 else 2 if _ == 1 else choice(range(1, 5))
while number_of_offices > len(offices):
office = choice(Office.query.all())
if office not in offices:
offices.append(office)
task = Task(name)
db.session.add(task)
db.session.commit()
task.offices = offices
# Add tasks initial tickets
db.session.add(Serial(number=100, office_id=office.id, task_id=task.id))
db.session.commit()
