main.py

#!/usr/bin/python3

from inky import InkyPHAT
from PIL import ImageFont, ImageDraw, Image
import textwrap
import requests
import time
from datetime import datetime
import yaml
from noaa_sdk import noaa
import requests
from requests.exceptions import HTTPError
import json
import logging
import os
import sys
import stockquotes


# Election
from urllib.request import urlopen
from xml.etree.ElementTree import parse
from datetime import datetime



logging.basicConfig(
    format='%(asctime)s.%(msecs)03d %(levelname)-8s %(message)s',
    level=logging.INFO,
    datefmt='%Y-%m-%d %H:%M:%S')

logging.info("""Status Dashboard for InkyPhat eInk display. Displays Weather, Covid-19 info, PiHole stats, Air Pollution from my personal station and top Hacker News stories. 
Press Ctrl+C to exit!
""")



font = ImageFont.truetype(
    "/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf", 10)
redBigFont = ImageFont.truetype(
    "/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf", 16)
bigFont = ImageFont.truetype(
    "/usr/share/fonts/truetype/dejavu/DejaVuSerif.ttf", 15)
medWFont = ImageFont.truetype(
    "/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf", 15)


''' Font Options
DejaVuSans-Bold.ttf      DejaVuSansMono.ttf  DejaVuSerif-Bold.ttf
DejaVuSansMono-Bold.ttf  DejaVuSans.ttf      DejaVuSerif.ttf
'''

'''
Lazy person code copying
sudo nano /usr/lib/systemd/system/eink.service
sudo systemctl start eink.service
sudo systemctl daemon-reload
journalctl -f
'''

# Display Setup
inky_display = InkyPHAT('red')

inky_display.set_border(inky_display.RED)

img = Image.new("P", (inky_display.WIDTH, inky_display.HEIGHT))
draw = ImageDraw.Draw(img)


# Setup When I want particular feeds to display

covidHours = [16,17,18,19,20,21]
stockHours = [8,9,10,11,12,13,14,15]
forecastHours = [6,7,8,9,10,17,18,19,20,12,22]
hnHours = [9,11,13,15,17,19,21]
piHours = [12,15,17]

#res1day = ""

# Utility Functions
def getConfig():
    # absolute to run as cron. PiTA
    with open('/home/pi/inkydash/config.yaml') as f:
        config = yaml.load(f, Loader=yaml.FullLoader)
        zipcode = config[0]["zip"]
        country = config[0]["country"]
        state = config[0]["state"]
        stocks = config[0]["stocks"]
        return zipcode, country, state, stocks


def fetchFeed(url):
    try:
        response = requests.get(url)
        response.raise_for_status()
        if 'application/json' in response.headers['content-type']:
          jsonResponse = response.json()
          return jsonResponse
        elif 'text/plain' in response.headers['content-type']:
          textResponse = response.text
          return textResponse
        else: 
          logging.warning("Unknown format in jquery assume json")
          jsonResponse = response.json()

    except HTTPError as http_err:
        print(f'HTTP error occurred: {http_err}')
        logging.warning(f'HTTP error occurred: {http_err}')

    except Exception as err:
        print(f'Other error occurred: {err}')
        logging.warning(f'Other error occurred: {err}')


def drawClean(color):
  if color == 'WHITE':
    draw.rectangle((0, 0, inky_display.WIDTH,
                        inky_display.HEIGHT), 
                        (inky_display.WHITE))
  elif color == 'BLACK':
    draw.rectangle((0, 0, inky_display.WIDTH,
                        inky_display.HEIGHT), 
                        (inky_display.BLACK))
  elif color == 'RED':
    draw.rectangle((0, 0, inky_display.WIDTH,
                        inky_display.HEIGHT), 
                        (inky_display.RED))
  else:
    draw.rectangle((0, 0, inky_display.WIDTH,
                        inky_display.HEIGHT), 
                        (inky_display.WHITE))

def drawScreen():
  inky_display.set_image(img.rotate(180))
  inky_display.show()


