def add_default_return(definition, processed_conditions, implementation): """ Modify source code to include a default return if no True key is present. :param definition: function def :param processed_conditions: obj containing dict with tables. :param implementation: source code :return: source code """ if processed_conditions.default: return get_returning_implementation(implementation, definition, processed_conditions.default) elif not h.has_true_key(processed_conditions.tables): return get_returning_implementation(implementation, definition, False) return implementation
def change_keys_from_bool_to_int(d, new_key): """ Changes the keys from booleans (True or False) to int(0 or 1) if a int(0 or 1) is present. :param d: dict :param new_key: a new key to be added to dict. :return: new dict """ if helpers.var_is_1(new_key) and helpers.has_true_key(d): d[1] = d.pop(True, None) if helpers.var_is_0(new_key) and helpers.has_false_key(d): d[0] = d.pop(False, None) return d