Google calendar module.

bumblebee_status/modules/contrib/gcalendar.py

@@ -0,0 +1,113 @@
+"""Displays first upcoming event in google calendar.
+
+Events that are set as 'all-day' will not be shown.
+
+Requires credentials.json from a google api application where the google calendar api is installed.
+On first time run the browser will open and google will ask for permission for this app to access the google calendar and then save a .gcalendar_token.json file to the credentials_path directory which stores this permission.
+
+A refresh is done every 15 minutes.
+
+Parameters:
+    * gcalendar.time_format: Format time output. Defaults to "%H:%M".
+    * gcalendar.date_format: Format date output. Defaults to "%d.%m.%y".
+    * gcalendar.credentials_path: Path to credentials.json. Defaults to "~/".
+
+Requires these pip packages:
+   * google-api-python-client 
+   * google-auth-httplib2 
+   * google-auth-oauthlib
+"""
+
+# This import belongs to the google code
+from __future__ import print_function
+
+from dateutil.parser import parse as dtparse
+
+import core.module
+import core.widget
+import core.decorators
+
+import datetime
+import os.path
+
+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
+
+class Module(core.module.Module):
+    @core.decorators.every(minutes=15)
+    def __init__(self, config, theme):
+        super().__init__(config, theme, core.widget.Widget(self.first_event))
+        self.__time_format = self.parameter("time_format", "%H:%M")
+        self.__date_format = self.parameter("date_format", "%d.%m.%y")
+        self.__credentials_path = os.path.expanduser(self.parameter("credentials_path", "~/"))
+        self.__credentials = os.path.join(self.__credentials_path, 'credentials.json')
+        self.__token = os.path.join(self.__credentials_path, '.gcalendar_token.json')
+
+    def first_event(self, widget):
+        SCOPES = ['https://www.googleapis.com/auth/calendar.readonly']
+
+        """Shows basic usage of the Google Calendar API.
+        Prints the start and name of the next 10 events on the user's calendar.
+        """
+        creds = None
+        # The file token.json 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(self.__token):
+            creds = Credentials.from_authorized_user_file(self.__token, SCOPES)
+        # 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(
+                    self.__credentials, SCOPES)
+                creds = flow.run_local_server(port=0)
+            # Save the credentials for the next run
+            with open(self.__token, 'w') as token:
+                token.write(creds.to_json())
+
+        try:
+            service = build('calendar', 'v3', credentials=creds)
+
+            # Call the Calendar API
+            now = datetime.datetime.utcnow().isoformat() + 'Z'  # 'Z' indicates UTC time
+            end = (datetime.datetime.utcnow() + datetime.timedelta(days=7)).isoformat() + 'Z'  # 'Z' indicates UTC time
+            # Get all calendars
+            calendar_list = service.calendarList().list().execute()
+            event_list = []
+            for calendar_list_entry in calendar_list['items']:
+                calendar_id = calendar_list_entry['id']
+                events_result = service.events().list(calendarId=calendar_id, timeMin=now,
+                                                      timeMax=end,
+                                                      singleEvents=True,
+                                                      orderBy='startTime').execute()
+                events = events_result.get('items', [])
+
+                if not events:
+                    return 'No upcoming events found.'
+
+                for event in events:
+                    start = dtparse(event['start'].get('dateTime', event['start'].get('date')))
+                    # Only add to list if not an whole day event
+                    if start.tzinfo:
+                        event_list.append({'date': start,
+                            'summary': event['summary'],
+                            'type': event['eventType']})
+            sorted_list = sorted(event_list, key=lambda t: t['date'])
+
+            for gevent in sorted_list:
+                if gevent['date'] >= datetime.datetime.now(datetime.timezone.utc):
+                    if gevent['date'].date() == datetime.datetime.utcnow().date():
+                        return str('%s %s' % (gevent['date'].astimezone().strftime(f'{self.__time_format}'), gevent['summary']))
+                    else:
+                        return str('%s %s' % (gevent['date'].astimezone().strftime(f'{self.__date_format} {__time_format}'), gevent['summary']))
+            return 'No upcoming events found.'
+
+        except:
+            return None
+
+# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4