def _session_user_mapping(session):
session_key = session['value']
session_list = [session_key]
session_key2 = '' if not session.has_key('value2') else session['value2']
if session_key2:
session_list.append(session_key2)
cfg.logger.warning('session_list: %s', session_list)
(error_code, db_results) = util.db_find2('session_user_map', {"session_key": {"$in": session_list}})
cfg.logger.warning('error_code: %s db_results: %s', error_code, db_results)
if error_code != S_OK:
return (error_code, {})
if not db_results:
return (S_ERR, {})
user_id = db_results[0].get('user_id', '')
user_info = util.db_find_one('user_info', {"user_id": user_id})
if not user_info:
return (S_ERR, {})
#if _is_to_refresh_google_token(user_info):
# _refresh_google_token(user_info)
_check_refresh_session(session, session_key, session_key2, user_info)
return (S_OK, user_info)
def _redirect_register(state):
login_info = util.db_find_one('login_info', {"state": state})
util.db_remove('login_info', {"state": state})
path_qs = login_info.get('url', '')
redirect_url = cfg.config.get('sitename', 'localhost') + path_qs
cfg.logger.warning('to redirect: redirect_url: %s', redirect_url)
redirect(redirect_url)
def _redirect_register(state):
login_info = util.db_find_one('login_info', {"state": state})
util.db_remove('login_info', {"state": state})
path_qs = login_info.get('url', '')
redirect_url = cfg.config.get('sitename', 'localhost') + path_qs
cfg.logger.warning('to redirect: redirect_url: %s', redirect_url)
redirect(redirect_url)
def g_thumbnail_handler(img_id):
db_result = util.db_find_one('bee_img', {'the_id': img_id}, {'_id': False, 'thumbnail_filename': True, 'content_type': True})
if not db_result:
return ('image/gif', util.empty_img())
the_filename = db_result.get('thumbnail_filename', '')
full_filename = '/data/thumbnail/bee/' + the_filename
cfg.logger.debug('full_filename: %s', full_filename)
with open(full_filename, 'r') as f:
content = f.read()
content_type = db_result.get('content_type', 'image/jpg')
return (content_type, content)
def post_ad_version_handler(params):
csv_key = params.get('csv_key', '')
if not csv_key:
return {"success": False, "error_msg": "no csv_key"}
ad_versions = params.get('ad_versions', [])
cfg.logger.debug('to update ad_versions: csv_key: %s ad_versions: %s', csv_key, ad_versions)
util.db_update('bee_csv', {'csv_key': csv_key}, {'ad_versions': ad_versions, "is_processed_ad_version": True})
db_result = util.db_find_one('bee_csv', {"csv_key": csv_key}, {"_id": False, "town": True, "count": True, "deliver_time": True, "deliver_date": True, "save_time": True, "geo": True, "county": True, "address": True, "user_name": True, "is_processed_ad_version": True, "is_processed_address": True, "csv_key": True, "ad_versions": True, "version_text": True, "memo": True, "deliver_status": True})
the_id = str(db_result.get('save_time', 0)) + "_" + util.uuid()
db_result['the_id'] = the_id
if db_result.get('is_processed_ad_version', False) and db_result.get('is_processed_address', False):
util.db_update('bee', {'csv_key': csv_key}, db_result)
def p_json_handler(data):
error_code = S_OK
error_msg = ''
save_timestamp = util.get_timestamp()
for each_data in data:
_infer_columns(each_data, save_timestamp)
the_id = each_data['the_id']
db_result = util.db_find_one('roadDB', {'the_id': the_id})
if db_result:
if not _is_same(db_result, each_data):
error_code = S_ERR
cfg.logger.error('data different: the_id: %s db_result: %s each_data: %s', the_id, db_result, each_data)
error_msg += 'data different: the_id: %s db_result: %s each_data: %s\n' % (the_id, db_result, each_data)
continue
util.db_insert_if_not_exist('roadDB', {'the_id': the_id}, each_data)
return {"success": True if error_code == S_OK else False, "error_msg": error_msg}
def g_thumbnail_handler(img_id):
db_result = util.db_find_one('bee_img', {'the_id': img_id}, {
'_id': False,
'thumbnail_filename': True,
'content_type': True
})
if not db_result:
return ('image/gif', util.empty_img())
the_filename = db_result.get('thumbnail_filename', '')
full_filename = '/data/thumbnail/bee/' + the_filename
cfg.logger.debug('full_filename: %s', full_filename)
with open(full_filename, 'r') as f:
content = f.read()
content_type = db_result.get('content_type', 'image/jpg')
return (content_type, content)
def _session_user_mapping(session):
session_key = session['value']
session_list = [session_key]
session_key2 = '' if not session.has_key('value2') else session['value2']
if session_key2:
session_list.append(session_key2)
cfg.logger.warning('session_list: %s', session_list)
(error_code,
db_results) = util.db_find2('session_user_map',
{"session_key": {
"$in": session_list
}})
cfg.logger.warning('error_code: %s db_results: %s', error_code, db_results)
if error_code != S_OK:
return (error_code, {})
if not db_results:
return (S_ERR, {})
user_id = db_results[0].get('user_id', '')
user_info = util.db_find_one('user_info', {"user_id": user_id})
if not user_info:
return (S_ERR, {})
#if _is_to_refresh_google_token(user_info):
# _refresh_google_token(user_info)
_check_refresh_session(session, session_key, session_key2, user_info)
return (S_OK, user_info)
