mirror of
https://github.com/cve-search/pyvariot.git
synced 2024-11-30 10:47:21 +00:00
46 lines
1.5 KiB
Python
46 lines
1.5 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from importlib.metadata import version
|
||
|
|
from typing import Any
|
||
|
|
from urllib.parse import urljoin, urlparse
|
||
|
|
|
||
|
|
import requests
|
||
|
|
|
||
|
|
|
||
|
|
class PyProject():
|
||
|
|
|
||
|
|
def __init__(self, root_url: str, useragent: str | None=None,
|
||
|
|
*, proxies: dict[str, str] | 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 instance.
|
||
|
|
:param proxies: The proxies to use to connect to theinstance - 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("pyproject")}'
|
||
|
|
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) -> dict[str, Any]:
|
||
|
|
'''Check if redis is up and running'''
|
||
|
|
r = self.session.get(urljoin(self.root_url, 'redis_up'))
|
||
|
|
return r.json()
|