lagerbidrag.py

# -*- coding: utf-8 -*-
"""
Calculate data needed and fill in forms for lägerbidragsansökan (hike grants).

The limits and data needed depends on region.
Göteborg has 2-14 nights, age 7-25, and includes som older people.
Stocholm has 2-7 days, age 7-20, and does not include older people.
Stockholm also needs postal address for each scout.
"""
from collections import namedtuple
from datetime import date
from data import Meeting, Troop
from datetime import datetime, date
from flask import render_template, make_response
from mailmerge import MailMerge
import io
import copy
import math
import urllib


RegionLimits = namedtuple('RegionLimits', ['min_days', 'max_days', 'min_age', 'max_age', 'count_over_max_age'])


LIMITS = {'gbg' : RegionLimits(3, 15, 7, 25, True),  # min 2 nights, max 14 nights, 7-25 yers + some older
          'sthlm' : RegionLimits(2, 7, 7, 20, False)} # min 2 days, max 7 days, 7-20 years

DATE_FORMAT = '%Y-%m-%d'
DATE_TIME_FORMAT = '%Y-%m-%d %H:%M'


def render_lagerbidrag(request, scoutgroup, context, sgroup_key=None, user=None, trooppersons=None, troop_key=None):
    if context == "group":
        assert sgroup_key is not None
        assert user is not None
        troops = Troop.getTroopsForUser(sgroup_key, user)
    elif context == "troop":
        assert trooppersons is not None
        assert troop_key is not None
    else:
        raise ValueError("Context %s unknown" % context)

    region = request.args.get('region')
    limits = LIMITS[region]
    bidrag=LagerBidrag(scoutgroup.getname())
    bidrag.foreningsID = scoutgroup.foreningsID
    bidrag.firmatecknare = scoutgroup.firmatecknare
    bidrag.firmatecknartelefon = scoutgroup.firmatecknartelefon
    bidrag.email = scoutgroup.epost
    bidrag.phone = scoutgroup.telefon
    bidrag.address = scoutgroup.adress
    bidrag.zipCode = scoutgroup.postadress
    bidrag.account = scoutgroup.bankkonto
    bidrag.contact = request.form['contactperson']
    bidrag.contactemail = request.form.get('contactemail', '')
    bidrag.contactphone = request.form.get('contactphone', '')
    bidrag.site = request.form['site']
    bidrag.dateFrom = request.form['fromDate']
    bidrag.dateTo = request.form['toDate']
    bidrag.hikeduringbreak = request.form.get('hikeduringbreak')
    try:
        if context == "group":
            bidragcontainer = createLagerbidragGroup(limits, scoutgroup, troops, bidrag)
        else:
            bidragcontainer = createLagerbidrag(limits, scoutgroup, trooppersons, troop_key, bidrag)

        if region == 'gbg':
            return response_gbg(bidragcontainer)
        elif region == 'sthlm':
            return response_sthlm(bidragcontainer)
        else:
            raise ValueError("Unknown region %s" % region)

    except ValueError as e:
        return render_template('error.html', error=str(e))


def response_gbg(bidragcontainer):
    result = render_template(
        'lagerbidrag_gbg.html',
        bidrag=bidragcontainer.bidrag,
        persons=bidragcontainer.persons,
        numbers=bidragcontainer.numbers)
    response = make_response(result)
    return response


