Skip to content

Commit

Permalink
Merge pull request #31 from Upsonic/system_information1
Browse files Browse the repository at this point in the history
feat: Added system diagnostic function
  • Loading branch information
onuratakan authored Oct 26, 2024
2 parents 426ae0e + 170c909 commit aa6713e
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 aa6713e

Please sign in to comment.