66fd7cf41c
Add the ability to print custom messages in override scripts. Adds the functions - `print` - `print_if_true` - `print_if_false` Will be used to make under-the-hood preset things more verbose.
84 lines
2.9 KiB
Python
84 lines
2.9 KiB
Python
from typing import Callable
|
|
from typing import Dict
|
|
|
|
from ytdl_sub.script.functions.array_functions import ArrayFunctions
|
|
from ytdl_sub.script.functions.boolean_functions import BooleanFunctions
|
|
from ytdl_sub.script.functions.conditional_functions import ConditionalFunctions
|
|
from ytdl_sub.script.functions.date_functions import DateFunctions
|
|
from ytdl_sub.script.functions.error_functions import ErrorFunctions
|
|
from ytdl_sub.script.functions.json_functions import JsonFunctions
|
|
from ytdl_sub.script.functions.map_functions import MapFunctions
|
|
from ytdl_sub.script.functions.numeric_functions import NumericFunctions
|
|
from ytdl_sub.script.functions.print_functions import PrintFunctions
|
|
from ytdl_sub.script.functions.regex_functions import RegexFunctions
|
|
from ytdl_sub.script.functions.string_functions import StringFunctions
|
|
from ytdl_sub.script.types.resolvable import Resolvable
|
|
from ytdl_sub.script.utils.exceptions import FunctionDoesNotExistRuntimeException
|
|
|
|
|
|
class Functions(
|
|
StringFunctions,
|
|
NumericFunctions,
|
|
ConditionalFunctions,
|
|
ArrayFunctions,
|
|
MapFunctions,
|
|
BooleanFunctions,
|
|
ErrorFunctions,
|
|
RegexFunctions,
|
|
DateFunctions,
|
|
JsonFunctions,
|
|
PrintFunctions,
|
|
):
|
|
_custom_functions: Dict[str, Callable[..., Resolvable]] = {}
|
|
|
|
@classmethod
|
|
def is_built_in(cls, name: str) -> bool:
|
|
"""
|
|
Returns
|
|
-------
|
|
True if the name exists as a built-in function or custom function. False otherwise.
|
|
"""
|
|
return hasattr(cls, name) or hasattr(cls, f"{name}_") or name in cls._custom_functions
|
|
|
|
@classmethod
|
|
def get(cls, name: str) -> Callable[..., Resolvable]:
|
|
"""
|
|
Returns
|
|
-------
|
|
The actual Python callable for the function of the given name.
|
|
|
|
Raises
|
|
------
|
|
FunctionDoesNotExistRuntimeException
|
|
If the function does not exist.
|
|
"""
|
|
if hasattr(cls, name):
|
|
return getattr(cls, name)
|
|
if hasattr(cls, f"{name}_"):
|
|
return getattr(cls, f"{name}_")
|
|
if name in cls._custom_functions:
|
|
return cls._custom_functions[name]
|
|
|
|
raise FunctionDoesNotExistRuntimeException(f"The function {name} does not exist")
|
|
|
|
@classmethod
|
|
def register_function(cls, function: Callable[..., Resolvable]) -> None:
|
|
"""
|
|
Adds a function to the suite of offered functions.
|
|
|
|
Parameters
|
|
----------
|
|
function
|
|
A static function whose name will be used as the offered function name.
|
|
|
|
Raises
|
|
------
|
|
ValueError
|
|
If the name already exists as a function.
|
|
"""
|
|
if cls.is_built_in(function.__name__):
|
|
raise ValueError(
|
|
f"Cannot register a function with name {function.__name__} "
|
|
f"because it already exists"
|
|
)
|
|
cls._custom_functions[function.__name__] = function
|