Add memory profiling and improve code formatting in asgi.py and core.py

blt/asgi.py

@@ -1,6 +1,7 @@
 # asgi.py
 
 import os
+import tracemalloc
 
 import django
 
@@ -12,6 +13,8 @@
 from django.core.asgi import get_asgi_application
 from django.urls import path
 
+tracemalloc.start()
+
 from website import consumers  # You will define a consumer for handling WebSockets
 
 application = ProtocolTypeRouter(

website/views/core.py

@@ -61,19 +61,21 @@
 # ----------------------------------------------------------------------------------
 
 
-def memory_usage_by_module(limit=10):
+def memory_usage_by_module(limit=100):
     """
     Returns a list of (filename, size_in_bytes) for the top
     `limit` files by allocated memory, using tracemalloc.
     """
-    tracemalloc.start()
+    # tracemalloc.start()
     snapshot = tracemalloc.take_snapshot()
+    print("Memory snapshot taken. and it is: ", snapshot)
 
     # Group memory usage by filename
     stats = snapshot.statistics("filename")
     module_usage = {}
 
     for stat in stats:
+        print("stat is: ", stat)
         if stat.traceback:
             filename = stat.traceback[0].filename
             # Accumulate memory usage
@@ -265,7 +267,7 @@ def check_status(request):
         # Memory usage by module (via tracemalloc)
         # -------------------------------------------------------
         print("Calculating memory usage by module...")
-        top_modules = memory_usage_by_module(limit=10)
+        top_modules = memory_usage_by_module(limit=100)
         status_data["memory_by_module"] = top_modules
 
         # -------------------------------------------------------
@@ -281,8 +283,11 @@ def check_status(request):
         # Redis stats
         # -------------------------------------------------------
         print("Getting Redis stats...")
-        redis_client = get_redis_connection("default")
-        status_data["redis_stats"] = redis_client.info()
+        try:
+            redis_client = get_redis_connection("default")
+            status_data["redis_stats"] = redis_client.info()
+        except Exception as e:
+            print(f"Redis error or not supported: {e}")
 
         # -------------------------------------------------------
         # Memory profiling info (current, peak) - optional