bartek/smnp-py
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Enable varargs in signatures

Browse Source
This commit is contained in:
Bartłomiej Pluta
2019-07-10 00:32:42 +02:00
parent 99647df061
commit 39eaa2b4d5
5 changed files with 63 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
smnp/function/signature.py
View File

@@ -1,5 +1,6 @@
# from smnp.type.model import Type
#
from smnp.type.model import Type
class Signature:
@@ -9,7 +10,7 @@ class Signature:
self.matchers = matchers
def varargSignature(varargMatcher, *basicSignature):
def varargSignature(varargMatcher, *basicSignature, wrapVarargInValue=False):
def check(args):
if len(basicSignature) > len(args):
return doesNotMatchVararg(basicSignature)
@@ -22,7 +23,10 @@ def varargSignature(varargMatcher, *basicSignature):
if not varargMatcher.match(args[i]):
return doesNotMatchVararg(basicSignature)
return True, (*args[:len(basicSignature)]), args[len(basicSignature):]
if wrapVarargInValue:
return True, (*args[:len(basicSignature)]), Type.list(args[len(basicSignature):])
else:
return True, (*args[:len(basicSignature)]), args[len(basicSignature):]
string = f"({', '.join([str(m) for m in basicSignature])}{', ' if len(basicSignature) > 0 else ''}{str(varargMatcher)}...)"
@@ -32,8 +36,6 @@ def varargSignature(varargMatcher, *basicSignature):
def doesNotMatchVararg(basicSignature):
return (False, *[None for n in basicSignature], None)
def signature(*signature):
def check(args):
if len(signature) != len(args):