mirror of
https://github.com/fuckpiracyshield/api.git
synced 2024-12-23 02:50:52 +01:00
35 lines
933 B
Python
35 lines
933 B
Python
|
from piracyshield_service.security.blacklist.exists_by_ip_address import SecurityBlacklistExistsByIPAddressService
|
||
|
|
|
||
|
|
from ioutils.errors import ErrorCode, ErrorMessage
|
||
|
|
|
||
|
|
class BlacklistInterceptor:
|
||
|
|
|
||
|
|
"""
|
||
|
|
Applies bans.
|
||
|
|
"""
|
||
|
|
|
||
|
|
security_blacklist_exists_by_ip_address_service = None
|
||
|
|
|
||
|
|
async def execute(self, r):
|
||
|
|
await self._prepare_modules()
|
||
|
|
|
||
|
|
if self.security_blacklist_exists_by_ip_address_service.execute(
|
||
|
|
ip_address = r.request.remote_ip
|
||
|
|
) == True:
|
||
|
|
raise BlacklistInterceptorException()
|
||
|
|
|
||
|
|
async def _prepare_modules(self):
|
||
|
|
self.security_blacklist_exists_by_ip_address_service = SecurityBlacklistExistsByIPAddressService()
|
||
|
|
|
||
|
|
class BlacklistInterceptorException(Exception):
|
||
|
|
|
||
|
|
"""
|
||
|
|
The IP address is temporary banned.
|
||
|
|
"""
|
||
|
|
|
||
|
|
status_code = 403
|
||
|
|
|
||
|
|
error_code = ErrorCode.IP_ADDRESS_BLACKLISTED
|
||
|
|
|
||
|
|
error_message = ErrorMessage.IP_ADDRESS_BLACKLISTED
|