RickRongen/WallE
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

feat: check if environ exists

Browse Source
This commit is contained in:
Rick Rongen
2020-08-27 17:22:10 +02:00
parent 24a36e156f
commit 408cda0aff
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
control/pca9685/pca9685.py
View File

@@ -17,6 +17,9 @@ import math
import os import os
MOCK_ENV = 'PCA9685_MOCK_SMBUS'
class MockSMBus: class MockSMBus:
def __init__(self, bus_number): def __init__(self, bus_number):
self.bus_number = bus_number self.bus_number = bus_number
@@ -72,7 +75,7 @@ class PWM:
self.bus_number = 1 self.bus_number = 1
else: else:
self.bus_number = bus_number self.bus_number = bus_number
if os.environ['PCA9685_MOCK_SMBUS'] == 'true': if MOCK_ENV in os.environ and os.environ[MOCK_ENV] == 'true':
self.bus = MockSMBus(self.bus_number) self.bus = MockSMBus(self.bus_number)
else: else:
self.bus = SMBus(self.bus_number) self.bus = SMBus(self.bus_number)