ДЗ от 05.12.2020

import socket

import threading

import tkinter as tk

from tkinter import scrolledtext

​

​

def registration():

    # Создаём надпись и крепим в 1 строку, 1 столбец

    tk.Label(mainWindow, text='Введите логин: ').grid(row=0, column=0)

    # Создаём поле для ввода логина и сохраняем в переменную

    # Так как после нам нужно будет достать из неё введенный текст

    user_name_input = tk.Entry(mainWindow)

    user_name_input.grid(row=0, column=1)

    # Создаём кнопку, по нажатии которой вызовется функций

    tk.Button(mainWindow, text='Подключиться', command=lambda: login(user_name_input.get())).grid(row=0, column=2)

​

​

def login(nickname):

    if nickname == '':

        return

    # Удаляем все ранее добавленные виджеты в окне: поле для ввода ника и кнопку

    for widget in mainWindow.winfo_children():

        widget.destroy()

    nickname = mes_code(nickname)

    # Отправляем на сервер уведомление что пользователь вошел в чат

    sock.sendto(nickname.encode(), server)

    # Сохраняем ник пользователя в глобальной переменной

    global user_name

    user_name = nickname

    # Создаём область для чата - прокручиваемый текст

    chat_area = scrolledtext.ScrolledText(mainWindow, height=15)

    chat_area.configure(state=tk.DISABLED)

    chat_area.pack()

    # Создаём область для ввода сообщений и отправки

    input_frame = tk.Frame(mainWindow)

    input_area = tk.Entry(input_frame)

    input_area.pack(side=tk.LEFT, fill=tk.X, expand=True, padx=(0, 5))

    send_button = tk.Button(input_frame, text='Отправить', command=lambda: send_message(input_area))

    send_button.pack(side=tk.LEFT)

    input_frame.pack(fill=tk.BOTH, expand=True, padx=5)

    # запускаем поток проверки сообщений

    get_mes = threading.Thread(target=get_message, args=(chat_area,))

    get_mes.start()

​

​

def send_message(input_field):

    # Получаем текст сообщения с помощью метода get

    mes = input_field.get()

    # Удаляем текст из поля для ввода (0 - начальный символ для удаления, tk.END - последний)

    input_field.delete(0, tk.END)

    # Шифруем отправляемое сообщение

    mes = mes_code(': ' + mes)

    # Объединяем имя пользователя и сообщение перед отправкой на сервер

    data = user_name + mes

    # Отправляем сообщение на сервер

    sock.sendto(data.encode(), server)

​

​

def get_message(chat_area):

    while True:

        # Ждём сообщение, а после записываем его в переменную

        new_mes = sock.recv(1024)

        # Снимаем блокировку на области чата

        chat_area.configure(state=tk.NORMAL)

        # Расшировываем сообщение

        new_mes = mes_decode(new_mes.decode())

        # Добавляем в чат на последнее место (tk.END) новое сообщение и делаем перевод строки

        chat_area.insert(tk.END, new_mes + '\n')

        # Снова накладываем блокировку на поле чата

        chat_area.configure(state=tk.DISABLED)

​

​

def mes_code(mes):

    code_mes = ''

    for s in mes:

        code_mes += sym[(sym.index(s) + 7) % len(sym)]

    return code_mes

​

​

def mes_decode(mes):

    decode_mes = ''

    for s in mes:

        decode_mes += sym[(sym.index(s) - 7) % len(sym)]

    return decode_mes

​

​

# Список доступных символов для шифровки

sym = 'ецDСonн0щйg-=KЬ+Уk3ч!ХЩдWкTФSъCЭMьТt_ЪБП8^@bVxcqiтэ$ЗЖ2юuхаШуjeI#и(РHовE56шлЧКUQзО*ИЁmhw17NY4rJпXЕ:АЫЯ;zДяМp,' \

      'ЮмсOВyB№ЦЛFГ)P/фdfбsрZAvёыRL?G&9Й.Нlж%aг '

# Запоминаем данные сервера

server = '176.121.241.20', 9090

# Создаем сокет для обмена информацией

sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)

# Привязываем сокет к порту 9090

sock.bind(('', 9090))

​

# Сохраняем имя пользователя глобально

user_name = ''

# Создаём окно, устанавливаем имя и размер

mainWindow = tk.Tk()

mainWindow.title("Супер чат")

mainWindow.geometry('400x300')

# Запускаем процесс регистрации (установку кнопки и поля для ввода имени)

registration()

# Запускаем основную программу, заставляем окно появится

mainWindow.mainloop()

# После закрытия окна, отправляем сообщение на сервер о выходе

sock.sendto('0'.encode(), server)