mirror of
https://github.com/fuckpiracyshield/api.git
synced 2024-05-20 05:56:19 +02:00
50 lines
1.6 KiB
Python
50 lines
1.6 KiB
Python
from piracyshield_component.log.logger import Logger
|
|
from piracyshield_component.exception import ApplicationException
|
|
from .response import ResponseHandler
|
|
|
|
import time
|
|
from collections import deque
|
|
from tornado.locks import Lock
|
|
|
|
from interceptors.logger import LoggerInterceptor
|
|
from interceptors.security.blacklist import BlacklistInterceptor
|
|
from interceptors.security.rate_limit import RateLimitInterceptor
|
|
|
|
from ioutils.errors import ErrorCode, ErrorMessage
|
|
|
|
class RequestHandler(ResponseHandler):
|
|
|
|
"""
|
|
Requests gateway.
|
|
|
|
The requestor gets blocked here before proceding any further.
|
|
Interceptors are implemented to allow certain features to take part of the action.
|
|
|
|
Current actived interceptors:
|
|
- security: takes care of the IP addresses exceeding limits (ex. applies the ban rules when triggered).
|
|
- rate limit: the IP will get a 429 when trying to perform too many requests on a given timeframe.
|
|
"""
|
|
|
|
interceptors = [
|
|
LoggerInterceptor,
|
|
BlacklistInterceptor,
|
|
RateLimitInterceptor
|
|
]
|
|
|
|
async def prepare(self):
|
|
await self.run_interceptors()
|
|
|
|
async def run_interceptors(self):
|
|
for interceptor in self.interceptors:
|
|
try:
|
|
await interceptor().execute(self)
|
|
|
|
except Exception as InterceptorException:
|
|
self.error(
|
|
status_code = InterceptorException.status_code,
|
|
error_code = InterceptorException.error_code,
|
|
message = InterceptorException.error_message
|
|
)
|
|
|
|
return
|