128 lines
4.8 KiB
Python
128 lines
4.8 KiB
Python
from app import app
|
|
from datetime import datetime, timedelta, timezone
|
|
import os
|
|
import caldav
|
|
import datetime
|
|
from icalendar import cal, Event
|
|
from flask import render_template, url_for, send_file, send_from_directory, request
|
|
from .models import Event
|
|
import requests
|
|
import imgkit
|
|
from PIL import Image, ImageOps
|
|
import io
|
|
caldav_url = os.getenv('caldav_url')
|
|
username = os.getenv('username')
|
|
password = os.getenv('password')
|
|
cal_id = os.getenv('cal_id')
|
|
from .weather import weather
|
|
|
|
def remove_emoji(string):
|
|
import re
|
|
emoji_pattern = re.compile("["
|
|
u"\U0001F600-\U0001F64F" # emoticons
|
|
u"\U0001F300-\U0001F5FF" # symbols & pictographs
|
|
u"\U0001F680-\U0001F6FF" # transport & map symbols
|
|
u"\U0001F1E0-\U0001F1FF" # flags (iOS)
|
|
u"\U00002500-\U00002BEF" # chinese char
|
|
u"\U00002702-\U000027B0"
|
|
u"\U00002702-\U000027B0"
|
|
u"\U000024C2-\U0001F251"
|
|
u"\U0001f926-\U0001f937"
|
|
u"\U00010000-\U0010ffff"
|
|
u"\u2640-\u2642"
|
|
u"\u2600-\u2B55"
|
|
u"\u200d"
|
|
u"\u23cf"
|
|
u"\u23e9"
|
|
u"\u231a"
|
|
u"\ufe0f" # dingbats
|
|
u"\u3030"
|
|
"]+", flags=re.UNICODE)
|
|
return emoji_pattern.sub(r'', string)
|
|
|
|
def daterange(start_date, end_date):
|
|
for n in range(int((end_date - start_date).days)):
|
|
yield datetime.datetime.date(start_date + timedelta(n))
|
|
|
|
@app.route('/')
|
|
def dashboard():
|
|
today = datetime.datetime.now(tz=datetime.datetime.now(timezone.utc).astimezone().tzinfo)
|
|
# today = datetime.datetime(2022,6,5, tzinfo=datetime.datetime.now(timezone.utc).astimezone().tzinfo)
|
|
start_of_week = today - timedelta(days=(today.weekday()+1)) # Monday
|
|
end_of_week = start_of_week + timedelta(days=8) # Sunday
|
|
|
|
events = []
|
|
|
|
for url in [
|
|
'https://www.calendarlabs.com/ical-calendar/ics/76/US_Holidays.ics',
|
|
]:
|
|
r = requests.get(url)
|
|
c = cal.Calendar.from_ical(r.content)
|
|
|
|
events += Event.fromIcalendar(c)
|
|
|
|
with caldav.DAVClient(url=caldav_url, username=username, password=password) as client:
|
|
my_principal = client.principal()
|
|
|
|
calendars = my_principal.calendars()
|
|
for id in [cal_id]:
|
|
calendar = my_principal.calendar(cal_id=id)
|
|
events += Event.fromCalDavEvents(calendar.date_search(
|
|
start=start_of_week, end=end_of_week, expand=False))
|
|
|
|
for url in [
|
|
'http://ical-cdn.teamsnap.com/team_schedule/5f1ddc9e-15b0-4912-84a2-11cc70e9e375.ics'
|
|
]:
|
|
r = requests.get(url)
|
|
c = cal.Calendar.from_ical(r.content)
|
|
|
|
events += Event.fromIcalendar(c)
|
|
|
|
days = []
|
|
for single_date in daterange(start_of_week, end_of_week):
|
|
days_events = []
|
|
for e in events:
|
|
if (e.dtstart.date() <= single_date <= e.dtend.date()):
|
|
e.summary = remove_emoji(e.summary)
|
|
days_events.append(e)
|
|
|
|
days.append((single_date, days_events))
|
|
|
|
extra_debug = request.args.get('extra', '')
|
|
# r = "<br>".join([event.vobject_instance.vevent.summary.value for event in events_fetched if event.vobject_instance.vevent.dtstart.value < datetime.now()])
|
|
return render_template("dashboard.html",
|
|
days=days,
|
|
today=today,
|
|
weather=weather(),
|
|
extra_debug=extra_debug,
|
|
HOST = os.getenv('HOST'),
|
|
HOST_PORT = os.getenv('HOST_PORT')
|
|
)
|
|
|
|
@app.route('/image')
|
|
def dashboard_image():
|
|
extension = "png"
|
|
page = dashboard()
|
|
out_file = os.path.join(os.path.dirname(__file__),
|
|
'static', f'out.{extension}'
|
|
)
|
|
|
|
s = imgkit.from_string(
|
|
page,
|
|
out_file,
|
|
options={
|
|
'width':600,
|
|
'height':800,
|
|
"disable-smart-width": "",
|
|
'enable-local-file-access': "",
|
|
'allow': os.path.join(os.path.dirname(__file__),
|
|
'static',
|
|
)
|
|
},
|
|
# css=os.path.join(os.path.dirname(__file__),'static','style.css')
|
|
)
|
|
image_file = Image.open(out_file) # open colour image
|
|
# image_file = ImageOps.grayscale(image_file)
|
|
image_file.save(out_file, 'BMP')
|
|
return send_from_directory(directory = 'static', path=f'out.{extension}', as_attachment=False, attachment_filename=f'out.{extension}')
|