def get_courts_to_search(row, courts_by_name): if 'courtName' not in row or row['courtName'] not in courts_by_name: return [court['fips_code'] for court in courts] return get_nearby_courts(row['courtName']) search_id = Database.insert_search() search_terms = 0 search_tasks = [] with open(sys.argv[1]) as csvfile: reader = csv.DictReader(csvfile) for row in reader: courts_to_search = get_courts_to_search(row, courts_by_name) name = row['last'] + ', ' + row['first'] search_terms += 1 print name, len(courts_to_search), 'courts' for fips_code in courts_to_search: search_tasks.append({ 'search_id': search_id, 'type': 'name', 'court_fips': fips_code, 'court_type': 'circuit', 'case_type': 'civil', 'term': name.upper(), 'case_details': True }) Database.insert_created_tasks('circuit', search_tasks) Database.update_search(search_id, search_terms, len(search_tasks))
[court['fips_code'] for court in nearby_courts] return courts_by_name[court_name]['nearbyCourts'] def get_courts_to_search(row, courts_by_name): if 'courtName' not in row or row['courtName'] not in courts_by_name: return [court['fips_code'] for court in courts] return get_nearby_courts(row['courtName']) search_id = Database.insert_search() search_terms = 0 search_tasks = [] with open(sys.argv[1]) as csvfile: reader = csv.DictReader(csvfile) for row in reader: courts_to_search = get_courts_to_search(row, courts_by_name) name = row['last'] + ', ' + row['first'] search_terms += 1 print name, len(courts_to_search), 'courts' for fips_code in courts_to_search: search_tasks.append({ 'search_id': search_id, 'type': 'name', 'court_fips': fips_code, 'court_type': 'circuit', 'case_type': 'civil', 'term': name.upper(), 'case_details': True }) Database.insert_created_tasks('circuit', search_tasks) Database.update_search(search_id, search_terms, len(search_tasks))