Files
smnp-py/smnp/library/function/flat.py
2019-07-04 15:23:57 +02:00

21 lines
544 B
Python

from smnp.library.model import Function
from smnp.library.signature import varargSignature, allTypes
from smnp.type.model import Type
from smnp.type.value import Value
_signature = varargSignature(allTypes())
def _function(env, vararg):
return Value(Type.LIST, doFlat(vararg, [])).decompose()
def doFlat(input, output=[]):
for item in input:
if item.type == Type.LIST:
doFlat(item.value, output)
else:
output.append(item)
return output
function = Function(_signature, _function, 'flat')