N.......L.......I
..I...I...N...V
....H.......O

NLI IIN VHO

def szyfruj_slowo(slowo, klucz=5):
    # test typu danych wejściowych
    if type (slowo) is not str:
        return False

    if klucz < 2:
        return False

    # Krok 1. Usuwamy spacje i zmieniamy wszystkie litery na WIELKIE
    tekst_jawny = "".join([litera.upper() for litera in slowo if litera != " "])

    # Krok 2. Tworzymy listę list - płotek.
    liczba_znakow = len(tekst_jawny)
    plotek = [[" " for _ in range(klucz)] for _ in range(liczba_znakow)]

    # Krok 3. Rozmieszczamy kolejne znaki w listach
    # flaga kierunek = 1 - wypełniamy płotek w dół
    # flaga kierunek = -1 - wypełniamy płotek w górę

    kierunek = 1  # rozpoczynamy wypełnianie z góry na dół
    indeks = 0 # indeks kolejnego znaku jest wykorzystywany jako indeks pierwszej listy
    wiersz = 0  # nr wiersza, w którym aktualnie wstawiamy znak
    # gdy osiągniemy ostatni wiersz (zależy od wartości parametru klucz)
    # zmieniamy wartość zmiennej flaga
    while tekst_jawny:
        znak_tekstu = tekst_jawny[0]  # pierwszy znak
        plotek[indeks][wiersz] = znak_tekstu
        indeks += 1

        # teraz sprawdzamy warunek dla flagi
        if wiersz == klucz-1:  # w pythonie indeks zaczynamy liczyć od zera, więc ostatni -1
            kierunek = -1
        elif wiersz == 0:
            kierunek = 1

        # teraz w zależności od flagi zwiększamy lub zmniejszamy
        wiersz += kierunek

        # przepisujemy resztę tekstu
        tekst_jawny = tekst_jawny[1:]  # reszta jawnego tekstu, bez pierwszego znaku


    # Krok 4. odczytujemy zaszyfrowaną wiadomość
    szyfrogram = ""
    for wiersz in range(klucz):
        for wiersz_plotka in plotek:
            znak = wiersz_plotka[wiersz]
            if znak != " ": # jeśli mamy znak zamiast spacji
                szyfrogram += znak # dodajemy go do szyfrogramu

    # Zwracamy rezultat
    return szyfrogram

print(szyfruj_slowo("NIHILNOVI", 2))
print(szyfruj_slowo("NIHILNOVI", 3))
print(szyfruj_slowo("NIHILNOVI", 4))
print(szyfruj_slowo("NIHILNOVI", 5))
print(szyfruj_slowo("NIHILNOVI", 6))
print(szyfruj_slowo("NIHILNOVI", 1))
print(szyfruj_slowo("NIHILNOVI", -3))
print(szyfruj_slowo(123, 6))

# NHLOIIINV
# NLIIINVHO
# NOINVHLII
# NIIVHOINL
# NIHIIVLON
# False
# False
# False

def odszyfruj_slowo(szyfrogram, klucz):
    # test typu danych wejściowych
    if type (szyfrogram) is not str:
        return False

    if klucz < 2:
        return False

    # Odszyfrowywuje słowo na podstawie klucza

    # Przechowujemy informację o długości szyfrogramu
    # Będziemy wykorzystywać ją w pętli while ... przy uzupełnianiu słowa
    szyfrogram_len = len(szyfrogram)
    jawny_tekst = ""    # Odszyfrowane słowo
    # przygotowujemy listę list dla znaków szyfrogramu
    plotek = [["-" for _ in range(klucz)] for _ in range(szyfrogram_len)]


    # Wypełniamy listę * w miejscach liter
    # flaga kierunek = 1 - wypełniamy płotek w dół
    # flaga kierunek = -1 - wypełniamy płotek w górę

    kierunek = 1  # rozpoczynamy wypełnianie z góry na dół
    indeks = 0 # indeks kolejnego znaku jest wykorzystywany jako indeks pierwszej listy
    wiersz = 0  # nr wiersza, w którym aktualnie wstawiamy znak
    ile_znakow = szyfrogram_len
    # gdy osiągniemy ostatni wiersz (zależy od wartości parametru klucz)
    # zmieniamy wartość zmiennej flaga
    while ile_znakow:
	    plotek[indeks][wiersz] = "*"
        indeks += 1

        # teraz sprawdzamy warunek dla flagi
        if wiersz == klucz-1:  # w pythonie indeks zaczynamy liczyć od zera, więc ostatni -1
            kierunek = -1
        elif wiersz == 0:
            kierunek = 1

        # teraz w zależności od flagi zwiększamy lub zmniejszamy
        wiersz += kierunek

        # zmniejszamy wielkość ilość znaków o 1
        ile_znakow -= 1

    # na kolejnych pozycjach gwiazdki wykładamy kolejne znaki

    while szyfrogram:
        for indeks in range(klucz):
            for wiersz in plotek:
                # ponieważ musimy iterować tyle razy, ile wynosi klucz,
                # natrafimy kilka razy na wiersz, w którym znak "*" zastąpiliśmy już
                # literą z szyfogramu; wtedy wiersz.index("*") zakończy się błędem
                # stosujemy konstrukcję try, aby nie przerwać skryptu błędem ValueError
                try:
                    if wiersz.index("*") == indeks: # miejsce "*" jest równe indeksowi klucza
                        wiersz[indeks] = szyfrogram[0] # wpisujemy do niego literę
                        szyfrogram = szyfrogram[1:] # pozostawiamy resztę szyfrogramu
                except:
                    pass
                    # przechodzimy dalej bez jakiegokolwiek komunikatu na ekranie
                    # aby nie zniekształcać obrazu

    # Uzupełniamy słowo analogicznie jak przy szyfrowaniu
    kierunek = 1  # rozpoczynamy wypełnianie z góry na dół
    indeks = 0 # indeks kolejnego znaku jest wykorzystywany jako indeks pierwszej listy
    wiersz = 0  # nr wiersza, w którym aktualnie wstawiamy znak
    ile_znakow = 0 # ile znakow odczytaliśmy, porównujemy do szyfrogram_len
    # gdy osiągniemy ostatni wiersz (zależy od wartości parametru klucz)
    # zmieniamy wartość zmiennej flaga
    while ile_znakow < szyfrogram_len:
		jawny_tekst += plotek[indeks][wiersz]
        indeks += 1

        # teraz sprawdzamy warunek dla flagi
        if wiersz == klucz-1:  # w pythonie indeks zaczynamy liczyć od zera, więc ostatni -1
            kierunek = -1
        elif wiersz == 0:
            kierunek = 1

        # teraz w zależności od flagi zwiększamy lub zmniejszamy
        wiersz += kierunek

        # zwiększamy ilość znaków zebranych
        ile_znakow += 1

    return jawny_tekst

