Yasuke/moto
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

insert functions and templates for query and rest-json protocol

Browse Source
This commit is contained in:
Toshiya Kawasaki 2017-09-22 00:01:01 +09:00
parent 6c33888b0f
commit d0154b8e71
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
scaffold.py
View File

@ -349,6 +349,15 @@ def insert_json_codes(service, operation):
print_progress('inserting code', models_path, 'green')
insert_code_to_class(models_path, BaseBackend, func_in_models)
def insert_restjson_codes(service, operation):
func_in_models = get_function_in_models(service, operation)
print_progress('skipping inserting code to responses.py', "dont't know how to implement", 'yellow')
# edit models.py
models_path = 'moto/{}/models.py'.format(service)
print_progress('inserting code', models_path, 'green')
insert_code_to_class(models_path, BaseBackend, func_in_models)
@click.command()
def main():
service, operation = select_service_and_operation()
@ -358,7 +367,8 @@ def main():
insert_query_codes(service, operation)
elif api_protocol == 'json':
insert_json_codes(service, operation)
pass
elif api_protocol == 'rest-json':
insert_restjson_codes(service, operation)
else:
print_progress('skip inserting code', 'api protocol "{}" is not supported'.format(api_protocol), 'yellow')