yta-video-frame-time 0.0.12__tar.gz → 0.0.16__tar.gz

{yta_video_frame_time-0.0.12 → yta_video_frame_time-0.0.16}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: yta-video-frame-time
-Version: 0.0.12
+Version: 0.0.16
 Summary: Youtube Autonomous Video Frame Time Module
 License-File: LICENSE
 Author: danialcala94

{yta_video_frame_time-0.0.12 → yta_video_frame_time-0.0.16}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "yta-video-frame-time"
-version = "0.0.12"
+version = "0.0.16"
 description = "Youtube Autonomous Video Frame Time Module"
 authors = [
     {name = "danialcala94",email = "danielalcalavalera@gmail.com"}
@@ -17,6 +17,7 @@ packages = [{include = "yta_video_frame_time", from = "src"}]
 
 [tool.poetry.group.dev.dependencies]
 pytest = "^8.3.5"
+yta_testing = ">=0.0.1"
 
 [build-system]
 requires = ["poetry-core>=2.0.0,<3.0.0"]

yta_video_frame_time-0.0.16/src/yta_video_frame_time/decorators.py

@@ -0,0 +1,99 @@
+"""
+TODO: I think this module should not be here, or maybe
+yes, but I need this decorator.
+"""
+from functools import wraps
+
+
+def parameter_to_time_interval(
+    param_name: str
+):
+    """
+    Force the parameter with the given `param_name` to
+    be a `TimeInterval` instance.
+
+    Values accepted:
+    - `TimeInterval` instance
+    - `tuple[float, float]` that will be `(start, end)`
+    """
+    def decorator(
+        func
+    ):
+        @wraps(func)
+        def wrapper(
+            *args,
+            **kwargs
+        ):
+            from inspect import signature
+            from yta_validation import PythonValidator
+            from yta_video_frame_time.interval import TimeInterval
+
+            sig = signature(func)
+            bound = sig.bind(*args, **kwargs)
+            bound.apply_defaults()
+
+            value = bound.arguments[param_name]
+
+            if PythonValidator.is_instance_of(value, TimeInterval):
+                pass
+            elif (
+                PythonValidator.is_tuple(value) and
+                len(value) == 2
+            ):
+                value = TimeInterval(*value)
+                bound.arguments[param_name] = value
+            else:
+                raise Exception(f'The "{param_name}" parameter must be a TimeInterval or a tuple[float, float].')
+
+            return func(*bound.args, **bound.kwargs)
+        
+        return wrapper
+    
+    return decorator
+
+# TODO: Remove when we have everything unified
+def parameter_to_new_time_interval(
+    param_name: str
+):
+    """
+    Force the parameter with the given `param_name` to
+    be a `NewTimeInterval` instance.
+
+    Values accepted:
+    - `NewTimeInterval` instance
+    - `tuple[float, float]` that will be `(start, end)`
+    """
+    def decorator(
+        func
+    ):
+        @wraps(func)
+        def wrapper(
+            *args,
+            **kwargs
+        ):
+            from inspect import signature
+            from yta_validation import PythonValidator
+            from yta_video_frame_time.interval import NewTimeInterval
+
+            sig = signature(func)
+            bound = sig.bind(*args, **kwargs)
+            bound.apply_defaults()
+
+            value = bound.arguments[param_name]
+
+            if PythonValidator.is_instance_of(value, NewTimeInterval):
+                pass
+            elif (
+                PythonValidator.is_tuple(value) and
+                len(value) == 2
+            ):
+                value = NewTimeInterval(*value)
+                bound.arguments[param_name] = value
+            else:
+                raise Exception(f'The "{param_name}" parameter must be a NewTimeInterval or a tuple[float, float].')
+
+            return func(*bound.args, **bound.kwargs)
+        
+        return wrapper
+    
+    return decorator