tobi-wan-kenobi
GitHub
commit
f64f71b6e2
1 changed files with 149 additions and 0 deletions
149
bumblebee_status/modules/contrib/gcalendar.py
Normal file
149
bumblebee_status/modules/contrib/gcalendar.py
Normal file
|
|
@ -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
|
||||||
Loading…
Reference in a new issue