from cryptography import fernet
from fernet import Fernet


def encode_decode_message(fernet, password, encode):
    # Dla encoda > 0 kodujemy, dla < 0 dekodujemy
    if encode > 0:
        return fernet.encrypt(password.encode())
    else:
        return fernet.decrypt(password).decode()


if __name__ == '__main__':
    #Ferneta musisz wjebac gdzies do globalnych zmiennych bo jak jest w funkcji to nie dziala
    crypto_key = Fernet.generate_key()
    fernet = Fernet(crypto_key)
    password = input("Please, input your password: ")
    encoded_message = encode_decode_message(fernet, password, 1)
    decoded_message = encode_decode_message(fernet, encoded_message, -1)
    print(encoded_message, decoded_message)