danvics/ytdl-sub
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

simplify generics

Browse source
This commit is contained in:
Jesse Bannon 2023-11-15 23:46:24 -08:00
parent d5ba12b265
commit e0b75b1514
3 changed files with 24 additions and 19 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
src/ytdl_sub/script/functions/special_functions.py
View file

@ -1,13 +1,15 @@
from typing import Union
from ytdl_sub.script.types.resolvable import AnyType_1
from ytdl_sub.script.types.resolvable import AnyType_2
from ytdl_sub.script.types.resolvable import AnyTypeReturnableA
from ytdl_sub.script.types.resolvable import AnyTypeReturnableB
from ytdl_sub.script.types.resolvable import Boolean
class SpecialFunctions:
@staticmethod
def if_(condition: Boolean, true: AnyType_1, false: AnyType_2) -> Union[AnyType_1, AnyType_2]:
def if_(
condition: Boolean, true: AnyTypeReturnableA, false: AnyTypeReturnableB
) -> Union[AnyTypeReturnableA, AnyTypeReturnableB]:
if condition.value:
return true
return false

15
src/ytdl_sub/script/types/function.py
View file

@ -14,9 +14,9 @@ from typing import Union
from typing import get_origin
from ytdl_sub.script.functions import Functions
from ytdl_sub.script.types.resolvable import AnyType_0
from ytdl_sub.script.types.resolvable import AnyType_1
from ytdl_sub.script.types.resolvable import AnyType_2
from ytdl_sub.script.types.resolvable import AnyTypeReturnable
from ytdl_sub.script.types.resolvable import AnyTypeReturnableA
from ytdl_sub.script.types.resolvable import AnyTypeReturnableB
from ytdl_sub.script.types.resolvable import ArgumentType
from ytdl_sub.script.types.resolvable import NamedType
from ytdl_sub.script.types.resolvable import Resolvable
@ -278,12 +278,9 @@ class BuiltInFunction(Function):
if is_union(output_type):
union_types_list = []
for union_type in output_type.__args__:
if union_type == AnyType_0:
union_types_list.append(self._arg_output_type(self.args[0]))
elif union_type == AnyType_1:
union_types_list.append(self._arg_output_type(self.args[1]))
elif union_type == AnyType_2:
union_types_list.append(self._arg_output_type(self.args[2]))
if union_type in (AnyTypeReturnable, AnyTypeReturnableA, AnyTypeReturnableB):
generic_arg_index = self.input_spec.args.index(union_type)
union_types_list.append(self._arg_output_type(self.args[generic_arg_index]))
else:
union_types_list.append(union_type)

20
src/ytdl_sub/script/types/resolvable.py
View file

@ -22,19 +22,25 @@ class ArgumentType(NamedType, ABC):
"""
class AnyType_0(NamedType, ABC):
pass
class AnyTypeReturnable(NamedType, ABC):
"""
AnyType to express generics in functions that are part of the return type
"""
class AnyType_1(NamedType, ABC):
pass
class AnyTypeReturnableA(NamedType, ABC):
"""
AnyType to express generics in functions when more than one are present (i.e. `if`)
"""
class AnyType_2(NamedType, ABC):
pass
class AnyTypeReturnableB(NamedType, ABC):
"""
AnyType to express generics in functions when more than one are present (i.e. `if`)
"""
class AnyType(ArgumentType, AnyType_0, AnyType_1, AnyType_2, ABC):
class AnyType(ArgumentType, AnyTypeReturnable, AnyTypeReturnableA, AnyTypeReturnableB, ABC):
"""
Human-readable name for FutureResolvable
"""