Calculer Les Tâches À Faire Sur Une Voiture

from pandas import DataFrame as df
import sys
import pandas as pd
from dateutil.relativedelta import relativedelta
import re
from datetime import datetime as dt
regex = re.compile(r'^((?P<years>\d+)y)?((?P<months>\d+)m)?$')
def parse_td(value):
   match = regex.match(value)
   if match is None:
       return None
   else:
       return relativedelta(**{
           key: int(value)
           for key, value in match.groupdict().items()
           if value
       })
v = df(releves[1:], columns=releves[0])
v["date"] = pd.to_datetime(v.stamp, unit="s")
s = {
    seuil[0]: seuil[1]
    for seuil in seuils[1:]
}
s2 = {
    seuil[0]: parse_td(seuil[2])
    for seuil in seuils[1:]
}
s3 = {
    seuil[0]: seuil[3]
    for seuil in seuils[1:]
}
sys.stderr.write("a\n")
a = df(
    [
        key + (value,)
        for key, values in
        v.set_index(["Km", "date"]).actions.items()
        for value in values.split(", ")
    ],
    columns=["km", "date", "action"]
)
sys.stderr.write(f"{a}\n")
a["diffkm"] = a["km"][0] - a["km"]
a["diffdate"] = a["date"][0] - a["date"]
res = []
now = dt.now()
for nom in s:
    km = s[nom]
    restekm = None
    sys.stderr.write(f"{nom}\n")
    if km:
         diffkm = a[a.action == nom]["diffkm"]
         sys.stderr.write(f"{diffkm}\n")
         diffkm = diffkm.iloc[0]
         restekm = km - diffkm
    tmp = s2[nom]
    nexttmp = None
    if tmp:
        nexttmp = a[a.action == nom]["date"].iloc[0] + tmp
    cur_res = [
        nom,
        "A FAIRE" if (restekm and (restekm < 0) or (nexttmp and nexttmp < now)) else "ok",
        f"{restekm} km",
        "{} days".format((nexttmp - now).days) if nexttmp else None,
        s3[nom],
    ]
    res.append(cur_res)
return sorted(res, key=lambda e: e[1])