Преглед на файлове

Raspberry pi check in setup.py as well

mendhak преди 2 години
родител
ревизия
2587516a8a
променени са 1 файла, в които са добавени 13 реда и са изтрити 1 реда
  1. 13 1
      RaspberryPi_JetsonNano/python/setup.py

+ 13 - 1
RaspberryPi_JetsonNano/python/setup.py

@@ -1,9 +1,21 @@
 import sys, os
 from setuptools import setup
+from pathlib import Path
+import re
 
 dependencies = ['Pillow']
 
-if os.path.exists('/sys/bus/platform/drivers/gpiomem-bcm2835'):
+def is_raspberry_pi():
+    # https://raspberrypi.stackexchange.com/a/139704/540
+    CPUINFO_PATH = Path("/proc/cpuinfo")
+
+    if not CPUINFO_PATH.exists():
+        return False
+    with open(CPUINFO_PATH) as f:
+        cpuinfo = f.read()
+    return re.search(r"^Model\s*:\s*Raspberry Pi", cpuinfo, flags=re.M) is not None
+
+if is_raspberry_pi():
     dependencies += ['RPi.GPIO', 'spidev']
 elif os.path.exists('/sys/bus/platform/drivers/gpio-x3'):
     dependencies += ['Hobot.GPIO', 'spidev']