From f19477c6fef4b8a4df4ad387f9fb373ca0e735ea Mon Sep 17 00:00:00 2001
From: BarsTiger <kit.barsic@gmail.com>
Date: Sat, 5 Feb 2022 14:20:10 +0200
Subject: [PATCH] Package editing

---
 modules/http_status.py  | 48 +++++++++++++++++++++++++++++++++++++++++
 modules/package_edit.py | 18 ++++++++++++++--
 2 files changed, 64 insertions(+), 2 deletions(-)
 create mode 100644 modules/http_status.py

diff --git a/modules/http_status.py b/modules/http_status.py
new file mode 100644
index 0000000..8218a9c
--- /dev/null
+++ b/modules/http_status.py
@@ -0,0 +1,48 @@
+CONTINUE = 100
+SWITCHING_PROTOCOLS = 101
+PROCESSING = 102
+EARLYHINTS = 103
+OK = 200
+CREATED = 201
+ACCEPTED = 202
+NON_AUTHORITATIVE_INFORMATION = 203
+NO_CONTENT = 204
+RESET_CONTENT = 205
+PARTIAL_CONTENT = 206
+AMBIGUOUS = 300
+MOVED_PERMANENTLY = 301
+FOUND = 302
+SEE_OTHER = 303
+NOT_MODIFIED = 304
+TEMPORARY_REDIRECT = 307
+PERMANENT_REDIRECT = 308
+BAD_REQUEST = 400
+UNAUTHORIZED = 401
+PAYMENT_REQUIRED = 402
+FORBIDDEN = 403
+NOT_FOUND = 404
+METHOD_NOT_ALLOWED = 405
+NOT_ACCEPTABLE = 406
+PROXY_AUTHENTICATION_REQUIRED = 407
+REQUEST_TIMEOUT = 408
+CONFLICT = 409
+GONE = 410
+LENGTH_REQUIRED = 411
+PRECONDITION_FAILED = 412
+PAYLOAD_TOO_LARGE = 413
+URI_TOO_LONG = 414
+UNSUPPORTED_MEDIA_TYPE = 415
+REQUESTED_RANGE_NOT_SATISFIABLE = 416
+EXPECTATION_FAILED = 417
+I_AM_A_TEAPOT = 418
+MISDIRECTED = 421
+UNPROCESSABLE_ENTITY = 422
+FAILED_DEPENDENCY = 424
+PRECONDITION_REQUIRED = 428
+TOO_MANY_REQUESTS = 429
+INTERNAL_SERVER_ERROR = 500
+NOT_IMPLEMENTED = 501
+BAD_GATEWAY = 502
+SERVICE_UNAVAILABLE = 503
+GATEWAY_TIMEOUT = 504
+HTTP_VERSION_NOT_SUPPORTED = 505
diff --git a/modules/package_edit.py b/modules/package_edit.py
index 1138f4e..9e4fb32 100644
--- a/modules/package_edit.py
+++ b/modules/package_edit.py
@@ -3,8 +3,8 @@ from modules.auth import get_auth
 from horsygui import login_ui, UiLoginWindow as Ui_LoginWindow
 import modules.gui as gui
 import requests
-from PyQt5 import QtWidgets
 import modules.vars as horsy_vars
+import modules.http_status as status
 import json
 
 
@@ -64,6 +64,20 @@ def edit(package, UiPackageWindow):
             'run': (lambda x: x if x != '' else None)(package_ui.main_exe_box.text())
         }
 
-        r = requests.put(f"{horsy_vars.protocol}{horsy_vars.server_url}/packages", json=body).text  # TODO: add error handling
+        r = requests.put(f"{horsy_vars.protocol}{horsy_vars.server_url}/packages", json=body)
+        if r.status_code == status.OK:
+            gui.cpopup('Success', 'Success, package edited')
+        elif r.status_code == status.UNAUTHORIZED:
+            gui.cpopup('Error', 'Unauthorized')
+        elif r.status_code == status.NOT_FOUND:
+            gui.cpopup('Error', 'Package not found or its not yours')
+        elif r.status_code == status.BAD_REQUEST:
+            gui.cpopup('Error', 'Bad request')
+        elif r.status_code == status.INTERNAL_SERVER_ERROR:
+            gui.cpopup('Error', 'Internal server error')
+        elif r.status_code == status.TOO_MANY_REQUESTS:
+            gui.cpopup('Error', 'Too many requests')
+        else:
+            gui.cpopup('Error', 'Unknown error')
 
     package_ui.update_button.clicked.connect(send)