Added some basic/crude graphics checks in hardware.py

This commit is contained in:
Anton Hvornum 2021-01-25 10:42:02 +01:00
parent 68adb3108f
commit 7358dc5a03
2 changed files with 28 additions and 1 deletions

1
.gitignore vendored
View File

@ -20,3 +20,4 @@ SAFETY_LOCK
**/**.network
**/**.target
**/**.qcow2
**/test.py

View File

@ -1,4 +1,5 @@
import os
from .general import sys_command
from .networking import list_interfaces, enrichIfaceTypes
def hasWifi():
@ -7,4 +8,29 @@ def hasWifi():
return False
def hasUEFI():
return os.path.isdir('/sys/firmware/efi')
return os.path.isdir('/sys/firmware/efi')
def graphicsDevices():
cards = {}
for line in sys_command(f"lspci"):
if b' VGA ' in line:
_, identifier = line.split(b': ',1)
cards[identifier.strip().lower().decode('UTF-8')] = line
return cards
def hasNvidiaGraphics():
if [x for x in graphicsDevices() if 'nvidia' in x]:
return True
return False
def hasAmdGraphics():
if [x for x in graphicsDevices() if 'amd' in x]:
return True
return False
def hasIntelGraphics():
if [x for x in graphicsDevices() if 'intel' in x]:
return True
return False
# TODO: Add more identifiers