remove cisco module and all refreneces

This commit is contained in:
Nicholas St. Germain 2019-01-05 15:45:31 -06:00
parent 5ba5e6eda1
commit 9fbd9c1ca0
5 changed files with 4 additions and 101 deletions

View file

@ -12,7 +12,6 @@ from logging import getLogger, StreamHandler, Formatter, DEBUG
from varken.ombi import OmbiAPI from varken.ombi import OmbiAPI
from varken.unifi import UniFiAPI from varken.unifi import UniFiAPI
from varken.cisco import CiscoAPI
from varken import VERSION, BRANCH from varken import VERSION, BRANCH
from varken.sonarr import SonarrAPI from varken.sonarr import SonarrAPI
from varken.radarr import RadarrAPI from varken.radarr import RadarrAPI
@ -131,11 +130,6 @@ if __name__ == "__main__":
if server.get_missing: if server.get_missing:
schedule.every(server.get_missing_run_seconds).seconds.do(threaded, SICKCHILL.get_missing) schedule.every(server.get_missing_run_seconds).seconds.do(threaded, SICKCHILL.get_missing)
if CONFIG.ciscoasa_enabled:
for firewall in CONFIG.ciscoasa_servers:
ASA = CiscoAPI(firewall, DBMANAGER)
schedule.every(firewall.get_bandwidth_run_seconds).seconds.do(threaded, ASA.get_bandwidth)
if CONFIG.unifi_enabled: if CONFIG.unifi_enabled:
for server in CONFIG.unifi_servers: for server in CONFIG.unifi_servers:
UNIFI = UniFiAPI(server, DBMANAGER) UNIFI = UniFiAPI(server, DBMANAGER)
@ -143,7 +137,7 @@ if __name__ == "__main__":
# Run all on startup # Run all on startup
SERVICES_ENABLED = [CONFIG.ombi_enabled, CONFIG.radarr_enabled, CONFIG.tautulli_enabled, CONFIG.unifi_enabled, SERVICES_ENABLED = [CONFIG.ombi_enabled, CONFIG.radarr_enabled, CONFIG.tautulli_enabled, CONFIG.unifi_enabled,
CONFIG.sonarr_enabled, CONFIG.ciscoasa_enabled, CONFIG.sickchill_enabled] CONFIG.sonarr_enabled, CONFIG.sickchill_enabled]
if not [enabled for enabled in SERVICES_ENABLED if enabled]: if not [enabled for enabled in SERVICES_ENABLED if enabled]:
vl.logger.error("All services disabled. Exiting") vl.logger.error("All services disabled. Exiting")
exit(1) exit(1)

View file

@ -3,7 +3,6 @@ sonarr_server_ids = 1,2
radarr_server_ids = 1,2 radarr_server_ids = 1,2
tautulli_server_ids = 1 tautulli_server_ids = 1
ombi_server_ids = 1 ombi_server_ids = 1
ciscoasa_server_ids = false
sickchill_server_ids = false sickchill_server_ids = false
unifi_server_ids = false unifi_server_ids = false
@ -88,15 +87,6 @@ verify_ssl = false
get_missing = true get_missing = true
get_missing_run_seconds = 300 get_missing_run_seconds = 300
[ciscoasa-1]
url = firewall.domain.tld
username = cisco
password = cisco
outside_interface = WAN
ssl = false
verify_ssl = false
get_bandwidth_run_seconds = 300
[unifi-1] [unifi-1]
url = unifi.domain.tld:8443 url = unifi.domain.tld:8443
username = ubnt username = ubnt

View file

