def get_yelp_item(coordinate, category, strategy=DistanceStrategy()):
extra_yelp_params = {}
yelp_item_repository = YelpItemRepository()
# for these strategies, use YelpAPI
# then reset strategy_name so that we can still call strategy module below
extra_yelp_params['radius_filters'] = 7500
if strategy.type == StrategyType.popularity:
extra_yelp_params['sort'] = 2
strategy = FirstStrategy()
elif strategy.type == StrategyType.distance:
extra_yelp_params['sort'] = 1
strategy = FirstRandomStrategy()
try:
search_results = query(yelpapi.search(coordinate.latitude,
coordinate.longitude,
category.search_term,
query(category.search_filters).select(lambda f: f.filter).to_list(),
**extra_yelp_params)).where(
lambda r: ['Food Trucks', 'foodtrucks'] not in r['categories'])
yelp_items = [YelpItem.create_from_dict(result) for result in search_results]
yelp_item = strategy.run_strategy(yelp_items)
yelp_item = yelp_item_repository.get_or_insert(yelp_item)
yelp_item_repository.save_changes()
return yelp_item
except:
return None
def from_json(json):
item = Item(
name=json.get('name'),
yelp_category_id=json.get('yelp_category', {}).get('id'),
yelp_category = YelpCategory.from_json(json.get('yelp_category')),
plan_id=json.get('plan_id'),
start_time=datetime.strptime(json.get('start_time'), "%Y-%m-%d %H:%M:%S %z") if json.get('start_time') is not None else None,
end_time=datetime.strptime(json.get('end_time'), "%Y-%m-%d %H:%M:%S %z") if json.get('end_time') is not None else None,
description=json.get("description"),
type=json.get("type"),
yelp_item_id=json.get('yelp_item', {}).get('id'),
yelp_item=YelpItem.from_json(json.get('yelp_item')),
location_id=json.get('location', {}).get('id'),
location=Location.from_json(json.get('location'))
)
return item
def get(self, **kwargs):
args = self.parser.parse_args()
search_results = yelpapi.search(args.latitude, args.longtitude,
args.term,args.categories.split(','),**kwargs)
yelp_items = [YelpItem.create_from_dict(result) for result in search_results]
return yelp_items
