Skip to content

Commit

Permalink
feat: Added system diagnostic function
Browse files Browse the repository at this point in the history
  • Loading branch information
hazargurbuzz committed Oct 26, 2024
1 parent c8452e4 commit 170c909
Showing 1 changed file with 12 additions and 0 deletions.
12 changes: 12 additions & 0 deletions upsonic/remote/on_prem.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,8 @@
import importlib.util
import cloudpickle

import psutil

from contextlib import contextmanager

import sys
Expand Down Expand Up @@ -303,6 +305,16 @@ def get_username(self):
except Exception as e:
return "to Upsonic"

def system_diagnostic(self):
diagnostic_data = {
"System Type": platform.system(),
"User Name": self.get_username(),
"Uptime (seconds)": round(time.time() - psutil.boot_time(), 2),
"CPU Usage (%)": psutil.cpu_percent(interval=1),
"RAM Usage (%)": psutil.virtual_memory().percent,
}
return json.dumps(diagnostic_data, indent=4)

def get_specific_version(self, package: str) -> int:
package_name = package.split("==")[0]
package_version = (
Expand Down

0 comments on commit 170c909

Please sign in to comment.