Capitalize enum keys

Signed-off-by: Jacob Stopak <[email protected]>

Author: initialcommit-io

git_sim/__main__.py

@@ -208,7 +208,7 @@ def main(
         config.background_color = WHITE
 
     if settings.transparent_bg:
-        settings.img_format = ImgFormat.png
+        settings.img_format = ImgFormat.PNG
 
     t = datetime.datetime.fromtimestamp(time.time()).strftime("%m-%d-%y_%H-%M-%S")
     config.output_file = "git-sim-" + ctx.invoked_subcommand + "_" + t + ".mp4"

git_sim/animations.py

@@ -11,12 +11,13 @@
 from manim.utils.file_ops import open_file
 
 from git_sim.settings import settings
+from git_sim.enums import VideoFormat
 
 
 def handle_animations(scene: Scene) -> None:
     scene.render()
 
-    if settings.video_format == "webm":
+    if settings.video_format == VideoFormat.WEBM:
         webm_file_path = str(scene.renderer.file_writer.movie_file_path)[:-3] + "webm"
         cmd = f"ffmpeg -y -i {scene.renderer.file_writer.movie_file_path} -hide_banner -loglevel error -c:v libvpx-vp9 -crf 50 -b:v 0 -b:a 128k -c:a libopus {webm_file_path}"
         print("Converting video output to .webm format...")

git_sim/enums.py

@@ -27,10 +27,10 @@ class StyleOptions(Enum):
 
 
 class VideoFormat(str, Enum):
-    mp4 = "mp4"
-    webm = "webm"
+    MP4 = "mp4"
+    WEBM = "webm"
 
 
 class ImgFormat(str, Enum):
-    jpg = "jpg"
-    png = "png"
+    JPG = "jpg"
+    PNG = "png"

git_sim/settings.py

@@ -14,7 +14,7 @@ class Settings(BaseSettings):
     n = 5
     files: Union[List[pathlib.Path], None] = None
     hide_first_tag = False
-    img_format: ImgFormat = ImgFormat.jpg
+    img_format: ImgFormat = ImgFormat.JPG
     INFO_STRING = "Simulating: git"
     light_mode = False
     transparent_bg = False
@@ -30,7 +30,7 @@ class Settings(BaseSettings):
     show_outro = False
     speed = 1.5
     title = "Git-Sim, by initialcommit.com"
-    video_format: VideoFormat = VideoFormat.mp4
+    video_format: VideoFormat = VideoFormat.MP4
     stdout = False
     output_only_path = False
     quiet = False