def response_sthlm(bidragcontainer):

    bidrag = bidragcontainer.bidrag
    persons = bidragcontainer.persons

    nr_persons = bidragcontainer.nr_persons_total
    persons = persons[:nr_persons]

    start = datetime.strptime(bidrag.dateFrom, DATE_FORMAT)
    end = datetime.strptime(bidrag.dateTo, DATE_FORMAT)
    nr_days = (end - start).days + 1

    data = {
        'kundnummer': bidrag.foreningsID,
        'foreningsnamn': bidrag.kar,
        'ledare': bidrag.contact,
        'ledartelefon': bidrag.contactphone,
        'ledaremail': bidrag.contactemail,
        'lov': u'X' if bidrag.hikeduringbreak else u'\u2610',
        'helg': u'\u2610' if bidrag.hikeduringbreak else u'X',
        'lagerplats': bidrag.site,
        'startdatum': bidrag.dateFrom,
        'slutdatum': bidrag.dateTo,
        'datum': date.today().strftime(DATE_FORMAT),
        'firmatecknare': bidrag.firmatecknare,
        'firmatecknartelefon': bidrag.firmatecknartelefon,
        'antalmedlemmar': str(bidragcontainer.nr_young_persons),
        'antaldagar': str(bidrag.days)  # Should be total number of days
    }

    today = date.today()
    period_data = generate_sthlm_period_data(today.year, today.month, today.day)
    data.update(period_data)

    persons_per_page = 16
    nr_pages, rest = divmod(nr_persons, persons_per_page)
    if rest > 0:
        nr_pages += 1
    pages = []
    for page_nr in range(nr_pages):
        page_data = copy.deepcopy(data)
        persons_in_page = persons[page_nr * persons_per_page:(page_nr + 1) * persons_per_page]
        for i, person in enumerate(persons_in_page):
            nr = i + 1
            page_data['namn%d' % nr] = person.name
            page_data['pa%d' % nr] = person.postal_address
            page_data['ar%d' % nr] = str(person.year)
        pages.append(page_data)

    document = MailMerge('templates/lagerbidragsmall_sthlm.docx')
    document.merge_pages(pages)

    bytesIO = io.BytesIO()
    document.write(bytesIO)
    resultbytes = bytesIO.getvalue()

    doc_name = "Lagerbidrag_%s_%s_%s.docx" % (bidrag.site, bidrag.dateFrom, bidrag.dateTo)
    ascii_doc_name = doc_name.encode('ascii', 'ignore')

    response = make_response(resultbytes)
    response.headers['Content-Type'] = 'application/vnd.openxmlformats-officedocument.wordprocessingml.document'
    response.headers['Content-Disposition'] = 'attachment; filename=' + urllib.quote(ascii_doc_name)
    return response


def generate_sthlm_period_data(year, month, day):
    """Generate dictionary with period data for right period.
    
    Dec 16 - Jun 30, the period is Dec 1 to May 31.
    Jul 1 - Dec 15, the period is Jun 1 to Nov. 30

    """

    if month <= 6 or (month == 12 and day > 15):  # Period 1
        if month <= 6:
            y = year
        else:
            y = year + 1
        pd = {
            'ansokningsar': str(y),
            'sistaansokan': '30 juni %d' % y,
            'periodstart': '1 december %d' % (y - 1),
            'periodslut': '31 maj %d' % y
        }
    else: # Period 2
        pd = {
            'ansokningsar': str(year),
            'sistaansokan': '15 december %d' % year,
            'periodstart': '1 juni %d' % year,
            'periodslut': '30 november %d' % year
        }
    return pd


def person_sort(a, b):
    if a.year == b.year:
        if a.name > b.name:
            return 1
        else:
            return -1
    elif a.year > b.year:
        return 1
    else:
        return -1


class LagerBidragContainer:
    bidrag = ""
    nr_young_persons = 0
    nr_older_persons = 0
    nr_under_min_age = 0
    nr_persons_total = 0

    def __init__(self):
        self.persons = []
        self.numbers = []


class LagerPerson():

    def __init__(self, person="", name="", year="", age=0, postal_address=""):
        self.person = person
        self.name = name
        self.year = year
        self.age = age
        self.postal_address = postal_address
        self.days = 0


class LagerBidrag():
    contact = ""
    contactemail = ""
    contactphone = ""
    kar = ""
    foreningsID = ""
    account = ""
    firmatecknare = ""
    firmatecknartelefon = ""
    dateFrom = ""
    dateTo = ""
    hikeduringbreak = None
    site = ""
    address = ""
    zipCode = ""
    phone = ""
    email = ""
    uptoMaxAge = 0
    overMaxAge = 0
    nights = 0
    days = 0
    divider = 25

    def __init__(self, kar):
        self.kar = kar


def _add_person(person, persons, year, days=None):
    "Create LagerPerson from person and add to persons lists."
    person_ob = person.get()
    postal_address = "%s %s" % (person_ob.zip_code, person_ob.zip_name)
    lager_person = LagerPerson(person, person_ob.getname(), person_ob.birthdate.year, person_ob.getyearsoldthisyear(year), postal_address)
    if days is not None:
        lager_person.days = days
    persons.append(lager_person)


