o3o3o / django-graphql-ratelimit Goto Github PK

In [1]: from django_graphql_ratelimit import ratelimit
---------------------------------------------------------------------------
ModuleNotFoundError                       Traceback (most recent call last)
<ipython-input-1-b33a7968dd74> in <module>
----> 1 from django_graphql_ratelimit import ratelimit

/usr/local/lib/python3.8/site-packages/django_graphql_ratelimit/__init__.py in <module>
----> 1 from .ratelimit import GQLRatelimitKey, ratelimit  # noqa

/usr/local/lib/python3.8/site-packages/django_graphql_ratelimit/ratelimit.py in <module>
      1 import logging
----> 2 from ratelimit import ALL
      3 from functools import wraps
      4 from ratelimit.exceptions import Ratelimited
      5 from ratelimit.utils import is_ratelimited

ModuleNotFoundError: No module named 'ratelimit'

from django_graphql_ratelimit import ratelimit
from django.utils.decorators import method_decorator

class CustomTokenAuth(ObtainJSONWebToken):

    @classmethod
    @method_decorator(ratelimit(key='gql:username', rate='0/s', block=True))
    def resolve(cls, root, info, **kwargs):

        user = info.context.user
        role = user.profile.role
        logger.info('USER_LOGIN: %s', user)

        if role == 'student':
            student = user.student
            student.check_login_permission()

        user.last_login = timezone.now()
        user.save()

        return cls()


class Mutation(
    api.schema.Mutation,
    # other mutations ...
):
    token_auth = CustomTokenAuth.Field()