def add_new_api_test(folder_root_path, name, route_path, route_name, filename, api_version):
print 'Generating api functional test script ... ',
try:
templateLoader = jinja2.FileSystemLoader(searchpath="/")
templateEnv = jinja2.Environment(loader=templateLoader)
TEMPLATE_FOLDER = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'template')
API_TEST_TEMPLATE_FILE = os.path.join(TEMPLATE_FOLDER, 'api_test_template.jinja2')
template = templateEnv.get_template(API_TEST_TEMPLATE_FILE)
test_file_path = os.path.join(os.path.join(folder_root_path, 'test'), 'api')
api_test_file_path = os.path.join(test_file_path, filename)
variables = {'api_classname': name[0].upper() + name[1:].lower() + 'API',
'name': name,
'path': route_path,
'api_version': api_version,
}
outputText = template.render(variables)
create_folders(os.path.join(folder_root_path, 'test'), ['api', 'model'])
with open(api_test_file_path, 'wb') as f:
f.write(outputText)
except Exception, e:
print 'fail'
exp = sys.exc_info()[0]
print exp
print e
return False
def add_new_async_api(folder_root_path, name, route_name, filename):
"""
1. add [celery] section in developement.ini
2. add 'pyramid_celery' library in setup.py
"""
print 'Generating api(async) ... ',
#1. check if [celery] section exist in development.ini and insert the [celery] section
dev_ini_path = os.path.join(os.path.dirname(folder_root_path), 'development.ini')
check_section = '[celery]'
section_line = ''
with open(dev_ini_path, 'r') as devf:
for line in devf.readlines():
if check_section in line:
section_line = line
if section_line == '':
section_title_to_add = '[celery]\n'
section_attributes_to_add = [
'BROKER_URL = redis://localhost/1\n',
'#CELERY_RESULT_BACKEND = redis://localhost/2'
]
insert_text = []
insert_text.append("\n")
insert_text.extend(section_title_to_add)
insert_text.extend(section_attributes_to_add)
AppendBottom(dev_ini_path, insert_text)
#2. check if 'pyramid_celery' exist in setup.py and insert it
setup_py_path = os.path.join(os.path.dirname(folder_root_path), 'setup.py')
check_lib = 'pyramid_celery'
lib_line = ''
with open(setup_py_path, 'r') as devf:
for line in devf.readlines():
if check_lib in line:
lib_line = line
if lib_line == '':
target_line = 'celery[redis]'
lib_to_add = "'%s',\n" % check_lib
InsertAbove(setup_py_path, [lib_to_add], target_line)
try:
templateLoader = jinja2.FileSystemLoader(searchpath="/")
templateEnv = jinja2.Environment(loader=templateLoader)
TEMPLATE_FOLDER = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'template')
API_TEMPLATE_FILE = os.path.join(TEMPLATE_FOLDER, 'async_api_template.jinja2')
template = templateEnv.get_template(API_TEMPLATE_FILE)
create_folders(folder_root_path, ['async_apis'])
init_py_path = os.path.join(os.path.join(folder_root_path,'async_apis'),'__init__.py')
if not os.path.isfile(init_py_path):
open(init_py_path, 'a').close()
api_file_path = os.path.join(os.path.join(folder_root_path, 'async_apis'), filename)
variables = {'api_classname': 'Async' + name[0].upper() + name[1:].lower() + 'API',
'name': name,
'service_name': name,
'route_name': route_name,
}
outputText = template.render(variables)
with open(api_file_path, 'wb') as f:
f.write(outputText)
except Exception, e:
print 'fail'
exp = sys.exc_info()[0]
print exp
print e
return False
