• Search by Module
• Search by Words
• Search Projects

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• fishroom-master
  • fishroom
    • telegram.py
    • gitter.py
    • db.py
    • photostore.py
    • xmpp.py
    • IRC.py
    • web
      • nickcolors.css
      • base.html
      • __main__.py
      • oauth.py
      • __init__.py
      • handlers.py
      • navbar.html
      • text_store.html
      • chat_log.html
    • telegram_tg.py
    • counter.py
    • textstore.py
    • fishroom.py
    • filestore.py
    • plugins
      • imglink.py
      • pia.py
      • hualao.py
      • stats.py
      • ratelimit.py
      • __init__.py
      • vote.py
    • models.py
    • dumpload.py
    • config.py.example
    • chatlogger.py
    • api_client.py
    • __init__.py
    • command.py
    • matrix.py
    • runner.py
    • bus.py
    • textformat.py
    • wechat.py
    • helpers.py
    • base.py
  • LICENSE
  • test
    • test_pastebin.py
    • __init__.py
    • test_vinergy.py
  • README.md
  • requirements.txt
  • Dockerfile
  • .gitignore

#!/usr/bin/env python
import pytz
from datetime import datetime

from ..db import get_redis
from ..config import config

tz = pytz.timezone(config.get("timezone", "utc"))


class RateLimiter(object):

    key = config["redis"]["prefix"] + ":rate_limit:{room}:{cmd}"

    def __init__(self):
        self.r = get_redis()

    def trigger(self, room, cmd):
        key = self.key.format(room=room, cmd=cmd)
        now_ts = datetime.now(tz=tz).strftime("%s")
        self.r.rpush(key, now_ts)

    def check(self, room, cmd, period=30, count=5):
        key = self.key.format(room=room, cmd=cmd)
        l = self.r.llen(key)
        if l < count:
            self.trigger(room, cmd)
            return True

        self.r.ltrim(key, -count, -1)
        first = int(self.r.lindex(key, 0))
        now_ts = int(datetime.now(tz=tz).strftime("%s"))
        if now_ts - first <= period:
            return False

        self.trigger(room, cmd)
        return True

# vim: ts=4 sw=4 sts=4 expandtab