ths-datenlogger/frontend_utils.py

#!/usr/bin/python3
import sys
import tkinter
import plot_main
import config_parse
from config_parse 	import CFG
from language 		import LAN
from datetime 		import datetime

l = LAN[CFG("language")]
timeformat = "%d.%m.%y %H:%M:%S (%A)"
def parse_date_from_user_input(s,end_of_day=False,datapoints=None):
    today  = datetime.now()
    day    = 0
    month  = 0
    year   = 0
    hour   = 0
    minute = 0
    second = 0
    
    ## EMPTY ##
    if s == None or s == "":
        return None

    ## TIME ##
    if len(s.split(" ")) > 1:
        time = s.split(" ")[1]
        time_a = time.split(":")
        if len(time_a) > 0:
            hour = int(time_a[0])
        elif end_of_day > 0:
            hour = 23
        if len(time_a) > 1:
            minute = int(time_a[1])
        elif end_of_day:
            minute = 59
        if len(time_a) > 2:
            second = int(time_a[2])
        elif end_of_day:
            second = 59
    elif end_of_day:
        hour = 23
        minute = 59
        second = 59

    ## DATE ##
    tmp = s.split(" ")[0]

    ## allow more speperators ##
    sep = None
    for c in ["-",".",","]:
        if c in tmp:
            sep = c
            break
    if sep == None:
        sep = "-"
    tmp = tmp.strip(sep)

    if len(tmp.split(sep)) == 0:
        raise ValueError("Invalid Date '%s'"%str(s))
    else:
        date_a = tmp.split(sep)
        if len(date_a) > 0:
            day = int(date_a[0])
        if len(date_a) > 1:
            month = int(date_a[1])
        if len(date_a) > 2:
            year = int(date_a[2])
    
    if year == 0:
        if today.month > month:
            year = today.year
        else:
            year = today.year-1
    if month == 0:
        if today.day > day and today.year == year:
            month = today.month
        else:
            month = today.month-1
            if month < 1:
                month = 12-month 
        ret = datetime(year,month,day,hour,minute,second)
        try:
            times = datapoints[CFG("plot_temperatur_key")].times
            if ( ret > max(times) or ret < min(times) ) and min(times).day < ret.day < max(times).day and min(times).month == max(times).month:
                month = min(times.month)
        except Exception as e:
            print("Warning, magic date selection failed for an unknown reason")

    ret = datetime(year,month,day,hour,minute,second)
    return ret

def info_list(datapoints):
    if len(datapoints.keys()) > 0:
        print("Erster Datensatz:  "+min(datapoints[list(datapoints.keys())[0]].times).strftime(timeformat))
        print("Letzer Datensatz:  "+max(datapoints[list(datapoints.keys())[0]].times).strftime(timeformat))
        print("Anzahl Datensätze: "+str(len(datapoints[list(datapoints.keys())[0]].times)))
    else:
        print("Keine Datesätze gefunden!")
    print_sep_line(True)

def input_date_repl(datapoints,startdate=True):
    date = None
    while True:
        try:
            if startdate:
                ret = input(l["input_first_date"])
            else:
                ret = input(l["input_second_date"])
        except EOFError:
            return (date,True)
        except KeyboardInterrupt:
            sys.exit(2)
        if ret in ["h","help","hilfe"]:
            if startdate:
                print(l["input_first_date_help"])
            else:
                print(l["input_second_date_help"])
            continue
        elif ret == "list":
            info_list(datapoints)
            continue
        else:
            try:
                if startdate:
                    date=parse_date_from_user_input(ret,datapoints=datapoints)
                else:
                    date=parse_date_from_user_input(ret,True,datapoints)
                return (date,True)
            except ValueError as e:
                print(l["cannot_parse_date"] + "( was: {} )\n".format(ret))
                return (None,False)

def print_sep_line(ln=False):
    if not ln:
        print("-----------------------------------------------")
    else:
        print("-----------------------------------------------",end='')


def check_dates(path,date1,date2,options=""):
    print_sep_line()
    if options!="":
        print("Config options: %s"%options)
    print("Datei: %s"%path)
    if date1 == None and date2 == None:
        print("Info:  Keine Zeitbeschränkung gewählt. Alle vorhandenen Daten werden verwendet.")
        return (True,True)
    elif date1 == None:
        print("Alle Werte vor %s"%date2.strftime(timeformat))
    elif date2 == None:
        print("Alle Werte nach %s"%date1.strftime(timeformat))
    else:
        print("Start: %s\nEnde:  %s"%(date1.strftime(timeformat),date2.strftime(timeformat)))

    FIRST=True
    while(True):
        try:
            if FIRST:
                ret = input("Stimmt das so?\n -> <ENTER> für ja/weiter\n -> 's'<ENTER> für Startzeit ändern\n -> 'e'<ENTER> für Endzeit ändern\n -> 'b'<ENTER> für beides ändern\n -> 'exit'<ENTER> to exit\n---> ")
            else:
                ret = input("Versuchen sie es nochmal: ")
        except EOFError:
            ret = ""

        if ret == 's':
            return (False,True)
        elif ret == 'e':
            return (True,False)
        elif ret == 'b':
            return (False,False)
        elif ret.startswith('c '):
            tmp = config_options(ret)
            if tmp == "":
                pass
            else:
                options += "\n     "+tmp
                check_dates(path,date1,date2,options)
        elif ret == "":
            return (True,True)
        elif ret == "exit":
            sys.exit(0)
        FIRST=False

def open_file():
    front_end_source_path = CFG("default_source_path")
    if CFG("use_input_filename"):
        f = front_end_source_path + CFG("input_filename")
        return f

    path=None
    path=tkinter.filedialog.askopenfilename(filetypes=(("DBF/XLS Files",("*.DBF","*.dbf","*.xls","*.XLS")),("All Files","*.*")))
    if path == None or path=="":
        print("Error: No file selected!")
        return None
    try:
        open(path,'r').close()
    except IOError:
        print("Error: Unable to open selected file, perhaps it does no longer exist or you have insufficient permissions to open it?")
        return None
    return path

def config_options(string):
        opt = ""
        arg = string.split(" ")
        if len(arg) == 2:
            for l in config_parse.get_keys(arg[1]):
                print(l)
        elif len(arg) == 3:
            if not config_parse.change_cfg(arg[1],arg[2]):
                print("Option %s does not exist."%str(arg[1]))
            else:
                opt += "set %s %s"%(arg[1],arg[2])
                print("set %s %s"%(arg[1],arg[2]))
        else:
            print("Ussage: c <configname> <value> / c <part_of_config_name> / c (= list all ), e to exit")
        return opt