refactor: use internal stgpytools

Author: Tohrusky

requirements.txt

@@ -1,3 +1,2 @@
 VapourSynth>=68
-stgpytools>=1.2.0
 rich>=12.6.0

vstools/enums/base.py

@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-from stgpytools import CustomEnum, CustomIntEnum, CustomStrEnum, SelfEnum
+from vstools.stgpytools import CustomEnum, CustomIntEnum, CustomStrEnum, SelfEnum
 
 __all__ = [
     'SelfEnum',

vstools/enums/color.py

@@ -3,7 +3,7 @@
 from typing import Any, NamedTuple, TypeAlias, Union
 
 import vapoursynth as vs
-from stgpytools import FuncExceptT, classproperty
+from vstools.stgpytools import FuncExceptT, classproperty
 
 from ..exceptions import (
     ReservedMatrixError, ReservedPrimariesError, ReservedTransferError, UndefinedMatrixError, UndefinedPrimariesError,

vstools/enums/funcs.py

@@ -2,7 +2,7 @@
 
 from typing import Literal, TypeAlias, Union
 
-from stgpytools import CustomIntEnum, CustomStrEnum
+from vstools.stgpytools import CustomIntEnum, CustomStrEnum
 
 __all__ = [
     'ConvMode',

vstools/enums/generic.py

@@ -3,7 +3,7 @@
 from typing import TYPE_CHECKING, Any, TypeAlias, Union
 
 import vapoursynth as vs
-from stgpytools import FuncExceptT
+from vstools.stgpytools import FuncExceptT
 
 from ..exceptions import (
     UndefinedChromaLocationError, UndefinedFieldBasedError, UnsupportedChromaLocationError,

vstools/enums/other.py

@@ -5,7 +5,7 @@
 from typing import Callable, Iterable, Literal, NamedTuple, TypeVar, overload
 
 import vapoursynth as vs
-from stgpytools import Coordinate, CustomIntEnum, CustomStrEnum, FuncExceptT, Position, Sentinel, Size
+from vstools.stgpytools import Coordinate, CustomIntEnum, CustomStrEnum, FuncExceptT, Position, Sentinel, Size
 
 from ..types import HoldsPropValueT
 

vstools/enums/stubs.py

@@ -4,7 +4,7 @@
 from typing import TYPE_CHECKING, Any, Iterable, TypeVar, overload
 
 import vapoursynth as vs
-from stgpytools import MISSING, CustomError, CustomIntEnum, FuncExceptT, classproperty
+from vstools.stgpytools import MISSING, CustomError, CustomIntEnum, FuncExceptT, classproperty
 
 __all__ = [
     'PropEnum',

vstools/exceptions/base.py

@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-from stgpytools import (
+from vstools.stgpytools import (
     CustomError, CustomIndexError, CustomKeyError, CustomNotImplementedError, CustomOverflowError,
     CustomPermissionError, CustomRuntimeError, CustomTypeError, CustomValueError
 )

vstools/exceptions/color.py

@@ -3,7 +3,7 @@
 from typing import Any
 
 import vapoursynth as vs
-from stgpytools import CustomPermissionError, CustomValueError, FuncExceptT, SupportsString
+from vstools.stgpytools import CustomPermissionError, CustomValueError, FuncExceptT, SupportsString
 
 __all__ = [
     'InvalidColorspacePathError',

vstools/exceptions/enum.py

@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-from stgpytools import CustomValueError, NotFoundEnumValue
+from vstools.stgpytools import CustomValueError, NotFoundEnumValue
 
 __all__ = [
     'UndefinedChromaLocationError',

vstools/exceptions/file.py

@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-from stgpytools import (
+from vstools.stgpytools import (
     FileIsADirectoryError, FileNotExistsError, FilePermissionError, FileTypeMismatchError, FileWasNotFoundError
 )
 

vstools/exceptions/generic.py

@@ -4,7 +4,7 @@
 from typing import TYPE_CHECKING, Any, Iterable, Sized
 
 import vapoursynth as vs
-from stgpytools import (
+from vstools.stgpytools import (
     CustomKeyError, CustomOverflowError, CustomValueError, FuncExceptT, MismatchError, MismatchRefError, SupportsString
 )
 

vstools/exceptions/module.py

@@ -2,7 +2,7 @@
 
 from typing import Any
 
-from stgpytools import CustomImportError, DependencyNotFoundError, FuncExceptT, SupportsString
+from vstools.stgpytools import CustomImportError, DependencyNotFoundError, FuncExceptT, SupportsString
 
 __all__ = [
     'CustomImportError',

vstools/functions/check.py

@@ -5,7 +5,7 @@
 from typing import Any, Callable, TypeGuard, cast, overload
 
 import vapoursynth as vs
-from stgpytools import CustomError, F, FuncExceptT
+from vstools.stgpytools import CustomError, F, FuncExceptT
 
 from ..exceptions import (
     FormatsRefClipMismatchError, ResolutionsRefClipMismatchError, VariableFormatError, VariableResolutionError

vstools/functions/file.py

@@ -3,7 +3,7 @@
 import inspect
 from pathlib import Path
 
-from stgpytools import CustomRuntimeError, SPath, get_script_path
+from vstools.stgpytools import CustomRuntimeError, SPath, get_script_path
 
 __all__ = [
     'PackageStorage'

vstools/functions/funcs.py

@@ -3,7 +3,7 @@
 from typing import Iterable, Sequence
 
 import vapoursynth as vs
-from stgpytools import FuncExceptT, T, cachedproperty, fallback, iterate, kwargs_fallback, normalize_seq, to_arr
+from vstools.stgpytools import FuncExceptT, T, cachedproperty, fallback, iterate, kwargs_fallback, normalize_seq, to_arr
 
 from vstools.exceptions.color import InvalidColorspacePathError
 

vstools/functions/heuristics.py

@@ -3,7 +3,7 @@
 from typing import Any, Literal, overload
 
 import vapoursynth as vs
-from stgpytools import KwargsT
+from vstools.stgpytools import KwargsT
 from vstools import PropEnum
 
 from ..enums import ChromaLocation, ColorRange, Matrix, Primaries, Transfer

vstools/functions/normalize.py

@@ -3,8 +3,8 @@
 from typing import Any, Iterable, Sequence, overload
 
 import vapoursynth as vs
-from stgpytools import T, norm_display_name, norm_func_name, normalize_list_to_ranges, to_arr
-from stgpytools import (
+from vstools.stgpytools import T, norm_display_name, norm_func_name, normalize_list_to_ranges, to_arr
+from vstools.stgpytools import (
     flatten as stg_flatten,
     invert_ranges as stg_invert_ranges,
     normalize_range as normalize_franges,

vstools/functions/packets.py

@@ -9,7 +9,7 @@
 from typing import Self, TypedDict
 
 import vapoursynth as vs
-from stgpytools import CustomValueError, DependencyNotFoundError, FuncExceptT, SPath, SPathLike
+from vstools.stgpytools import CustomValueError, DependencyNotFoundError, FuncExceptT, SPath, SPathLike
 
 from .file import PackageStorage
 from .timecodes import Keyframes

vstools/functions/render.py

@@ -7,8 +7,8 @@
 from typing import BinaryIO, Callable, Literal, overload
 
 import vapoursynth as vs
-from stgpytools import CustomRuntimeError, CustomValueError, Sentinel, T, normalize_list_to_ranges
-from stgpytools.types.funcs import SentinelDispatcher
+from vstools.stgpytools import CustomRuntimeError, CustomValueError, Sentinel, T, normalize_list_to_ranges
+from vstools.stgpytools.types.funcs import SentinelDispatcher
 
 from ..exceptions import InvalidColorFamilyError
 from .progress import get_render_progress

vstools/functions/timecodes.py

@@ -7,7 +7,7 @@
 from typing import Any, ClassVar, Iterable, NamedTuple, TypeVar, overload
 
 import vapoursynth as vs
-from stgpytools import CustomValueError, FilePathType, FuncExceptT, LinearRangeLut, Sentinel, SPath, inject_self
+from vstools.stgpytools import CustomValueError, FilePathType, FuncExceptT, LinearRangeLut, Sentinel, SPath, inject_self
 
 from ..enums import Matrix, SceneChangeMode
 from ..exceptions import FramesLengthError, InvalidTimecodeVersionError

vstools/functions/utils.py

@@ -5,7 +5,7 @@
 from weakref import WeakValueDictionary
 
 import vapoursynth as vs
-from stgpytools import CustomIndexError, CustomStrEnum, CustomValueError
+from vstools.stgpytools import CustomIndexError, CustomStrEnum, CustomValueError
 
 from ..enums import ColorRange, ColorRangeT, Matrix
 from ..exceptions import ClipLengthError, InvalidColorFamilyError

vstools/stgpytools/__init__.py

@@ -0,0 +1,5 @@
+from .enums import *  # noqa: F401, F403
+from .exceptions import *  # noqa: F401, F403
+from .functions import *  # noqa: F401, F403
+from .types import *  # noqa: F401, F403
+from .utils import *  # noqa: F401, F403

vstools/stgpytools/_metadata.py

@@ -0,0 +1,12 @@
+"""Collection of stuff that's useful in general python programming"""
+
+__version__ = '1.2.2'
+
+__author__ = 'Setsugen no ao <[email protected]>'
+__maintainer__ = __author__
+
+__author_name__, __author_email__ = [x[:-1] for x in __author__.split('<')]
+__maintainer_name__, __maintainer_email__ = [x[:-1] for x in __maintainer__.split('<')]
+
+if __name__ == '__github__':
+    print(__version__)

vstools/stgpytools/enums/__init__.py

@@ -0,0 +1,2 @@
+from .base import *  # noqa: F401, F403
+from .other import *  # noqa: F401, F403

vstools/stgpytools/enums/base.py

@@ -0,0 +1,78 @@
+from __future__ import annotations
+
+from enum import Enum
+from typing import Any, TypeVar
+
+from ..exceptions import CustomValueError, NotFoundEnumValue
+from ..types import FuncExceptT
+
+__all__ = [
+    'SelfEnum',
+    'CustomEnum', 'CustomIntEnum', 'CustomStrEnum'
+]
+
+
+class CustomEnum(Enum):
+    """Base class for custom enums."""
+
+    @classmethod
+    def _missing_(cls: type[SelfEnum], value: Any) -> SelfEnum | None:
+        return cls.from_param(value)
+
+    @classmethod
+    def from_param(cls: type[SelfEnum], value: Any, func_except: FuncExceptT | None = None) -> SelfEnum | None:
+        """
+        Return the enum value from a parameter.
+
+        :param value:               Value to instantiate the enum class.
+        :param func_except:         Exception function.
+
+        :return:                    Enum value.
+
+        :raises NotFoundEnumValue:   Variable not found in the given enum.
+        """
+
+        if value is None:
+            return None
+
+        if func_except is None:
+            func_except = cls.from_param
+
+        if isinstance(value, cls):
+            return value
+
+        if value is cls:
+            raise CustomValueError('You must select a member, not pass the enum!', func_except)
+
+        try:
+            return cls(value)
+        except Exception:
+            pass
+
+        if isinstance(func_except, tuple):
+            func_name, var_name = func_except
+        else:
+            func_name, var_name = func_except, cls
+
+        raise NotFoundEnumValue(
+            'The given value for "{var_name}" argument must be a valid {enum_name}, not "{value}"!\n'
+            'Valid values are: [{readable_enum}].', func_name,
+            var_name=var_name, enum_name=cls, value=value,
+            readable_enum=iter([f'{x.name} ({x.value})' for x in cls]),
+            reason=value
+        )
+
+
+class CustomIntEnum(int, CustomEnum):
+    """Base class for custom int enums."""
+
+    value: int
+
+
+class CustomStrEnum(str, CustomEnum):
+    """Base class for custom str enums."""
+
+    value: str
+
+
+SelfEnum = TypeVar('SelfEnum', bound=CustomEnum)

vstools/stgpytools/enums/other.py

@@ -0,0 +1,59 @@
+from __future__ import annotations
+
+from typing import TypeVar, overload
+
+__all__ = [
+    'Coordinate',
+    'Position',
+    'Size'
+]
+
+
+class Coordinate:
+    """
+    Positive set of (x, y) coordinates.
+
+    :raises ValueError:     Negative values were passed.
+    """
+
+    x: int
+    """Horizontal coordinate."""
+
+    y: int
+    """Vertical coordinate."""
+
+    @overload
+    def __init__(self: SelfCoord, other: tuple[int, int] | SelfCoord, /) -> None:
+        ...
+
+    @overload
+    def __init__(self: SelfCoord, x: int, y: int, /) -> None:
+        ...
+
+    def __init__(self: SelfCoord, x_or_self: int | tuple[int, int] | SelfCoord, y: int | None = None, /) -> None:
+        from ..exceptions import CustomValueError
+
+        if isinstance(x_or_self, int):
+            x = x_or_self
+        else:
+            x, y = x_or_self if isinstance(x_or_self, tuple) else (x_or_self.x, x_or_self.y)
+
+        if y is None:
+            raise CustomValueError("y coordinate must be defined!", self.__class__)
+
+        if x < 0 or y < 0:
+            raise CustomValueError("Values can't be negative!", self.__class__)
+
+        self.x = x
+        self.y = y
+
+
+SelfCoord = TypeVar('SelfCoord', bound=Coordinate)
+
+
+class Position(Coordinate):
+    """Positive set of an (x,y) offset relative to the top left corner of an area."""
+
+
+class Size(Coordinate):
+    """Positive set of an (x,y), (horizontal,vertical), size of an area."""

vstools/stgpytools/exceptions/__init__.py

@@ -0,0 +1,5 @@
+from .base import *  # noqa: F401, F403
+from .enum import *  # noqa: F401, F403
+from .file import *  # noqa: F401, F403
+from .generic import *  # noqa: F401, F403
+from .module import *  # noqa: F401, F403