mirror of
https://github.com/ChronosX88/PyNesca.git
synced 2024-10-18 14:41:00 +00:00
33 lines
1.3 KiB
Python
33 lines
1.3 KiB
Python
|
def complitable_functions(output_function, input_function, convert_table):
|
||
|
|
input_keys = set(value for key, value in
|
||
|
|
input_function.__annotations__.items() if key != 'return')
|
||
|
|
return_keys = set(output_function.__annotations__["return"])
|
||
|
|
all_possible_return_keys = return_keys.union(
|
||
|
|
convert_table.all_possible_conversions(return_keys)
|
||
|
|
)
|
||
|
|
print("return:%s\npossible_output:%s" % (return_keys,
|
||
|
|
all_possible_return_keys))
|
||
|
|
return input_keys.issubset(all_possible_return_keys)
|
||
|
|
|
||
|
|
def get_converted_arguments(function, simple_arg_dict, convert_table):
|
||
|
|
if simple_arg_dict == None:
|
||
|
|
return [None for key in function.__annotations__.keys() if key != 'return']
|
||
|
|
result = []
|
||
|
|
for key, value in function.__annotations__.items():
|
||
|
|
if key != 'return':
|
||
|
|
converted_arg = None
|
||
|
|
try:
|
||
|
|
converted_arg = simple_arg_dict[value]
|
||
|
|
except KeyError:
|
||
|
|
key_to_convert, convert_function = convert_table.get_converter(
|
||
|
|
simple_arg_dict.keys(),
|
||
|
|
value
|
||
|
|
)
|
||
|
|
key_to_convert = list(key_to_convert)[0]
|
||
|
|
converted_arg = convert_function(
|
||
|
|
simple_arg_dict[key_to_convert]
|
||
|
|
)[value]
|
||
|
|
result.append(converted_arg)
|
||
|
|
return result
|
||
|
|
|