WIDGETS!!11111!!!!

Author: MF366-Coding

core/database_handler.py

@@ -1,6 +1,10 @@
 import json
 from typing import Any
 import requests
+from PIL import Image
+import io
+
+# pylint: disable=W0718
 
 
 class StatusCodeNot200(Exception): ...
@@ -46,3 +50,44 @@ def get_database(source: str, error_handler: Any, mode: str = 'online') -> dict
 
     return error_handler(8, f"An unknown error happened when trying to access database located at:\n\n{source}\n\nMake sure the URL is correct.") # [!] Unknown error
 
+
+def get_image(source: str, handler: Any, **kw):
+    _ = kw.pop('url', None)
+    _ = kw.pop('timeout', None)
+    _ = kw.pop('allow_redirects', None)
+    del _
+    
+    try:
+        response = requests.get(url=source, allow_redirects=False, timeout=1.3)
+        
+        if response.status_code != 200:
+            raise StatusCodeNot200(f"status code is {response.status_code} and not 200")
+        
+        img_bytes: bytes | bytearray = response.content
+
+    except StatusCodeNot200:
+        return handler(31, f"The Status Code of URL...\n\n{source}\n\n...is not 200, meaning DoomMapGuesser is unable to collect the image.")
+    
+    except (requests.exceptions.ConnectionError, requests.exceptions.HTTPError):
+        return handler(24, "Unable to connect via HTTP. This might be caused by a bad Internet connection.")
+    
+    except (TimeoutError, requests.exceptions.Timeout):
+        return handler(21, f"The image's URL...\n\n({source})\n\n...took too long to respond to the GET request.")
+    
+    except (requests.exceptions.InvalidURL, requests.exceptions.InvalidSchema, requests.exceptions.MissingSchema):
+        return handler(27, f"The URL...\n\n({source})\n\n...is invalid.")
+    
+    except Exception as e:
+        return handler(28, f"Unknown error when trying to obtain the chosen image with URL:\n\n{source}\n\nError Details:\n{e}")
+    
+    try:
+        image_data = io.BytesIO(img_bytes)
+        image = Image.open(image_data, "r")
+        
+    except ValueError as e:
+        return handler(26, f"The collected image is not in its RAW form or isn't even an image, therefore cannot be used by DoomMapGuesser.\n\nError Details:\n{e}")
+
+    except Exception as e:
+        return handler(28, f"Unknown error when trying to read the chosen image with URL with PIL:\n\n{source}\n\nError Details:\n{e}")
+    
+    return image

main.py

@@ -10,10 +10,9 @@
 import simple_webbrowser
 from PIL import ImageTk, Image
 from core import level_db, utils_constants
-from core.database_handler import get_database
+from core.database_handler import get_database, get_image
 from core.settings import SettingsObject
 import math
-import io
 import requests
 import json
 
@@ -324,7 +323,7 @@ class __InvalidButtonAction(Exception): ...
 PH_DATA: list[str] | None = None
 POINTS: int = 0
 GEN_SF: int = 0
-CUR_IMAGE = None
+CUR_IMG_LINK = None
 
 
 class Database:
@@ -772,9 +771,7 @@ def add_database(source: str, *_, index: int | None = None) -> Database | bool:
 DATABASES[0].use()
 
 
-def generate_new_screenshot(**kw) -> None:
-    global CUR_DATA, GEN_SF
-
+def generate_new_map_data(**kw) -> list[str] | int:
     db: Database = kw.get('database', CUR_DB)
 
     warrens: list[str] | None = db.warrens
@@ -783,18 +780,15 @@ def generate_new_screenshot(**kw) -> None:
     # [*] 1st Case: WARRENS and HELL_KEEP are undefined
     # [i] this code assumes Database.verify() has been ran
     if warrens is None and hell_keep is None:
-        GEN_SF += 4
-        CUR_DATA = db.generate()
-        return
+        return db.generate()
 
     # [*] 2nd Case: WARRENS and HELL_KEEP can be both shown - in this case, we don't need to do anything
     if settings.exclude_rule_for_e3m1_e3m9 == 'both':
-        GEN_SF += 4
-        CUR_DATA = db.generate()
-        return
+        return db.generate()
 
     is_warrens = False
     is_hk = False
+    data = None
 
     # [*] 3rd Case: we loop around 20 times, which should be enough to get rid of WARRENS and HELL_KEEP generated data
     # [i] ofc that if we get valid data before the last loop, we end it, duh?!
@@ -823,16 +817,21 @@ def generate_new_screenshot(**kw) -> None:
                 if is_warrens is False:
                     break
 
-            case _:
+            case None | 'null' | 'none':
                 if is_hk is False and is_warrens is False:
                     break
+                
+            case _:
+                data = None
+                continue
+            
+    if data is None:
+        return handle_error(52, "Failed to get an image that respects the chosen exclusion rule for HELL_KEEP and WARRENS.")
 
-    GEN_SF += 4
-    CUR_DATA = data
-    return
+    return data
 
 
-def generate_new_image(**kw) -> str:
+def generate_new_image(data: list[str], **kw) -> str:
     """
     # generate_new_image
 
@@ -841,21 +840,65 @@ def generate_new_image(**kw) -> str:
     :param database: overwrite for CUR_DB *(should be of type Database)*
 
     Returns:
-        str: the link to the new image *(it also saves it as a global so it might not be necessary to get the returned value)*
+        str: the link to the new image
     """
 
