def generate_projects():
organisations = [org for org in Organisation.query('ORG', limit=4)]
project_types = ['freelance', 'contract', 'full_time']
project_status = ['pending', 'in_progress', 'on_hold', 'complete']
for _ in range(40):
print("Generating a project")
organisation = random.choice(organisations)
serializer = ProjectSerializer(
data=dict(organisation_id=organisation.model_id,
name=fake.word(),
project_type=random.choice(project_types),
status=random.choice(project_status)))
if serializer.is_valid():
serializer.save()
def generate_employees():
organisations = [org for org in Organisation.query('ORG', limit=4)]
for _ in range(20):
print("Generating Employee")
organisation = random.choice(organisations)
days = [_ for _ in range(1, 29)]
months = [_ for _ in range(1, 13)]
years = [_ for _ in range(1993, 2001)]
serializer = EmployeeSerializer(
data=dict(organisation_id=organisation.model_id,
name=f"{fake.first_name()} {fake.last_name()}",
email=fake.email(),
date_of_birth=timezone.datetime(
year=random.choice(years),
month=random.choice(months),
day=random.choice(days)).strftime('%Y-%m-%d')))
if serializer.is_valid():
serializer.save()
else:
print(serializer.errors)
