Added options for getting spleeter version

This commit is contained in:
romi1502
2021-01-29 17:12:59 +01:00
parent 9981dad5c3
commit f388df3129
2 changed files with 16 additions and 2 deletions

View File

@@ -11,7 +11,6 @@
command function scope to avoid heavy import on CLI evaluation, command function scope to avoid heavy import on CLI evaluation,
leading to large bootstraping time. leading to large bootstraping time.
""" """
import json import json
from functools import partial from functools import partial
from glob import glob from glob import glob
@@ -34,6 +33,12 @@ spleeter: Typer = Typer(add_completion=False)
""" CLI application. """ """ CLI application. """
@spleeter.callback()
def default(
version: bool = VersionOption,
) -> None:
pass
@spleeter.command() @spleeter.command()
def train( def train(
adapter: str = AudioAdapterOption, adapter: str = AudioAdapterOption,

View File

@@ -6,7 +6,7 @@
from os.path import join from os.path import join
from tempfile import gettempdir from tempfile import gettempdir
from typer import Argument, Option from typer import Argument, Option, echo, Exit
from typer.models import ArgumentInfo, OptionInfo from typer.models import ArgumentInfo, OptionInfo
from .audio import Codec, STFTBackend from .audio import Codec, STFTBackend
@@ -126,3 +126,12 @@ TrainingDataDirectoryOption: OptionInfo = Option(
) )
VerboseOption: OptionInfo = Option(False, "--verbose", help="Enable verbose logs") VerboseOption: OptionInfo = Option(False, "--verbose", help="Enable verbose logs")
def version_callback(value: bool):
if value:
from importlib.metadata import version
echo(f"Spleeter Version: {version('spleeter')}")
raise Exit()
VersionOption: OptionInfo = Option(None, "--version", callback=version_callback, is_eager=True, help="Return Spleeter version")