Update orangepizero2.py

[Adafruit_Blinka-hackapet.git] / src / adafruit_blinka / microcontroller / generic_linux / spi.py

diff --git a/src/adafruit_blinka/microcontroller/generic_linux/spi.py b/src/adafruit_blinka/microcontroller/generic_linux/spi.py
index 01b885bb50c35a629d72af9474767ced2b71d9c8..454909210377ae7cdb8dd215f9dfa3518958a726 100755 (executable)
--- a/src/adafruit_blinka/microcontroller/generic_linux/spi.py
+++ b/src/adafruit_blinka/microcontroller/generic_linux/spi.py
@@ -1,8 +1,11 @@
+"""Generic Linux SPI class using PureIO's SPI class"""

 import Adafruit_PureIO.spi as spi
 import Adafruit_PureIO.spi as spi

-import time

 from adafruit_blinka.agnostic import detector
 
 from adafruit_blinka.agnostic import detector
 

+

 class SPI:
 class SPI:

+    """SPI Class"""
+

     MSB = 0
     LSB = 1
     CPHA = 1
     MSB = 0
     LSB = 1
     CPHA = 1


@@ -17,9 +20,24 @@ class SPI:
             self._spi = spi.SPI(device=portid)
         else:
             self._spi = spi.SPI(device=(portid, 0))
             self._spi = spi.SPI(device=portid)
         else:
             self._spi = spi.SPI(device=(portid, 0))

+        self.clock_pin = None
+        self.mosi_pin = None
+        self.miso_pin = None
+        self.chip = None

 
 

-    def init(self, baudrate=100000, polarity=0, phase=0, bits=8,
-                  firstbit=MSB, sck=None, mosi=None, miso=None):
+    # pylint: disable=too-many-arguments,unused-argument
+    def init(
+        self,
+        baudrate=100000,
+        polarity=0,
+        phase=0,
+        bits=8,
+        firstbit=MSB,
+        sck=None,
+        mosi=None,
+        miso=None,
+    ):
+        """Initialize SPI"""

         mode = 0
         if polarity:
             mode |= self.CPOL
         mode = 0
         if polarity:
             mode |= self.CPOL


@@ -35,52 +53,66 @@ class SPI:
         self.mosi_pin = mosi
         self.miso_pin = miso
 
         self.mosi_pin = mosi
         self.miso_pin = miso
 

+    # pylint: enable=too-many-arguments,unused-argument
+
+    # pylint: disable=unnecessary-pass

     def set_no_cs(self):
     def set_no_cs(self):

+        """Setting so that SPI doesn't automatically set the CS pin"""

         # No kernel seems to support this, so we're just going to pass
         pass
 
         # No kernel seems to support this, so we're just going to pass
         pass
 

+    # pylint: enable=unnecessary-pass
+

     @property
     def frequency(self):
     @property
     def frequency(self):

+        """Return the current baudrate"""

         return self.baudrate
 
     def write(self, buf, start=0, end=None):
         return self.baudrate
 
     def write(self, buf, start=0, end=None):

+        """Write data from the buffer to SPI"""

         if not buf:
             return
         if end is None:
             end = len(buf)
         try:
         if not buf:
             return
         if end is None:
             end = len(buf)
         try:

-            #self._spi.open(self._port, 0)
+            # self._spi.open(self._port, 0)

             self.set_no_cs()
             self._spi.max_speed_hz = self.baudrate
             self._spi.mode = self.mode
             self._spi.bits_per_word = self.bits
             self._spi.writebytes(buf[start:end])
             self.set_no_cs()
             self._spi.max_speed_hz = self.baudrate
             self._spi.mode = self.mode
             self._spi.bits_per_word = self.bits
             self._spi.writebytes(buf[start:end])

-            #self._spi.close()
-        except FileNotFoundError as not_found:
+            # self._spi.close()
+        except FileNotFoundError:

             print("Could not open SPI device - check if SPI is enabled in kernel!")
             raise
 
     def readinto(self, buf, start=0, end=None, write_value=0):
             print("Could not open SPI device - check if SPI is enabled in kernel!")
             raise
 
     def readinto(self, buf, start=0, end=None, write_value=0):

+        """Read data from SPI and into the buffer"""

         if not buf:
             return
         if end is None:
             end = len(buf)
         try:
         if not buf:
             return
         if end is None:
             end = len(buf)
         try:

-            #self._spi.open(self._port, 0)
-            #self.set_no_cs()
+            # self._spi.open(self._port, 0)
+            # self.set_no_cs()

             self._spi.max_speed_hz = self.baudrate
             self._spi.mode = self.mode
             self._spi.bits_per_word = self.bits
             self._spi.max_speed_hz = self.baudrate
             self._spi.mode = self.mode
             self._spi.bits_per_word = self.bits

-            data = self._spi.transfer([write_value]*(end-start))
-            for i in range(end-start):  # 'readinto' the given buffer
-              buf[start+i] = data[i]
-            #self._spi.close()
-        except FileNotFoundError as not_found:
+            data = self._spi.transfer([write_value] * (end - start))
+            for i in range(end - start):  # 'readinto' the given buffer
+                buf[start + i] = data[i]
+            # self._spi.close()
+        except FileNotFoundError:

             print("Could not open SPI device - check if SPI is enabled in kernel!")
             raise
 
             print("Could not open SPI device - check if SPI is enabled in kernel!")
             raise
 

-    def write_readinto(self, buffer_out, buffer_in, out_start=0,
-                       out_end=None, in_start=0, in_end=None):
+    # pylint: disable=too-many-arguments
+    def write_readinto(
+        self, buffer_out, buffer_in, out_start=0, out_end=None, in_start=0, in_end=None
+    ):
+        """Perform a half-duplex write from buffer_out and then
+        read data into buffer_in
+        """

         if not buffer_out or not buffer_in:
             return
         if out_end is None:
         if not buffer_out or not buffer_in:
             return
         if out_end is None:


@@ -88,17 +120,19 @@ class SPI:
         if in_end is None:
             in_end = len(buffer_in)
         if out_end - out_start != in_end - in_start:
         if in_end is None:
             in_end = len(buffer_in)
         if out_end - out_start != in_end - in_start:

-            raise RuntimeError('Buffer slices must be of equal length.')
+            raise RuntimeError("Buffer slices must be of equal length.")

         try:
         try:

-            #self._spi.open(self._port, 0)
-            #self.set_no_cs()
+            # self._spi.open(self._port, 0)
+            # self.set_no_cs()

             self._spi.max_speed_hz = self.baudrate
             self._spi.mode = self.mode
             self._spi.bits_per_word = self.bits
             self._spi.max_speed_hz = self.baudrate
             self._spi.mode = self.mode
             self._spi.bits_per_word = self.bits

-            data = self._spi.transfer(list(buffer_out[out_start:out_end+1]))
+            data = self._spi.transfer(list(buffer_out[out_start : out_end + 1]))

             for i in range((in_end - in_start)):
             for i in range((in_end - in_start)):

-                buffer_in[i+in_start] = data[i]
-            #self._spi.close()
-        except FileNotFoundError as not_found:
+                buffer_in[i + in_start] = data[i]
+            # self._spi.close()
+        except FileNotFoundError:

             print("Could not open SPI device - check if SPI is enabled in kernel!")
             raise
             print("Could not open SPI device - check if SPI is enabled in kernel!")
             raise

+
+    # pylint: enable=too-many-arguments