def __init__(self,
counter,
host,
port=6379,
db=0,
password=None,
socket_connect_timeout=1,
socket_timeout=5,
ssl=False,
ssl_cert_reqs='required'):
if not host:
raise ConfigurationError(
'Redis wrapper improperly configured. Valid redis host is required!'
)
self._counter = counter('')
self.__con = redis.StrictRedis(
host,
port,
db,
password,
socket_connect_timeout=socket_connect_timeout,
socket_timeout=socket_timeout,
ssl=ssl,
ssl_cert_reqs=ssl_cert_reqs)
def __init__(self, counter, host, port=6379, db=0, socket_connect_timeout=1, socket_timeout=5):
if not host:
raise ConfigurationError('Redis wrapper improperly configured. Valid redis host is required!')
self._counter = counter('')
self.__con = redis.StrictRedis(
host, port, db, socket_connect_timeout=socket_connect_timeout, socket_timeout=socket_timeout)
def __init__(
self,
host=None,
tls=True,
krb5=False,
user='******',
password='',
timeout=60,
logger=None,
counter=None,
):
'''expects ready to use "partial" for counter (CounterWrapper)'''
self.logger = logger or logging.getLogger()
self._counter = counter('')
self.use_tls = tls
self.use_krb5 = krb5
self.user_base_dn = ','.join(explode_dn(user)[1:])
self.user_filter = '(' + explode_dn(user)[0] + ')'
self.user_attrib = ['dn']
# password auth
self.bind_dn = user
self.__password = password
self.session = None
self.host = host
if self.host and len(self.host.split('.')) <= 1:
# FIXME
self.host += '.zalando'
if not password and not krb5 and tls:
raise ConfigurationError('LDAP password is not set properly!')
def __init__(self, read_key):
self.__numeric_url = 'https://www.scalyr.com/api/numericQuery'
self.__timeseries_url = 'https://www.scalyr.com/api/timeseriesQuery'
self.__facet_url = 'https://www.scalyr.com/api/facetQuery'
if not read_key:
raise ConfigurationError('Scalyr read key is not set.')
self.read_key = read_key
def __init__(self, host, timeout=10):
if not host:
raise ConfigurationError(
'Whois wrapper improperly configured. Valid host is required!')
self.host = host
self.timeout = timeout
def __init__(self, counter, host, port=11211, socket_connect_timeout=1):
if not host:
raise ConfigurationError(
'Memcached wrapper improperly configured. Valid memcached host is required!'
)
self.__con = Client((host, port))
self._counter = counter('')
def __init__(self, host, port):
if not host:
raise ConfigurationError(
'ZMON wrapper improperly configured. Valid redis host is required!'
)
self.__redis = redis.StrictRedis(host, port)
self.logger = logger
def __init__(self, host, port=27017):
if not host:
raise ConfigurationError(
'MongoDB improperly configured. host is invalid. Check entity["host"] value or set valid host.'
)
self.host = host
self.port = port
def __init__(self, url=None, timeout=10, oauth2=False):
if not url:
raise ConfigurationError(
'Elasticsearch plugin improperly configured. URL is required!')
self.url = url
self.timeout = timeout
self.oauth2 = oauth2
self._headers = {'User-Agent': get_user_agent()}
def __init__(self, url, oauth2=False):
if not url:
raise ConfigurationError('KairosDB wrapper improperly configured. URL is missing!')
self.url = url
self.__session = requests.Session()
if oauth2:
self.__session.headers.update({'Authorization': 'Bearer {}'.format(tokens.get('uid'))})
def query(self, host=None, recordtype=None, nameserver=None):
if not self.host and not host:
raise ConfigurationError('DNS wrapper improperly configured. Host is required!')
if not recordtype:
raise ConfigurationError('DNS wrapper improperly configure. Recordtype is required!')
resolver = dns.resolver.Resolver()
# Allow to use custom nameserver
if nameserver:
resolver.nameservers = nameserver
query = resolver.query(host, recordtype)
result = []
for response in query:
result.append(str(response).replace('"', ''))
return result
def resolve(self, host=None):
if not self.host and not host:
raise ConfigurationError('DNS wrapper improperly configure. Host is required!')
if not host:
host = self.host
try:
result = socket.gethostbyname(host)
except Exception, e:
result = 'ERROR: ' + str(e)
def __init__(self, host, instance, http, jmx, counter):
'''expects ready to use "partials" for http, jmx and counter'''
if not host or not instance:
raise ConfigurationError('Zomcat wrapper improperly configured. Valid host & instance are required!')
self.host = host
self.instance = instance
self._http = http
self._jmx = jmx
# initialize counter with Redis connection
self._counter = counter('')
def __init__(self, read_key, scalyr_region=None):
scalyr_prefix = SCALYR_URL_PREFIX_US
if scalyr_region == 'eu':
scalyr_prefix = SCALYR_URL_PREFIX_EU
self.__query_url = '{}/query'.format(scalyr_prefix)
self.__numeric_url = '{}/numericQuery'.format(scalyr_prefix)
self.__timeseries_url = '{}/timeseriesQuery'.format(scalyr_prefix)
self.__facet_url = '{}/facetQuery'.format(scalyr_prefix)
if not read_key:
raise ConfigurationError('Scalyr read key is not set.')
self.__read_key = read_key
def __init__(self, jmxqueryhost, jmxqueryport, host, port, timeout=5):
if not jmxqueryhost or not jmxqueryport:
raise ConfigurationError(
'JMX wrapper improperly configured. Missing jmxqueryhost & jmxqueryport.'
)
# jmxquery running where?
self.jmxquery_host = jmxqueryhost
self.jmxquery_port = jmxqueryport
self.host = host
self.port = port
self.timeout = timeout
self._queries = []
def __init__(self, host, port, sid, user='******', password='', enabled=True):
self._stmt = None
self._dsn_tns = None
self._enabled = enabled
self.__cx_Oracle = None
self.__conn = None
self.__cursor = None
if not host:
raise ConfigurationError('Oracle wrapper improperly configured. Valid host is required!')
if self._enabled:
self.__cx_Oracle = _import_db_driver()
port = (int(port) if port and str(port).isdigit() else DEFAULT_PORT)
try:
self._dsn_tns = self.__cx_Oracle.makedsn(host, port, sid)
self.__conn = self.__cx_Oracle.connect(user, password, self._dsn_tns)
self.__cursor = self.__conn.cursor()
except Exception, e:
raise DbError(str(e), operation='Connect to dsn={}'.format(self._dsn_tns)), None, sys.exc_info()[2]
def __init__(self,
service_url,
infrastructure_account,
verify=True,
oauth2=False):
if not service_url:
raise ConfigurationError(
'EntitiesWrapper improperly configured. URL is missing!')
self.infrastructure_account = infrastructure_account
self.__service_url = urlparse.urljoin(service_url, 'api/v1/')
self.__session = requests.Session()
self.__session.headers.update({'User-Agent': get_user_agent()})
self.__session.verify = verify
if oauth2:
self.__session.headers.update(
{'Authorization': 'Bearer {}'.format(tokens.get('uid'))})
def __init__(self, url=None, check_id='', entities=None, oauth2=False):
if not url:
raise ConfigurationError(
'History wrapper improperly configured. URL is required.')
self.url = os.path.join(url, DATAPOINTS_ENDPOINT)
self.check_id = check_id
if not entities:
self.entities = []
elif type(entities) == list:
self.entities = entities
else:
self.entities = [entities]
self.__session = requests.Session()
self.__session.headers.update({'Content-Type': 'application/json'})
if oauth2:
self.__session.headers.update(
{'Authorization': 'Bearer {}'.format(tokens.get('uid'))})
def __init__(
self,
url,
method='GET',
params=None,
base_url=None,
timeout=10,
max_retries=0,
allow_redirects=None,
verify=True,
oauth2=False,
oauth2_token_name='uid',
headers=None,
):
if method.lower() not in ('get', 'head'):
raise CheckError(
'Invalid method. Only GET and HEAD are supported!')
if not base_url and not absolute_http_url(url):
# More verbose error message!
raise ConfigurationError(
'HTTP wrapper improperly configured. Invalid base_url. Check entity["url"] or call with absolute url.'
)
self.url = (base_url + url if not absolute_http_url(url) else url)
self.clean_url = None
self.params = params
self.timeout = timeout
self.max_retries = max_retries
self.verify = verify
self._headers = headers or {}
self.oauth2 = oauth2
self.oauth2_token_name = oauth2_token_name
self.__method = method.lower()
self.allow_redirects = True if allow_redirects is None else allow_redirects
if self.__method == 'head' and allow_redirects is None:
self.allow_redirects = False
self.__r = None
def __init__(self, bigquery_key, location='EU'):
self._location = location
if not bigquery_key:
raise ConfigurationError('Bigquery key (bigquery_key) is not set.')
self._bigquery_key = bigquery_key