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)
