From 4f7d76d5d58c4d5ce0fd4c1b6859e4bfb4dba63d Mon Sep 17 00:00:00 2001
From: nex <me@nexy7574.co.uk>
Date: Tue, 2 Apr 2024 02:38:20 +0100
Subject: [PATCH] add common resolutions to screenshot

---
 src/cogs/screenshot.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/src/cogs/screenshot.py b/src/cogs/screenshot.py
index b04801a..9390ba0 100644
--- a/src/cogs/screenshot.py
+++ b/src/cogs/screenshot.py
@@ -112,6 +112,18 @@ class ScreenshotCog(commands.Cog):
             )
             driver.set_page_load_timeout(load_timeout)
             if resolution:
+                common = {
+                    "8k": "7680x4320",
+                    "4k": "3840x2160",
+                    "1440p": "2560x1440",
+                    "1080p": "1920x1080",
+                    "720p": "1280x720",
+                    "480p": "854x480",
+                    "360p": "640x360",
+                    "240p": "426x240",
+                    "144p": "256x144"
+                }
+                resolution = common.get(resolution.lower(), resolution)
                 try:
                     width, height = map(int, resolution.split("x"))
                     driver.set_window_size(width, height)