def update_pol_beers(conn_str=None): db = Database(conn_str) beers_polet, updated = read_pol_beers() # ok, unwrap generator beers_polet = [b for b in beers_polet] breweries_rb = db.get_rb_breweries() breweries_pol = get_breweries_polet(beers_polet) matched_pol_rb = match_pol_breweries(breweries_pol, breweries_rb) matched_beers = [] for rb_beer_id, pol_breweries in matched_pol_rb.iteritems(): rb_brewery = find_in_list(breweries_rb, 'id', rb_beer_id)['name'] rb_beers_for_brewery = db.get_rb_beers_for_brewery(rb_beer_id) matched_beers += match_pol_beers( rb_brewery, rb_beers_for_brewery, pol_breweries, beers_polet, db ) save_pol_beers(matched_beers, db) db.add_log('pol_beers', updated)
def update_pol_beers(conn_str=None): db = Database(conn_str) beers_polet, updated = read_pol_beers() breweries_rb = db.get_rb_breweries() breweries_pol = get_breweries_polet(beers_polet) matched_pol_rb = match_pol_breweries(breweries_pol, breweries_rb) matched_beers = [] for rb_beer_id, pol_breweries in matched_pol_rb.iteritems(): rb_brewery = find_in_list(breweries_rb, 'id', rb_beer_id)['name'] rb_beers_for_brewery = db.get_rb_beers_for_brewery(rb_beer_id) matched_beers += match_pol_beers(rb_brewery, rb_beers_for_brewery, pol_breweries, beers_polet, db) save_pol_beers(matched_beers, db) db.add_log('pol_beers', updated)
# -*- coding: utf-8 -*- import sys import json import os from beertools import read_pol_beers if __name__ == '__main__': outfile = sys.argv[1] path = os.path.join(os.getcwd(), outfile) with open(path, 'w') as out: beers, updated = read_pol_beers() out.write(json.dumps(beers, indent=4))