-    global CUR_IMAGE
-    
     db: Database = kw.get('database', CUR_DB)
 
-    x = random.choice(db.structure[CUR_DATA[0]][CUR_DATA[1]][CUR_DATA[2]]['screenshots'].remove(CUR_IMAGE))
-    CUR_IMAGE = x
-    del x
+    x = random.choice(db.structure[data[0]][data[1]][data[2]]['screenshots'].remove(CUR_IMG_LINK))
+    return x
+
+
+def get_selected_image(img_link: str, **kw) -> Image.Image | int:
+    return get_image(img_link, handle_error, **kw)
+
+
+def generate_new_round(*_, first: dict[str, Any] = None, second: dict[str, Any] = None, third: dict[str, Any] = None):
+    global CUR_DATA, GEN_SF, CUR_IMG_LINK
+    
+    # [<] ik one letter vars are not good but screw it
+    if first is None:
+        a = generate_new_map_data()
+        
+    else:   
+        a = generate_new_map_data(**first)
+    
+    if isinstance(a, int):
+        return # [!?] Cancel the operation, since an error happened
+    
+    if second is None:
+        b = generate_new_image(a)
+    
+    else:
+        b = generate_new_image(b, **second)    
+    
+    if third is None:
+        c = get_selected_image(b)
 
-    return CUR_IMAGE
+    else:
+        c = get_selected_image(b, **third)
+
+    GEN_SF += 4
+    CUR_DATA = a
+    CUR_IMG_LINK = b
+    PLAY_ITEMS.cur_img = resize_image(c, settings.image_width, settings.use_width_as_height, settings.image_ratio)
+
+
+def update_game_widgets():
+    return handle_error(11, "Not Implemented!")
 
 
 def setup_play_screen():
+    global CUR_DATA, POINTS, GEN_SF, CUR_IMG_LINK, PH_DATA
+    
+    PH_DATA = [random.randint(10, 99) for _ in range(3)]
+    CUR_DATA = PH_DATA.copy()
+    
+    PLAY_ITEMS.selected_game = tk.StringVar(root, list(CUR_DB.structure.keys())[0])
+    PLAY_ITEMS.selected_episode = tk.StringVar(root, list(CUR_DB.structure[PLAY_ITEMS.selected_game].keys())[0])
+    PLAY_ITEMS.selected_map = tk.StringVar(root, list(CUR_DB.structure[PLAY_ITEMS.selected_game][PLAY_ITEMS.selected_episode].keys())[0])
+    PLAY_ITEMS.selected_secrets = tk.IntVar(root, 0)
+
     PLAY_ITEMS.heading = ttk.Label(PLAY_ITEMS, text='Play', font=HEADING1)
     PLAY_ITEMS.f1 = ttk.Frame(PLAY_ITEMS)
     PLAY_ITEMS.f2 = ttk.Frame(PLAY_ITEMS)
@@ -884,13 +927,20 @@ def setup_play_screen():
     PLAY_ITEMS.img_widget = ttk.Button(PLAY_ITEMS.f1, image=PLAY_ITEMS.cur_img, command=lambda:
         send_dialog('warning', 'TODO', 'Sadly, not done yet.')) # TODO
 
-    PLAY_ITEMS.generation_butt = ttk.Button(PLAY_ITEMS.f2, text="Generate", command=lambda:
-        send_dialog('warning', 'TODO', 'Sadly, not done yet.')) # TODO
+    PLAY_ITEMS.generation_butt = ttk.Button(PLAY_ITEMS.f2, text="Generate", command=generate_new_round)
 
     PLAY_ITEMS.guessing_butt = ttk.Button(PLAY_ITEMS.f2, text='Guess', command=lambda:
         send_dialog('warning', 'TODO', 'Sadly, not done yet.')) # TODO
 
-    # TODO: after this it's just the game, episode, map and thingies
+    
+    PLAY_ITEMS.game_ch = ttk.OptionMenu(PLAY_ITEMS.f5, PLAY_ITEMS.selected_game, PLAY_ITEMS.selected_game.get(), *list(CUR_DB.structure.keys()), command=lambda:
+        update_game_widgets('game'))
+    PLAY_ITEMS.episode_ch = ttk.OptionMenu(PLAY_ITEMS.f6, PLAY_ITEMS.selected_episode, PLAY_ITEMS.selected_episode.get(), *list(CUR_DB.structure[PLAY_ITEMS.selected_game].keys()), command=lambda:
+        update_game_widgets('episode'))
+    PLAY_ITEMS.map_ch = ttk.OptionMenu(PLAY_ITEMS.f7, PLAY_ITEMS.selected_map, PLAY_ITEMS.selected_map.get(), *list(CUR_DB.structure[PLAY_ITEMS.selected_game][PLAY_ITEMS.selected_episode].keys()), command=lambda:
+        update_game_widgets('map'))
+    
+    PLAY_ITEMS.secrets_plus = None # TODO
 
 
 # [*] Sidebar Buttons