# klient_szyfrowany.py
# Wysyłanie wiadomość

import os
from socket import *
from cryptography.fernet import Fernet


host = "127.0.0.1" # Adres IP docelowego komputera
port = 8080
addr = (host, port)
UDPSock = socket(AF_INET, SOCK_DGRAM)
key = input("Wprowadź klucz: ")
f = Fernet(key)

def encrypt(plaintext):
    msg = f.encrypt(plaintext)
    return msg

while True:
    data = str(input("Wprowadź wiadomość, zmień klucz, wpisując 'newkey', lub zakończ działanie, wpisując 'exit': ")).encode()
    ciphertext = encrypt(data)
    UDPSock.sendto(ciphertext, addr)
    if data == b'exit':
        break
    if data == b'newkey':
        key = input("Wprowadź nowy klucz: ")
        f = Fernet(key)
UDPSock.close()
os._exit(0)


