Reorganize project structure

2024-11-19 09:22:39 +01:00
parent 48d06341ef
commit 7d76e217f1
13 changed files with 142 additions and 68 deletions
@@ -0,0 +1,92 @@
+from dataclasses import dataclass
+import gspread
+import csv
+import locale
+import datetime
+import yaml
+import os
+
+#TODO: Upload automatically!
+
+@dataclass
+class Export: # all necessary fields for ELSAP-importfiles
+    sum_col: str
+    desc_col: str
+    week: str
+    bestellnummer: str
+    bestellposition: str
+    leistungszeilennummer: str
+    kurztext: str
+    leistungszeilentext: str
+    username: str
+    name: str
+    kontierung: str
+    ansprechpartner_username: str
+    ansprechpartner_name: str
+
+def export_timesheet(csvexport_filename, ex: Export):
+    with open(f"{csvexport_filename}.csv", 'w') as file:
+        writer = csv.writer(file, delimiter=';', lineterminator="\n")
+     
+        writer.writerow(["Bestellnummer","Bestellpositionsnr","Leistungszeilennummer","Kurztext","Leistungszeilentext","Username","Name","Arbeitstag","Stunden","Beschreibung","Kontierung","Ansprechpartner Username","Ansprechpartner","Status"])
+        # iterate
+        for time_entry in list_of_dicts:
+            if time_entry.get(ex.sum_col) and time_entry.get(ex.desc_col) and time_entry.get("week"):
+                if (int(time_entry.get("week")) in week) or week is None:
+                #if week == int(time_entry.get("week")) or week is None:
+                    date_str_de_de = time_entry.get("date")
+                    datetime_object = datetime.datetime.strptime(date_str_de_de, '%a., %d. %B %Y')
+                    if ex.kontierung == '':
+                        sub_kontierung = time_entry.get("pos_service").split("-")[1]
+                        writer.writerow([ex.bestellnummer,ex.bestellposition,ex.leistungszeilennummer,ex.kurztext,ex.leistungszeilentext,ex.username,ex.name,datetime_object.strftime('%d.%m.%Y'),time_entry.get(ex.sum_col),time_entry.get(ex.desc_col),sub_kontierung,ex.ansprechpartner_username,ex.ansprechpartner_name,"OFFEN"])
+                    else:
+                        writer.writerow([ex.bestellnummer,ex.bestellposition,ex.leistungszeilennummer,ex.kurztext,ex.leistungszeilentext,ex.username,ex.name,datetime_object.strftime('%d.%m.%Y'),time_entry.get(ex.sum_col),time_entry.get(ex.desc_col),ex.kontierung,ex.ansprechpartner_username,ex.ansprechpartner_name,"OFFEN"])
+
+    # Check if the file size is small (indicating only the header)
+    file_path = f"{csvexport_filename}.csv"
+
+    # Use os.path.getsize to get the file size
+    if os.path.getsize(file_path) <= 190:  # Adjust this size threshold based on your expected header size
+        os.remove(file_path)
+        print(f"File {file_path} was deleted because it only contained the header.")
+    else:
+        print(f"File {file_path} contains data and was not deleted.")
+
+locale.setlocale(locale.LC_ALL, 'de_DE')
+
+with open("./app/config.yaml", encoding='utf8') as f:
+    cfg = yaml.load(f, Loader=yaml.FullLoader)
+
+gc = gspread.service_account()
+sh = gc.open(cfg['spreadsheet_name'])
+worksheet = sh.worksheet(cfg['worksheet_name'])
+week = cfg['week']    # 10, 11, 12 or None
+
+username = cfg['username']
+name = cfg['name']
+ansprechpartner_username = cfg['ansprechpartner_username']
+ansprechpartner_name = cfg['ansprechpartner_name']
+
+# read the whole sheet
+list_of_dicts = worksheet.get_all_records(numericise_ignore=['all'])
+
+for key in ["wgwr3", "wgwr_service", "dkarb", "bauwb", "service"]:
+    elsap_id = list_of_dicts[2].get(f"sum_{key}")
+    if elsap_id != "": 
+        csvexport_filename = f"{datetime.date.today().strftime('%Y_%m')}_01_{elsap_id}_{username}"
+
+        ex1 = Export(sum_col=f"sum_{key}",
+                     desc_col=f"desc_{key}",
+                     week=week,
+                     bestellnummer=elsap_id.split('_')[0],
+                     bestellposition=elsap_id.split('_')[1],
+                     leistungszeilennummer=elsap_id.split('_')[2],
+                     kurztext=list_of_dicts[1].get(f"sum_{key}"),
+                     leistungszeilentext=list_of_dicts[3].get(f"sum_{key}"),
+                     username=username,
+                     name=name,
+                     kontierung=list_of_dicts[4].get(f"sum_{key}"),
+                     ansprechpartner_username=ansprechpartner_username,
+                     ansprechpartner_name=ansprechpartner_name)
+
+        export_timesheet(csvexport_filename=csvexport_filename, ex=ex1)