encryptare + adaugare de imagini

messenger_gui/app/views/chat.py

@@ -1,7 +1,10 @@
+import base64
 import json
+import os
 import tkinter as tk
 from app.http import request_builder
 from app.views.message import MessageView
+import tkinter.filedialog
 
 
 class ChatWindow(tk.Toplevel):
@@ -9,13 +12,16 @@ class ChatWindow(tk.Toplevel):
     This view is opened in a separate window and takes care of the communication between the two users. It receives
     both the users' information and retrieves the conversation_url.
     """
-    def __init__(self, master=None, user_data=None, current_logged_user=None):
+
+    def __init__(self, master=None, encryptor=None, user_data=None, current_logged_user=None):
         super().__init__(master)
 
         self.user_data = user_data
         self.current_logged_user = current_logged_user
         self.define_get_conversation_url_params()
 
+        self.encryptor = encryptor
+
         self.geometry("700x700")
 
         self.conversation_url = self.get_conversation_url()
@@ -46,7 +52,7 @@ class ChatWindow(tk.Toplevel):
         self.params = {}
         self.body = {
             'conversation_url': self.conversation_url,
-            'message': message,
+            'message': self.encryptor.encrypt(message),
             'message_type': message_type,
             'sender': self.current_logged_user['id']
         }
@@ -59,6 +65,7 @@ class ChatWindow(tk.Toplevel):
 
         # Display existing conversation
         for message in conversation:
+            message['message'] = self.encryptor.decrypt(message['message'])
             message_view = MessageView(self.messages_frame, message, self.current_logged_user, self.user_data)
             message_view.pack()
 
@@ -74,6 +81,28 @@ class ChatWindow(tk.Toplevel):
         self.send_button = tk.Button(self.input_frame, text="Send", command=lambda: self.send_message())
         self.send_button.pack(side=tk.RIGHT)
 
+        self.select_photo_button = tk.Button(self.input_frame, text="Select Photo", command=self.select_photo)
+        self.select_photo_button.pack(side=tk.LEFT)
+
+    def select_photo(self):
+        initial_dir = os.path.expanduser("~")
+        file_path = tkinter.filedialog.askopenfilename(
+            initialdir=initial_dir, filetypes=[("Image files", "*.png;*.jpg;*.jpeg;*.gif;*.bmp")])
+        with open(file_path, 'rb') as file:
+            file_content = file.read()
+            image_base64 = base64.b64encode(file_content).decode('utf-8')
+            self.define_send_message_params(image_base64, "image")
+
+            request_builder.set_url(self.url)
+            request_builder.set_body(self.body)
+            request_builder.set_headers(self.headers)
+            request_builder.set_method(self.method)
+            request_builder.set_params(self.params)
+
+            request_builder.build().make_request()
+
+            self.update_chat()
+
     def update_chat(self):
         """
         This function is scheduled each second to make a request to the API to retrieve the latest chat between the
@@ -85,6 +114,7 @@ class ChatWindow(tk.Toplevel):
 
         self.define_get_conversation_params()
         for message in self.get_conversation():
+            message['message'] = self.encryptor.decrypt(message['message'])
             message_view = MessageView(self.messages_frame, message, self.current_logged_user, self.user_data)
             message_view.pack()