aimslite.py

import requests
import os.path
import json
from typing import List
import tkinter as tk
from tkinter import ttk
from tkinter import messagebox
from tkinter import filedialog

import aimslib.common.types as T
import aimslib.detailed_roster.process as dr

from aimslib.output.csv import csv
from aimslib.output.ical import ical

from aimslite_version import VERSION

SETTINGS_FILE = os.path.expanduser("~/.aimsgui")



class ModeSelector(ttk.Frame):

    def __init__(self, parent, initial_role, with_header):
        ttk.Frame.__init__(self, parent)
        self.role = tk.StringVar()
        self.role.set(initial_role or 'captain')
        self.output_type = tk.StringVar()
        self.output_type.set('csv')
        self.with_header = tk.BooleanVar()
        self.with_header.set(with_header)
        self.frm_csv_options = None
        self.frm_csv_settings = None
        self.__make_widgets()


    def __make_widgets(self):
        frm_output_type = ttk.LabelFrame(self, text="Output type")
        frm_output_type.pack(fill=tk.X, expand=True, ipadx=5, pady=5)
        csv_output = ttk.Radiobutton(
            frm_output_type, text=" Logbook (csv)",
            variable=self.output_type, value='csv',
            command=self.output_type_changed)
        csv_output.pack(fill=tk.X)
        ical_output = ttk.Radiobutton(
            frm_output_type, text=" Roster (iCal)",
            variable=self.output_type, value='ical',
            command=self.output_type_changed)
        ical_output.pack(fill=tk.X)

        self.frm_csv_settings = ttk.LabelFrame(self, text="Role")
        captain = ttk.Radiobutton(
            self.frm_csv_settings, text="Captain",
            variable=self.role, value='captain',
            command=self.role_changed)
        captain.pack(fill=tk.X)
        fo = ttk.Radiobutton(
            self.frm_csv_settings, text="FO",
            variable=self.role, value='fo',
            command=self.role_changed)
        fo.pack(fill=tk.X)
        self.frm_csv_settings.pack(fill=tk.X, expand=True, ipadx=5, pady=5)

        self.frm_csv_options = ttk.LabelFrame(self, text="Options")
        with_header = ttk.Checkbutton(self.frm_csv_options,
                                      text="Header",
                                      variable = self.with_header,
                                      command=self.options_changed)
        with_header.pack(fill=tk.X)
        self.frm_csv_options.pack(fill=tk.X, expand=True, ipadx=5, pady=5)


    def output_type_changed(self):
        assert self.output_type.get() in ('csv', 'ical')
        if self.output_type.get() == 'csv':
            self.frm_csv_settings.pack(fill=tk.X, expand=True, ipadx=5, pady=5)
            self.frm_csv_options.pack(fill=tk.X, expand=True, ipadx=5, pady=5)
        else:
            self.frm_csv_settings.pack_forget()
            self.frm_csv_options.pack_forget()
        self.event_generate("<<ModeChange>>", when="tail")


    def role_changed(self):
        self.event_generate("<<ModeChange>>", when="tail")


    def options_changed(self):
        self.event_generate("<<OptionChange>>", when="tail")


class Actions(ttk.Frame):

    def __init__(self, parent):
        ttk.Frame.__init__(self, parent)
        self.__make_widgets()


    def __make_widgets(self):
        frm_1 = ttk.Frame(self)
        frm_1.pack(fill=tk.X)
        btn_convert = ttk.Button(
            frm_1, text="Import", width=0,
            command=lambda: self.event_generate("<<Action-Import>>"))
        btn_convert.pack(fill=tk.X)

        frm_2 = ttk.Frame(self)
        frm_2.pack(fill=tk.X, pady=10)
        btn_save = ttk.Button(
            frm_2, text="Save", width=0,
            command=lambda: self.event_generate("<<Action-Save>>"))
        btn_save.pack(fill=tk.X, pady=2)
        self.btn_copy = ttk.Button(
            frm_2, text="Copy All", width=0,
            command=lambda: self.event_generate("<<Action-Copy>>"))
        self.btn_copy.pack(fill=tk.X, pady=2)

        frm_3 = ttk.Frame(self)
        frm_3.pack(fill=tk.X)
        btn_quit = ttk.Button(
            frm_3, text="Quit", width=0,
            command=lambda: self.event_generate("<<Action-Quit>>"))
        btn_quit.pack(fill=tk.X)


    def set_copy_selected(self, selected):
        if selected:
            self.btn_copy.config(text="Copy Selected")
        else:
            self.btn_copy.config(text="Copy All")


