Initial addition

Author: dastels

Twitter_API/binascii.py

@@ -0,0 +1,99 @@
+
+PAD = '='
+
+table_a2b_base64 = [
+    -1,-1,-1,-1, -1,-1,-1,-1, -1,-1,-1,-1, -1,-1,-1,-1,
+    -1,-1,-1,-1, -1,-1,-1,-1, -1,-1,-1,-1, -1,-1,-1,-1,
+    -1,-1,-1,-1, -1,-1,-1,-1, -1,-1,-1,62, -1,-1,-1,63,
+    52,53,54,55, 56,57,58,59, 60,61,-1,-1, -1,-1,-1,-1, # Note PAD->-1 here
+    -1, 0, 1, 2,  3, 4, 5, 6,  7, 8, 9,10, 11,12,13,14,
+    15,16,17,18, 19,20,21,22, 23,24,25,-1, -1,-1,-1,-1,
+    -1,26,27,28, 29,30,31,32, 33,34,35,36, 37,38,39,40,
+    41,42,43,44, 45,46,47,48, 49,50,51,-1, -1,-1,-1,-1,
+    -1,-1,-1,-1, -1,-1,-1,-1, -1,-1,-1,-1, -1,-1,-1,-1,
+    -1,-1,-1,-1, -1,-1,-1,-1, -1,-1,-1,-1, -1,-1,-1,-1,
+    -1,-1,-1,-1, -1,-1,-1,-1, -1,-1,-1,-1, -1,-1,-1,-1,
+    -1,-1,-1,-1, -1,-1,-1,-1, -1,-1,-1,-1, -1,-1,-1,-1,
+    -1,-1,-1,-1, -1,-1,-1,-1, -1,-1,-1,-1, -1,-1,-1,-1,
+    -1,-1,-1,-1, -1,-1,-1,-1, -1,-1,-1,-1, -1,-1,-1,-1,
+    -1,-1,-1,-1, -1,-1,-1,-1, -1,-1,-1,-1, -1,-1,-1,-1,
+    -1,-1,-1,-1, -1,-1,-1,-1, -1,-1,-1,-1, -1,-1,-1,-1,
+]
+def _transform(n):
+    if n == -1:
+        return '\xff'
+    else:
+        return chr(n)
+table_a2b_base64 = ''.join(map(_transform, table_a2b_base64))
+assert len(table_a2b_base64) == 256
+
+def a2b_base64(ascii):
+    "Decode a line of base64 data."
+
+    res = []
+    quad_pos = 0
+    leftchar = 0
+    leftbits = 0
+    last_char_was_a_pad = False
+
+    for c in ascii:
+        c = chr(c)
+        if c == PAD:
+            if quad_pos > 2 or (quad_pos == 2 and last_char_was_a_pad):
+                break      # stop on 'xxx=' or on 'xx=='
+            last_char_was_a_pad = True
+        else:
+            n = ord(table_a2b_base64[ord(c)])
+            if n == 0xff:
+                continue    # ignore strange characters
+            #
+            # Shift it in on the low end, and see if there's
+            # a byte ready for output.
+            quad_pos = (quad_pos + 1) & 3
+            leftchar = (leftchar << 6) | n
+            leftbits += 6
+            #
+            if leftbits >= 8:
+                leftbits -= 8
+                res.append((leftchar >> leftbits).to_bytes(1, 'big'))
+                leftchar &= ((1 << leftbits) - 1)
+            #
+            last_char_was_a_pad = False
+    else:
+        if leftbits != 0:
+            raise Exception("Incorrect padding")
+
+    return b''.join(res)
+
+# ____________________________________________________________
+
+table_b2a_base64 = (
+    "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/")
+
+def b2a_base64(bin_data):
+    "Base64-code line of data."
+
+    newlength = (len(bin_data) + 2) // 3
+    newlength = newlength * 4 + 1
+    res = []
+
+    leftchar = 0
+    leftbits = 0
+    for c in bin_data:
+        # Shift into our buffer, and output any 6bits ready
+        leftchar = (leftchar << 8) | c
+        leftbits += 8
+        res.append(table_b2a_base64[(leftchar >> (leftbits-6)) & 0x3f])
+        leftbits -= 6
+        if leftbits >= 6:
+            res.append(table_b2a_base64[(leftchar >> (leftbits-6)) & 0x3f])
+            leftbits -= 6
+    #
+    if leftbits == 2:
+        res.append(table_b2a_base64[(leftchar & 3) << 4])
+        res.append(PAD)
+        res.append(PAD)
+    elif leftbits == 4:
+        res.append(table_b2a_base64[(leftchar & 0xf) << 2])
+        res.append(PAD)
+    return ''.join(res).encode('ascii')

Twitter_API/code.py

