def createuser(self, **kwargs):
"""
Create user without admin status
"""
user = User(**kwargs)
user.password = kwargs['password']
user.create()
def test_create_user(db):
user = User.create("test", "password")
assert user.name == "test"
assert User.exists("test")
with pytest.raises(User.UserExistsError) as excinfo:
user = User.create("test", "password")
assert "test" in str(excinfo)
def arguments_wrapper(*args, **kwargs):
try:
changed = False
user = User.get(telegram_user_id=args[0].from_user.id)
if not user.has_messages_after_notification:
user.has_messages_after_notification = True
changed = True
if user.username != args[0].from_user.username:
user.username = args[0].from_user.username
changed = True
if user.first_name != args[0].from_user.first_name:
user.first_name = args[0].from_user.first_name
changed = True
if user.surname != args[0].from_user.last_name:
user.surname = args[0].from_user.last_name
changed = True
if changed:
user.save()
except Exception as e:
user = User.create(
username=args[0].from_user.username,
telegram_chat_id=args[0].chat.id,
first_name=args[0].from_user.first_name,
surname=args[0].from_user.last_name,
# phone=args[0].contact.phone_number,
telegram_user_id=args[0].from_user.id)
obj = tele_func(*args, **kwargs, user=user)
return obj
def test_user_id_is_null(self): try: self.user = User.create(user_id=None) except IntegrityError: assert True else: assert False
def test_user_id_is_not_unique(self): try: self.user = User.create(user_id='vk_1234') except IntegrityError: assert True else: assert False
def add_user_(nickname):
"""
Добавить пользователя
:param nickname: никнейм
:return: json ответ с полями нового пользователя
"""
try:
with db.atomic():
user = User.create(nickname=nickname, active=True)
user.save()
user_json = __create_json_auth_user(user)
except Exception as e:
error_message = str(e).split('\n')[0]
user_json = {'error': error_message}
return user_json
def setup_class(self): self.user = User.create(user_id='vk_1234')
def setup_class(self):
self.user = User.create(user_id='id_1234')
self.theory = Theory.create(text='lala', theme='lala')
self.readed_theory = ReadedTheory(user=self.user, theory=self.theory)
