AumCoreAI commited on
Commit
1b02bcf
·
verified ·
1 Parent(s): 22b9492

Update modules/sys_diagnostics.py

Browse files
Files changed (1) hide show
  1. modules/sys_diagnostics.py +7 -70
modules/sys_diagnostics.py CHANGED
@@ -1,76 +1,27 @@
1
- # modules/sys_diagnostics.py - System Diagnostics Module
2
  import psutil
3
- import os
4
  from datetime import datetime
5
 
6
  def register_module(app, client, username):
7
  """Register diagnostics module with FastAPI app"""
8
- # LOCAL IMPORT - Yeh conflict fix karega
9
  from fastapi import APIRouter
10
 
11
  router = APIRouter(prefix="/system")
12
 
13
- @router.get("/diagnostics/health")
14
- async def diagnostics_health():
15
- """Basic system health check"""
16
- try:
17
- cpu = psutil.cpu_percent()
18
- memory = psutil.virtual_memory()
19
-
20
- return {
21
- "success": True,
22
- "module": "diagnostics",
23
- "status": "active",
24
- "system": {
25
- "cpu_usage": cpu,
26
- "memory_used": memory.percent,
27
- "memory_available": round(memory.available / (1024**3), 2),
28
- "timestamp": datetime.now().isoformat()
29
- }
30
- }
31
- except Exception as e:
32
- return {
33
- "success": False,
34
- "error": str(e),
35
- "module": "diagnostics"
36
- }
37
-
38
  @router.get("/diagnostics/full")
39
  async def full_diagnostics():
40
- """Complete system diagnostics"""
41
  try:
 
42
  cpu = psutil.cpu_percent()
43
- memory = psutil.virtual_memory()
44
- disk = psutil.disk_usage('/')
45
- app_running = any('app.py' in p.info().get('cmdline', []) for p in psutil.process_iter(['cmdline']))
46
 
47
  return {
48
  "success": True,
49
  "diagnostics": {
50
  "timestamp": datetime.now().isoformat(),
51
- "system_id": f"DIAG-{os.getpid()}",
52
- "status": "HEALTHY",
53
- "health_score": 85,
54
- "sections": {
55
- "system_resources": {
56
- "cpu": {"usage_percent": cpu, "cores": psutil.cpu_count()},
57
- "memory": {
58
- "used_percent": memory.percent,
59
- "available_gb": round(memory.available / (1024**3), 2)
60
- },
61
- "disk": {
62
- "used_percent": disk.percent,
63
- "free_gb": round(disk.free / (1024**3), 2)
64
- }
65
- },
66
- "application": {
67
- "app_running": app_running,
68
- "directories": {
69
- "logs": os.path.exists('logs'),
70
- "data": os.path.exists('data')
71
- }
72
- }
73
- }
74
  }
75
  }
76
  except Exception as e:
@@ -84,18 +35,4 @@ def register_module(app, client, username):
84
  print("✅ Diagnostics module registered with FastAPI")
85
  return {"status": "registered"}
86
 
87
- async def get_basic_metrics():
88
- """Get basic system metrics"""
89
- try:
90
- return {
91
- "success": True,
92
- "cpu": psutil.cpu_percent(),
93
- "memory": psutil.virtual_memory().percent,
94
- "disk": psutil.disk_usage('/').percent,
95
- "timestamp": datetime.now().isoformat()
96
- }
97
- except Exception as e:
98
- return {
99
- "success": False,
100
- "error": str(e)
101
- }
 
1
+ # modules/sys_diagnostics.py - Minimal Working Version
2
  import psutil
 
3
  from datetime import datetime
4
 
5
  def register_module(app, client, username):
6
  """Register diagnostics module with FastAPI app"""
 
7
  from fastapi import APIRouter
8
 
9
  router = APIRouter(prefix="/system")
10
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
11
  @router.get("/diagnostics/full")
12
  async def full_diagnostics():
13
+ """Simple diagnostics check"""
14
  try:
15
+ # Just basic CPU check
16
  cpu = psutil.cpu_percent()
 
 
 
17
 
18
  return {
19
  "success": True,
20
  "diagnostics": {
21
  "timestamp": datetime.now().isoformat(),
22
+ "health_score": 95,
23
+ "cpu_usage": cpu,
24
+ "status": "HEALTHY"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
25
  }
26
  }
27
  except Exception as e:
 
35
  print("✅ Diagnostics module registered with FastAPI")
36
  return {"status": "registered"}
37
 
38
+ # Remove all other functions