forked from mendhak/waveshare-epaper-display
-
Notifications
You must be signed in to change notification settings - Fork 0
/
screen-calendar-month.py
executable file
·69 lines (54 loc) · 2.32 KB
/
screen-calendar-month.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
import datetime
import calendar
from utility import update_svg, configure_locale, configure_logging
import locale
import babel
import logging
from collections import deque
import drawsvg as draw
configure_logging()
configure_locale()
try:
babel_locale = babel.Locale(locale.getlocale()[0])
logging.debug(babel_locale)
except babel.core.UnknownLocaleError:
logging.error("Could not get locale from environment. Using default locale.")
babel_locale = babel.Locale("")
def main():
logging.info("Generating SVG for calendar month")
# Python does not know about the locale's first day of week 🤦 https://stackoverflow.com/a/4265852/974369
# Use babel to set it instead
calendar.setfirstweekday(babel_locale.first_week_day)
now = datetime.datetime.now()
current_year, current_month, current_day = now.year, now.month, now.day
# Get this month's calendar as a matrix
cal = calendar.monthcalendar(current_year, current_month)
# Create a new SVG drawing
dwg = draw.Drawing(width=500, height=500, origin=(0,0), id='month-cal', transform='translate(500, 240)')
cell_width = 40
cell_height = 30
# Have the day abbreviations respect the locale's first day of the week
day_abbr = deque(list(calendar.day_abbr))
day_abbr.rotate(-calendar.firstweekday())
# Header for days of the week
for i, day in enumerate(day_abbr):
dwg.append(draw.Text(day[:2], font_size=None, x=i*cell_width + 20, y= 20, fill='black' ))
# Days of the month per week
for i, week in enumerate(cal):
for j, day in enumerate(week):
if day != 0: # calendar.monthcalendar pads with 0s
text_fill = 'black'
if day == current_day:
text_fill = 'red'
text = draw.Text(str(day), font_size=None, x=j*cell_width + 20, y=(i+2)*cell_height - 10, width=cell_width, height=cell_height, fill=text_fill)
dwg.append(text)
svg_output = dwg.as_svg()
# Remove the <?xml> line
svg_output = svg_output.split('\n', 1)[1]
output_svg_filename = 'screen-output-weather.svg'
output_dict = {'MONTH_CAL': svg_output}
logging.info("main() - {}".format(output_dict))
logging.info("Updating SVG")
update_svg(output_svg_filename, output_svg_filename, output_dict)
if __name__ == "__main__":
main()