abuseipdb.py•927 B
import requests
from string import Template
from urllib.parse import urljoin
import jinja2
from services.base import BaseService
class AbuseIPDB(BaseService):
name: str = "abuseripdb"
host: str = "https://api.abuseipdb.com/api/v2/"
class AbuseIPDBIP(AbuseIPDB):
endpoint: Template = Template("check")
def get(self, ipaddress: str) -> requests.PreparedRequest:
return requests.Request(
method=self.method,
url=urljoin(self.host, self.endpoint.substitute()),
params={"ipAddress": ipaddress},
headers=self.headers,
).prepare()
def parse_response(self, response: requests.Response, template: jinja2.Template) -> str:
if response and response.ok and response.json():
return template.render(
name=self.name,
**response.json().get("data")
)
else:
return ""