study-faraphel/M1-Palto-Server
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

fixed Absence factory not working

Browse source
This commit is contained in:
faraphel 2023-12-04 20:00:55 +01:00
parent 93036887c4
commit 7acc292bad
1 changed files with 3 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
Palto/Palto/factories.py
View file

@ -172,24 +172,10 @@ class FakeAbsenceFactory(factory.django.DjangoModelFactory):
message: str = factory.Faker("paragraph")
department: models.Department = factory.SubFactory(FakeDepartmentFactory)
student: models.User = factory.SubFactory(FakeUserFactory)
@factory.post_generation
def sessions(self, create, extracted, **kwargs):
if not create:
return
if extracted is not None:
self.sessions.add(*extracted)
else:
# all the sessions should be in the same department
department = FakeDepartmentFactory()
# create a group of between 1 and 8 sessions from the department
self.sessions.add(*[
FakeTeachingSessionFactory.create(unit__department=department)
for _ in range(random.randint(1, 8))
])
start: datetime = factory.Faker("date_time", tzinfo=timezone.get_current_timezone())
end: datetime = factory.LazyAttribute(lambda obj: obj.start + timedelta(days=random.randint(1, 8)))
class FakeAbsenceAttachmentFactory(factory.django.DjangoModelFactory):