From dcef22a46d6331befae31d51a977ed714868366f Mon Sep 17 00:00:00 2001
From: Goldy <153996346+g0ldyy@users.noreply.github.com>
Date: Sat, 13 Jul 2024 21:10:01 +0200
Subject: [PATCH] fix: divising by 0

---
 comet/utils/general.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/comet/utils/general.py b/comet/utils/general.py
index 237a151..f876f5d 100644
--- a/comet/utils/general.py
+++ b/comet/utils/general.py
@@ -202,6 +202,8 @@ def get_debrid_extension(debridService: str):
         debrid_extension = "TB"
     elif debridService == "debridlink":
         debrid_extension = "DL"
+    elif debridService == "offcloud":
+        debrid_extension = "OC"
 
     return debrid_extension
 
@@ -387,10 +389,10 @@ async def get_balanced_hashes(hashes: dict, config: dict):
             hashes_by_resolution[resolution_key] = []
         hashes_by_resolution[resolution_key].append(hash)
 
-    if max_results == 0:
-        return hashes_by_resolution
-
     total_resolutions = len(hashes_by_resolution)
+    if max_results == 0 or total_resolutions == 0:
+        return hashes_by_resolution
+    
     hashes_per_resolution = max_results // total_resolutions
     extra_hashes = max_results % total_resolutions