interface-udp/interface.py at main · apfdamascena/interface-udp · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
from tkinter import *
from tkinter import filedialog
from client import ClientUDP
from ImageScreen import ImageScreen
from ExtensionType import ExtensionType
from MessageCheck import MessageCheck
from TypeFile import TypeFile
import os
import sys

class GUI:
    def __init__(self, width, height):
        self.__window = Tk()
        self.__window.title('4 Atividade')
        self.__window.bind('<Return>', self.__send)

        self.__canva = Canvas(self.__window, width=width, height=height)
        self.__canva.grid(columnspan=4)

        self.__message_checker = MessageCheck()

        self.__createWidgets()

        self.__client = ClientUDP(28886, self)
        self.__client.listenning()

        self.__video_counter = 0
        self.__audio_counter = 0

    def __createWidgets(self):
        self.__txt_area = Text(self.__canva, border=1)
        self.__image_on_screen = ImageScreen(self.__txt_area)

        self.__txt_field = Entry(self.__canva, width=80, border=1, bg='white')

        self.__attachment = Button(self.__canva, text='Attachment', command=self.__open_dialog_with_files)
        self.__send_button = Button(self.__canva, text='Send', padx=30, command=self.__send)
        self.__video_button = Button(self.__canva, text='Play Video', padx=30, command=self.__play_video)
        self.__audio_button = Button(self.__canva, text='Play Audio', padx=30, command=self.__play_audio)
        self.__clear_button = Button(self.__canva, text='Clear', padx=30, command=self.__clear)

        self.__listaudio = Listbox(selectmode = SINGLE, width = 20)
        self.__listvideo = Listbox(selectmode = SINGLE, width = 20)

        self.__listvideo.grid(column=2, row=0, columnspan=2)
        self.__listaudio.grid(column=3, row=0, columnspan=2)
        self.__send_button.grid(column=2, row=2)
        self.__clear_button.grid(column=3, row=2)
        self.__attachment.grid(column=4, row=2)
        self.__video_button.grid(column=5, row=2)
        self.__audio_button.grid(column=6,row=2)
        self.__txt_area.grid(column=0, row=0)
        self.__txt_field.grid(column=0, row=2)

        self.__txt_area.config(background='#abd3eb')

    def __get_open_command(self, filename):
        operational_system = sys.platform
        if operational_system in ['linux', 'linux2']:
            filename = filename.replace(' ', '\ ')
            return f'xdg-open {filename}'
        elif operational_system in ['win32', 'win64']:
            return f'"{filename}"'
        else:
            return f'open {filename}'

    def __open_dialog_with_files(self):
        filename = filedialog.askopenfilename()
        extension_type = ExtensionType().check_extension(filename)

        self.__image_on_screen.add(extension_type, filename)

        if extension_type == TypeFile.AUDIO:
           self.__add_list_audio(filename)

        if extension_type == TypeFile.VIDEO:
           self.__add_list_video(filename)

        self.__send_file(filename)

    def __send_file(self, filename):
        name = filename.split('/')[-1]
        header = f'{name}:file'
        self.__client.send_message(header)

        file = open(filename, 'rb')
        parts_bytes_file = True
        while parts_bytes_file:
            parts_bytes_file = file.read(ClientUDP.BUFFER_SIZE)
            self.__client.send_message(parts_bytes_file)
        file.close()

    def __play_video(self):
        chosen_video = self.__listvideo.get(self.__listvideo.curselection())
        os.system(self.__get_open_command(chosen_video))

    def __play_audio(self):
        chosen_audio= self.__listaudio.get(self.__listaudio.curselection())
        os.system(self.__get_open_command(chosen_audio))

    def receive_and_show_at_screen(self, message):
        if self.__message_checker.is_printable(message):
            self.__txt_area.insert(END, f'Another User: {message}')
            self.__txt_field.delete(0, END)

    def receive_and_show_image(self, filename):
        extension_type = ExtensionType().check_extension(filename)
        self.__image_on_screen.add(extension_type, filename)

        if extension_type == TypeFile.AUDIO:
            self.__add_list_audio(filename)

        if extension_type == TypeFile.VIDEO:
            self.__add_list_video(filename)

    def __send(self, event=None):
        text = self.__message_checker.is_valid_input(self.__txt_field.get())
        self.__client.send_message(text)
        self.__txt_area.insert(END, f'You: {text}')
        self.__txt_field.delete(0, END)

    def __clear(self):
        self.__txt_area.delete('1.0', END)

    def start(self):
        self.__window.mainloop()

    def __add_list_audio(self, filename):
        self.__listaudio.insert(self.__audio_counter, filename)
        self.__audio_counter += 1

    def __add_list_video(self, filename):
        self.__listvideo.insert(self.__video_counter, filename)
        self.__video_counter += 1


if __name__ == '__main__':
    interface = GUI(800, 600).start()