Removing old logging

This commit is contained in:
samwiseg0 2018-12-04 19:16:00 -05:00
parent 50c1be7e41
commit aa04ea84b7
6 changed files with 19 additions and 20 deletions

View file

@ -12,6 +12,7 @@ from varken.tautulli import TautulliAPI
from varken.radarr import RadarrAPI from varken.radarr import RadarrAPI
from varken.ombi import OmbiAPI from varken.ombi import OmbiAPI
from varken.dbmanager import DBManager from varken.dbmanager import DBManager
from varken.varkenlogger import VarkenLogger
def threaded(job): def threaded(job):
thread = threading.Thread(target=job) thread = threading.Thread(target=job)
@ -19,6 +20,9 @@ def threaded(job):
if __name__ == "__main__": if __name__ == "__main__":
vl = VarkenLogger()
vl.logger.info('Starting Varken...')
parser = ArgumentParser(prog='varken', parser = ArgumentParser(prog='varken',
description='Command-line utility to aggregate data from the plex ecosystem into InfluxDB', description='Command-line utility to aggregate data from the plex ecosystem into InfluxDB',
formatter_class=RawTextHelpFormatter) formatter_class=RawTextHelpFormatter)
@ -84,4 +88,3 @@ if __name__ == "__main__":
while True: while True:
schedule.run_pending() schedule.run_pending()
sleep(1) sleep(1)

View file

@ -1,5 +1,8 @@
import logging
from influxdb import InfluxDBClient from influxdb import InfluxDBClient
logger = logging.getLogger('Varken')
class DBManager(object): class DBManager(object):
def __init__(self, server): def __init__(self, server):
@ -13,4 +16,5 @@ class DBManager(object):
self.influx.create_retention_policy('varken 30d/1h', '30d', '1', 'varken', False, '1h') self.influx.create_retention_policy('varken 30d/1h', '30d', '1', 'varken', False, '1h')
def write_points(self, data): def write_points(self, data):
logger.debug('Writing Data to InfluxDB {}'.format(data))
self.influx.write_points(data) self.influx.write_points(data)

View file

