[display] use dict instead of judge

Author: ColdWindScholar

library/display.py

@@ -55,20 +55,17 @@ def _get_theme_orientation() -> Orientation:
 
 
 def _get_theme_size() -> tuple[int, int]:
-    if config.THEME_DATA["display"].get("DISPLAY_SIZE", '') == '0.96"':
-        return 80, 160
-    if config.THEME_DATA["display"].get("DISPLAY_SIZE", '') == '2.1"':
-        return 480, 480
-    elif config.THEME_DATA["display"].get("DISPLAY_SIZE", '') == '3.5"':
-        return 320, 480
-    elif config.THEME_DATA["display"].get("DISPLAY_SIZE", '') == '5"':
-        return 480, 800
-    elif config.THEME_DATA["display"].get("DISPLAY_SIZE", '') == '8.8"':
-        return 480, 1920
-    else:
+    sizes = {
+        '0.96"':(80, 160),
+        '2.1"':(480, 480),
+        '3.5"':(320, 480),
+        '5"':(480, 800),
+        '8.8"':(480, 1920),
+    }
+    if config.THEME_DATA["display"].get("DISPLAY_SIZE", '') not in sizes.keys():
         logger.warning(
-            f'Cannot find valid DISPLAY_SIZE property in selected theme {config.CONFIG_DATA["config"]["THEME"]}, defaulting to 3.5"')
-        return 320, 480
+        f'Cannot find valid DISPLAY_SIZE property in selected theme {config.CONFIG_DATA["config"]["THEME"]}, defaulting to 3.5"')
+    return sizes.get(config.THEME_DATA["display"].get("DISPLAY_SIZE", ''), (320, 480))
 
 
 class Display: