def register(self, request=(UserMessageP,)):
if getattr(request, 'username') is not None and getattr(request, 'password') is not None:
user_key = ndb.Key('User', getattr(request, 'email'))
if user_key.get() is None:
user = User(
key=ndb.Key('User', getattr(request, 'email')),
username=getattr(request, 'username'),
password=getattr(request, 'password'),
email=getattr(request, 'email'))
user.put()
if user is not None:
return protopigeon.to_message(user, UserMessageP)
else:
raise f3.BadRequestException("Incorrect credentials")
else:
raise f3.BadRequestException("You already an account")
else:
raise f3.BadRequestException("user or password doesn't specified")
def login(self, request=(UserLogginMessage,)):
if getattr(request, 'username') is not None and getattr(request, 'password') is not None:
query = User.query(
User.username == getattr(request, 'username'),
User.password == getattr(request, 'password'))
user = query.get()
if user is not None:
if user.is_active:
return UserMessage(
username=user.username,
email=user.email,
user=user.key.urlsafe())
else:
raise f3.BadRequestException("User is Inactive")
else:
raise f3.BadRequestException("Incorrect credentials")
else:
raise f3.BadRequestException("user or password doesn't specified")
