import pandas as pd
import icecream as ic

dfdict = {}


def savetoDict(df, tablename):
    global dfdict
    dfdict[tablename] = df

def search(excel, letter):

    df1 = excel[excel['Schwedisch'].str.startswith((letter, letter.upper()))]
    sorted = (df1.sort_values(['Schwedisch', 'Deutsch'], key=lambda col: col.str.lower()))
    #unique_values_set = set(sorted['Schwedisch'])
    return sorted

excel = pd.read_excel("Schwedisch_Woerter.xlsx")

sverige = ['a','b','c','d','e','f','g','h','i','j','k','l','m','n','o','p','q','r','s','t','u','v','w','x','y','z','å','ä','ö']

for l in sverige:
    lf = search(excel, l)
    savetoDict(lf, l.upper())


#df1a = excel[excel['Schwedisch'].str.startswith('E')]
#df1b = excel[excel['Schwedisch'].str.startswith('e')]
#df1 = pd.concat([df1a, df1b], ignore_index=True)


#sorted = df1.sort_values(['Schwedisch', 'Deutsch'], key=lambda col: col.str.lower())
# df1.concat(df1b, ignore_index=True)
#print(sorted)
# if excel.columns["Schwedisch"].startswith("a"): #| excel.columns["Schwedisch"].startswith("A"):
# if excel[excel["Schwedisch"]]:
#    ic(excel.columns)
# dftest = pd.DataFrame([["schwedisches Wort", "Übersetzung", "sollte eine Bemerkung werden"]], columns=["Schwedisch", "Deutsch", "Bemerkung"])

#df2 = pd.DataFrame([["schwedisches Wort", "Übersetzung", "sollte eine Bemerkung werden"], ["Bla", "Blu", "Bli"]],
#                   columns=["Schwedisch", "Deutsch", "Bemerkung"])


for key, value in dfdict.items():
    print(f'{key}: {value}')

with pd.ExcelWriter('SchwedischSorted.xlsx') as writer:
    for sheet_name, content in dfdict.items():
        df = pd.DataFrame(content, columns=["Schwedisch", "Deutsch","Aussprache", "Bemerkung"])  # Umwandlung in DataFrame
        content.to_excel(writer, sheet_name=sheet_name, index=False)



# with pd.ExcelWriter("SchwedischSorted.xlsx") as writer:
#    df1.to_excel(writer, sheet_name="E",index=False)