oncall-engine/engine/apps/public_api/serializers/schedules_ical.py
Ildar Iskhakov 498e7ed5f3
Allow editing assigned team via public api (#1619)
# What this PR does

For example, changing the integration's team
```
curl "localhost:8080/api/v1/integrations/CLYV3QBVHDV3T/" \
  --request PUT \
  --header "Authorization: meow" \
  --header "Content-Type: application/json" --data '{"team_id": "TWP6JJJN6LYZX"}'
```

## Which issue(s) this PR fixes

## Checklist

- [ ] Unit, integration, and e2e (if applicable) tests updated
- [ ] Documentation added (or `pr:no public docs` PR label added if not
required)
- [ ] `CHANGELOG.md` updated (or `pr:no changelog` PR label added if not
required)
2023-04-05 06:39:30 +00:00

73 lines
2.5 KiB
Python

from apps.public_api.serializers.schedules_base import ScheduleBaseSerializer
from apps.schedules.models import OnCallScheduleICal
from apps.schedules.tasks import (
drop_cached_ical_task,
schedule_notify_about_empty_shifts_in_schedule,
schedule_notify_about_gaps_in_schedule,
)
from common.api_helpers.custom_fields import TeamPrimaryKeyRelatedField
from common.api_helpers.utils import validate_ical_url
class ScheduleICalSerializer(ScheduleBaseSerializer):
class Meta:
model = OnCallScheduleICal
fields = [
"id",
"team_id",
"name",
"ical_url_primary",
"ical_url_overrides",
"slack",
"on_call_now",
]
extra_kwargs = {
"ical_url_primary": {"required": True, "allow_null": False},
"ical_url_overrides": {"required": False, "allow_null": True},
}
def validate_ical_url_primary(self, url):
return validate_ical_url(url)
def validate_ical_url_overrides(self, url):
return validate_ical_url(url)
class ScheduleICalUpdateSerializer(ScheduleICalSerializer):
team_id = TeamPrimaryKeyRelatedField(required=False, allow_null=True, source="team")
class Meta:
model = OnCallScheduleICal
fields = [
"id",
"team_id",
"name",
"ical_url_primary",
"ical_url_overrides",
"slack",
"on_call_now",
]
extra_kwargs = {
"name": {"required": False},
"ical_url_primary": {"required": False, "allow_null": False},
"ical_url_overrides": {"required": False, "allow_null": True},
}
def update(self, instance, validated_data):
ical_changed = False
validated_data = self._correct_validated_data(validated_data)
if "ical_url_primary" in validated_data and validated_data["ical_url_primary"] != instance.ical_url_primary:
ical_changed = True
if (
"ical_url_overrides" in validated_data
and validated_data["ical_url_overrides"] != instance.ical_url_overrides
):
ical_changed = True
if ical_changed:
drop_cached_ical_task.apply_async(
(instance.pk,),
)
schedule_notify_about_empty_shifts_in_schedule.apply_async((instance.pk,))
schedule_notify_about_gaps_in_schedule.apply_async((instance.pk,))
return super().update(instance, validated_data)