✨ built-in files

Author: userElaina

src/badapple/__main__.py

@@ -7,20 +7,9 @@
 from .audio import help_audio
 from .players import get_names
 from .util import get_info
+from .builtin_files import BA_BA, BA_MP4,BA_MP3, BA_WAV, ba_get
 
 if __name__ == "__main__":
-    D_DIR = os.path.dirname(__file__)
-    D_MP3 = '_BA_MP3'
-    D_MP4 = '_BA_MP4'
-    D_WAV = '_BA_WAV'
-    D_BA = '_BA_BA'
-    D_FILES = {
-        D_MP3: os.path.join(D_DIR, 'badapple.mp3'),
-        D_MP4: os.path.join(D_DIR, 'badapple.mp4'),
-        D_WAV: os.path.join(D_DIR, 'badapple.wav'),
-        D_BA: os.path.join(D_DIR, 'badapple.badapple'),
-    }
-
     parser = argparse.ArgumentParser(
         'badapple',
         'badapple [options] ... ',
@@ -36,10 +25,10 @@
     parser.add_argument(
         '-i', '--input',
         help='video file (use %s or %s to load built-in video)' % (
-            D_MP4,
-            D_BA,
+            BA_MP4,
+            BA_BA,
         ),
-        default=D_MP4
+        default=BA_MP4
     )
     parser.add_argument(
         '-o', '--output',
@@ -54,9 +43,10 @@
     )
     parser.add_argument(
         '--audio',
-        help='audio file (use %s or %s to load built-in audio)' % (
-            D_WAV,
-            D_MP3,
+        help='audio file (use %s, %s or %s to load built-in audio)' % (
+            BA_WAV,
+            BA_MP3,
+            BA_MP4,
         ),
         default=''
     )
@@ -102,7 +92,7 @@
 
     a = parser.parse_args()
 
-    if a.avaliable_player:
+    if a.help_audio:
         help_audio()
         sys.exit(0)
 
@@ -113,8 +103,8 @@
 
     p_list: list[Process] = list()
 
-    video = D_FILES.get(a.input, a.input)
-    audio = D_FILES.get(a.audio, a.audio)
+    video = ba_get(a.input)
+    audio = ba_get(a.audio)
 
     try:
         play(

src/badapple/builtin_files.py

@@ -0,0 +1,18 @@
+import os
+
+BA_DIR = os.path.dirname(__file__)
+BA_MP3 = '_BA_MP3'
+BA_MP4 = '_BA_MP4'
+BA_WAV = '_BA_WAV'
+BA_BA = '_BA_BA'
+
+__FILES = {
+    BA_MP3: os.path.join(BA_DIR, 'badapple.mp3'),
+    BA_MP4: os.path.join(BA_DIR, 'badapple.mp4'),
+    BA_WAV: os.path.join(BA_DIR, 'badapple.wav'),
+    BA_BA: os.path.join(BA_DIR, 'badapple.badapple'),
+}
+
+
+def ba_get(x: str) -> str:
+    return __FILES.get(x, x)