def getWeather():
    try:
        n = noaa.NOAA()
        res = n.get_forecasts(config[0], config[1], False)
        res1day = res[0]["detailedForecast"]
        # display on InkyPHAT
        # Take string of text and wrap it at 38 chars
        txt = textwrap.fill(res1day, 36)

        # Get size of text
        w, h = draw.multiline_textsize(txt, font)

        # Center Center the text
        x = (inky_display.WIDTH / 2) - (w / 2)
        y = (inky_display.HEIGHT / 2) - (h / 2)

        drawClean('WHITE')


        # Draw multiline text on scren
        if (w * h) <= 11000:
          txt = textwrap.fill(res1day, 24)
          draw.multiline_text((x, 0), txt, inky_display.BLACK, medWFont)
        elif (w * h) > 11000:
          draw.multiline_text((x, 0), txt, inky_display.BLACK, font)
        
        drawScreen()

    except HTTPError as http_err:
        print(f'HTTP error occurred: {http_err}')
        logging.warning(f'HTTP error occurred: {http_err}')

    except Exception as err:
        print(f'Other error occurred: {err}')
        logging.warning(f'Other error occurred: {err}')


def getCovid():
  drawClean('WHITE')
  jsonResponse = fetchFeed(f'https://api.covidtracking.com/v1/states/{config[2]}/daily.json')
  todayCovid = jsonResponse[0]["hospitalizedCurrently"]
  yesterdayCovid = jsonResponse[1]["hospitalizedCurrently"]
  currentlyHospitalized = "Hospitalized: " + "{:,}".format(todayCovid)
  hospitalizationChange = str(
      todayCovid - yesterdayCovid) + " new hospitalizations"
  newCases = "{:,}".format(jsonResponse[0]["positiveIncrease"]) + " new positivity"
  covidUpdate = datetime.now().strftime('%a %b %d %-I:%M %p')
  draw.text((0, 0), "CO Covid Update", inky_display.BLACK, bigFont)
  draw.text((0, 20), currentlyHospitalized, inky_display.BLACK, bigFont)
  draw.text((0, 40), hospitalizationChange, inky_display.RED, bigFont)
  draw.text((0, 60), newCases, inky_display.RED, bigFont)
  draw.text((0, 88), covidUpdate, inky_display.BLACK, font)

  drawScreen()


def getCurrentConditions():
  drawClean('WHITE')

  # AQI
  aqiResponse = fetchFeed(
      'https://io.adafruit.com/api/v2/drkpxl/feeds/pollution.aqi')
  printAqi = "Current AQI: " + aqiResponse["last_value"]
  draw.text((0, 0), printAqi, inky_display.BLACK, bigFont)

  # Current Temp
  tempResponse = fetchFeed(
      'https://io.adafruit.com/api/v2/drkpxl/feeds/temp')
  printTemp = "Current Temp: " + tempResponse["last_value"] + " °F"
  draw.text((0, 20), printTemp, inky_display.BLACK, bigFont)

  # Current Humidity
  humResponse = fetchFeed(
      'https://io.adafruit.com/api/v2/drkpxl/feeds/humidity')
  printHum = "Current Humidity: " + humResponse["last_value"] + " %"
  draw.text((0, 40), printHum, inky_display.BLACK, bigFont)

  # Feed Last Update
  lastUpdate = aqiResponse["updated_at"]
  draw.text((0, 88), lastUpdate, inky_display.BLACK, font)

  drawScreen()


def getHackerNews():
  topics = fetchFeed('https://hacker-news.firebaseio.com/v0/topstories.json')
  toptopics = [topics[0], topics[1], topics[2]]
  topic = ""

  for x in toptopics:
    r = fetchFeed(
        'https://hacker-news.firebaseio.com/v0/item/' + str(x) + '.json')
    topic = (r['title'])
    score = str(r['score']) + " pts"

    # Take string of text and wrap it at 38 chars
    txt = textwrap.fill(topic, 24)

    # Get size of text
    w, h = draw.multiline_textsize(txt, bigFont)

    # Center Center the text
    x = (inky_display.WIDTH / 2) - (w / 2)
    y = (inky_display.HEIGHT / 2) - (h / 2)

    drawClean('BLACK')

    # Draw multiline text on scren
    draw.multiline_text((x, y), txt, inky_display.WHITE, bigFont)
    draw.text((0, 0), score, inky_display.RED, redBigFont)
    
    drawScreen()

    time.sleep(60)


