def mutate(self, info, **kwargs): if "account" in kwargs: if not account_validation(kwargs["account"]): raise GraphQLError("Incorrect Account Format") if User.objects.filter(account=kwargs["account"]): raise GraphQLError("Already Exist Account") if "phone_number" in kwargs: if not phone_number_validation(kwargs["phone_number"]): raise GraphQLError("Incorrect Phone Number Format") if User.objects.filter(phone_number=kwargs["phone_number"]): raise GraphQLError("Already Exist Phone_Number") if "email" in kwargs: if User.objects.filter(email=kwargs["email"]): raise GraphQLError("Already Exist Email") user = User.objects.get(id=globals.user) kwargs["updated_at"] = datetime.now() for key, value in kwargs.items(): setattr(user, key, value) if "password" in kwargs: if not password_validation(kwargs["password"]): raise GraphQLError("Incorrect Password Format") user.set_password(kwargs["password"]) user.save() token = get_token(user) refresh_token = create_refresh_token(user) return UpdateUser(user=user, token=token, refresh_token=refresh_token)
def mutate(self, info, phone, provider=None, auth_code=None, user_info=None): try: phone = parse_phone(phone) except exceptions.InvalidPhone as e: raise exceptions.GQLError(e.message) try: shop_user = ShopUser.objects.get(phone=phone) except ShopUser.DoesNotExist: shop_user = ShopUser.objects.create_user(phone=phone) if provider and auth_code: try: shop_user.bind_third_account(provider, auth_code) except ( exceptions.AlreadyBinded, exceptions.DoNotSupportBindType, exceptions.CodeBeUsed, ) as e: raise exceptions.GQLError(e.message) if user_info: for k, v in user_info.items(): setattr(shop_user, k, v) shop_user.save(update_fields=[k]) token = get_token(shop_user.user) refresh_token = create_refresh_token(shop_user.user) return SignUp(me=shop_user, token=token, refresh_token=refresh_token)
def setUp(self): super(ClearTokensTests, self).setUp() self.refresh_token = create_refresh_token(self.user) if sys.version_info[0] < 3: self.f = io.BytesIO() else: self.f = io.StringIO()
def resolve_after_signup(self, info): """To login user from frontend after finish sigunp process.""" user = info.context.user if user.has_configured_image_analysis: return { 'token': get_token(user), 'refresh_token': create_refresh_token(user) } return {'token': None, 'refresh_token': None}
def mutate(self, info, username, password, email): user = get_user_model()( username=username, email=email, ) user.set_password(password) user.save() profile_obj = profile.objects.get(user=user.id) token = get_token(user) refresh_token = create_refresh_token(user) return CreateUser(user=user, profile=profile_obj, token=token, refresh_token=refresh_token)
def mutate(root, info, username, password, email=""): user = get_user_model()(username=username, email=email) user.set_password(password) user.save() profile_obj = Profile.objects.get(owner=user.id) token = get_token(user) refresh_token = create_refresh_token(user) return RegisterUser(user=user, profile=profile_obj, token=token, refresh_token=refresh_token)
def mutate(self, info, username, password, email): user = get_user_model()( username=username, email=email, ) user.set_password(password) user.save() token = get_token(user) refresh_token = create_refresh_token(user) return CreateUser(user=user, token=token, refresh_token=refresh_token)
def mutate(cls, root, info, **input): user = authenticate(email=input['input']['email'], password=input['input']['password']) if user is None: # Please enter a correct email and password raise Exception('100') if not user.is_active: # It seems your account has been disabled raise Exception('101') return cls(user=user, token=get_token(user), refresh_token=create_refresh_token(user))
def mutate(self, info, **kwargs): if not account_validation(kwargs["account"]): raise GraphQLError("Incorrect Account Format") if not password_validation(kwargs["password"]): raise GraphQLError("Incorrect Password Format") if not phone_number_validation(kwargs["phone_number"]): raise GraphQLError("Incorrect Phone Number Format") if User.objects.filter(account=kwargs["account"]): raise GraphQLError("Already Exist Account") if User.objects.filter(email=kwargs["email"]): raise GraphQLError("Already Exist Email") if User.objects.filter(phone_number=kwargs["phone_number"]): raise GraphQLError("Already Exist Phone_Number") user = User(**kwargs) user.set_password(kwargs["password"]) user.save() token = get_token(user) refresh_token = create_refresh_token(user) return CreateUser(user=user, token=token, refresh_token=refresh_token)
def _login(user: UserModel) -> dict: jti = user.jtis.create(value=crypto.create_jwt_id()) user.jwt_salt = jti.value user.last_login = timezone.now() user.save(update_fields=["last_login", "jwt_salt"]) success = True token = shortcuts.get_token(user) p = shortcuts.get_payload(token) payload = { "email":p.get("email", ""), "exp":p.get("exp", 0), "origIat":p.get("origIat", 0) } refreshtkn = shortcuts.create_refresh_token(user) return { "success":success, "token":token, "refresh_token": refreshtkn, "user":user, "payload":payload }
def mutate(self, info, username, password, email, **kwargs): user = get_user_model()( username=username, email=email, ) user.set_password(password) fname = kwargs.pop("firstname") lname = kwargs.pop("lastname") user.first_name = fname user.last_name = lname # print(user.firstName) user.save() profile_obj = Profile.objects.get(user=user.id) for i in kwargs.keys(): setattr(profile_obj, i, kwargs[i]) profile_obj.save() token = get_token(user) refresh_token = create_refresh_token(user) user = User.objects.get(id=user.id) return CreateUser( user=user, profile=profile_obj, token=token, refresh_token=refresh_token )
def setUp(self): super().setUp() self.refresh_token = create_refresh_token(self.user)
def setUp(self): super().setUp() RefreshToken = get_refresh_token_model() self.refresh_token = create_refresh_token(self.user) self.refresh_token_admin = admin.RefreshTokenAdmin(RefreshToken, site)
def setUp(self): super().setUp() self.refresh_token = create_refresh_token(self.user) self.f = io.StringIO()
def setUp(self): super(AbstractRefreshTokenTests, self).setUp() self.refresh_token = create_refresh_token(self.user)
def test_get_refresh_token(self): refresh_token = shortcuts.create_refresh_token(self.user) user = shortcuts.get_refresh_token(refresh_token).user self.assertEqual(user, self.user)
def setUp(self): super(RefreshTokenMixin, self).setUp() self.refresh_token = create_refresh_token(self.user)