noham/prometheus-qbittorrent-exporter
1
0
Fork 0
mirror of https://github.com/NohamR/prometheus-qbittorrent-exporter.git synced 2025-05-23 16:49:29 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #20 from jsawatzky/master

Browse Source 
Return metrics on connection error
This commit is contained in:
Esteban Sánchez 2023-05-09 15:52:59 +02:00 committed by GitHub
commit 940b5c6242
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
qbittorrent_exporter/exporter.py
View File

@ -29,7 +29,6 @@ class QbittorrentMetricsCollector():
def __init__(self, config):
self.config = config
self.torrents = None
self.client = Client(
host=config["host"],
port=config["port"],
@ -38,12 +37,6 @@ class QbittorrentMetricsCollector():
)
def collect(self):
try:
self.torrents = self.client.torrents.info()
except Exception as e:
logger.error(f"Couldn't get server info: {e}")
return None
metrics = self.get_qbittorrent_metrics()
for metric in metrics:
@ -75,11 +68,8 @@ class QbittorrentMetricsCollector():
try:
response = self.client.transfer.info
version = self.client.app.version
self.torrents = self.client.torrents.info()
except APIConnectionError as e:
logger.error(f"Couldn't get server info: {e.error_message}")
except Exception:
logger.error(f"Couldn't get server info")
except Exception as e:
logger.error(f"Couldn't get server info: {e}")
return [
{
@ -120,17 +110,15 @@ class QbittorrentMetricsCollector():
def get_qbittorrent_torrent_tags_metrics(self):
try:
categories = self.client.torrent_categories.categories
torrents = self.client.torrents.info()
except Exception as e:
logger.error(f"Couldn't fetch categories: {e}")
return []
if not self.torrents:
logger.error(f"Couldn't fetch torrent info: {e}")
return []
metrics = []
categories.Uncategorized = AttrDict({'name': 'Uncategorized', 'savePath': ''})
for category in categories:
category_torrents = [t for t in self.torrents if t['category'] == category or (category == "Uncategorized" and t['category'] == "")]
category_torrents = [t for t in torrents if t['category'] == category or (category == "Uncategorized" and t['category'] == "")]
for status in self.TORRENT_STATUSES:
status_prop = f"is_{status}"
@ -211,7 +199,7 @@ def main():
logger.error("No host specified, please set QBITTORRENT_HOST environment variable")
sys.exit(1)
if not config["port"]:
logger.error("No post specified, please set QBITTORRENT_PORT environment variable")
logger.error("No port specified, please set QBITTORRENT_PORT environment variable")
sys.exit(1)
# Register our custom collector