class TextWithSyntaxHighlighting(tk.Text):

    def __init__(self, parent=None, **kwargs):
        tk.Text.__init__(self, parent, background='white', **kwargs)
        self.highlight_mode = None
        self.tag_configure("grayed", foreground="#909090")
        self.tag_configure("keyword", foreground="green")
        self.tag_configure("datetime", foreground="blue")
        self.bind(
            '<KeyRelease>',
            lambda *args: self.edit_modified() and self.highlight_syntax())

    def insert(self, idx, text, mode=None, *args):
        tk.Text.insert(self, idx, text, *args)
        if mode:
            self.highlight_mode = mode
            self.highlight_syntax()


    def highlight_syntax(self):
        if not self.highlight_mode: return
        for tag in ("keyword", "datetime", "grayed"):
            self.tag_remove(tag, "1.0", "end")
        if self.highlight_mode == 'ical':
            self.highlight_vcalendar()
        elif self.highlight_mode == 'csv':
            self.highlight_csv()
        self.edit_modified(False)


    def highlight_vcalendar(self):
        count = tk.IntVar()
        start_idx = "1.0"
        while True:
            new_idx = self.search(
                "(BEGIN|END):VEVENT",
                start_idx, count=count, regexp=True,
                stopindex = "end")
            if not new_idx: break
            start_idx = f"{new_idx} + {count.get()} chars"
            self.tag_add("keyword", new_idx, start_idx)
        start_idx = "1.0"
        while True:
            new_idx = self.search(
                r"^[\w-]+:",
                start_idx, count=count, regexp=True,
                stopindex = "end")
            if not new_idx: break
            start_idx = f"{new_idx} + {count.get()} chars"
            self.tag_add("grayed", new_idx, start_idx)
        start_idx = "1.0"
        while True:
            new_idx = self.search(
                r"\d{4}-\d{2}-\d{2}T\d{2}:\d{2}",
                start_idx, count=count, regexp=True,
                stopindex = "end")
            if not new_idx: break
            start_idx = f"{new_idx} + {count.get()} chars"
            self.tag_add("datetime", new_idx, start_idx)


    def highlight_csv(self):
        count = tk.IntVar()
        start_idx = "1.0"
        while True:
            new_idx = self.search(
                r"\d{4}-\d{2}-\d{2} \d{2}:\d{2}",
                start_idx, count=count, regexp=True,
                stopindex = "end")
            if not new_idx: break
            start_idx = f"{new_idx} + {count.get()} chars"
            self.tag_add("datetime", new_idx, start_idx)
        start_idx = "1.0"
        while True:
            new_idx = self.search(
                r'(:00)?[",]+',
                start_idx, count=count, regexp=True,
                stopindex = "end")
            if not new_idx: break
            start_idx = f"{new_idx} + {count.get()} chars"
            self.tag_add("grayed", new_idx, start_idx)


