diff --git a/aleksis/apps/chronos/schema/__init__.py b/aleksis/apps/chronos/schema/__init__.py index 253f56f183a9391bebf1cc2c4b1322864a4de48b..279003489d17818eda4487867de799fb1eeef213 100644 --- a/aleksis/apps/chronos/schema/__init__.py +++ b/aleksis/apps/chronos/schema/__init__.py @@ -2,13 +2,13 @@ from datetime import timezone import graphene from graphene_django import DjangoObjectType -from graphene_django_cud.mutations import ( - DjangoBatchCreateMutation, - DjangoBatchDeleteMutation, - DjangoBatchPatchMutation, -) from aleksis.core.models import Group, Person, Room +from aleksis.core.schema.base import ( + BaseBatchCreateMutation, + BaseBatchDeleteMutation, + BaseBatchPatchMutation, +) from ..models import LessonEvent from ..util.chronos_helpers import get_groups, get_rooms, get_teachers @@ -80,7 +80,7 @@ class DatetimeTimezoneMixin: return value -class AmendLessonBatchCreateMutation(DatetimeTimezoneMixin, DjangoBatchCreateMutation): +class AmendLessonBatchCreateMutation(DatetimeTimezoneMixin, BaseBatchCreateMutation): class Meta: model = LessonEvent permissions = ("chronos.edit_substitution_rule",) @@ -98,12 +98,13 @@ class AmendLessonBatchCreateMutation(DatetimeTimezoneMixin, DjangoBatchCreateMut @classmethod def before_save(cls, root, info, input, created_objects): # noqa: A002 + super().before_save(root, info, input, created_objects) for obj in created_objects: obj.timezone = obj.amends.timezone return created_objects -class AmendLessonBatchPatchMutation(DatetimeTimezoneMixin, DjangoBatchPatchMutation): +class AmendLessonBatchPatchMutation(DatetimeTimezoneMixin, BaseBatchPatchMutation): class Meta: model = LessonEvent permissions = ("chronos.edit_substitution_rule",) @@ -111,12 +112,13 @@ class AmendLessonBatchPatchMutation(DatetimeTimezoneMixin, DjangoBatchPatchMutat @classmethod def before_save(cls, root, info, input, updated_objects): # noqa: A002 + super().before_save(root, info, input, updated_objects) for obj in updated_objects: obj.timezone = obj.amends.timezone return updated_objects -class AmendLessonBatchDeleteMutation(DjangoBatchDeleteMutation): +class AmendLessonBatchDeleteMutation(BaseBatchDeleteMutation): class Meta: model = LessonEvent permissions = ("chronos.delete_substitution_rule",)