def add_record(record): time = datetime.now() trash_list = [record[GARBAGE_NAME]] if record[SIMILAR_WORDS] != "": trash_list.extend(record[SIMILAR_WORDS].split(' ')) if record[OTHER_RULE] != "": category_list = [record[RULE], record[OTHER_RULE]] else: category_list = [record[RULE]] for trash_name in trash_list: # register trash trash = Trash() trash.name = trash_name trash.detail = record[HOW_TO] trash.created_at = time trash.updated_at = time db_session.add(trash) db_session.commit() # insert categories and throw_rules for category_name in category_list: category = db_session.query(Category).filter_by( name=category_name).first() if (category is None): category = Category() category.name = category_name category.created_at = time category.updated_at = time db_session.add(category) db_session.commit() throw_rule = ThrowRule() throw_rule.trash_id = trash.trash_id throw_rule.category_id = category.category_id throw_rule.created_at = time throw_rule.updated_at = time db_session.add(throw_rule) db_session.commit()
def insert_data(key, data_list): time = datetime.now() if key == 'trashes': for data in data_list: trash = Trash() trash.trash_id = data['trash_id'] trash.name = data['name'] trash.reading = data['reading'] trash.detail = data['detail'] trash.created_at = time trash.updated_at = time db_session.add(trash) db_session.commit() elif key == 'categories': for data in data_list: category = Category() category.category_id = data['category_id'] category.name = data['name'] category.created_at = time category.updated_at = time db_session.add(category) db_session.commit() elif key == 'towns': for data in data_list: town = Town() town.town_id = data['town_id'] town.area_id = data['area_id'] town.name = data['name'] town.created_at = time town.updated_at = time db_session.add(town) db_session.commit() elif key == 'areas': for data in data_list: area = Area() area.area_id = data['area_id'] area.name = data['name'] area.created_at = time area.updated_at = time db_session.add(area) db_session.commit() elif key == 'collection_rules': for data in data_list: collection_rule = CollectionRule() collection_rule.collection_rule_id = data['collection_rule_id'] collection_rule.town_id = data['town_id'] collection_rule.category_id = data['category_id'] collection_rule.data = data['data'] collection_rule.created_at = time collection_rule.updated_at = time db_session.add(collection_rule) db_session.commit() elif key == 'throw_rules': for data in data_list: throw_rules = ThrowRule() throw_rules.throw_rule_id = data['throw_rule_id'] throw_rules.trash_id = data['trash_id'] throw_rules.category_id = data['category_id'] throw_rules.created_at = time throw_rules.updated_at = time db_session.add(throw_rules) db_session.commit()