class MainWindow(ttk.Frame):

    def __init__(self, parent):
        ttk.Frame.__init__(self, parent)
        self.parent = parent
        try:
            with open(SETTINGS_FILE) as f:
                self.settings = json.load(f)
        except:
            self.settings = {}
        self.__make_widgets()
        self.txt.insert(tk.END, f"Version: {VERSION}")
        self.copy_mode = "all"


    def __make_widgets(self):
        self.columnconfigure(1, weight=1)
        self.rowconfigure(0, weight=1)
        sb = ttk.Scrollbar(self)
        sb.grid(row=0, column=2, sticky=tk.NS)
        sidebar = ttk.Frame(self, width=0)
        sidebar.grid(row=0, column=0, sticky=tk.NS, padx=5, pady=5)
        self.txt = TextWithSyntaxHighlighting(self)
        self.txt.grid(row=0, column=1, sticky=tk.NSEW)
        sb.config(command=self.txt.yview)
        self.txt.config(yscrollcommand=sb.set)
        self.txt.bind("<<Selection>>", self.__on_selection_change)

        sidebar.rowconfigure(1, weight=1)
        self.ms = ModeSelector(sidebar,
                               self.settings.get('Role', None),
                               self.settings.get('Header', True))
        self.ms.bind("<<ModeChange>>", self.__on_mode_change)
        self.ms.bind("<<OptionChange>>", self.__on_option_change)
        self.ms.grid(row=0, sticky=tk.N)
        self.act = Actions(sidebar)
        self.act.grid(row=1, sticky=(tk.EW + tk.S))
        for event, func in (
                ("<<Action-Import>>", self.__import),
                ("<<Action-Copy>>", self.__copy),
                ("<<Action-Save>>", self.__save),
                ("<<Action-Quit>>", lambda _: self.parent.destroy())):
            self.act.bind(event, func)
        self.act.set_copy_selected(False)


    def __on_mode_change(self, _):
        self.settings['Role'] = self.ms.role.get()
        self.txt.delete('1.0', tk.END)

    def __on_option_change(self, _):
        self.settings['Header'] = self.ms.with_header.get()
        self.txt.delete('1.0', tk.END)

    def __on_selection_change(self, _):
        if self.txt.tag_ranges("sel"):
            if self.copy_mode == "sel": return
            self.copy_mode = "sel"
            self.act.set_copy_selected(True)
        else:
            self.copy_mode = "all"
            self.act.set_copy_selected(False)


    def __import(self, _):
        assert self.ms.output_type.get() in ('csv', 'ical')
        try:
            if self.ms.output_type.get() == 'csv':
                self.__csv()
            else:
                self.__ical()
        except dr.DetailedRosterException as e:
            self.txt.delete('1.0', tk.END)
            messagebox.showerror("Error", str(e))


    def __roster_html(self):
        retval = ""
        path = self.settings.get('openPath')
        fn = filedialog.askopenfilename(
            filetypes=(
                ("HTML file", "*.htm"),
                ("HTML file", "*.html"),
                ("All", "*.*")),
            initialdir=path)
        if fn:
            self.settings['openPath'] = os.path.dirname(fn)
            try:
                with open(fn) as f:
                    retval = f.read()
            except:
                raise dr.InputFileException
        return retval


    def __csv(self):
        txt = ""
        dutylist, crewlist_map = [], {}
        html = self.__roster_html()
        if not html: return
        self.txt.delete('1.0', tk.END)
        self.txt.insert(tk.END, "Getting registration and type info...")
        self.txt.update()
        dutylist = update_dutylist_from_flightinfo(dr.duties(html))
        if not dutylist:
            messagebox.showinfo('Duties', 'No relevant duties found')
            self.txt.delete('1.0', tk.END)
            return
        crewlist_map = dr.crew(html, dutylist)
        fo = True if self.ms.role.get() == 'fo' else False
        txt = csv(dutylist, crewlist_map, fo)
        if self.ms.with_header.get() == False:
            txt = txt.split("\n", 1)[1]
        self.txt.delete('1.0', tk.END)
        self.txt.insert(tk.END, txt, 'csv')


    def __ical(self):
        dutylist = []
        html = self.__roster_html()
        if not html: return
        dutylist = dr.duties(html)
        if not dutylist:
            self.txt.delete('1.0', tk.END)
            messagebox.showinfo('Duties', 'No relevant duties found')
            return
        #note: normalise newlines for Text widget - will restore on output
        txt = ical(dutylist).replace("\r\n", "\n")
        self.txt.delete('1.0', tk.END)
        self.txt.insert(tk.END, txt, 'ical')


    def __copy(self, _):
        self.clipboard_clear()
        if self.copy_mode == "all":
            start, end = '1.0', tk.END
        else:
            start, end = self.txt.tag_ranges("sel")
        text = self.txt.get(start, end)
        if self.ms.output_type == 'ical': #ical requires DOS style line endings
            text = text.replace("\n", "\r\n")
        self.clipboard_append(text)
        messagebox.showinfo('Copy', 'Text copied to clipboard.')


    def __save(self, _):
        output_type = self.ms.output_type.get()
        assert  output_type in ('csv', 'ical')
        if output_type == 'csv':
            pathtype = 'csvSavePath'
            filetypes = (("CSV file", "*.csv"),
                         ("All", "*.*"))
            default_ext = '.csv'
        else:
            pathtype = 'icalSavePath'
            filetypes = (("ICAL file", "*.ics"),
                         ("ICAL file", "*.ical"),
                         ("All", "*.*"))
            default_ext = '.ics'
        path = self.settings.get(pathtype)
        fn = filedialog.asksaveasfilename(
            initialdir=path,
            filetypes=filetypes,
            defaultextension=default_ext)
        if fn:
            self.settings[pathtype] = os.path.dirname(fn)
            with open(fn, "w", encoding="utf-8", newline='') as f:
                text = self.txt.get('1.0', tk.END)
                if output_type == 'ical': #ical needs DOS style line endings
                    text = text.replace("\n", "\r\n")
                f.write(text)
                messagebox.showinfo('Saved', f'Save complete.')


    def destroy(self):
        with open(SETTINGS_FILE, "w") as f:
            json.dump(self.settings, f, indent=4)
        ttk.Frame.destroy(self)




def update_dutylist_from_flightinfo(dutylist: List[T.Duty]) -> List[T.Duty]:
    retval: List[T.Duty] = []
    ids = []
    for duty in dutylist:
        ids.extend([f'{X.sched_start:%Y%m%dT%H%M}F{X.name}'
                    for X in duty.sectors
                    if X.flags == T.SectorFlags.NONE])
    try:
        r = requests.post(
            f"https://efwj6ola8d.execute-api.eu-west-1.amazonaws.com/default/reg",
            json={'flights': ids},
            timeout=5)
        r.raise_for_status()
        regntype_map = r.json()
    except requests.exceptions.RequestException:
        return dutylist #if anything goes wrong, just return input
    for duty in dutylist:
        updated_sectors: List[T.Sector] = []
        for sec in duty.sectors:
            flightid = f'{sec.sched_start:%Y%m%dT%H%M}F{sec.name}'
            if flightid in regntype_map:
                reg, type_ = regntype_map[flightid]
                updated_sectors.append(sec._replace(reg=reg, type_=type_))
            else:
                updated_sectors.append(sec)
        retval.append(duty._replace(sectors=updated_sectors))
    return retval


def main():
    root = tk.Tk()
    root.title("aimsgui")
    mw = MainWindow(root)
    mw.pack(fill=tk.BOTH, expand=True)
    root.mainloop()

if __name__ == "__main__":
    main()