mirror of
https://github.com/TurTaskProject/TurTaskWeb.git
synced 2025-12-19 14:04:07 +01:00
74 lines
2.8 KiB
Python
74 lines
2.8 KiB
Python
from datetime import datetime, timedelta
|
|
|
|
from django.utils import timezone
|
|
|
|
from rest_framework import viewsets
|
|
from rest_framework.response import Response
|
|
from rest_framework.permissions import IsAuthenticated
|
|
|
|
from tasks.utils import get_service
|
|
from tasks.models import Todo, RecurrenceTask
|
|
from tasks.serializers import TodoUpdateSerializer, RecurrenceTaskUpdateSerializer
|
|
|
|
|
|
class GoogleCalendarEventViewset(viewsets.ViewSet):
|
|
permission_classes = (IsAuthenticated,)
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__()
|
|
self.current_time = datetime.now(tz=timezone.utc).isoformat()
|
|
self.event_fields = 'items(id,summary,description,created,recurringEventId,updated,start,end)'
|
|
|
|
def _validate_serializer(self, serializer):
|
|
if serializer.is_valid():
|
|
serializer.save()
|
|
return Response("Validate Successfully", status=200)
|
|
return Response(serializer.errors, status=400)
|
|
|
|
def post(self, request):
|
|
service = get_service(request)
|
|
events = service.events().list(calendarId='primary', fields=self.event_fields).execute()
|
|
for event in events.get('items', []):
|
|
if event.get('recurringEventId'):
|
|
continue
|
|
event['start_datetime'] = event.get('start').get('dateTime')
|
|
event['end_datetime'] = event.get('end').get('dateTime')
|
|
event.pop('start')
|
|
event.pop('end')
|
|
try:
|
|
task = Todo.objects.get(google_calendar_id=event['id'])
|
|
serializer = TodoUpdateSerializer(instance=task, data=event)
|
|
return self._validate_serializer(serializer)
|
|
except Todo.DoesNotExist:
|
|
serializer = TodoUpdateSerializer(data=event, user=request.user)
|
|
return self._validate_serializer(serializer)
|
|
|
|
def list(self, request, days=7):
|
|
max_time = (datetime.now(tz=timezone.utc) + timedelta(days=days)).isoformat()
|
|
|
|
service = get_service(request)
|
|
events = []
|
|
next_page_token = None
|
|
|
|
while True:
|
|
query = service.events().list(
|
|
calendarId='primary',
|
|
timeMin=self.current_time,
|
|
timeMax=max_time,
|
|
maxResults=200,
|
|
singleEvents=True,
|
|
orderBy='startTime',
|
|
pageToken=next_page_token,
|
|
fields='items(id,summary,description,created,recurringEventId,updated,start,end)',
|
|
)
|
|
|
|
page_results = query.execute()
|
|
page_events = page_results.get('items', [])
|
|
|
|
events.extend(page_events)
|
|
next_page_token = page_results.get('nextPageToken')
|
|
|
|
if next_page_token is None:
|
|
break
|
|
|
|
return Response(events, status=200) |