@@ -0,0 +1,134 @@
+"""
+Halloween countdown for PyPortal.
+
+Adafruit invests time and resources providing this open source code.
+Please support Adafruit and open source hardware by purchasing
+products from Adafruit!
+
+Written by Dave Astels for Adafruit Industries
+Copyright (c) 2019 Adafruit Industries
+Licensed under the MIT license.
+
+All text above must be included in any redistribution.
+"""
+
+import time
+import random
+import board
+import busio
+import binascii
+import json
+from adafruit_pyportal import PyPortal
+from adafruit_bitmap_font import bitmap_font
+from adafruit_display_text.label import Label
+from digitalio import DigitalInOut
+from adafruit_esp32spi import adafruit_esp32spi
+import adafruit_esp32spi.adafruit_esp32spi_requests as requests
+import adafruit_logging as logging
+
+username = 'codewisdom'
+
+logger = logging.getLogger('main')
+logger.setLevel(logging.DEBUG)
+try:
+    from secrets import secrets
+except ImportError:
+    logger.critical("""WiFi settings are kept in secrets.py, please add them there!
+the secrets dictionary must contain 'ssid' and 'password' at a minimum""")
+    raise
+
+def halt_and_catch_fire(message, *args):
+    """Log a critical error and stall the system."""
+    logger.critical(message, *args)
+    while True:
+        pass
+
+def connect_esp():
+    """Connect the ESP to the network."""
+    pyportal.neo_status((0, 0, 100))
+    while not esp.is_connected:
+        # secrets dictionary must contain 'ssid' and 'password' at a minimum
+        if secrets['ssid'] == 'CHANGE ME' or secrets['ssid'] == 'CHANGE ME':
+            change_me = "\n"+"*"*45
+            change_me += "\nPlease update the 'secrets.py' file on your\n"
+            change_me += "CIRCUITPY drive to include your local WiFi\n"
+            change_me += "access point SSID name in 'ssid' and SSID\n"
+            change_me += "password in 'password'. Then save to reload!\n"
+            change_me += "*"*45
+            raise OSError(change_me)
+        pyportal.neo_status((100, 0, 0)) # red = not connected
+        try:
+            esp.connect(secrets)
+        except RuntimeError as error:
+            logger.error("Could not connect to internet: %s", error)
+            logger.error("Retrying in 3 seconds...")
+            time.sleep(3)
+
+#pylint:disable=redefined-outer-name
+def get_bearer_token():
+    """Get the bearer authentication token from twitter."""
+    raw_key = secrets['twitter_api_key'] + ':' + secrets['twitter_secret_key']
+    encoded_key = binascii.b2a_base64(bytes(raw_key, 'utf8'))
+    string_key = bytes.decode(encoded_key)
+    headers= {'Authorization': 'Basic ' + string_key,
+              'Content-Type': 'application/x-www-form-urlencoded;charset=UTF-8'}
+    response = requests.post('https://api.twitter.com/oauth2/token',
+                             headers=headers,
+                             data='grant_type=client_credentials')
+    response_dict = json.loads(response.content)
+    if response_dict['token_type'] != 'bearer':
+        halt_and_catch_fire('Wrong token type from twitter: %s', response_dict['token_type'])
+    return response_dict['access_token']
+#pylint:enable=redefined-outer-name
+
+#setup esp interface
+esp32_ready = DigitalInOut(board.ESP_BUSY)
+esp32_gpio0 = DigitalInOut(board.ESP_GPIO0)
+esp32_reset = DigitalInOut(board.ESP_RESET)
+esp32_cs = DigitalInOut(board.ESP_CS)
+spi = busio.SPI(board.SCK, board.MOSI, board.MISO)
+esp = adafruit_esp32spi.ESP_SPIcontrol(spi, esp32_cs, esp32_ready,
+                                       esp32_reset, esp32_gpio0)
+
+# determine the current working directory
+# needed so we know where to find files
+cwd = ("/"+__file__).rsplit('/', 1)[0]
+
+
+# Initialize the pyportal object and let us know what data to fetch and where
+# to display it
+pyportal = PyPortal(status_neopixel=board.NEOPIXEL,
+                    default_bg=cwd + '/twitter_background.bmp',
+                    external_spi=spi,
+                    esp=esp,
+                    text_font=cwd+'/fonts/Helvetica-Bold-16.bdf',
+                    text_position=(20, 60),
+                    text_color=0xFFFFFF,
+                    caption_text='@' + username,
+                    caption_font=cwd+'/fonts/Helvetica-Bold-16.bdf',
+                    caption_position=(5, 210),
+                    caption_color=0x808080)
+
+logger.debug("Connecting to AP %s...", secrets['ssid'])
+connect_esp()
+logger.debug('Connected')
+
+logger.debug('Getting bearer token...')
+bearer_token = get_bearer_token()
+logger.debug('Got it')
+
+headers = {'Authorization': 'Bearer ' + bearer_token}
+
+while True:
+    logger.debug('fetching latest tweet from @%s', username)
+    response = requests.get('https://api.twitter.com/1.1/statuses/user_timeline.json?count=1&screen_name=' + username,
+                            headers=headers)
+    if response.status_code != 200:
+        logger.error('Tweet fetch status: %d', response.status_code)
+    else:
+        tweet = json.loads(response.content)[0]['text']
+        logger.debug('Tweet: %s', tweet)
+        lines = PyPortal.wrap_nicely(tweet, 35)
+        pyportal.set_text('\n'.join(lines))
+    # check every hour
+    time.sleep(3600)