@ -3,17 +3,20 @@ import time
import tarfile import tarfile
import hashlib import hashlib
import geoip2.database import geoip2.database
import logging
from json.decoder import JSONDecodeError from json.decoder import JSONDecodeError
from os.path import abspath, join from os.path import abspath, join
from requests.exceptions import InvalidSchema, SSLError from requests.exceptions import InvalidSchema, SSLError
from urllib.request import urlretrieve from urllib.request import urlretrieve
logger = logging.getLogger('Varken')
def geoip_download(): def geoip_download():
tar_dbfile = abspath(join('.', 'data', 'GeoLite2-City.tar.gz')) tar_dbfile = abspath(join('.', 'data', 'GeoLite2-City.tar.gz'))
url = 'http://geolite.maxmind.com/download/geoip/database/GeoLite2-City.tar.gz' url = 'http://geolite.maxmind.com/download/geoip/database/GeoLite2-City.tar.gz'
urlretrieve(url, tar_dbfile) urlretrieve(url, tar_dbfile)
tar = tarfile.open(tar_dbfile, "r:gz") tar = tarfile.open(tar_dbfile, 'r:gz')
for files in tar.getmembers(): for files in tar.getmembers():
if 'GeoLite2-City.mmdb' in files.name: if 'GeoLite2-City.mmdb' in files.name:
files.name = os.path.basename(files.name) files.name = os.path.basename(files.name)
@ -56,20 +59,20 @@ def connection_handler(session, request, verify):
try: try:
get = s.send(r, verify=v) get = s.send(r, verify=v)
if get.status_code == 401: if get.status_code == 401:
print("Your api key is incorrect for {}".format(r.url)) logger.info('Your api key is incorrect for {}'.format(r.url))
elif get.status_code == 404: elif get.status_code == 404:
print("This url doesnt even resolve: {}".format(r.url)) logger.info('This url doesnt even resolve: {}'.format(r.url))
elif get.status_code == 200: elif get.status_code == 200:
try: try:
return_json = get.json() return_json = get.json()
except JSONDecodeError: except JSONDecodeError:
print("No JSON response... BORKED! Let us know in discord") logger.info('No JSON response... BORKED! Let us know in discord')
except InvalidSchema: except InvalidSchema:
print("You added http(s):// in the config file. Don't do that.") logger.info('You added http(s):// in the config file. Don't do that.')
except SSLError as e: except SSLError as e:
print("Either your host is unreachable or you have an ssl issue.") logger.info('Either your host is unreachable or you have an ssl issue.')
print("The issue was: {}".format(e)) logger.info('The issue was: {}'.format(e))
return return_json return return_json

View file

@ -1,7 +1,6 @@
from requests import Session, Request from requests import Session, Request
from datetime import datetime, timezone from datetime import datetime, timezone
from varken.logger import logging
from varken.helpers import connection_handler from varken.helpers import connection_handler
from varken.structures import OmbiRequestCounts from varken.structures import OmbiRequestCounts
@ -15,7 +14,6 @@ class OmbiAPI(object):
self.session = Session() self.session = Session()
self.session.headers = {'Apikey': self.server.api_key} self.session.headers = {'Apikey': self.server.api_key}
@logging
def get_total_requests(self): def get_total_requests(self):
self.now = datetime.now(timezone.utc).astimezone().isoformat() self.now = datetime.now(timezone.utc).astimezone().isoformat()
tv_endpoint = '/api/v1/Request/tv' tv_endpoint = '/api/v1/Request/tv'
@ -49,7 +47,6 @@ class OmbiAPI(object):
self.dbmanager.write_points(influx_payload) self.dbmanager.write_points(influx_payload)
@logging
def get_request_counts(self): def get_request_counts(self):
self.now = datetime.now(timezone.utc).astimezone().isoformat() self.now = datetime.now(timezone.utc).astimezone().isoformat()
endpoint = '/api/v1/Request/count' endpoint = '/api/v1/Request/count'

View file

@ -1,7 +1,6 @@
from requests import Session, Request from requests import Session, Request
from datetime import datetime, timezone from datetime import datetime, timezone
from varken.logger import logging
from varken.helpers import hashit, connection_handler from varken.helpers import hashit, connection_handler
from varken.structures import Movie, Queue from varken.structures import Movie, Queue
@ -15,7 +14,6 @@ class RadarrAPI(object):
self.session = Session() self.session = Session()
self.session.headers = {'X-Api-Key': self.server.api_key} self.session.headers = {'X-Api-Key': self.server.api_key}
@logging
def get_missing(self): def get_missing(self):
endpoint = '/api/movie' endpoint = '/api/movie'
self.now = datetime.now(timezone.utc).astimezone().isoformat() self.now = datetime.now(timezone.utc).astimezone().isoformat()
@ -60,7 +58,6 @@ class RadarrAPI(object):
self.dbmanager.write_points(influx_payload) self.dbmanager.write_points(influx_payload)
@logging
def get_queue(self): def get_queue(self):
endpoint = '/api/queue' endpoint = '/api/queue'
self.now = datetime.now(timezone.utc).astimezone().isoformat() self.now = datetime.now(timezone.utc).astimezone().isoformat()

View file

@ -1,11 +1,9 @@
from requests import Session, Request from requests import Session, Request
from datetime import datetime, timezone, date, timedelta from datetime import datetime, timezone, date, timedelta
from varken.logger import logging
from varken.helpers import hashit, connection_handler from varken.helpers import hashit, connection_handler
from varken.structures import Queue, TVShow from varken.structures import Queue, TVShow
class SonarrAPI(object): class SonarrAPI(object):
def __init__(self, server, dbmanager): def __init__(self, server, dbmanager):
# Set Time of initialization # Set Time of initialization
@ -18,7 +16,6 @@ class SonarrAPI(object):
self.session.headers = {'X-Api-Key': self.server.api_key} self.session.headers = {'X-Api-Key': self.server.api_key}
self.session.params = {'pageSize': 1000} self.session.params = {'pageSize': 1000}
@logging
def get_missing(self): def get_missing(self):
endpoint = '/api/calendar' endpoint = '/api/calendar'
last_days = str(date.today() + timedelta(days=-self.server.missing_days)) last_days = str(date.today() + timedelta(days=-self.server.missing_days))
@ -66,7 +63,6 @@ class SonarrAPI(object):
self.dbmanager.write_points(influx_payload) self.dbmanager.write_points(influx_payload)
@logging
def get_future(self): def get_future(self):
endpoint = '/api/calendar/' endpoint = '/api/calendar/'
self.now = datetime.now(timezone.utc).astimezone().isoformat() self.now = datetime.now(timezone.utc).astimezone().isoformat()
@ -115,7 +111,6 @@ class SonarrAPI(object):
self.dbmanager.write_points(influx_payload) self.dbmanager.write_points(influx_payload)
@logging
def get_queue(self): def get_queue(self):
influx_payload = [] influx_payload = []
endpoint = '/api/queue' endpoint = '/api/queue'