def _insert_data_vm_to_db(data, type_poster): """ """ for i in range(len(data['street'])): flat = {} street_obj = db_util.get_street(data['street'][i]) if len(street_obj) >= 1: district_obj = db_util.get_district_by_id(street_obj[0].district_id) city = db_util.get_city_by_id(district_obj.city_id) region = db_util.get_region_by_id(city.region_id) flat['street'] = street_obj[0] flat['district'] = district_obj flat['city'] = city flat['region'] = region flat['type_room'] = data['type_room'][i] flat['count_room'] = data['count_room'][i] flat['floor'] = data['floor'][i] flat['max_floor'] = data['max_floor'][i] flat['price'] = data['price'][i] flat['full_price'] = data['full_price'][i] flat['s_live'] = data['s_live'][i] flat['s_cook'] = data['s_cook'][i] flat['description'] = data['description'][i] flat['phone'] = data['phone'][i] flat['image'] = data['image'][i] flat['type_poster'] = type_poster db_util.set_flat(flat)
def main(): parser = argparse.ArgumentParser(description='Parse data from file.') parser.add_argument('-id', '--city_id', dest='city_id', type=int, required=True, help='please write city_id from your db') parser.add_argument('-f','--file', dest='filename', type=str, required=True, help='please write path to file') parser.add_argument('-v','--verbosity', dest='verbosity', type=int, choices=[0, 1, 2, 3], default=3, help='please chouse verbosity level (0, 1, 2, 3)') args = parser.parse_args() util.change_verbosity(args.verbosity) logger = util.get_logger() logger.info('Start process...') args.filename = os.path.abspath(args.filename) if not os.path.exists(args.filename): logger.error('File %s doesn\'t exist' % args.filename) elif db_util.get_city_by_id(args.city_id) is None: logger.error('Object with id = %s doesn\'t exist' % args.city_id) else: parse_content.parse_district(args.filename, args.city_id)