Merge pull request #869 from arivarton/main

Google calendar module

bumblebee_status/modules/contrib/gcalendar.py

@@ -0,0 +1,149 @@
+"""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