fixed usage

Author: khoroshevskyi

bedhost/_version.py

@@ -1 +1 @@
-__version__ = "0.10.4"
+__version__ = "0.10.5"

bedhost/helpers.py

@@ -94,6 +94,9 @@ def count_requests(
     def decorator(func):
         @wraps(func)
         async def wrapper(*args, **kwargs):
+
+            function_result = await func(*args, **kwargs)
+
             if event == "files":
                 file_path = kwargs.get("file_path")
                 if "bed" in file_path or "bigbed" in file_path.lower():
@@ -128,7 +131,7 @@ async def wrapper(*args, **kwargs):
                     usage_data.bedset_meta[bedset_id] = 1
             else:
                 raise ValueError(f"Unknown event type: {event}")
-            return await func(*args, **kwargs)
+            return function_result
 
         return wrapper
 

bedhost/main.py

@@ -172,11 +172,14 @@ def upload_usage():
             Upload usage data to the database and reset the usage data
             """
 
-            print("Running uploading of the usage")
+            _LOGGER.info("Running uploading of the usage")
             usage_data.date_to = datetime.datetime.now() + datetime.timedelta(
                 days=USAGE_RECORD_DAYS
             )
-            bbagent.add_usage(usage_data)
+            try:
+                bbagent.add_usage(usage_data)
+            except Exception as e:
+                _LOGGER.error(f"Error while uploading usage data: {e}")
 
             usage_data.bed_meta = {}
             usage_data.bedset_meta = {}