formatted by black && isort

jingfelix · May 19, 2024 · 32382da · 32382da
1 parent 1abc3b0
commit 32382da
Show file tree

Hide file tree

Showing 3 changed files with 29 additions and 10 deletions.
diff --git a/src/bilifm/audio.py b/src/bilifm/audio.py
@@ -5,7 +5,7 @@
 import requests
 import typer
 
-from .util import get_signed_params, AudioQualityEnums, audio_quality_map
+from .util import AudioQualityEnums, audio_quality_map, get_signed_params
 
 
 class Audio:
@@ -79,12 +79,16 @@ def download(self):
                 ).json()
 
                 if json["data"] is None:
-                    typer.echo(f" `data` field is not valid with url: {self.playUrl} and params : {params}")
+                    typer.echo(
+                        f" `data` field is not valid with url: {self.playUrl} and params : {params}"
+                    )
                     return
 
                 audio = json["data"]["dash"]["audio"]
                 if not audio:
-                    typer.echo(f" `audio` field is empty with url: {self.playUrl} and params : {params}")
+                    typer.echo(
+                        f" `audio` field is empty with url: {self.playUrl} and params : {params}"
+                    )
                     return
 
                 base_url = None

diff --git a/src/bilifm/command.py b/src/bilifm/command.py
@@ -6,19 +6,27 @@
 from .fav import Fav
 from .season import Season
 from .user import User
-from .util import Directory, Path, AudioQuality, AudioQualityEnums
+from .util import AudioQuality, AudioQualityEnums, Directory, Path
 
 app = typer.Typer()
 
 
 @app.command()
-def bv(bv: str, directory: Directory = None, audio_quality: AudioQuality = AudioQualityEnums.k64):
+def bv(
+    bv: str,
+    directory: Directory = None,
+    audio_quality: AudioQuality = AudioQualityEnums.k64,
+):
     audio = Audio(bv, audio_quality)
     audio.download()
 
 
 @app.command()
-def uid(uid: str, directory: Directory = None, audio_quality: AudioQuality = AudioQualityEnums.k64):
+def uid(
+    uid: str,
+    directory: Directory = None,
+    audio_quality: AudioQuality = AudioQualityEnums.k64,
+):
     user = User(uid)
 
     for video in user.videos:
@@ -34,7 +42,7 @@ def fav(
     media_id: str,
     cookies_path: str = Path,
     directory: Directory = None,
-    audio_quality: AudioQuality = AudioQualityEnums.k64
+    audio_quality: AudioQuality = AudioQualityEnums.k64,
 ):
     with open(cookies_path, "r") as f:
         cookies = f.read()
@@ -49,7 +57,12 @@ def fav(
 
 
 @app.command()
-def season(uid: str, sid: str, directory: Directory = None, audio_quality: AudioQuality = AudioQualityEnums.k64):
+def season(
+    uid: str,
+    sid: str,
+    directory: Directory = None,
+    audio_quality: AudioQuality = AudioQualityEnums.k64,
+):
     sea = Season(uid, sid)
     ret = sea.get_videos()
     if not ret:

diff --git a/src/bilifm/util.py b/src/bilifm/util.py
@@ -96,6 +96,7 @@
     52,
 ]
 
+
 class AudioQualityEnums(str, Enum):
     k64 = "64"
     k132 = "132"
@@ -206,5 +207,6 @@ def check_path(path: str):
 Directory = Annotated[str, typer.Option("-o", "--directory", callback=change_directory)]
 Path = typer.Argument(callback=check_path)
 AudioQuality = Annotated[
-    AudioQualityEnums, typer.Option("--quality", "-q", help="audio quality",
-                                    case_sensitive=False)]
+    AudioQualityEnums,
+    typer.Option("--quality", "-q", help="audio quality", case_sensitive=False),
+]