def test_func_account_is_old(self): business1 = Business(name='name', password='******', email='email', confirmed=True) business2 = Business(name='name', password='******', email='email', confirmed=True) business2.registered_on -= timedelta(days=32) db.session.add(business1) db.session.add(business2) db.session.commit() self.assertFalse(Business.account_is_old(business1.id)) self.assertTrue(business2.id)
def confirm_email(token): try: email = Business.confirm_token(token) except: return redirect('/business/feedback/Ссылка не действительна') business = Business.query.filter_by(email=email).first_or_404() if business.confirmed: return redirect('/business/feedback/Аккаунт был подтвержден. Пожалуйста, войдите.') else: if Business.account_is_old(business.id): return redirect('/feedback/Ссылка не действительна! Попробуйте зарегистрироваться еще раз!') business.confirmed = True business.confirmed_on = datetime.now() db.session.add(business) db.session.commit() return redirect('/business/feedback/Аккаунт был подтвержден. Пожалуйста, войдите.')