def authentication(self):
try:
if self.args['type'] == "API":
if self.auth['authentication_type'] == 'token':
url = "{0}{1}".format(self.auth['host'],
self.auth['token_url'])
auth = requests.post(url,
auth=HTTPBasicAuth(
username=self.auth['username'],
password=self.auth['password']),
verify=False)
return auth.json()
else:
return None
elif self.args['type'] == "Database":
conn = DatabaseUtil(self.auth['host'], self.auth['username'],
self.auth['password'],
self.auth['db_name'])
test_conn = conn.get_connection(api=True)
return conn
except Exception as err:
raise ValueError('Cannot connect to source {0} - {1}'.format(
self.args['type'], self.auth['host']))
class start_polling:
def __init__(self, poller_id=0):
self.poller_id = poller_id
# self.conn = DatabaseUtil(os.environ.get("DB_CONN"), os.environ.get("DB_USER"), os.environ.get("DB_PASSWORD"), os.environ.get("DB_NAME"))
try:
self.conn = DatabaseUtil(os.environ.get("DB_CONN"),
os.environ.get("DB_USER"),
os.environ.get("DB_PASSWORD"),
os.environ.get("DB_NAME"))
conn = self.conn.get_connection(interval=60)
conn.close()
except Exception as err:
print(err)
sys.exit()
self.logger = None
self.pid_util = None
atexit.register(self.exit_handler)
def poller_event(self, status):
status = int(status)
try:
self.conn.jinja_update_query(
sql_utils.sql_templates["update_event"].value, {
"status": status,
"id": self.poller_id
})
if status is not 1:
os.kill(os.getpid(), signal.SIGINT)
except Exception as err:
self.logger.log(
'An error has occured while updating poller status: {0}'.
format(err), 'critical')
sys.exit(1)
def get_poller_config(self):
sql_query = sql_utils.sql_templates["poller_config"].value
return self.conn.jinja_select_query(sql_query,
{'poll_id': self.poller_id})
def get_selected_oid(self, id, brand):
oid_list = {"oid_list": []}
oid_inner = dict()
sql_query = sql_utils.sql_templates["oid_config"].value
oid_raw = self.conn.jinja_select_query(sql_query, {
'poll_id': id,
'brand': brand
})
for key, value in enumerate(oid_raw):
oid_inner[value['oid_key']] = value['oid']
oid_list['oid_list'] = [oid_inner]
return oid_list
def get_ip_address(self, table):
ip_list = list()
raw_list = self.conn.select_query(
"""SELECT ip_address from {0}""".format(table))
for key, items in enumerate(raw_list):
ip_list.append(items["ip_address"])
return ip_list
def insert_update(self, device_detail, table_name):
ip_address = device_detail['ip_address']
del device_detail['ip_address']
placeholder = ', '.join('?' * len(device_detail))
update_util = sql_utils.sql_templates["poll_update_up"].value
update_query = update_util.format(
table_name,
', '.join("{0}='{1}'".format(value, device_detail[value])
for key, value in enumerate(device_detail)), ip_address,
self.poller_id)
self.conn.update_query(update_query)
def exit_handler(self):
if self.pid_util:
if str(self.pid_util.read_pid()) == str(os.getpid()):
try:
self.pid_util.delete_pid()
except Exception as err:
self.logger.log(err, 'critical')
def signal_handler(self, signum, frame):
self.exit_handler()
if self.logger:
self.logger.log('Service stopped.')
sys.exit(1)
def create_directory(self, directory):
if not os.path.exists(os.path.join(os.getcwd(), directory)):
os.mkdir(os.path.join(os.getcwd(), directory))
print(directory, 'Directory created.')
def run(self):
self.create_directory(logs_directory)
self.create_directory(pid_directory)
poller_config = self.get_poller_config()
if poller_config:
poller_config = poller_config[0]
subnet = poller_config["subnet"]
community_string = poller_config["community_string"]
interval = poller_config["interval"]
table_name = poller_config["table_name"]
poll_name = poller_config["poll_name"]
self.logger = Logger(logs_directory, self.poller_id, table_name,
'snmp_poller_logs')
self.pid_util = ProcessIdUtil(pid_directory, self.poller_id,
table_name, 'snmp_poller', None)
self.logger.config_logging()
if self.pid_util.is_process_running():
self.logger.log('Already running.')
sys.exit(1)
else:
self.logger.log("[{0}] : Poller running...".format(poll_name))
self.poller_event(1)
ip_list = self.get_ip_address(table_name)
try:
self.pid_util.create_pid()
self.pid_util.save_pid()
except Exception as err:
self.logger.log(err, 'critical')
sys.exit(1)
runner = True
while runner:
# start
for ip_address in ip_list:
mdi_runner = main_device_info(ip_address,
community_string)
mdi_output = mdi_runner.run()
if mdi_output["is_valid"]:
try:
mdi_data = mdi_output["main_info"]
for_mdd = self.get_selected_oid(
self.poller_id, mdi_data['brand'])
mdd_runner = main_device_details(
mdi_data["ip_address"], for_mdd,
community_string)
mdd_output = mdd_runner.run()
mdd_output["ip_address"] = mdi_data[
"ip_address"]
self.insert_update(mdd_output, table_name)
except Exception as err:
self.logger.log(
"[{0}] : Stopped due to an error : {1}".
format(poll_name, err))
self.poller_event('-1')
else:
update_util = sql_utils.sql_templates[
"poll_update_down"].value
update_query = update_util.format(
table_name, ip_address)
self.conn.update_query(update_query)
self.logger.log(
"[{0}] : No SNMP response for {1}".format(
poll_name, ip_address))
time.sleep(2)
#end
time.sleep(interval)
else:
print('Poller does not exist.')