laipeng/reflex
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

opt for powershell as opposed to wmic

Browse Source
This commit is contained in:
Khaleel Al-Adhami 2025-02-22 09:55:02 -08:00
parent 98f50811f9
commit 35ad1c9d73
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
reflex/utils/prerequisites.py
View File

@ -1936,13 +1936,13 @@ def get_cpu_info() -> CpuInfo | None:
cpuinfo = {} cpuinfo = {}
try: try:
if platform_os == "Windows": if platform_os == "Windows":
cmd = "wmic cpu get addresswidth,caption,manufacturer /FORMAT:csv" cmd = 'powershell -Command "Get-CimInstance Win32_Processor | Select-Object AddressWidth,Manufacturer,Name | ConvertTo-Json"'
output = processes.execute_command_and_return_output(cmd) output = processes.execute_command_and_return_output(cmd)
if output: if output:
val = output.splitlines()[-1].split(",")[1:] cpu_data = json.loads(output)
cpuinfo["manufacturer_id"] = val[2] cpuinfo["address_width"] = cpu_data["AddressWidth"]
cpuinfo["model_name"] = val[1].split("Family")[0].strip() cpuinfo["manufacturer_id"] = cpu_data["Manufacturer"]
cpuinfo["address_width"] = format_address_width(val[0]) cpuinfo["model_name"] = cpu_data["Name"]
elif platform_os == "Linux": elif platform_os == "Linux":
output = processes.execute_command_and_return_output("lscpu") output = processes.execute_command_and_return_output("lscpu")
if output: if output:
@ -1968,6 +1968,7 @@ def get_cpu_info() -> CpuInfo | None:
) )
except Exception as err: except Exception as err:
console.error(f"Failed to retrieve CPU info. {err}") console.error(f"Failed to retrieve CPU info. {err}")
raise typer.Exit(1) from err # TODO: REMOVE THIS AFTER MAKING SURE IT WORKS
return None return None
return ( return (