def createLagerbidragGroup(limits, scoutgroup, troops, bidrag):

    from_date_time = datetime.strptime(bidrag.dateFrom + " 00:00", DATE_TIME_FORMAT)
    to_date_time = datetime.strptime(bidrag.dateTo + " 23:59", DATE_TIME_FORMAT)
    year = to_date_time.year
    persons = []
    person_days = {}

    validateLagetbidragInput(from_date_time, to_date_time, limits)

    for troop in troops:
        # Count number of days participating
        for meeting in Meeting.query(Meeting.datetime >= from_date_time, Meeting.datetime <= to_date_time, Meeting.troop == troop.key).fetch():
            for person in meeting.attendingPersons:
                if not person in person_days:
                    person_days[person] = 1
                else:
                    person_days[person] += 1

    # Collect number of persons
    for person, days in person_days.iteritems():
        _add_person(person, persons, year, days)

    return createLagerbidragReport(limits, scoutgroup, persons, bidrag)


def createLagerbidrag(limits, scoutgroup, trooppersons, troopkey_key, bidrag):

    from_date_time = datetime.strptime(bidrag.dateFrom + " 00:00", DATE_TIME_FORMAT)
    to_date_time = datetime.strptime(bidrag.dateTo + " 23:59", DATE_TIME_FORMAT)
    year = to_date_time.year
    persons = []

    validateLagetbidragInput(from_date_time, to_date_time, limits)

    # Collect number of persons
    for troopperson in trooppersons:
        _add_person(troopperson.person, persons, year)

    # Count number of days participating
    for meeting in Meeting.query(Meeting.datetime >= from_date_time, Meeting.datetime <= to_date_time, Meeting.troop == troopkey_key).fetch():
        for person in persons:
            is_attending = person.person in meeting.attendingPersons
            if is_attending:
                person.days += 1

    return createLagerbidragReport(limits, scoutgroup, persons, bidrag)


def validateLagetbidragInput(from_date_time, to_date_time, limits):
    "Check the number of days compared to limits."
    delta = to_date_time.date() - from_date_time.date()
    nr_days = delta.days + 1  # The 00:00 - 23:59 does not give any extra day
    if nr_days > limits.max_days:
        raise ValueError('Lägret får max vara %d dagar' % limits.max_days)
    if nr_days < limits.min_days:
        raise ValueError('Lägret måsta vara minst %d dagar' % limits.min_days)


def createLagerbidragReport(limits, scoutgroup, persons, bidrag):
    "Create a container with lagerbidrag data. The input bidrag instance will be enhanced."
    container = LagerBidragContainer()

    container.persons = persons

    # Filter out persons participation at least limits.min_days days
    container.persons = [p for p in container.persons if p.days >= limits.min_days]

    # Sort by number of days to get the persons over limits.max_age with most days first
    container.persons.sort(key=lambda x: x.days, reverse=True)

    # sum number of persons and days for person under max_age
    for person in container.persons:
        if person.age > limits.max_age:
            bidrag.overMaxAge = bidrag.overMaxAge + 1
            container.nr_older_persons += 1
        elif person.age >= limits.min_age:
            bidrag.uptoMaxAge = bidrag.uptoMaxAge + 1
            bidrag.nights += person.days - 1
            bidrag.days += person.days
            container.nr_young_persons += 1
        else:
            container.nr_under_min_age += 1

    if limits.count_over_max_age:
        # Count number of days for person over max_age
        allowed_over_max_age = math.floor(bidrag.uptoMaxAge / 3)
        count_over_max_age = 0
        for person in container.persons:
            if count_over_max_age == allowed_over_max_age:
                break
            if person.age > limits.max_age:
                count_over_max_age += 1
                bidrag.nights += person.days - 1
                bidrag.days += person.days

    container.persons.sort(person_sort)

    number_of_persons = container.nr_young_persons + container.nr_older_persons + container.nr_under_min_age
    assert number_of_persons == len(container.persons)
    container.nr_persons_total = number_of_persons

    # Set up divider to get two columns of data (used by Gbg template)
    tmp_divider = int(math.ceil(number_of_persons / 2.0))
    bidrag.divider = 25 if tmp_divider < 25 else tmp_divider
    # Add empty persons
    container.persons.extend([LagerPerson() for i in range(0, bidrag.divider * 2 - len(container.persons))])
    container.numbers = range(0, bidrag.divider)

    container.bidrag = bidrag
    return container