FTP Upload Implementation

Author: PoulDev

README.md

@@ -40,6 +40,6 @@
 
 > ## Client Configuration
 > At this point you will have to set the client to use the personal server, after loading the server folder with the [configuration executed previously](https://github.com/PoulDev/PyScreenSaver#server-configuration) we open the configs.json file in the "data" folder and modify some values.
-> - The first value to set is "use_server", which will have to be set to true
+> - The first value to set is "enabled", which will have to be set to true
 > - The second value is "server_password", and you have to enter the chosen password in the server configuration
-> - The last important value to change is "server_ip", where you have to put the ip of your server.
+> - The last important value to change is "server_ip", where you have to put the ip of your server.

data/configs.json

@@ -5,9 +5,17 @@
         "outline_rgb": [255, 255, 255]
     },
     "server": {
-        "use_server": false,
+        "enabled": false,
         "server_ip": "127.0.0.1",
         "server_port": 8482,
         "server_password": "Your Super Secret Password"
+    },
+    "ftp_server": {
+        "enabled": false,
+        "address": "",
+        "username": "",
+        "password": "",
+        "directory": "",
+        "url": "http://127.0.0.1/"
     }
-}
+}

main.py

@@ -1,5 +1,6 @@
 import ctypes
 import hashlib
+from typing import Any, Callable, Union
 from PySide6.QtCore import *
 from PySide6.QtGui import QPixmap
 from PySide6.QtWidgets import *
@@ -12,14 +13,21 @@
 import sys
 import mss
 import json
-import time
+import win32con
+import win32gui
+import ctypes
 import keyboard
 import pyperclip
 import requests
+import random
+import string
 import win32clipboard
+from ftplib import FTP
+
 
 from data.selector import Ui_MainWindow
 
+
 configs = json.loads(open('data\\configs.json', 'r').read())
 
 OUTLINE = tuple(configs['selector']['outline_rgb'])
@@ -46,6 +54,73 @@ def uploadToServer():
         headers = {"authentication" : hashlib.sha256(configs['server']['server_password'].encode()).hexdigest()}
     ).json()
 
+class UploadMethod:
+    INSTANCE: Any = None
+
+    def __init__(self):
+        if UploadMethod.INSTANCE:
+            raise RuntimeError('UploadMethod is a singletone Object.')
+        UploadMethod.INSTANCE = self
+        self.save_image = self._find_upload_method()
+
+    def animate_cursor(self):
+        cursor = win32gui.LoadImage(0, 32512, win32con.IMAGE_CURSOR, 
+            0, 0, win32con.LR_SHARED)
+        self.system_cursor = ctypes.windll.user32.CopyImage(cursor, win32con.IMAGE_CURSOR, 
+            0, 0, win32con.LR_COPYFROMRESOURCE)
+
+        cursor = win32gui.LoadImage(0, "working.ani", win32con.IMAGE_CURSOR, 
+                0, 0, win32con.LR_LOADFROMFILE);
+        ctypes.windll.user32.SetSystemCursor(cursor, 32512)
+        ctypes.windll.user32.DestroyCursor(cursor);
+
+
+    def restore_cursor(self):
+        ctypes.windll.user32.SetSystemCursor(self.system_cursor, 32512)
+        ctypes.windll.user32.DestroyCursor(self.system_cursor);
+
+    def _find_upload_method(self) -> Callable:
+        for upload_method, method_function in {'ftp_server': self._ftp_upload, 'server': self._server_upload}.items():
+            if configs[upload_method]['enabled']:
+                return method_function
+        return self._clipboard_save
+
+    def _ftp_upload(self, image: Image.Image) -> None:
+        filename = 'screenshot_' + ''.join(random.choices(string.ascii_letters, k=15)) + '.png'
+        image.save(filename)
+        with FTP(
+                configs['ftp_server']['address'],
+                configs['ftp_server']['username'],
+                configs['ftp_server']['password']) as ftp, open(filename, 'rb') as file:
+            ftp.cwd(configs['ftp_server']['directory'])
+            ftp.storbinary(f'STOR {filename}', file)
+
+        pyperclip.copy(f'{configs["ftp_server"]["url"]}/{filename}')
+        try:
+            os.remove(filename)
+        except: pass
+
+    def _server_upload(self, image: Image.Image) -> None:
+        image.save('screenshot.png')
+        res = uploadToServer()
+        if not 'uid' in res:
+            ctypes.windll.user32.MessageBoxW(0, res['error'], "Upload Error", 0)
+        else:
+            pyperclip.copy(f'http://{configs["server"]["server_ip"]}:{configs["server"]["server_port"]}/{res["uid"]}')
+        try:
+            os.remove('screenshot.png')
+        except: pass
+
+    def _clipboard_save(self, image: Image.Image) -> None:
+        output = io.BytesIO()
+        image.convert("RGB").save(output, "BMP")
+        data = output.getvalue()[14:]
+        output.close()
+        send_to_clipboard(win32clipboard.CF_DIB, data)
+
+
+UploadMethod()
+
 class SelectorWindow(QMainWindow):
     def __init__(self):
         QMainWindow.__init__(self)
@@ -100,24 +175,10 @@ def mouseReleaseEvent(self, event):
                 print(e)
                 return self.ui.image.setPixmap(QPixmap.fromImage(ImageQt(self.originalImg)))
 
-            # Copy screenshot to clipboard
             self.close()
-            if configs['server']['use_server']:
-                cropped_im.save('screenshot.png')
-                res = uploadToServer()
-                if not 'uid' in res:
-                    ctypes.windll.user32.MessageBoxW(0, res['error'], "Upload Error", 0)
-                else:
-                    pyperclip.copy(f'http://{configs["server"]["server_ip"]}:{configs["server"]["server_port"]}/{res["uid"]}')
-                try:
-                    os.remove('screenshot.png')
-                except: pass
-            else:
-                output = io.BytesIO()
-                cropped_im.convert("RGB").save(output, "BMP")
-                data = output.getvalue()[14:]
-                output.close()
-                send_to_clipboard(win32clipboard.CF_DIB, data)
+            UploadMethod.INSTANCE.animate_cursor()
+            UploadMethod.INSTANCE.save_image(cropped_im)
+            UploadMethod.INSTANCE.restore_cursor()
 
         self.startpos = None
         super().mouseReleaseEvent(event)