def generate_posts():
"""
Generate post pages
"""
for file in _post_file_names:
y, m, d, name = file.split('-', 3)
post_dir = os.path.join(_deploy_dir, 'post', y, m, d, name)
os.makedirs(post_dir)
with open(os.path.join(post_dir, 'index.html'), 'w', encoding='utf-8') as f:
f.write(handler.post(y, m, d, name))
def generate_posts():
"""
Generate post pages
"""
for file in _post_file_names:
y, m, d, name = file.split('-', 3)
post_dir = os.path.join(_deploy_dir, 'post', y, m, d, name)
os.makedirs(post_dir)
with open(os.path.join(post_dir, 'index.html'), 'w', encoding='utf-8') as f:
f.write(handler.post(y, m, d, name))
def process(args):
session, msg = args
log("get message: " + msg)
_b = time.time()
params = msg.split("::")
try:
# <0.652.0>:[email protected]:<0.486.0>:1573:429363:10167::{"res_code":201}
msg = json.loads(params[1])
_uri = msg.get("uri")
_headers = msg.get("headers")
_method = msg.get("method")
_args = msg.get("args")
_data = msg.get("data")
resp = ""
if _method == 'get':
resp = handler.get(session, _uri, _headers, _args, _data)
elif _method == 'post':
resp = handler.post(session, _uri, _headers, _args, _data)
elif _method == 'put':
resp = handler.put(_uri, _headers, _args, _data)
elif _method == 'patch':
resp = handler.patch(_uri, _headers, _args, _data)
elif _method == 'delete':
resp = handler.delete(_uri, _headers, _args, _data)
send("{}::{}".format(params[0], resp))
log("message send: {} ".format(time.time() - _b))
except Exception as e:
send("{}::{}".format(params[0], json.dumps({"x_res_code": 500,
"x_res_body": str(e)})))
def post(year, month, day, post_name):
return handler.post(year, month, day, post_name)
def test_post_returns_http_201(self, mock_boto3, mock_os):
response = post({'body': '{"url":"https://www.test.com"}'}, None)
self.assertTrue(response['statusCode'], 201)
self.assertIsNotNone(response['headers']['Location'])
def test_post_returns_http_400_with_no_url_supplied(self):
response = post({'body': "{}"}, None)
self.assertTrue(response['statusCode'], 400)
def test_post_returns_http_400_with_invalid_url(self):
response = post({'body': '{"url":"foobar"}'}, None)
self.assertTrue(response['statusCode'], 400)
def post(year, month, day, post_name):
return handler.post(year, month, day, post_name)
