Adding a script to test retrieval of calendar events
This commit is contained in:
parent
bbfa4921a7
commit
5919bca2ca
1
.gitignore
vendored
1
.gitignore
vendored
@ -1 +1,2 @@
|
|||||||
credentials.json
|
credentials.json
|
||||||
|
token.json
|
@ -10,7 +10,7 @@ from googleapiclient.discovery import build
|
|||||||
from googleapiclient.errors import HttpError
|
from googleapiclient.errors import HttpError
|
||||||
from pprint import pprint
|
from pprint import pprint
|
||||||
|
|
||||||
SCOPES = ['https://www.googleapis.comd/auth/calendar.readonly']
|
SCOPES = ['https://www.googleapis.com/auth/calendar.readonly']
|
||||||
|
|
||||||
def authorize():
|
def authorize():
|
||||||
creds = None
|
creds = None
|
||||||
|
89
get_calendar_events.py
Normal file
89
get_calendar_events.py
Normal file
@ -0,0 +1,89 @@
|
|||||||
|
from __future__ import print_function
|
||||||
|
|
||||||
|
import datetime
|
||||||
|
import os.path
|
||||||
|
import argparse
|
||||||
|
|
||||||
|
from google.auth.transport.requests import Request
|
||||||
|
from google.oauth2.credentials import Credentials
|
||||||
|
from google_auth_oauthlib.flow import InstalledAppFlow
|
||||||
|
from googleapiclient.discovery import build
|
||||||
|
from googleapiclient.errors import HttpError
|
||||||
|
from pprint import pprint
|
||||||
|
|
||||||
|
SCOPES = ['https://www.googleapis.com/auth/calendar.readonly']
|
||||||
|
|
||||||
|
def authorize():
|
||||||
|
creds = None
|
||||||
|
|
||||||
|
if os.path.exists('token.json'):
|
||||||
|
creds = Credentials.from_authorized_user_file('token.json', SCOPES)
|
||||||
|
|
||||||
|
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('credentials.json', SCOPES)
|
||||||
|
creds = flow.run_local_server(port=7384)
|
||||||
|
|
||||||
|
with open('token.json', 'w') as token:
|
||||||
|
token.write(creds.to_json())
|
||||||
|
|
||||||
|
return creds
|
||||||
|
|
||||||
|
def construct_calendar_service(creds):
|
||||||
|
return build('calendar', 'v3', credentials=creds)
|
||||||
|
|
||||||
|
def list_all_calendars(calendar_service):
|
||||||
|
page_token = None
|
||||||
|
list_of_entries = []
|
||||||
|
|
||||||
|
while True:
|
||||||
|
calendar_list = calendar_service.calendarList().list(pageToken=page_token).execute()
|
||||||
|
|
||||||
|
list_of_entries.extend(calendar_list['items'])
|
||||||
|
|
||||||
|
page_token = calendar_list.get('nextPageToken')
|
||||||
|
|
||||||
|
if not page_token:
|
||||||
|
break
|
||||||
|
|
||||||
|
return list_of_entries
|
||||||
|
|
||||||
|
def get_calendar_events(calendar_service, calendar_id, num_events):
|
||||||
|
now = datetime.datetime.utcnow().isoformat() + "Z"
|
||||||
|
|
||||||
|
events_result = calendar_service.events().list(
|
||||||
|
calendarId=calendar_id,
|
||||||
|
timeMin=now,
|
||||||
|
maxResults=num_events,
|
||||||
|
singleEvents=True,
|
||||||
|
orderBy='startTime'
|
||||||
|
).execute()
|
||||||
|
|
||||||
|
events = events_result.get('items', [])
|
||||||
|
|
||||||
|
return events
|
||||||
|
|
||||||
|
def main():
|
||||||
|
argument_parser = argparse.ArgumentParser()
|
||||||
|
argument_parser.add_argument('calendarname', help="The calendar name you want to see events from")
|
||||||
|
argument_parser.add_argument('numberofevents', help="The number of events to return")
|
||||||
|
parsed_args = argument_parser.parse_args()
|
||||||
|
|
||||||
|
creds = authorize()
|
||||||
|
|
||||||
|
service = construct_calendar_service(creds)
|
||||||
|
|
||||||
|
all_calendars = list_all_calendars(service)
|
||||||
|
|
||||||
|
for element in all_calendars:
|
||||||
|
if element["summary"] == parsed_args.calendarname:
|
||||||
|
events = get_calendar_events(service, element["id"], parsed_args.numberofevents)
|
||||||
|
|
||||||
|
pprint(events)
|
||||||
|
|
||||||
|
break
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
main()
|
Loading…
Reference in New Issue
Block a user