reminder/app/index.py

56 lines
1.9 KiB
Python
Raw Normal View History

2018-12-30 09:24:29 +01:00
from flask import Blueprint, render_template, send_file
from icalendar import Calendar, Alarm
from flask_wtf import FlaskForm
2018-12-30 09:24:29 +01:00
from flask_wtf.file import FileField, FileRequired
from wtforms import IntegerField, SubmitField
from wtforms.widgets.html5 import NumberInput
import tempfile
2018-12-30 09:24:29 +01:00
index = Blueprint('index', __name__, template_folder='templates')
2018-12-30 09:24:29 +01:00
@index.route("/", methods=['GET', 'POST'])
def addreminder():
form = IcsForm()
content = render_template('index.html', form=form)
file = form.icsfile.data
if not file:
return content
hours = form.hours.data
minutes = form.minutes.data
try:
calendar = Calendar.from_ical(file.read())
except ValueError:
error = 'can\'t read file'
return render_template('index.html', form=form, error=error)
for component in calendar.walk('VEVENT'):
valarm_found = False
for k, v in component.property_items():
if k == 'BEGIN' and v == 'VALARM':
valarm_found = True
if not valarm_found:
alarm = Alarm()
alarm.add('ACTION', 'DISPLAY')
alarm.add('DESCRIPTION', component.get('SUMMARY'))
alarm.add('TRIGGER;VALUE=DURATION', '-PT%dH%dM' % (hours, minutes))
component.add_component(alarm)
new_ics = tempfile.TemporaryFile()
new_ics.write(calendar.to_ical())
new_ics.seek(0)
new_filename = file.filename.rstrip('.ics')+'_with_reminders'+'.ics'
return send_file(new_ics, as_attachment=True,
attachment_filename=new_filename)
class IcsForm(FlaskForm):
2018-12-30 09:24:29 +01:00
icsfile = FileField('ICS File', validators=[FileRequired()])
hours = IntegerField('Hours', default=0,
widget=NumberInput(step=1, min=0, max=24))
minutes = IntegerField('Minutes', default=0,
widget=NumberInput(step=5, min=0, max=55))
submit = SubmitField('Submit')