print(odszyfruj_slowo("NHLOIIINV", 2))
print(odszyfruj_slowo("NLIIINVHO", 3))
print(odszyfruj_slowo("NOINVHLII", 4))
print(odszyfruj_slowo("NIIVHOINL", 5))
print(odszyfruj_slowo("NIHIIVLON", 6))
print(odszyfruj_slowo("NIHIIVLON", 1))
print(odszyfruj_slowo(124, 6))
print(odszyfruj_slowo("NIHIIVLON", -4))

# NIHILNOVI
# NIHILNOVI
# NIHILNOVI
# NIHILNOVI
# NIHILNOVI
# False
# False
# False

for klucz in range(2, 11):
	tekst = odszyfruj_slowo("FAINTNOOIRPSOUCS", klucz)
	print(f"Odszyfrowany tekst dla klucza {klucz} to: {tekst}")

	
# Odszyfrowany tekst dla klucza 2 to: FIARIPNSTONUOCOS
# Odszyfrowany tekst dla klucza 3 to: FTONAOUOIICRNPSS
# Odszyfrowany tekst dla klucza 4 to: FNIURTANPCSOIOOS
# Odszyfrowany tekst dla klucza 5 to: FIOPCSONATIOSURN
# Odszyfrowany tekst dla klucza 6 to: FINISCORONATOPUS
# Odszyfrowany tekst dla klucza 7 to: FINISUSCORONATOP
# Odszyfrowany tekst dla klucza 8 to: FINORSUSCOPIONAT
# Odszyfrowany tekst dla klucza 9 to: FANNORSUSCOPIOTI
# Odszyfrowany tekst dla klucza 10 to: FAINNORSUSCOPIOT

import PySimpleGUI as sg
import sys

try:
    from funkcje_szyfrujace import *
except ModuleNotFoundError:
    print("Brak pliku: 'funkcje_szyfrujace.py' - popraw to.")
    # koniec programu z Error Code 2 (FileNotFound)
    sys.exit(2)
    
# jeśli funkcje zostały zaimportowane, działamy dalej...    

uklad = [[sg.Text("Podaj tekst jawny lub szyfrogram (max. 35 znaków)")],      
         [sg.Input(key="dane")],
         [sg.Radio("Szyfruj",1 , key="s", default=True),
          sg.Radio("Odszyfruj",1 , key="o"),
          sg.Text("Wartość klucza"), sg.Input(size=(3, 1), key="k")],
         [sg.Text("Tekst wynikowy: ")],
         [sg.Text(" " * 40, size=(40,1), auto_size_text=True, key="-OUT-")],
         [sg.Button('Wykonaj'), sg.Exit()]]
okno = sg.Window('Szyfr płotkowy', uklad)

while True:                             
    event, values = okno.read() 
    # wykonujemy próbę zmiany wartości na liczbę
    try:
        klucz = int(values["k"])
    # w przypadku niepowodzenia przyjmujemy 3 jako wartość klucza
    except:
        klucz = 3
        
    if event == 'Exit' or event is None:
        break
    if event == 'Wykonaj':
        if values['s']:
            wynik = szyfruj_slowo(values["dane"], klucz)
        if values['o']:
            wynik = odszyfruj_slowo(values["dane"], klucz)
        okno["-OUT-"].update(wynik)
okno.close()