pylokid/lodur.py

#!/usr/bin/env python3

""" Small Lodur Library for the Module 36 - Einsatzrapport """

import re
import logging
from datetime import datetime
import mechanicalsoup

class Lodur:
    """ Lodur """

    def __init__(self, url, username, password):
        self.logger = logging.getLogger(__name__)
        self.logger.info('Connecting to Lodur')

        self.url = url

        # MechanicalSoup initialization and login to Lodur
        self.browser = mechanicalsoup.StatefulBrowser()
        # The login form is located in module number 9
        self.browser.open(self.url + '?modul=9')
        self.browser.select_form()

        self.browser['login_member_name'] = username
        self.browser['login_member_pwd'] = password
        self.browser.submit_selected()

        # Check if login succeeded by finding the img with
        # alt text LOGOUT
        page = self.browser.get_current_page()
        if page.find(alt='LOGOUT'):
            self.logger.info('Login to Lodur succeeded')
        else:
            self.logger.fatal('Login to Lodur failed - exiting')
            raise SystemExit(1)

    def einsatzprotokoll(self, lodur_id, pdf_data):
        """ Prepare Einsatzprotokoll to be sent to Lodur 
        TODO This doesn't work as Lodur doesn't add the values directly in to HTML but
        uses JavaScript to dynamically populate the form data.
        To be able to update the form we would need to have access to the existing data
        or else it won't work.
        Ideas: Somehow store the RAW data in a JSON file and reuse this
        to update the form in Lodur
        """

        # when PDF parsing fails, pdf_data is false. fill with tbd when this happens
        if pdf_data:
            zh_alarmierung = datetime.strptime(
                pdf_data['disposition'],
                '%H:%M',
            )
            zh_fw_ausg = datetime.strptime(
                pdf_data['ausgerueckt'],
                '%H:%M',
            )
            zh_am_schad = datetime.strptime(
                pdf_data['anort'],
                '%H:%M',
            )
        else:
            # Do nothing when no PDF data - we don't have anything to do then
            return False

        # Prepare the form
        self.browser.open(self.url + '?modul=36&what=144&edit=1&event=' + lodur_id)
        self.browser.select_form('#einsatzrapport_main_form')
        print(self.browser.get_current_form().print_summary())

        # Fill in form data
        self.browser['zh_alarmierung_h'] = zh_alarmierung.hour # 12. Alarmierung
        self.browser['zh_alarmierung_m'] = zh_alarmierung.minute # 12. Alarmierung
        self.browser['zh_fw_ausg_h'] = zh_fw_ausg.hour # 13. FW ausgerückt
        self.browser['zh_fw_ausg_m'] = zh_fw_ausg.minute # 13. FW ausgerückt
        self.browser['zh_am_schad_h'] = zh_am_schad.hour # 14. Am Schadenplatz
        self.browser['zh_am_schad_m'] = zh_am_schad.minute # 14. Am Schadenplatz
        # The following fields are currently unknown as PDF parsing is hard for these
        #self.browser['zh_fw_einge_h'] = 'UNKNOWN' # 15. FW eingerückt
        #self.browser['zh_fw_einge_m'] = 'UNKNOWN' # 15. FW eingerückt
        #self.browser['eins_erst_h'] = 'UNKNOWN' # 16. Einsatzbereitschaft erstellt
        #self.browser['eins_erst_m'] = 'UNKNOWN' # 16. Einsatzbereitschaft erstellt

        # Submit the form
        #print(self.browser.get_current_form().print_summary())
        self.logger.info('Submitting form Einsatzrapport')
        self.browser.submit_selected()

    def einsatzrapport(self, f_id, pdf_data):
        """ Form in module 36 - Einsatzrapport """

        # when PDF parsing fails, pdf_data is false. fill with tbd when this happens
        if pdf_data:
            date = datetime.strptime(
                pdf_data['datum'],
                '%d.%m.%Y',
            )
            time = datetime.strptime(
                pdf_data['zeit'],
                '%H:%M',
            )
            eins_ereig = pdf_data['einsatz']
            adr = pdf_data['strasse'] + ', ' + pdf_data['plzort']
        else:
            date = datetime.now()
            time = datetime.now()
            eins_ereig = 'UNKNOWN'
            adr = 'UNKNOWN'

        # Prepare the form
        self.browser.open(self.url + '?modul=36')
        self.browser.select_form('#einsatzrapport_main_form')

        # Fill in form data
        self.browser['e_r_num'] = f_id # 01. Einsatzrapportnummer
        self.browser['eins_stat_kantone'] = '1' # 02. Einsatzart FKS
        self.browser['emergency_concept_id'] = '2' # 03. Verrechnungsart
        self.browser['ver_sart'] = 'ab' # 03. Verrechnungsart internal: ab, th, uh, ak, tt
        self.browser['dtv_d'] = str(date.day) # 04. Datum von
        self.browser['dtv_m'] = str(date.month) # 04. Datum von
        self.browser['dtv_y'] = str(date.year) # 04. Datum von
        self.browser['dtb_d'] = str(date.day) # 04. Datum bis - we dont know yet the end date
        self.browser['dtb_m'] = str(date.month) # 04. Datum bis - assume the same day
        self.browser['dtb_y'] = str(date.year) # 04. Datum bis
        self.browser['ztv_h'] = str(time.hour) # 05. Zeit von
        self.browser['ztv_m'] = str(time.minute) # 05. Zeit von
        self.browser['ztb_h'] = str(time.hour + 1) # 05. Zeit bis - we dont know yet the end time
        self.browser['ztb_m'] = str(time.minute) # 05. Zeit bis - just add 1 hour and correct later
        self.browser['e_ort_1'] = '306' # 06. Einsatzort: Urdorf 306, Birmensdorf 298
        self.browser['eins_ereig'] = eins_ereig.encode('iso-8859-1') # 07. Ereignis
        self.browser['adr'] = adr.encode('iso-8859-1') # 08. Adresse
        self.browser['ang_sit'] = 'TBD1' # 17. Angetroffene Situation
        self.browser['mn'] = 'TBD2' # 19. Massnahmen
        self.browser['bk'] = 'TBD3' # 20. Bemerkungen
        self.browser['en_kr_feuwehr'] = '1' # 21. Einsatzkräfte
        self.browser['ali_io'] = '1' # 24. Alarmierung
        self.browser['kopie_gvz'] = '1' # 31. Kopie innert 10 Tagen an GVZ
        self.browser['mannschaftd_einsa'] = '70' # 32. Einsatzleiter|in

        # Submit the form
        self.logger.info('Submitting form Einsatzrapport')
        response = self.browser.submit_selected()

        # very ugly way to find the assigned event id by lodur
        # lodur adds a script element at the bottom of the returned html
        # with the location to reload the page - containing the assigned event id
        lodur_id = re.search('modul=36&event=([0-9].*)&edit=1&what=144', response.text).group(1)
        self.logger.info('Lodur assigned the id ' + lodur_id + ' to ' + f_id)
        return lodur_id

    def einsatzrapport_alarmdepesche(self, lodur_id, file_path):
        """ Upload a file to Alarmdepesche """

        # Prepare the form
        self.browser.open(self.url + '?modul=36&what=828&event=' + lodur_id)
        self.browser.select_form('#frm_alarmdepesche')

        # Fill in form data
        self.logger.info('Submitting Alarmdepesche to Lodur id ' + lodur_id)
        self.browser['alarmdepesche'] = open(file_path, 'rb')

        # Submit the form
        self.browser.submit_selected()
        self.logger.info('Alarmdepesche submitted')