Exemple #1
0
def generate_project_employees():
    organisation = Organisation.get(
        "ORG", range_key="ORG#65a7b51dee5049565f65dc6839aa423d")
    projects = [
        project for project in Project.query(organisation.sk,
                                             Project.sk.startswith('PRO'))
    ]

    employees = [
        employee for employee in Employee.query(organisation.sk,
                                                Employee.sk.startswith('EMP'))
    ]

    for employee in employees:
        project = projects[0]
        serializer = ProjectEmployeeSerializer(
            data=dict(organisation_id=organisation.model_id,
                      employee_id=employee.model_id,
                      project_id=project.model_id))

        if serializer.is_valid():
            serializer.save()

        else:
            print(serializer.errors)

    for project in projects:
        employee = employees[0]
        serializer = ProjectEmployeeSerializer(
            data=dict(organisation_id=organisation.model_id,
                      employee_id=employee.model_id,
                      project_id=project.model_id))

        if serializer.is_valid():
            serializer.save()

        else:
            print(serializer.errors)