Allow selection of i2c bus for older (rev 1) Pi

Extend __init__ method with default bus number and set in the init method.
sslayer · Nov 25, 2017 · 31a409c · 31a409c
1 parent b0e43ed
commit 31a409c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/mpu6050/mpu6050.py b/mpu6050/mpu6050.py
@@ -12,7 +12,7 @@ class mpu6050:
     # Global Variables
     GRAVITIY_MS2 = 9.80665
     address = None
-    bus = smbus.SMBus(1)
+    bus = None
 
     # Scale Modifiers
     ACCEL_SCALE_MODIFIER_2G = 16384.0
@@ -53,9 +53,9 @@ class mpu6050:
     ACCEL_CONFIG = 0x1C
     GYRO_CONFIG = 0x1B
 
-    def __init__(self, address):
+    def __init__(self, address, bus=1):
         self.address = address
-
+        self.bus = smbus.SMBus(bus)
         # Wake up the MPU-6050 since it starts in sleep mode
         self.bus.write_byte_data(self.address, self.PWR_MGMT_1, 0x00)