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')