def set_settings(self, db_session: Session, setting_to_set: SettingSet): new_setting = Setting(welcome_speech=setting_to_set.welcome_speech, color_button=setting_to_set.color_button, community_id=setting_to_set.community_id) db_session.add(new_setting) db_session.commit() db_session.refresh(new_setting) return new_setting
def add_time_table(self, db_session: Session, time_to_set: TimeTableSet): new_time_table = TimeTable(manager_id=time_to_set.manager_id, day_of_the_week=time_to_set.day_of_the_week, start_work=time_to_set.start_work, end_work=time_to_set.end_work) db_session.add(new_time_table) db_session.commit() db_session.refresh(new_time_table) return new_time_table
def update_settings(self, db_session: Session, community_id: int, setting_to_update: SettingUpdate): update_setting = db_session.query(Setting).filter_by( community_id=community_id).first() if update_setting is None: raise NoResultFound('Setting not found') update_setting.welcome_speech = setting_to_update.welcome_speech update_setting.color_button = setting_to_update.color_button db_session.add(update_setting) db_session.commit() db_session.refresh(update_setting) return update_setting
def add_manager(self, db_session: Session, community_id: int, manager_to_add: ManagerCreate): community = (db_session.query(Community).options( lazyload("managers")).get(community_id)) manager = Manager( phone=manager_to_add.phone, name=manager_to_add.name, is_blocked=manager_to_add.is_blocked, ) community.managers.append(manager) db_session.add(community) db_session.commit() db_session.refresh(community) return community
def update_time_table(self, db_session: Session, time_table_id: int, time_to_update: TimeTableUpdate): update_time = db_session.query(TimeTable).filter_by( id=time_table_id).first() if update_time is None: raise NoResultFound('Time Table not found') update_time.day_of_the_week = time_to_update.day_of_the_week update_time.start_work = time_to_update.start_work update_time.end_work = time_to_update.end_work db_session.add(update_time) db_session.commit() db_session.refresh(update_time) return update_time
def download_recipe_meta_data(cls): """Downloads recipe metadata""" session = Session() # loading bar based on recipe count recipe_count = 1 recipe_batch_count = cls.get_recipe_batch_count() cls.logger.info('Parsing and saving recipe meta data') with tqdm(total=recipe_batch_count, unit=' recipes') as pbar: for page in range(1, cls.max_api_page): req = requests.get(RECIPE_API, params={'page': page}) soup = BeautifulSoup(req.text, 'html.parser') section = soup.find('section', {'id': 'archive-recipes'}) recipes = section.findAll('li') for recipe_html in recipes: if recipe_count > RECIPE_BATCH_COUNT: return slug = recipe_html.a['href'].replace( '/plant-based-recipes/', '') recipe = get_or_create(session, Recipe, slug=slug) recipe.name = recipe_html.img['title'] cls.logger.info('Parsing recipe: ' + recipe.name) recipe.slug = slug recipe.url = 'https://' + cls.hostname + \ str(recipe_html.a['href']) recipe.origin = SERVICE_CODE recipe.country = 'US' assets = [] thumbnail_url = recipe_html.img['src'] thumbnail = get_or_create( session, Asset, url=thumbnail_url) thumbnail.type = 'thumbnail' thumbnail.url = thumbnail_url assets.append(thumbnail) recipe.assets = assets session.add(recipe) session.commit() pbar.update(1) recipe_count += 1 page += 1
def create(self, db_session: Session, community_to_create: CommunityCreate, user_id: int): info = services.vk_service.get_community_info( community_to_create.api_key, community_to_create.community_vk_id) community = Community( community_vk_id=community_to_create.community_vk_id, avatar_url=info["photo_200"], name=info["name"], ) user = db_session.query(User).get(user_id) if user is not None: community.admins.append(user) db_session.add(community) db_session.commit() db_session.refresh(community) return community
def create_user(self, db_session: Session, user_to_create: UserCreate): existing_user = self.get_by_username(db_session, user_to_create.username) if existing_user is not None: raise Exception('User already exists') new_user = User( username=user_to_create.username, first_name=user_to_create.first_name, last_name=user_to_create.last_name, is_admin=user_to_create.is_admin, vk_id=user_to_create.vk_id, avatar_url=user_to_create.avatar_url, email=user_to_create.email, phone=user_to_create.phone, password=user_to_create.password, ) new_user.set_password(user_to_create.password) db_session.add(new_user) db_session.commit() db_session.refresh(new_user) return new_user