import datetime from rental.pricing import get_reservation_price from django.shortcuts import get_object_or_404 from rental.models import Reservation, Place import datetime from google.auth.transport.requests import Request from google_auth_oauthlib.flow import InstalledAppFlow from googleapiclient.discovery import build import os.path import pickle def get_bookings(place): """ returns a list of all related place reservations """ booked_dates = Reservation.objects.all() return [booking for booking in booked_dates if booking.place.name == f"{place.name}"] def check_availability(place, start_date, end_date): """ check if the related place is available during a given period """ bookings = get_bookings(place) for booking in bookings: if (booking.start <= start_date <= booking.end) or (booking.start <= end_date <= booking.end): return False return True def synchronize_calendars(): """ Get a complete list of existing bookings in calendar Creates reservation if not in db, update if already in db Delete from db reservation deleted from cal """ creds = None # If modifying these scopes, delete the file token.pickle. SCOPES = ['https://www.googleapis.com/auth/calendar.readonly'] # The file token.pickle stores the user's access and refresh tokens, and is # created automatically when the authorization flow completes for the first # time. if os.path.exists('token.pickle'): with open('token.pickle', 'rb') as token: creds = pickle.load(token) # If there are no (valid) credentials available, let the user log in. if not creds or not creds.valid: if creds and creds.expired and creds.refresh_token: creds.refresh(Request()) else: flow = InstalledAppFlow.from_client_secrets_file( 'client_secrets.json', scopes=SCOPES, redirect_uri="http://localhost:8080/") creds = flow.run_local_server() # Save the credentials for the next run with open('token.pickle', 'wb') as token: pickle.dump(creds, token) service = build('calendar', 'v3', credentials=creds) calendars = { 'T2': "burik7aclvhc7vsboh06c179uo@group.calendar.google.com", 'T3': "fu7h30p0gk4a2p4nvo7nsbgpok@group.calendar.google.com" } now = datetime.datetime.utcnow().isoformat() + 'Z' # 'Z' indicates UTC time for calendar in calendars: print(f"Upcoming {calendar} events:") events_result = service.events().list( calendarId=calendars[calendar], timeMin=now, singleEvents=True, orderBy='startTime').execute() events = events_result.get('items', []) if not events: print('No upcoming events found.') else: reservation = {} for index, event in enumerate(events): reservation[index] = { 'place': calendar, 'guest': event['summary'], 'start': event['start'].get('dateTime', event['start'].get('date')), 'end': event['end'].get('dateTime', event['end'].get('date')) } print(reservation[index]) # if booking not in db -> create # if booking in db and modification_date_cal > update_date_db -> update place = get_object_or_404(Place, name=calendar) price = get_reservation_price( place, reservation['start'], reservation['end']) # trouver si guest existe déjà, créer sinon guest = Guest.objects.create(name=reservation['guest']) Reservation.objects.create( place=place, guest=guest, start=start, end=end, price=price ) if __name__ == '__main__': synchronize_calendars()