58 lines
2 KiB
Python
58 lines
2 KiB
Python
|
from flask import Flask, render_template, request, send_file
|
||
|
|
from icalendar import Calendar, Alarm
|
||
|
|
from flask_wtf import FlaskForm
|
||
|
|
#from flask_wtf.file import FileField, FileRequired
|
||
|
|
from wtforms import FileField, IntegerField, SubmitField
|
||
|
|
from wtforms.validators import InputRequired
|
||
|
|
from wtforms.widgets.html5 import NumberInput
|
||
|
|
import tempfile
|
||
|
|
|
||
|
|
application = Flask(__name__)
|
||
|
|
application.config['SECRET_KEY'] = 'secret'
|
||
|
|
|
||
|
|
|
||
|
|
@application.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):
|
||
|
|
icsfile = FileField('ICS File', validators=[InputRequired()])
|
||
|
|
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')
|