mirror of
https://github.com/cve-search/PyVulnerabilityLookup.git
synced 2024-11-29 02:07:21 +00:00
50 lines
1.8 KiB
Python
50 lines
1.8 KiB
Python
#!/usr/bin/env python3
|
|
|
|
from __future__ import annotations
|
|
|
|
from importlib.metadata import version
|
|
from pathlib import Path
|
|
from typing import Any
|
|
from urllib.parse import urljoin, urlparse
|
|
|
|
import requests
|
|
|
|
|
|
class PyVulnerabilityLookup():
|
|
|
|
def __init__(self, root_url: str, useragent: str | None=None,
|
|
*, proxies: dict[str, str] | None=None) -> None:
|
|
'''Query a specific instance.
|
|
|
|
:param root_url: URL of the instance to query.
|
|
:param useragent: The User Agent used by requests to run the HTTP requests against the vulnerability lookup instance
|
|
:param proxies: The proxies to use to connect to the vulnerability lookup instance - More details: https://requests.readthedocs.io/en/latest/user/advanced/#proxies
|
|
'''
|
|
self.root_url = root_url
|
|
|
|
if not urlparse(self.root_url).scheme:
|
|
self.root_url = 'http://' + self.root_url
|
|
if not self.root_url.endswith('/'):
|
|
self.root_url += '/'
|
|
self.session = requests.session()
|
|
self.session.headers['user-agent'] = useragent if useragent else f'PyProject / {version("pyvulnerabilitylookup")}'
|
|
if proxies:
|
|
self.session.proxies.update(proxies)
|
|
|
|
@property
|
|
def is_up(self) -> bool:
|
|
'''Test if the given instance is accessible'''
|
|
try:
|
|
r = self.session.head(self.root_url)
|
|
except requests.exceptions.ConnectionError:
|
|
return False
|
|
return r.status_code == 200
|
|
|
|
def redis_up(self) -> bool:
|
|
'''Check if redis is up and running'''
|
|
r = self.session.get(urljoin(self.root_url, 'redis_up'))
|
|
return r.json()
|
|
|
|
def get_vulnerability(self, vulnerability_id: str) -> dict[str, Any]:
|
|
r = self.session.get(urljoin(self.root_url, str(Path('vulnerability', vulnerability_id))))
|
|
return r.json()
|