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 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)