@ -1,60 +0,0 @@
from logging import getLogger
from requests import Session, Request
from datetime import datetime, timezone
from varken.helpers import connection_handler
class CiscoAPI(object):
def __init__(self, firewall, dbmanager):
self.dbmanager = dbmanager
self.firewall = firewall
# Create session to reduce server web thread load, and globally define pageSize for all requests
self.session = Session()
self.session.auth = (self.firewall.username, self.firewall.password)
self.logger = getLogger()
self.get_token()
def __repr__(self):
return f"<ciscoasa-{self.firewall.id}>"
def get_token(self):
endpoint = '/api/tokenservices'
req = self.session.prepare_request(Request('POST', self.firewall.url + endpoint))
post = connection_handler(self.session, req, self.firewall.verify_ssl)
if not post:
return
self.session.headers = {'X-Auth-Token': post}
def get_bandwidth(self):
now = datetime.now(timezone.utc).astimezone().isoformat()
endpoint = '/api/monitoring/device/interfaces/' + self.firewall.outside_interface
if not self.session.headers:
return
req = self.session.prepare_request(Request('GET', self.firewall.url + endpoint))
get = connection_handler(self.session, req, self.firewall.verify_ssl)
if not get:
return
influx_payload = [
{
"measurement": "Cisco ASA",
"tags": {
"interface": self.firewall.outside_interface
},
"time": now,
"fields": {
"upload_bitrate": get['outputBitRate'],
"download_bitrate": get['inputBitRate']
}
}
]
self.dbmanager.write_points(influx_payload)

View file

@ -7,7 +7,7 @@ from configparser import ConfigParser, NoOptionError, NoSectionError
from varken.varkenlogger import BlacklistFilter from varken.varkenlogger import BlacklistFilter
from varken.structures import SickChillServer, UniFiServer from varken.structures import SickChillServer, UniFiServer
from varken.helpers import clean_sid_check, rfc1918_ip_check from varken.helpers import clean_sid_check, rfc1918_ip_check
from varken.structures import SonarrServer, RadarrServer, OmbiServer, TautulliServer, InfluxServer, CiscoASAFirewall from varken.structures import SonarrServer, RadarrServer, OmbiServer, TautulliServer, InfluxServer
class INIParser(object): class INIParser(object):
@ -15,7 +15,7 @@ class INIParser(object):
self.config = None self.config = None
self.data_folder = data_folder self.data_folder = data_folder
self.filtered_strings = None self.filtered_strings = None
self.services = ['sonarr', 'radarr', 'ombi', 'tautulli', 'sickchill', 'ciscoasa', 'unifi'] self.services = ['sonarr', 'radarr', 'ombi', 'tautulli', 'sickchill', 'unifi']
self.logger = getLogger() self.logger = getLogger()
self.influx_server = InfluxServer() self.influx_server = InfluxServer()
@ -248,20 +248,9 @@ class INIParser(object):
verify_ssl=verify_ssl, get_missing=get_missing, verify_ssl=verify_ssl, get_missing=get_missing,
get_missing_run_seconds=get_missing_run_seconds) get_missing_run_seconds=get_missing_run_seconds)
if service in ['ciscoasa', 'unifi']: if service == 'unifi':
username = self.config.get(section, 'username') username = self.config.get(section, 'username')
password = self.config.get(section, 'password') password = self.config.get(section, 'password')
if service == 'ciscoasa':
outside_interface = self.config.get(section, 'outside_interface')
get_bandwidth_run_seconds = self.config.getint(section, 'get_bandwidth_run_seconds')
server = CiscoASAFirewall(id=server_id, url=scheme + url, verify_ssl=verify_ssl,
username=username, password=password,
outside_interface=outside_interface,
get_bandwidth_run_seconds=get_bandwidth_run_seconds)
if service == 'unifi':
site = self.config.get(section, 'site').lower() site = self.config.get(section, 'site').lower()
usg_name = self.config.get(section, 'usg_name') usg_name = self.config.get(section, 'usg_name')
get_usg_stats_run_seconds = self.config.getint(section, 'get_usg_stats_run_seconds') get_usg_stats_run_seconds = self.config.getint(section, 'get_usg_stats_run_seconds')

View file

@ -76,16 +76,6 @@ class SickChillServer(NamedTuple):
verify_ssl: bool = False verify_ssl: bool = False
class CiscoASAFirewall(NamedTuple):
get_bandwidth_run_seconds: int = 30
id: int = None
outside_interface: str = None
password: str = 'cisco'
url: str = '192.168.1.1'
username: str = 'cisco'
verify_ssl: bool = False
class UniFiServer(NamedTuple): class UniFiServer(NamedTuple):
get_usg_stats_run_seconds: int = 30 get_usg_stats_run_seconds: int = 30
id: int = None id: int = None