diff --git a/src/bilifm/audio.py b/src/bilifm/audio.py index 3d4e6a2..d304cec 100644 --- 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 index 198e1e2..beb9fcc 100644 --- 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 index 3ba1a23..f53f84d 100644 --- 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)] \ No newline at end of file + AudioQualityEnums, + typer.Option("--quality", "-q", help="audio quality", case_sensitive=False), +]