add Uploading sensor

chris-mc1 · Dec 30, 2023 · 8c8dcfc · 8c8dcfc
1 parent 461b4a8
commit 8c8dcfc
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 5 deletions.
diff --git a/custom_components/qbittorrent_alt/coordinator.py b/custom_components/qbittorrent_alt/coordinator.py
@@ -40,6 +40,7 @@ async def _async_update_data(self) -> dict[str, Any]:
             seeding = 0
             paused = 0
             stalled = 0
+            uploading = 0
             longest_eta = 0
             for torrent in main_data.torrents.values():
                 if (
@@ -48,12 +49,10 @@ async def _async_update_data(self) -> dict[str, Any]:
                     or torrent["state"] == "forcedDL"
                 ):
                     downloading += 1
-                if (
-                    torrent["state"] == "uploading"
-                    or torrent["state"] == "stalledUP"
-                    or torrent["state"] == "forcedUP"
-                ):
+                if torrent["state"] == "stalledUP" or torrent["state"] == "forcedUP":
                     seeding += 1
+                if torrent["state"] == "uploading":
+                    uploading += 1
                 if torrent["state"] == "pausedUP" or torrent["state"] == "pausedDL":
                     paused += 1
                 if torrent["state"] == "stalledDL":
@@ -65,6 +64,7 @@ async def _async_update_data(self) -> dict[str, Any]:
                 "preferences": await self.client.app.preferences(),
                 "downloading": downloading,
                 "seeding": seeding,
+                "uploading": uploading,
                 "paused": paused,
                 "stalled": stalled,
                 "total": len(main_data.torrents),

diff --git a/custom_components/qbittorrent_alt/sensor.py b/custom_components/qbittorrent_alt/sensor.py
@@ -168,6 +168,13 @@ class QBittorrentSensorEntityDescription(SensorEntityDescription, QBittorrentMix
         state_class=SensorStateClass.MEASUREMENT,
         value_fn=lambda coordinator: coordinator.data["seeding"],
     ),
+    QBittorrentSensorEntityDescription(
+        key="uploading",
+        name="Uploading",
+        icon="mdi:upload",
+        state_class=SensorStateClass.MEASUREMENT,
+        value_fn=lambda coordinator: coordinator.data["uploading"],
+    ),
     QBittorrentSensorEntityDescription(
         key="paused",
         name="Paused",