瀏覽代碼

Merge pull request #149 from missionfloyd/missionfloyd-install_requires

Add dependencies to setup.py
SSYYL 4 年之前
父節點
當前提交
af9694c602
共有 1 個文件被更改,包括 10 次插入1 次删除
  1. 10 1
      RaspberryPi_JetsonNano/python/setup.py

+ 10 - 1
RaspberryPi_JetsonNano/python/setup.py

@@ -1,10 +1,19 @@
-import sys
+import sys, os
 from setuptools import setup
+
+dependencies = ['Pillow', 'numpy']
+
+if os.path.exists('/sys/bus/platform/drivers/gpiomem-bcm2835'):
+    dependencies += ['RPi.GPIO', 'spidev']
+else:
+    dependencies += ['Jetson.GPIO']
+
 setup(
     name='waveshare-epd',
     description='Waveshare e-Paper Display',
     author='Waveshare',
     package_dir={'': 'lib'},
     packages=['waveshare_epd'],
+    install_requires=dependencies,
 )