def getPihole():
  drawClean('BLACK')

  piholeResponse = fetchFeed('http://pi.hole/admin/api.php')
  printPercentBlockedToday = "PiHoled: " + str(round(piholeResponse["ads_percentage_today"])) + " %"
  printClientsSeen = "Total Clients: " + str(piholeResponse["clients_ever_seen"])
  printQueriesToday = "Queries Today: {:,}".format(piholeResponse['dns_queries_today'])

  #('{:,}'.format(value)) 
  printStatus = "PiHole Status: " + piholeResponse['status']
  draw.text((0, 0), printStatus, inky_display.WHITE, bigFont)
  draw.text((0, 20), printQueriesToday, inky_display.WHITE, bigFont)
  draw.text((0, 40), printPercentBlockedToday, inky_display.RED, bigFont)
  draw.text((0, 60), printClientsSeen, inky_display.WHITE, bigFont)

  # Get Public IP, both work.
  #ipResponse = fetchFeed('https://api.ipify.org?format=json')
  ipResponse = fetchFeed('https://icanhazip.com')
  
  printIpResonse = "Public IP: " + ipResponse
  draw.text((0, 80), printIpResonse, inky_display.WHITE, bigFont)
  
  drawScreen()

def getStocks():

    stockSymbols = config[3]


    for x in stockSymbols:
      drawClean('WHITE')
      symbolData = stockquotes.Stock(x)
      printName = symbolData.name + " (" + symbolData.symbol + ")"
      # Get currnet price, adjust value to have $, comma, and 2 decimal points, also crop photo to show a max char of 7
      printCurrentPrice = str("${:,.2f}".format(symbolData.current_price))[0:7]
      printIncreasePercent = str(symbolData.increase_percent) + "%"
      printHighLow = "H: " + str("${:,.2f}".format(symbolData.historical[0]['high']))[0:7] + " / L: " + str("${:,.2f}".format(symbolData.historical[0]['low']))[0:7]

      draw.text((0, 20), printName, inky_display.BLACK, bigFont)
      draw.text((0, 40), printCurrentPrice, inky_display.BLACK, bigFont)
      if symbolData.increase_percent <= 0:
        draw.text((68, 40), ("▼ " + printIncreasePercent), inky_display.RED, bigFont)
      else:
        draw.text((68, 40), ("▲ " + printIncreasePercent), inky_display.BLACK, bigFont)
      draw.text((0, 60), printHighLow, inky_display.BLACK, bigFont)
      
      drawScreen()

      time.sleep(30)



def getElection():
  var_url = urlopen('https://raw.githubusercontent.com/alex/nyt-2020-election-scraper/master/battleground-state-changes.xml')
  xmldoc = parse(var_url)
  for item in xmldoc.iterfind('channel/item'):
      description = item.findtext('description')
      date = item.findtext('pubDate')
      print(description)
      # Take string of text and wrap it at 38 chars
      txt = textwrap.fill(description, 24)

      # Get size of text
      w, h = draw.multiline_textsize(txt, bigFont)

      # Center Center the text
      x = (inky_display.WIDTH / 2) - (w / 2)
      y = (inky_display.HEIGHT / 2) - (h / 2)

      drawClean('WHITE')
      
      currentLead = (int(''.join(list(filter(str.isdigit, description)))))
      currentLead = "{:,}".format(currentLead)
      if "Trump" in txt:
        draw.multiline_text((x, y), txt, inky_display.RED, bigFont)
        
        #logging.info(f'Trump Current Lead: {currentLead}')
      else:
        draw.multiline_text((x, y), txt, inky_display.BLACK, bigFont)
        #logging.info(f'Biden Current Lead: {currentLead}')

      
      drawScreen()

      time.sleep(20)
    



# Get initial YAML
config = getConfig()

# Main Loop
while True:
  try:
    currentTime = ((datetime.now()).hour)
    getElection()
    if currentTime in covidHours:
      getCovid()
      time.sleep(180)
    if currentTime in stockHours:
      getStocks()
    if currentTime in hnHours:
      getHackerNews()
    if currentTime in piHours:
      getPihole()
      time.sleep(180)
    getCurrentConditions()
    time.sleep(180)
    if currentTime in forecastHours:
      getWeather()
      time.sleep(180)
  except (KeyboardInterrupt, SystemExit):
    print('\nkeyboardinterrupt found!')
    sys.exit(0)
    print('\n...Program Stopped Manually!')
    raise