def search_restaurants(restaurant_ids=None, is_valid=None,
city_ids=None, region_group_ids=None,
region_ids=None, is_premium=None,
order_mode=None, offset=None, limit=None):
query = thirdparty_svc.ers.TRestaurantSearchFilterQuery()
query.city_ids = city_ids
query.region_group_ids = region_group_ids
query.region_ids = region_ids
query.is_valid = is_valid
query.is_premium = is_premium
query.order_mode = order_mode
query.offset = offset
query.size = limit
query.order_by_id = 'desc'
if restaurant_ids is not None:
query.restaurant_ids = restaurant_ids
with thrift_client('ers') as ers:
restaurant_json = ers.search_filter_restaurant(query)
if not restaurant_json:
raise_zeus_exc(ZEUS_DATABASE_ERROR)
restaurant_info = json.loads(restaurant_json)
total_num = restaurant_info['total']
if total_num == 0:
return [], 0
restaurants = [rest['_source'] for rest in restaurant_info['hits']]
return restaurants, total_num
def zeus_session():
session = ZeusMySession()
try:
yield session
session.commit()
except Exception as e:
session.rollback()
raise_zeus_exc(ZEUS_DATABASE_ERROR, exc=e)
finally:
session.close()
def zeus_session():
session = ZeusMySession()
try:
yield session
session.commit()
except Exception as e:
session.rollback()
raise_zeus_exc(ZEUS_DATABASE_ERROR, exc=e)
finally:
session.close()
def upload_file(file,
sized=False,
watermark=False,
file_size=FILE_SIZE,
isprivate=False):
file_name = file.filename
extension_name = os.path.splitext(file_name)[1].lstrip('.')
file_hash = file_upload_raw(file.stream.read(), extension_name, sized,
watermark, file_size, isprivate)
if not file_hash:
raise_zeus_exc(FILE_UPLOAD_ERR, file_name=file_name)
return file_hash
def wrapper(*args, **kwargs):
session = ZeusMySession()
func.func_globals['session'] = session
try:
ret = func(*args, **kwargs)
except SQLAlchemyError as se:
session.rollback()
raise_zeus_exc(ZEUS_DATABASE_ERROR, exc=se)
except Exception as e:
session.rollback()
raise_zeus_exc(ZEUS_DATABASE_ERROR, exc=e)
finally:
session.close()
func.func_globals.pop('session')
return ret
def wrapper(*args, **kwargs):
session = ZeusMySession()
func.func_globals["session"] = session
try:
ret = func(*args, **kwargs)
except SQLAlchemyError as se:
session.rollback()
raise_zeus_exc(ZEUS_DATABASE_ERROR, exc=se)
except Exception as e:
session.rollback()
raise_zeus_exc(ZEUS_DATABASE_ERROR, exc=e)
finally:
session.close()
func.func_globals.pop("session")
return ret
def upload_file(file, sized=False, watermark=False, file_size=FILE_SIZE,isprivate=False):
file_name = file.filename
extension_name = os.path.splitext(file_name)[1].lstrip('.')
file_hash = file_upload_raw(
file.stream.read(),
extension_name,
sized,
watermark,
file_size,
isprivate
)
if not file_hash:
raise_zeus_exc(FILE_UPLOAD_ERR, file_name=file_name)
return file_hash
