1680 lines
62 KiB
Python
1680 lines
62 KiB
Python
# Copyright (C) 2020 Daniel Richardson richardson.daniel@hotmail.co.uk
|
||
#
|
||
# This file is part of stockTicker project for justinodunn.
|
||
#
|
||
# stockTicker can not be copied and/or distributed without the express
|
||
# permission of Daniel Richardson
|
||
|
||
|
||
import sys, select
|
||
import os
|
||
import threading
|
||
from PIL import Image, ImageDraw, ImageFont
|
||
|
||
Image.init()
|
||
print(Image.SAVE.keys())
|
||
import time
|
||
import csv
|
||
import requests
|
||
import pexpect
|
||
from rgbmatrix import RGBMatrix, RGBMatrixOptions
|
||
from rgbmatrix.graphics import *
|
||
from multiprocessing import Process
|
||
import traceback
|
||
import json
|
||
from datetime import datetime
|
||
import matplotlib.colors as mcolors
|
||
|
||
def getInput(Block=False):
|
||
if Block or select.select([sys.stdin], [], [], 0) == ([sys.stdin], [], []):
|
||
msg = sys.stdin.read(1)
|
||
#sys.stdin.flush()
|
||
else:
|
||
msg = ''
|
||
return msg
|
||
|
||
|
||
|
||
class StockTicker():
|
||
def __init__(self):
|
||
#Define global resources
|
||
self.symbols = []
|
||
|
||
self.greenORred = (255, 255, 255)
|
||
#self.blank = Image.open('logos/blank.png')
|
||
self.blank = Image.new('RGB', (10, 32))
|
||
self.running = True
|
||
self.brightness = 1.0
|
||
self.delay = 0.02
|
||
|
||
# Configuration for the matrix
|
||
options = RGBMatrixOptions()
|
||
options.rows = 32
|
||
options.cols = 64
|
||
options.chain_length = 2
|
||
options.parallel = 1
|
||
options.hardware_mapping = 'adafruit-hat' # If you have an Adafruit HAT: 'adafruit-hat'
|
||
options.gpio_slowdown = 3
|
||
self.matrix = RGBMatrix(options = options)
|
||
self.points = True # display crypto change in points or percent
|
||
|
||
|
||
def openImage(self, image_file):
|
||
image = Image.open(image_file)
|
||
# Make image fit our screen.
|
||
#image.thumbnail((self.matrix.width, self.matrix.height), Image.ANTIALIAS)
|
||
|
||
#image = image.convert('RGB')
|
||
return image
|
||
|
||
def degreesToCompass(self, deg):
|
||
|
||
dirs = ['N', 'NE', 'E', 'SE', 'S', 'SW', 'W', 'NW']
|
||
|
||
return dirs[int((deg+22.5)//45)%8]
|
||
|
||
|
||
|
||
def setImage(self, image, offset_x = 0, offset_y = 0, unsafe=False, min_x = 0, max_x = 128, min_y = 0, max_y = 32):
|
||
|
||
|
||
if (image.mode != "RGB"):
|
||
image = image.convert('RGB')
|
||
|
||
if unsafe:
|
||
#In unsafe mode we directly acceshow to send commands to running python processs the underlying PIL image array
|
||
#in cython, which is considered unsafe pointer accecss,
|
||
#however it's super fast and seems to work fine
|
||
#https://groups.google.com/forum/#!topic/cython-users/Dc1ft5W6KM4
|
||
img_width, img_height = image.size
|
||
|
||
self.matrix.SetPixelsPillow(offset_x, offset_y, img_width, img_height, image)
|
||
else:
|
||
# First implementation of a SetImage(). OPTIMIZE_ME: A more native
|
||
# implementation that directly reads the buffer and calls the underlying
|
||
# C functions can certainly be faster.
|
||
img_width, img_height = image.size
|
||
pixels = image.load()
|
||
for y in range(max(0, -offset_y), min(img_height, self.matrix.height - offset_y)):
|
||
for x in range(max(0,-offset_x), min(img_width, self.matrix.width - offset_x)):
|
||
|
||
|
||
(r, g, b) = pixels[x, y]
|
||
if min_x <= x + offset_x <= max_x and min_y <= y + offset_y <= max_y:
|
||
self.matrix.SetPixel(x + offset_x, y + offset_y, r*self.brightness, g*self.brightness, b*self.brightness)
|
||
|
||
def scrollImage(self, image, offset_x = 0, offset_y = 0):
|
||
img_width, img_height = image.size
|
||
|
||
|
||
|
||
while offset_x > -img_width:
|
||
offset_x -= 1
|
||
|
||
self.setImage(image, offset_x = offset_x, offset_y = offset_y)
|
||
|
||
# remove the ppixels behind the image, to stop trailing
|
||
for x in range(offset_x + img_width, 128):
|
||
for y in range(self.matrix.height):
|
||
|
||
|
||
self.matrix.SetPixel(x , y , 0,0,0)
|
||
|
||
|
||
kill = self.checkKilled()
|
||
|
||
time.sleep(self.delay)
|
||
return False
|
||
|
||
def scrollImageStacked(self, image, offset_x = 0, offset_y = 0):
|
||
img_width, img_height = image.size
|
||
|
||
|
||
|
||
while offset_x > -img_width - 128:
|
||
offset_x -= 1
|
||
|
||
self.setImage(image, offset_x = offset_x+128, offset_y = offset_y)
|
||
self.setImage(image, offset_x = offset_x, offset_y = offset_y+20)
|
||
|
||
kill = self.checkKilled()
|
||
|
||
time.sleep(self.delay)
|
||
return kill
|
||
|
||
def scrollImageTransition(self, image_files, offset_x = 0, offset_y = 0, stocks = True):
|
||
# use two image files and switch between them with a seemless transition
|
||
current_img = 1
|
||
image1 = self.openImage(image_files[0])
|
||
image2 = self.openImage(image_files[1])
|
||
kill = False
|
||
while True:
|
||
|
||
if current_img == 1:
|
||
|
||
if stocks:
|
||
update_process = Process(target = self.getFullStockImage, args = (1,))
|
||
update_process.start()
|
||
image1 = self.openImage(image_files[0])
|
||
image2 = self.openImage(image_files[1])
|
||
|
||
elif current_img == 2:
|
||
|
||
if stocks:
|
||
update_process = Process(target = self.getFullStockImage, args = (2,))
|
||
update_process.start()
|
||
|
||
image1 = self.openImage(image_files[1])
|
||
image2 = self.openImage(image_files[0])
|
||
|
||
img_width, img_height = image1.size
|
||
|
||
|
||
while offset_x > -img_width:
|
||
offset_x -= 1
|
||
|
||
self.setImage(image1, offset_x = offset_x, offset_y = offset_y)
|
||
|
||
if offset_x + img_width < self.matrix.width: # if the image is ending
|
||
self.setImage(image2, offset_x = offset_x + img_width, offset_y = offset_y)
|
||
|
||
|
||
time.sleep(self.delay)
|
||
kill = self.checkKilled()
|
||
if kill: break
|
||
|
||
if stocks:
|
||
image1.close()
|
||
image2.close()
|
||
|
||
if kill: break
|
||
|
||
if stocks:
|
||
update_process.join()
|
||
if current_img == 1:
|
||
current_img = 2
|
||
elif current_img == 2:
|
||
current_img = 1
|
||
offset_x = 0
|
||
|
||
|
||
def updateMultiple(self, options):
|
||
|
||
for option in options:
|
||
print(option)
|
||
print('display_images/' + option + '.ppm')
|
||
if option not in ['display_gif']:
|
||
img = self.functions[option]()
|
||
img.save('./display_images/'+ option+ '.ppm')
|
||
|
||
def checkKilled(self):
|
||
|
||
kill = False
|
||
try:
|
||
msg = getInput()
|
||
|
||
if msg == 'K':
|
||
self.resetMatrix()
|
||
kill = True
|
||
self.process_msg(msg)
|
||
except KeyboardInterrupt:
|
||
sys.stdout.flush()
|
||
pass
|
||
return kill
|
||
|
||
def scrollFunctionsAnimated(self, options, animation = 'continuous'):
|
||
# scrolls trhough all functions with animation. Updates functions and remakes images when each function not being dispplayed
|
||
|
||
self.updateMultiple([options[0]])
|
||
|
||
|
||
kill = False
|
||
i = 0 # keep track of which image we are displaying
|
||
while True:
|
||
|
||
update_process = Process(target = self.updateMultiple, args = ([options[(i+1) % len(options)]],))
|
||
update_process.start()
|
||
|
||
|
||
image = self.openImage('./display_images/' + options[i % len(options)] +'.ppm')
|
||
|
||
if animation == 'continuous':
|
||
image2 = self.openImage('./display_images/' + options[(i + 1) % len(options)] +'.ppm')
|
||
|
||
img_width, img_height = image.size
|
||
offset_x = 0
|
||
if animation in ['traditional', 'continuous']:
|
||
offset_x = 128
|
||
elif animation in ['up', 'down']:
|
||
offset_x = max(0, 128-img_width)
|
||
|
||
|
||
offset_y = 0
|
||
#first scroll image in from bottom
|
||
|
||
frames = 10 #controls how fast gifs run
|
||
frame = 0
|
||
|
||
|
||
pause_frames = int(0.5/self.delay)
|
||
if animation == 'up':
|
||
offset_y = 33
|
||
|
||
|
||
while offset_y > 0:
|
||
|
||
# for animation in gifs
|
||
if offset_y%frames == 0:
|
||
print(frame)
|
||
try:
|
||
|
||
image.seek(frame)
|
||
|
||
except EOFError:
|
||
print('finished')
|
||
frame = 0
|
||
image.seek(frame)
|
||
frame +=1
|
||
|
||
|
||
offset_y -= 1
|
||
self.setImage(image.convert('RGB'), offset_x = offset_x, offset_y = offset_y)
|
||
|
||
time.sleep(self.delay)
|
||
kill = self.checkKilled()
|
||
if kill: break
|
||
|
||
while pause_frames > 0:
|
||
if pause_frames%frames == 0:
|
||
|
||
try:
|
||
|
||
image.seek(frame)
|
||
|
||
except EOFError:
|
||
|
||
frame = 0
|
||
image.seek(frame)
|
||
frame +=1
|
||
|
||
pause_frames -=1
|
||
self.setImage(image.convert('RGB'), offset_x = offset_x, offset_y = offset_y)
|
||
|
||
time.sleep(self.delay)
|
||
kill = self.checkKilled()
|
||
if kill: break
|
||
|
||
elif animation == 'down':
|
||
|
||
offset_y = -33
|
||
while offset_y < 0:
|
||
# for animation in gifs
|
||
if offset_y%frames == 0:
|
||
|
||
try:
|
||
|
||
image.seek(frame)
|
||
|
||
except EOFError:
|
||
|
||
frame = 0
|
||
image.seek(frame)
|
||
frame +=1
|
||
|
||
offset_y += 1
|
||
self.setImage(image.convert('RGB'), offset_x = offset_x, offset_y = offset_y)
|
||
|
||
time.sleep(self.delay)
|
||
kill = self.checkKilled()
|
||
if kill: break
|
||
|
||
|
||
|
||
while pause_frames > 0:
|
||
if pause_frames%frames == 0:
|
||
|
||
try:
|
||
|
||
image.seek(frame)
|
||
|
||
except EOFError:
|
||
|
||
frame = 0
|
||
image.seek(frame)
|
||
frame +=1
|
||
|
||
pause_frames -=1
|
||
self.setImage(image.convert('RGB'), offset_x = offset_x, offset_y = offset_y)
|
||
|
||
time.sleep(self.delay)
|
||
kill = self.checkKilled()
|
||
if kill: break
|
||
|
||
|
||
while offset_x > -img_width:
|
||
|
||
# for animation in gifs
|
||
if offset_x%frames == 0:
|
||
|
||
try:
|
||
|
||
image.seek(frame)
|
||
|
||
except EOFError:
|
||
|
||
frame = 0
|
||
image.seek(frame)
|
||
frame +=1
|
||
|
||
#image = image.convert('RGB')
|
||
offset_x -= 1
|
||
|
||
|
||
# remove the ppixels behind the image, to stop trailing
|
||
for x in range(0,offset_x ):
|
||
for y in range(self.matrix.height):
|
||
self.matrix.SetPixel(x , y , 0,0,0)
|
||
self.setImage(image.convert('RGB'), offset_x = offset_x, offset_y = offset_y)
|
||
|
||
|
||
buff = 0
|
||
if offset_x + img_width + buff< self.matrix.width and animation == 'continuous': # if the image is ending
|
||
self.setImage(image2, offset_x = offset_x + img_width + buff, offset_y = offset_y)
|
||
else:
|
||
# remove the ppixels behind the image, to stop trailing
|
||
for x in range(offset_x + img_width, 128):
|
||
for y in range(self.matrix.height):
|
||
|
||
|
||
self.matrix.SetPixel(x , y , 0,0,0)
|
||
|
||
time.sleep(self.delay)
|
||
kill = self.checkKilled()
|
||
if kill: break
|
||
|
||
if kill: break
|
||
|
||
|
||
update_process.join()
|
||
i+=1
|
||
|
||
|
||
def textImage(self, text, font, r = 255, g = 255, b = 255, matrix_height = False, w_buff = 3, h_buff = 3):
|
||
'''
|
||
creates and returns a ppm image containing the text in the supplied font and colour
|
||
'''
|
||
|
||
width, height = self.get_text_dimensions(text, font)
|
||
|
||
if matrix_height:
|
||
height = 32
|
||
img = Image.new('RGB', (width + w_buff, height + h_buff))
|
||
d = ImageDraw.Draw(img)
|
||
|
||
d.text((0, 0), text, fill=(r, g, b), font=font)
|
||
return img
|
||
|
||
|
||
def getUserText(self):
|
||
'''
|
||
displays the text entered in the webpage by the user.
|
||
'''
|
||
|
||
f = open('csv/scroll_text.csv', 'r')
|
||
|
||
CSV = csv.reader(f)
|
||
text, r, g, b = next(CSV)
|
||
|
||
f.close()
|
||
|
||
title_img = self.openImage('feature_titles/message.png')
|
||
|
||
font = ImageFont.load("./fonts/texgyre-27.pil")
|
||
|
||
img = self.textImage(text, font, int(r), int(g), int(b), True, w_buff = 50)
|
||
return self.stitchImage([title_img, img])
|
||
|
||
def displayGIF(self, gif):
|
||
# To iterate through the entire gif
|
||
i = 0
|
||
while True:
|
||
try:
|
||
gif.seek(i)
|
||
except EOFError:
|
||
print('finished')
|
||
i = 0
|
||
gif.seek(i)
|
||
# do something to im
|
||
self.setImage(gif.convert('RGB'))
|
||
time.sleep(0.5)
|
||
i += 1
|
||
try:
|
||
msg = getInput()
|
||
|
||
if msg == 'K':
|
||
gif.close()
|
||
self.resetMatrix()
|
||
|
||
break
|
||
|
||
self.process_msg(msg)
|
||
except KeyboardInterrupt:
|
||
sys.stdout.flush()
|
||
pass
|
||
|
||
def scrollGIF(self, gif, offset_x = 0, offset_y = 0):
|
||
# To iterate through the entire gif
|
||
i = 0
|
||
|
||
img_width, img_height = gif.size
|
||
|
||
while offset_x > -img_width:
|
||
offset_x -= 1
|
||
|
||
try:
|
||
gif.seek(i)
|
||
except EOFError:
|
||
print('finished')
|
||
i = 0
|
||
gif.seek(i)
|
||
# do something to im
|
||
self.setImage(gif.convert('RGB'), offset_x = offset_x)
|
||
time.sleep(self.delay)
|
||
|
||
if offset_x % 20 == 0:
|
||
i += 1
|
||
|
||
for x in range(offset_x + img_width, 128):
|
||
for y in range(self.matrix.height):
|
||
|
||
self.matrix.SetPixel(x , y , 0,0,0)
|
||
|
||
try:
|
||
msg = getInput()
|
||
|
||
if msg == 'K':
|
||
gif.close()
|
||
self.resetMatrix()
|
||
return True
|
||
|
||
self.process_msg(msg)
|
||
except KeyboardInterrupt:
|
||
sys.stdout.flush()
|
||
pass
|
||
return False
|
||
|
||
#Using change between min and day price give appropriate arrow
|
||
#and set the overall change colour
|
||
def getArrow(self, CHANGE):
|
||
self.greenORred
|
||
logos_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'logos')
|
||
if(CHANGE>0):
|
||
Arrow = Image.open(os.path.join(logos_path, 'up.png'))
|
||
self.greenORred = (0, 255, 0)
|
||
return Arrow, CHANGE
|
||
Arrow = Image.open(os.path.join(logos_path, 'down.png'))
|
||
self.greenORred = (255, 0, 0)
|
||
CHANGE = (CHANGE * -1)
|
||
return Arrow, CHANGE
|
||
|
||
def get_text_dimensions(self, text_string, font):
|
||
canvas = Image.new('RGB', (10000,100))
|
||
|
||
draw = ImageDraw.Draw(canvas)
|
||
monospace = font
|
||
|
||
text = text_string
|
||
white = (255,255,255)
|
||
draw.text((0, 0), text, font=monospace, fill=white)
|
||
|
||
bbox = canvas.getbbox()
|
||
width = bbox[2]-bbox[0]
|
||
height = bbox[3]-bbox[1]
|
||
|
||
return width,height
|
||
|
||
#Draw Ticker, current and change onto one image
|
||
def textToImage(self, TICKER, CURRENT, CHANGE, ARROW):
|
||
font = ImageFont.load("./fonts/10x20.pil")
|
||
text_width_current, text_height = self.get_text_dimensions(CURRENT, font)
|
||
|
||
img = Image.new('RGB', (text_width_current +100 , 32))
|
||
d = ImageDraw.Draw(img)
|
||
|
||
d.text((4, 0), TICKER, fill=(255, 255, 255), font=font)
|
||
d.text((4, 16), CURRENT, fill=self.greenORred, font=font)
|
||
|
||
|
||
|
||
img.paste(ARROW, ((text_width_current + 9),18))
|
||
d.text(((text_width_current+29), 16), CHANGE, fill=self.greenORred, font=font)
|
||
|
||
text_width_change, text_height = self.get_text_dimensions(CHANGE, font)
|
||
|
||
newWidth = text_width_current + text_width_change +30
|
||
|
||
img = img.crop((0,0,newWidth,32))
|
||
return img
|
||
|
||
#Stitch the logo & prices picture into one image
|
||
def stitchImage(self, image_list):
|
||
widths, heights = zip(*(i.size for i in image_list))
|
||
total_width = sum(widths)
|
||
max_height = max(heights)
|
||
new_im = Image.new('RGB', (total_width, max_height))
|
||
x_offset = 0
|
||
for im in image_list:
|
||
new_im.paste(im, (x_offset,0))
|
||
x_offset += im.size[0]
|
||
|
||
|
||
return new_im
|
||
|
||
def resetMatrix(self):
|
||
for x in range(self.matrix.width):
|
||
for y in range(self.matrix.height):
|
||
|
||
|
||
self.matrix.SetPixel(x , y , 0,0,0)
|
||
|
||
|
||
def getCryptoImage(self):
|
||
|
||
title_img = self.openImage('feature_titles/crypto.png')
|
||
image_list = [title_img]
|
||
image_list.append(self.blank)
|
||
|
||
start = time.time()
|
||
self.readCryptoCSV()
|
||
|
||
for i, coin in enumerate(self.coins):
|
||
|
||
info = self.coin_info[coin]
|
||
|
||
change = float(info[3]) #TEXT
|
||
ticker = info[0] #TEXT
|
||
current = float(info[2]) #TEXT
|
||
base = info[1].upper()
|
||
|
||
if self.points:
|
||
# convert percent to points
|
||
change = change/100 * current
|
||
|
||
current = '%.2f' % current
|
||
|
||
|
||
arrow, change = self.getArrow(change)
|
||
change = '%.2f' % change
|
||
midFrame = self.textToImage(ticker + '(' + base + ')', current, change, arrow) #IMAGE THE TEXT
|
||
|
||
|
||
try:
|
||
logos_path = os.path.join(os.path.join(os.path.dirname(os.path.abspath(__file__)), 'logos'), 'crypto')
|
||
|
||
logo = Image.open(os.path.join(logos_path, ticker + '.png'))
|
||
stitchedStock = self.stitchImage([logo,midFrame])
|
||
except:
|
||
|
||
stitchedStock = midFrame
|
||
|
||
|
||
image_list.append(stitchedStock)
|
||
|
||
image_list.append(self.blank)
|
||
|
||
|
||
|
||
|
||
finalDisplayImage = self.stitchImage(image_list)
|
||
|
||
return finalDisplayImage
|
||
|
||
def getForexImage(self):
|
||
title_img = self.openImage('feature_titles/forex.png')
|
||
image_list = [title_img]
|
||
image_list.append(self.blank)
|
||
|
||
start = time.time()
|
||
base, currency_info = json.load(open('csv/currency.json', 'r'))
|
||
currencies = ['AUD', 'CAD', 'CHF', 'EUR', 'GBP', 'JPY', 'NZD']
|
||
|
||
for i, currency in enumerate(currencies):
|
||
|
||
current, yesterday = currency_info[currency]
|
||
|
||
change = 1/current - 1/yesterday
|
||
|
||
current = 1/current
|
||
current = '%.3f' % current
|
||
|
||
|
||
arrow, change = self.getArrow(change)
|
||
change = '%.6f' % change
|
||
midFrame = self.textToImage(currency + '(' + base + ')', current, change, arrow) #IMAGE THE TEXT
|
||
|
||
|
||
try:
|
||
logos_path = os.path.join(os.path.join(os.path.dirname(os.path.abspath(__file__)), 'logos'), 'currencies')
|
||
|
||
logo = Image.open(os.path.join(logos_path, currency.lower() + '.png'))
|
||
bse = Image.open(os.path.join(logos_path, base.lower() + '.png'))
|
||
|
||
new_im = Image.new('RGB', (32, 32))
|
||
|
||
|
||
new_im.paste(bse, (0,10), bse.convert('RGBA'))
|
||
new_im.paste(logo, (10,0), logo.convert('RGBA'))
|
||
|
||
stitchedStock = self.stitchImage([new_im, midFrame])
|
||
image_list.append(new_im)
|
||
|
||
|
||
|
||
|
||
except Exception as e:
|
||
print(e)
|
||
|
||
image_list.append(midFrame)
|
||
|
||
|
||
|
||
|
||
image_list.append(self.blank)
|
||
|
||
print('image', len(image_list))
|
||
|
||
finalDisplayImage = self.stitchImage(image_list)
|
||
|
||
|
||
|
||
return finalDisplayImage
|
||
#Connect all the pieces togeather creating 1 long final stock image
|
||
def getStockImage(self):
|
||
title_img = self.openImage('feature_titles/stocks.png')
|
||
image_list = [title_img]
|
||
image_list.append(self.blank)
|
||
|
||
start = time.time()
|
||
self.readStocksCSV()
|
||
|
||
|
||
for i, symbol in enumerate(self.symbols):
|
||
|
||
|
||
info = self.stock_info[symbol]
|
||
|
||
change = float(info[0])-float(info[1]) #TEXT
|
||
ticker = symbol #TEXT
|
||
current = '%.2f' % float(info[0]) #TEXT
|
||
|
||
arrow, change = self.getArrow(change)
|
||
change = '%.2f' % change
|
||
midFrame = self.textToImage(ticker, current, change, arrow) #IMAGE THE TEXT
|
||
|
||
|
||
try:
|
||
logos_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'logos')
|
||
|
||
logo = Image.open(os.path.join(logos_path, ticker + '.png'))
|
||
stitchedStock = self.stitchImage([logo,midFrame])
|
||
except:
|
||
|
||
stitchedStock = midFrame
|
||
|
||
image_list.append(stitchedStock)
|
||
|
||
|
||
image_list.append(self.blank)
|
||
|
||
|
||
finalDisplayImage = self.stitchImage(image_list)
|
||
|
||
|
||
|
||
return finalDisplayImage
|
||
|
||
|
||
def getNewsImage(self):
|
||
|
||
title_img = self.openImage('feature_titles/news.png')
|
||
|
||
|
||
|
||
headline_font = ImageFont.load("./fonts/10x20.pil")
|
||
source_font = ImageFont.load("./fonts/10x20.pil")
|
||
|
||
image_list = [title_img]
|
||
|
||
headlines = []
|
||
source_date_times = []
|
||
|
||
f = open('csv/news.csv', 'r')
|
||
CSV = csv.reader(f)
|
||
next(CSV)
|
||
for row in CSV:
|
||
headline, source, date, time = row
|
||
headlines.append(headline)
|
||
source_date_times.append(source + ': ' + date + ' ' + time)
|
||
|
||
f.close()
|
||
|
||
|
||
|
||
for i, headline in enumerate(headlines):
|
||
headline = headline.replace("^", ",")
|
||
headline = headline.replace("’", "'")
|
||
headline = headline.replace("‘", "'")
|
||
headline = headline.replace('“', '"')
|
||
headline = headline.replace('”', '"')
|
||
headline = headline.replace('—', '-')
|
||
headline = headline.replace('–', '-')
|
||
|
||
headline_img = self.textImage(headline, headline_font, matrix_height = True)
|
||
source_img = self.textImage(source_date_times[i], source_font, r=255, g=255, b=0, matrix_height = True)
|
||
|
||
|
||
try:
|
||
logos_path = os.path.join(os.path.join(os.path.dirname(os.path.abspath(__file__)), 'logos'), 'news_logos')
|
||
|
||
logo = Image.open(os.path.join(logos_path, 'techcrunch' + '.png'))
|
||
|
||
img = Image.new('RGB', (headline_img.size[0], 32))
|
||
img.paste(headline_img, (2, -3))
|
||
img.paste(source_img, (2,16))
|
||
|
||
img= self.stitchImage([logo,img])
|
||
|
||
except Exception as e:
|
||
print(e)
|
||
img = Image.new('RGB', (headline_img.size[0], 32))
|
||
img.paste(headline_img, (0,0))
|
||
img.paste(source_img, (0,16))
|
||
|
||
image_list.append(img)
|
||
|
||
news_image = self.stitchImage(image_list)
|
||
|
||
return news_image
|
||
|
||
def getLeagueImage(self, league=False, time = 'past'):
|
||
|
||
if not league:
|
||
f = open( "csv/league.txt", 'r' )
|
||
league = f.read().replace('\n', '')
|
||
f.close()
|
||
|
||
img = Image.new('RGB', (10000, 32))
|
||
league_info = json.load(open('csv/sports/{}/{}_games.json'.format(league, time), 'r'))
|
||
|
||
print(league_info[0].keys())
|
||
|
||
small_font = ImageFont.load("./fonts/5x7.pil")
|
||
med_font = ImageFont.load("./fonts/7x14B.pil")
|
||
large_font = ImageFont.load("./fonts/9x18B.pil")
|
||
|
||
|
||
|
||
sports_info = self.readSportsCSV(league)
|
||
|
||
|
||
|
||
buff_size = 25
|
||
x_offset = 0
|
||
print(len(league_info))
|
||
|
||
for match in league_info[-15:]:
|
||
home_team = match['home_team']
|
||
away_team = match['away_team']
|
||
|
||
home_score = match['home_score']
|
||
away_score = match['away_score']
|
||
|
||
date = match['date'].replace('-', '.')
|
||
|
||
#rond = match['round']
|
||
|
||
|
||
try:
|
||
|
||
home_logo = Image.open('logos/sports/{}/{}'.format(league, sports_info[home_team]['logo']))
|
||
|
||
except Exception as e:
|
||
home_logo = self.textImage(home_team.replace(' ', '\n'), small_font, r = 255, g = 255, b = 255)
|
||
print(e)
|
||
|
||
try:
|
||
|
||
away_logo = Image.open('logos/sports/{}/{}'.format(league, sports_info[away_team]['logo']))
|
||
|
||
except Exception as e:
|
||
away_logo = self.textImage(away_team.replace(' ', '\n'), small_font, r = 255, g = 255, b = 255)
|
||
print(e)
|
||
|
||
date_timage = self.textImage(date, small_font, r = 255, g = 255, b = 255)
|
||
|
||
img.paste(home_logo, (x_offset,0))
|
||
|
||
x_offset += home_logo.size[0] + 2
|
||
|
||
|
||
if time == 'future':
|
||
img.paste(date_timage, (x_offset+5, 0))
|
||
|
||
h_colour = mcolors.to_rgb(sports_info[home_team]['colour'].replace(' ', ''))
|
||
a_colour = mcolors.to_rgb(sports_info[away_team]['colour'].replace(' ', ''))
|
||
|
||
|
||
hc_timage = self.textImage(sports_info[home_team]['code'], med_font, r = int(h_colour[0]*255), g = int(h_colour[1]*255), b = int(h_colour[2]*255))
|
||
ac_timage = self.textImage(sports_info[away_team]['code'], med_font, r = int(a_colour[0]*255), g = int(a_colour[1]*255), b = int(a_colour[2]*255))
|
||
vs_timage = self.textImage('vs', med_font, r = 255, g = 255, b = 255)
|
||
|
||
img.paste(hc_timage, (x_offset, 9))
|
||
img.paste(vs_timage, (x_offset + hc_timage.size[0], 9))
|
||
img.paste(ac_timage, (x_offset + hc_timage.size[0] + vs_timage.size[0], 9))
|
||
x_offset += max( date_timage.size[0], hc_timage.size[0] + vs_timage.size[0] + ac_timage.size[0])
|
||
else:
|
||
|
||
score_image = self.textImage(home_score + '-' + away_score, large_font, h_buff = 5, r = 255, g = 255, b = 255)
|
||
|
||
#vs_timage = self.textImage(sports_info[home_team]['code'] + 'vs' + sports_info[away_team]['code'], small_font, r = 255, g = 255, b = 255)
|
||
|
||
h_colour = mcolors.to_rgb(sports_info[home_team]['colour'].replace(' ', ''))
|
||
a_colour = mcolors.to_rgb(sports_info[away_team]['colour'].replace(' ', ''))
|
||
|
||
hc_timage = self.textImage(sports_info[home_team]['code'], small_font, r = int(h_colour[0]*255), g = int(h_colour[1]*255), b = int(h_colour[2]*255))
|
||
ac_timage = self.textImage(sports_info[away_team]['code'], small_font, r = int(a_colour[0]*255), g = int(a_colour[1]*255), b = int(a_colour[2]*255))
|
||
vs_timage = self.textImage('vs', small_font, r = 255, g = 255, b = 255)
|
||
|
||
if date_timage.size[0] > score_image.size[0]:
|
||
img.paste(date_timage, (x_offset+2, 0))
|
||
img.paste(hc_timage, (x_offset+6, 9))
|
||
img.paste(vs_timage, (x_offset+5 + hc_timage.size[0], 9))
|
||
img.paste(ac_timage, (x_offset+6 + hc_timage.size[0] + vs_timage.size[0], 9))
|
||
img.paste(score_image, (x_offset + 2 + int((date_timage.size[0] - score_image.size[0])/2), 15))
|
||
else:
|
||
|
||
img.paste(date_timage, (x_offset+1+int((score_image.size[0] - date_timage.size[0] )/2), 0))
|
||
vs_size = hc_timage.size[0] + vs_timage.size[0] + ac_timage.size[0]
|
||
img.paste(hc_timage, (x_offset + 1 + int((score_image.size[0] - vs_size)/2), 9))
|
||
img.paste(vs_timage, (x_offset + int((score_image.size[0] - vs_size)/2) + hc_timage.size[0], 9))
|
||
img.paste(ac_timage, (x_offset+1 + int((score_image.size[0] - vs_size)/2) + hc_timage.size[0] + vs_timage.size[0], 9))
|
||
|
||
img.paste(score_image, (x_offset+1, 15))
|
||
|
||
|
||
x_offset += max( date_timage.size[0]+4, hc_timage.size[0] + vs_timage.size[0] + ac_timage.size[0]+4, 2 + int(score_image.size[0]))
|
||
|
||
#img.paste(vs_timage, (x_offset+4, 9))
|
||
|
||
#if league == 'NHL':
|
||
#
|
||
|
||
|
||
|
||
#img.paste(round_timage, (x_offset+ 7, 8))
|
||
|
||
|
||
#x_offset += max(home_timage.size[0], away_timage.size[0], date_timage.size[0], round_timage.size[0], score_image.size[0])
|
||
|
||
|
||
|
||
|
||
img.paste(away_logo, (x_offset,0))
|
||
|
||
x_offset += away_logo.size[0]
|
||
x_offset += buff_size
|
||
img = img.crop((0,0,x_offset ,32))
|
||
|
||
font = ImageFont.load("./fonts/texgyre-27.pil")
|
||
title_img = self.textImage(time.upper() + ' GAMES', font, matrix_height = True)
|
||
|
||
return self.stitchImage([title_img, img])
|
||
|
||
def getLeagueTableImage(self, league = False):
|
||
|
||
|
||
if not league:
|
||
f = open( "csv/table_league.txt", 'r' )
|
||
league = f.read().replace('\n', '')
|
||
f.close()
|
||
img = Image.new('RGB', (10000, 32))
|
||
team_info = json.load(open('csv/sports/{}/team_stats.json'.format(league), 'r'))
|
||
|
||
small_font = ImageFont.load("./fonts/5x7.pil")
|
||
med_font = ImageFont.load("./fonts/8x13B.pil")
|
||
large_font = ImageFont.load("./fonts/10x20.pil")
|
||
|
||
if league =='NHL': # read the NHl info from the csv, prem will need this as well
|
||
sports_info = self.readSportsCSV(league)
|
||
|
||
buff_size = 20
|
||
x_offset = 0
|
||
for team in team_info:
|
||
|
||
try:
|
||
if league == 'NHL':
|
||
|
||
logo = Image.open('logos/sports/{}/{}'.format(league, sports_info[team['name']]['logo']))
|
||
elif league == 'premier_league':
|
||
|
||
logo = Image.open('logos/sports/{}/{}.png'.format(league, team['name']))
|
||
img.paste(logo, (x_offset, 0))
|
||
x_offset += logo.size[0] + 2
|
||
except Exception as e:
|
||
print('no logo for:', team['name'])
|
||
print(e)
|
||
|
||
name_timage = self.textImage(team['name'], med_font, r = 255, g = 255, b = 0)
|
||
wins_timage = self.textImage('Wins:' + team['wins'], small_font, r = 0, g = 255, b = 0)
|
||
loss_timage = self.textImage('Losses:' + team['loss'], small_font, r = 255, g = 0, b = 0)
|
||
draw_timage = self.textImage('Draws:' + team['draw'], small_font, r = 0, g = 0, b = 255)
|
||
points_timage = self.textImage('Points:' + team['points'], small_font, r = 255, g = 255, b = 255)
|
||
standing_timage = self.textImage('Standing:' + team['standing'], small_font, r = 255, g = 255, b = 255)
|
||
|
||
|
||
|
||
img.paste(name_timage, (x_offset, -1))
|
||
img.paste(wins_timage, (x_offset, 12))
|
||
img.paste(loss_timage, (x_offset, 19))
|
||
img.paste(draw_timage, (x_offset, 26))
|
||
|
||
x_offset += max( wins_timage.size[0], loss_timage.size[0], draw_timage.size[0])
|
||
img.paste(points_timage, (x_offset, 14))
|
||
img.paste(standing_timage, (x_offset, 22))
|
||
|
||
x_offset += max(points_timage.size[0], standing_timage.size[0]) + buff_size
|
||
|
||
|
||
img = img.crop((0,0,x_offset ,32))
|
||
|
||
font = ImageFont.load("./fonts/texgyre-27.pil")
|
||
title_img = self.textImage('LEAGUE TABLE', font, matrix_height = True)
|
||
|
||
return self.stitchImage([title_img, img])
|
||
|
||
|
||
def getTodayWeatherImage(self):
|
||
f = open( "csv/weather_location.txt", 'r' )
|
||
line = next(f)
|
||
locations = line.split(',')
|
||
f.close()
|
||
|
||
|
||
title_img = self.openImage('feature_titles/weather.png')
|
||
imgs = [title_img]
|
||
for location in locations:
|
||
img = Image.new('RGB', (200, 32))
|
||
|
||
current_weather = json.load(open('csv/current_weather.json', 'r'))
|
||
|
||
small_font = ImageFont.load("./fonts/5x7.pil")
|
||
large_font = ImageFont.load("./fonts/10x20.pil")
|
||
|
||
location_img = self.textImage(location, small_font, r = 255, g = 255, b = 0)
|
||
|
||
img.paste(location_img, (5,0))
|
||
|
||
main = current_weather['main_weather']
|
||
if main == 'Clouds':
|
||
main = current_weather['description']
|
||
weather_ids = {'Clear': '01', 'few clouds': '02', 'scattered clouds': '03', 'broken clouds':'04', 'overcast clouds':'04', 'Drizzle':'09',
|
||
'Rain':'10', 'Thunderstorm':'11', 'Snow':'13', 'Mist': '50', 'Smoke': '50', 'Haze': '50', 'Dust': '50', 'Fog': '50',
|
||
'Sand': '50', 'Ash': '50', 'Squall': '50', 'Tornado': '50'}
|
||
|
||
weather_dir = './logos/weather_icons'
|
||
|
||
weather_img = Image.open(weather_dir + '/weather_type_icons/' + weather_ids[main] + '.png')
|
||
img.paste(weather_img, (5,9))
|
||
|
||
temp_img = self.textImage(str("{0:.0f}".format(current_weather['temp'])), large_font)
|
||
img.paste(temp_img, (36,9))
|
||
|
||
deg_img = self.textImage('o', small_font)
|
||
|
||
img.paste(deg_img, (56, 8))
|
||
|
||
main = current_weather['main_weather']
|
||
main_img = self.textImage(main, small_font)
|
||
img.paste(main_img, (34, 26))
|
||
|
||
feels_img = self.textImage('Feels like:' + str("{0:.0f}".format(current_weather['feels_like'])), small_font)
|
||
img.paste(feels_img, (location_img.size[0] + 10, 0))
|
||
|
||
min_img = self.textImage( "{0:.0f}".format(current_weather['min_temp']), small_font, r=0, g=0, b=255)
|
||
img.paste(min_img, (66, 12))
|
||
|
||
max_img = self.textImage( "{0:.0f}".format(current_weather['max_temp']), small_font, r=255, g=0, b=0)
|
||
img.paste(max_img, (66, 21))
|
||
|
||
hum_img = Image.open(weather_dir + '/humidity.png')
|
||
img.paste(hum_img, ( 82, 8))
|
||
|
||
htext_img = self.textImage(str(current_weather['humidity']) + '%', small_font)
|
||
img.paste(htext_img, (95, 10))
|
||
|
||
uv_img = Image.open(weather_dir + '/uv.png')
|
||
img.paste(uv_img, ( 82, 22))
|
||
|
||
utext_img = self.textImage(str(current_weather['uv']) , small_font)
|
||
img.paste(utext_img, (95, 23))
|
||
|
||
weekdays = ['Mon', 'Tue', 'Wed', 'Thu', 'Fri', 'Sat', 'Sun']
|
||
months =['Jan', 'Feb', 'Mar', 'Apr', 'May', 'Jun', 'Jul', 'Aug', 'Sep', 'Oct', 'Nov', 'Dec']
|
||
month = months[int(datetime.now().strftime('%m'))]
|
||
date = str(int(datetime.now().strftime('%d')))
|
||
|
||
weekday = weekdays[datetime.today().weekday()]
|
||
|
||
date_img = self.textImage(month + ' ' + date + ',' + weekday, small_font)
|
||
img.paste(date_img, (132, 0))
|
||
|
||
rain_img = Image.open(weather_dir + '/rain-chance.png')
|
||
img.paste(rain_img, (118,8))
|
||
|
||
|
||
rtext_img = self.textImage(str(int(current_weather['rain_chance']*100)) + '%', small_font)
|
||
img.paste(rtext_img, (131, 10))
|
||
|
||
cloud_img = Image.open(weather_dir + '/clouds.png')
|
||
img.paste(cloud_img, (118,20))
|
||
|
||
ctext_img = self.textImage(str(current_weather['clouds']) + '%', small_font)
|
||
img.paste(ctext_img, (131, 22))
|
||
|
||
wind_img = Image.open(weather_dir + '/wind.png')
|
||
img.paste(wind_img, (154,8))
|
||
|
||
wtext_img = self.textImage("{0:.0f}".format(current_weather['wind_speed']) + 'm/s', small_font)
|
||
img.paste(wtext_img, (168, 10))
|
||
|
||
wdir_img = self.textImage(self.degreesToCompass(current_weather['wind_direction']), small_font)
|
||
img.paste(wdir_img, (183, 10))
|
||
|
||
vis_img = Image.open(weather_dir + '/visibility.png')
|
||
img.paste(vis_img, (154,20))
|
||
|
||
vtext_img = self.textImage(str(current_weather['visibility']/1000) + 'km', small_font)
|
||
img.paste(vtext_img, (168, 22))
|
||
|
||
imgs.append(img)
|
||
|
||
|
||
|
||
return self.stitchImage(imgs)
|
||
|
||
def getDailyWeatherImageAlt(self):
|
||
|
||
|
||
f = open( "csv/weather_location.txt", 'r' )
|
||
location = f.read()
|
||
f.close()
|
||
img = Image.new('RGB', (128, 32))
|
||
|
||
current_weather = json.load(open('csv/current_weather.json', 'r'))
|
||
|
||
small_font = ImageFont.load("./fonts/5x7.pil")
|
||
extra_small_font = ImageFont.load("./fonts/4x6.pil")
|
||
large_font = ImageFont.load("./fonts/10x20.pil")
|
||
|
||
|
||
location_img = self.textImage(location, extra_small_font, r = 255, g = 255, b = 0)
|
||
|
||
|
||
|
||
main = current_weather['main_weather']
|
||
if main == 'Clouds':
|
||
main = current_weather['description']
|
||
weather_ids = {'Clear': '01', 'few clouds': '02', 'scattered clouds': '03', 'broken clouds':'04', 'overcast clouds':'04', 'Drizzle':'09',
|
||
'Rain':'10', 'Thunderstorm':'11', 'Snow':'13', 'Mist': '50', 'Smoke': '50', 'Haze': '50', 'Dust': '50', 'Fog': '50',
|
||
'Sand': '50', 'Ash': '50', 'Squall': '50', 'Tornado': '50'}
|
||
|
||
weather_dir = './logos/weather_icons'
|
||
|
||
weather_img = Image.open(weather_dir + '/weather_type_icons/' + weather_ids[main] + '.png')
|
||
|
||
|
||
temp_img = self.textImage(str("{0:.0f}".format(current_weather['temp'])), large_font)
|
||
|
||
|
||
deg_img = self.textImage('o', small_font)
|
||
|
||
min_img = self.textImage( "{0:.0f}".format(current_weather['min_temp']), small_font, r=0, g=0, b=255)
|
||
|
||
|
||
max_img = self.textImage( "{0:.0f}".format(current_weather['max_temp']), small_font, r=255, g=0, b=0)
|
||
|
||
|
||
main = current_weather['main_weather']
|
||
main_img = self.textImage(main, small_font)
|
||
|
||
weekdays = ['Mon', 'Tue', 'Wed', 'Thu', 'Fri', 'Sat', 'Sun']
|
||
months =['Jan', 'Feb', 'Mar', 'Apr', 'May', 'Jun', 'Jul', 'Aug', 'Sep', 'Oct', 'Nov', 'Dec']
|
||
month = months[int(datetime.now().strftime('%m'))]
|
||
date = str(int(datetime.now().strftime('%d')))
|
||
|
||
weekday = weekdays[datetime.today().weekday()]
|
||
|
||
date_img = self.textImage(month + ' ' + date + ',' + weekday, extra_small_font)
|
||
|
||
rain_img = Image.open(weather_dir + '/rain-chance.png')
|
||
|
||
rtext_img = self.textImage(str(int(current_weather['rain_chance']*100)) + '%', extra_small_font)
|
||
|
||
|
||
hum_img = Image.open(weather_dir + '/humidity.png')
|
||
|
||
|
||
htext_img = self.textImage(str(current_weather['humidity']) + '%', extra_small_font)
|
||
|
||
|
||
uv_img = Image.open(weather_dir + '/uv.png')
|
||
|
||
|
||
utext_img = self.textImage(str(current_weather['uv']) , extra_small_font)
|
||
|
||
wind_img = Image.open(weather_dir + '/wind.png')
|
||
|
||
wtext_img = self.textImage(str(current_weather['wind_speed']) + 'm/s', extra_small_font)
|
||
|
||
|
||
img.paste(location_img, (0,0))
|
||
img.paste(weather_img, (0,12))
|
||
img.paste(temp_img, (30,9))
|
||
img.paste(deg_img, (50, 8))
|
||
img.paste(min_img, (55, 10))
|
||
img.paste(main_img, (30, 26))
|
||
img.paste(max_img, (55, 19))
|
||
img.paste(date_img, (location_img.size[0], 1))
|
||
|
||
|
||
img.paste(rain_img, (75,0))
|
||
img.paste(rtext_img, (88, 1))
|
||
img.paste(hum_img, (102, 0))
|
||
img.paste(htext_img, (113, 1))
|
||
|
||
#img.paste(uv_img, ( 96, 0))
|
||
#img.paste(utext_img, (109, 0))
|
||
#img.paste(wind_img, (96,0))
|
||
#img.paste(wtext_img, (109,0))
|
||
|
||
|
||
|
||
img0 = img
|
||
|
||
img = Image.new('RGB', (1000, 32))
|
||
|
||
|
||
daily_weather = json.load(open('csv/daily_weather.json', 'r'))
|
||
|
||
#img.paste(date_img, (70, 0))
|
||
|
||
|
||
x_offset = 64
|
||
|
||
for i in range(0,len(daily_weather)-1):
|
||
|
||
|
||
weekday = weekdays[(datetime.today().weekday() + i)%7]
|
||
|
||
day_img = self.textImage( weekday, small_font)
|
||
weather = daily_weather[i]
|
||
main = weather['main_weather']
|
||
|
||
|
||
if main == 'Clouds':
|
||
main = weather['description']
|
||
|
||
weather_img = Image.open(weather_dir + '/small_icons/' + weather_ids[main] + '.png')
|
||
min_img = self.textImage( "{0:.0f}".format(weather['min_temp']), small_font, r=0, g=0, b=255)
|
||
|
||
|
||
max_img = self.textImage( "{0:.0f}".format(weather['max_temp']), small_font, r=255, g=0, b=0)
|
||
|
||
img.paste(day_img, (x_offset +5, 9))
|
||
img.paste(weather_img, (x_offset +5, 16))
|
||
img.paste(min_img, (x_offset + 25, 10))
|
||
img.paste(max_img, (x_offset + 25, 20))
|
||
|
||
x_offset += 35
|
||
|
||
img1 = img.crop((64,0,x_offset ,32))
|
||
|
||
img1.save('display_images/weather.ppm')
|
||
|
||
return img0, img1
|
||
|
||
|
||
def getDailyWeatherImage(self):
|
||
|
||
|
||
f = open( "csv/weather_location.txt", 'r' )
|
||
line = next(f)
|
||
locations = line.split(',')
|
||
f.close()
|
||
|
||
title_img = self.openImage('feature_titles/weather.png')
|
||
imgs = [title_img]
|
||
|
||
for location in locations:
|
||
img = Image.new('RGB', (1000, 32))
|
||
|
||
current_weather = json.load(open('csv/current_weather.json', 'r'))
|
||
|
||
small_font = ImageFont.load("./fonts/5x7.pil")
|
||
extra_small_font = ImageFont.load("./fonts/4x6.pil")
|
||
large_font = ImageFont.load("./fonts/10x20.pil")
|
||
|
||
|
||
location_img = self.textImage(location, extra_small_font, r = 255, g = 255, b = 0)
|
||
|
||
|
||
|
||
main = current_weather['main_weather']
|
||
if main == 'Clouds':
|
||
main = current_weather['description']
|
||
weather_ids = {'Clear': '01', 'few clouds': '02', 'scattered clouds': '03', 'broken clouds':'04', 'overcast clouds':'04', 'Drizzle':'09',
|
||
'Rain':'10', 'Thunderstorm':'11', 'Snow':'13', 'Mist': '50', 'Smoke': '50', 'Haze': '50', 'Dust': '50', 'Fog': '50',
|
||
'Sand': '50', 'Ash': '50', 'Squall': '50', 'Tornado': '50'}
|
||
|
||
weather_dir = './logos/weather_icons'
|
||
|
||
weather_img = Image.open(weather_dir + '/weather_type_icons/' + weather_ids[main] + '.png')
|
||
|
||
|
||
temp_img = self.textImage(str("{0:.0f}".format(current_weather['temp'])), large_font)
|
||
|
||
|
||
deg_img = self.textImage('o', small_font)
|
||
|
||
min_img = self.textImage( "{0:.0f}".format(current_weather['min_temp']), small_font, r=0, g=0, b=255)
|
||
|
||
|
||
max_img = self.textImage( "{0:.0f}".format(current_weather['max_temp']), small_font, r=255, g=0, b=0)
|
||
|
||
|
||
main = current_weather['main_weather']
|
||
main_img = self.textImage(main, small_font)
|
||
|
||
weekdays = ['Mon', 'Tue', 'Wed', 'Thu', 'Fri', 'Sat', 'Sun']
|
||
months =['Jan', 'Feb', 'Mar', 'Apr', 'May', 'Jun', 'Jul', 'Aug', 'Sep', 'Oct', 'Nov', 'Dec']
|
||
month = months[int(datetime.now().strftime('%m'))]
|
||
date = str(int(datetime.now().strftime('%d')))
|
||
|
||
weekday = weekdays[datetime.today().weekday()]
|
||
|
||
date_img = self.textImage(month + ' ' + date + ',' + weekday, extra_small_font)
|
||
|
||
rain_img = Image.open(weather_dir + '/rain-chance.png')
|
||
|
||
|
||
|
||
rtext_img = self.textImage(str(int(current_weather['rain_chance']*100)) + '%', extra_small_font)
|
||
|
||
|
||
hum_img = Image.open(weather_dir + '/humidity.png')
|
||
|
||
|
||
htext_img = self.textImage(str(current_weather['humidity']) + '%', extra_small_font)
|
||
|
||
|
||
uv_img = Image.open(weather_dir + '/uv.png')
|
||
|
||
|
||
utext_img = self.textImage(str(current_weather['uv']) , extra_small_font)
|
||
|
||
wind_img = Image.open(weather_dir + '/wind.png')
|
||
|
||
wtext_img = self.textImage(str(current_weather['wind_speed']) + 'm/s', extra_small_font)
|
||
|
||
|
||
|
||
uv_img = Image.open(weather_dir + '/uv.png')
|
||
|
||
|
||
utext_img = self.textImage(str(current_weather['uv']) , small_font)
|
||
|
||
|
||
cloud_img = Image.open(weather_dir + '/clouds.png')
|
||
|
||
|
||
ctext_img = self.textImage(str(current_weather['clouds']) + '%', small_font)
|
||
|
||
|
||
wind_img = Image.open(weather_dir + '/wind.png')
|
||
|
||
|
||
wtext_img = self.textImage("{0:.0f}".format(current_weather['wind_speed']) + 'm/s', small_font)
|
||
|
||
|
||
wdir_img = self.textImage(self.degreesToCompass(current_weather['wind_direction']), small_font)
|
||
|
||
|
||
vis_img = Image.open(weather_dir + '/visibility.png')
|
||
|
||
|
||
vtext_img = self.textImage(str(current_weather['visibility']/1000) + 'km', small_font)
|
||
|
||
|
||
img.paste(location_img, (0,0))
|
||
img.paste(weather_img, (0,12))
|
||
img.paste(temp_img, (30,9))
|
||
img.paste(deg_img, (50, 8))
|
||
img.paste(min_img, (55, 10))
|
||
img.paste(main_img, (30, 26))
|
||
img.paste(max_img, (55, 19))
|
||
img.paste(date_img, (location_img.size[0], 1))
|
||
|
||
|
||
img.paste(rain_img, (90,0))
|
||
img.paste(rtext_img, (103, 1))
|
||
|
||
img.paste(hum_img, (117, 0))
|
||
img.paste(htext_img, (128, 1))
|
||
|
||
img.paste(uv_img, ( 143, 0))
|
||
img.paste(utext_img, (156, 1))
|
||
|
||
img.paste(cloud_img, (169,0))
|
||
img.paste(ctext_img, (182, 1))
|
||
|
||
img.paste(wind_img, (205,0))
|
||
img.paste(wtext_img, (218, 1))
|
||
img.paste(wdir_img, (231, 1))
|
||
|
||
img.paste(vis_img, (243,0))
|
||
img.paste(vtext_img, (256, 1))
|
||
#img.paste(uv_img, ( 96, 0))
|
||
#img.paste(utext_img, (109, 0))
|
||
#img.paste(wind_img, (96,0))
|
||
#img.paste(wtext_img, (109,0))
|
||
|
||
|
||
daily_weather = json.load(open('csv/daily_weather.json', 'r'))
|
||
|
||
#img.paste(date_img, (70, 0))
|
||
|
||
|
||
x_offset = 70
|
||
|
||
for i in range(1,len(daily_weather)-1):
|
||
|
||
|
||
weekday = weekdays[(datetime.today().weekday() + i)%7]
|
||
|
||
day_img = self.textImage( weekday, small_font)
|
||
weather = daily_weather[i]
|
||
main = weather['main_weather']
|
||
|
||
|
||
if main == 'Clouds':
|
||
main = weather['description']
|
||
|
||
weather_img = Image.open(weather_dir + '/small_icons/' + weather_ids[main] + '.png')
|
||
min_img = self.textImage( "{0:.0f}".format(weather['min_temp']), small_font, r=0, g=0, b=255)
|
||
|
||
|
||
max_img = self.textImage( "{0:.0f}".format(weather['max_temp']), small_font, r=255, g=0, b=0)
|
||
|
||
img.paste(day_img, (x_offset +5, 9))
|
||
img.paste(weather_img, (x_offset +5, 16))
|
||
img.paste(min_img, (x_offset + 25, 10))
|
||
img.paste(max_img, (x_offset + 25, 20))
|
||
|
||
x_offset += 40
|
||
|
||
img1 = img.crop((0,0,x_offset ,32))
|
||
imgs.append(img1)
|
||
# add the image text
|
||
|
||
|
||
|
||
return self.stitchImage(imgs)
|
||
|
||
|
||
#Retrieve symbols and stock info from the csv file
|
||
def readStocksCSV(self):
|
||
|
||
self.symbols = []
|
||
self.stock_info = {}
|
||
f = open('csv/tickers.csv', 'r')
|
||
CSV = csv.reader(f)
|
||
next(CSV)
|
||
for row in CSV:
|
||
|
||
try:
|
||
symbol, current_price, opening_price = row
|
||
self.symbols.append(symbol)
|
||
self.stock_info[symbol] = [current_price, opening_price]
|
||
except:
|
||
pass # we dont want to include incomplete information
|
||
|
||
f.close()
|
||
|
||
def readCryptoCSV(self):
|
||
self.coins = []
|
||
self.coin_info = {}
|
||
f = open('csv/crypto.csv', 'r')
|
||
CSV = csv.reader(f)
|
||
next(CSV)
|
||
for row in CSV:
|
||
|
||
try:
|
||
symbol, coin, base, current_price, day_change = row
|
||
self.coins.append(coin)
|
||
self.coin_info[coin] = [symbol, base, current_price, day_change]
|
||
except:
|
||
pass
|
||
|
||
f.close()
|
||
|
||
def readSportsCSV(self, league):
|
||
|
||
team_info = {}
|
||
f = open('csv/sports/{}/team_info.csv'.format(league), 'r')
|
||
CSV = csv.reader(f)
|
||
next(CSV)
|
||
for row in CSV:
|
||
|
||
team_info[row[0]] = {}
|
||
team_info[row[0]]['id'] = row[1]
|
||
team_info[row[0]]['code'] = row[2]
|
||
team_info[row[0]]['logo'] = row[4]
|
||
team_info[row[0]]['colour'] = row[3]
|
||
|
||
return team_info
|
||
|
||
def displayDailyWeatherAlt(self):
|
||
img0, img1 = self.getDailyWeatherImageAlt()
|
||
#img = stock_ticker.getTodayWeatherImage()
|
||
|
||
|
||
while True:
|
||
|
||
self.setImage(img0, offset_x = 0, offset_y = 0)
|
||
image = img1
|
||
|
||
img_width, img_height = image.size
|
||
|
||
offset_x = 64
|
||
offset_y = 0
|
||
|
||
while offset_x > 64-img_width:
|
||
offset_x -= 1
|
||
|
||
|
||
self.setImage(image, offset_x = offset_x, offset_y = offset_y, min_x = 64, min_y = 9)
|
||
if offset_x + img_width < self.matrix.width: # if the image is ending
|
||
self.setImage(image, offset_x = offset_x + img_width, offset_y = offset_y, min_x = 64, min_y = 9)
|
||
|
||
|
||
|
||
|
||
try:
|
||
msg = getInput()
|
||
if msg == 'K':
|
||
self.resetMatrix()
|
||
return True
|
||
|
||
|
||
self.process_msg(msg)
|
||
except KeyboardInterrupt:
|
||
sys.stdout.flush()
|
||
pass
|
||
|
||
time.sleep(self.delay*1.1)
|
||
|
||
def getUserImage(self):
|
||
|
||
title_img = self.openImage('feature_titles/images.png')
|
||
|
||
image = self.openImage(os.path.join(os.path.dirname(os.path.abspath(__file__)), 'display_images/user_image.ppm'))
|
||
return self.stitchImage([title_img, image])
|
||
|
||
def getUserGIF(self):
|
||
gif = Image.open(os.path.join(os.path.dirname(os.path.abspath(__file__)), 'display_images/display_gif.ppm'))
|
||
return gif
|
||
|
||
def displayStocks(self):
|
||
|
||
self.scrollImageTransition(['final.ppm', 'final.ppm'], offset_x = 0, offset_y = 0)
|
||
|
||
def process_msg(self, msg):
|
||
|
||
if msg == 'S': # stocks
|
||
img = self.getStockImage()
|
||
|
||
|
||
img.save('display_images/stocks.ppm')
|
||
self.scrollImageTransition(['display_images/stocks.ppm', 'display_images/stocks.ppm'], stocks = False)
|
||
|
||
elif msg == 'N': #news
|
||
img = self.getNewsImage()
|
||
|
||
|
||
img.save('display_images/news.ppm')
|
||
self.scrollImageTransition(['display_images/news.ppm', 'display_images/news.ppm'], stocks = False)
|
||
|
||
# speed settings
|
||
elif msg == 's':
|
||
self.delay = 0.03
|
||
|
||
elif msg == 'm':
|
||
self.delay = 0.01
|
||
|
||
elif msg == 'f':
|
||
self.delay = 0.005
|
||
|
||
|
||
elif msg in ['0', '1', '2', '3', '4', '5', '6', '7', '8', '9']: # birghtness ettings
|
||
|
||
self.brightness = min(1.0, float(msg)/10 + 0.1)
|
||
|
||
|
||
elif msg == 'T':# text
|
||
|
||
img = self.getUserText()
|
||
|
||
img.save('display_images/scroll_text.ppm')
|
||
self.scrollImageTransition(['display_images/scroll_text.ppm', 'display_images/scroll_text.ppm'], offset_x = 128, offset_y = 0, stocks = False)
|
||
|
||
elif msg == 'I': # image
|
||
|
||
image = self.openImage(os.path.join(os.path.dirname(os.path.abspath(__file__)), 'display_images/display_image'))
|
||
|
||
#self.setImage( image)
|
||
while True:
|
||
kill = self.scrollImage(image, offset_x = 128)
|
||
|
||
if kill:
|
||
break
|
||
|
||
elif msg == 'G': # gif
|
||
gif = Image.open(os.path.join(os.path.dirname(os.path.abspath(__file__)), 'display_images/display_gif.ppm'))
|
||
#self.displayGIF(gif)
|
||
while True:
|
||
kill = self.scrollGIF(gif, offset_x = 128)
|
||
if kill:
|
||
break
|
||
|
||
elif msg == 'W': # weather
|
||
img = self.getTodayWeatherImage()
|
||
img.save('display_images/weather.ppm')
|
||
self.scrollImageTransition(['display_images/weather.ppm', 'display_images/weather.ppm'], stocks = False)
|
||
|
||
elif msg == 'D': # daily weather
|
||
#self.displayDailyWeatherAlt()
|
||
|
||
img = self.getDailyWeatherImage()
|
||
img.save('display_images/weather.ppm')
|
||
#self.setImage( image)
|
||
self.scrollImageTransition(['display_images/weather.ppm', 'display_images/weather.ppm'], stocks = False)
|
||
|
||
elif msg == 'P': # past league
|
||
img = self.getLeagueImage('NBA', 'past')
|
||
img.save('display_images/league.ppm')
|
||
|
||
stock_ticker.scrollImageTransition(['display_images/league.ppm', 'display_images/league.ppm'], stocks = False)
|
||
|
||
elif msg == 'F': # future league
|
||
img = self.getLeagueImage('NHL', 'future')
|
||
img.save('display_images/league.ppm')
|
||
|
||
stock_ticker.scrollImageTransition(['display_images/league.ppm', 'display_images/league.ppm'], stocks = False)
|
||
|
||
elif msg == 'L': # live game
|
||
img = self.getLeagueImage('NBA', 'live')
|
||
img.save('display_images/league.ppm')
|
||
|
||
stock_ticker.scrollImageTransition(['display_images/league.ppm', 'display_images/league.ppm'], stocks = False)
|
||
|
||
elif msg == 't': #legue tble
|
||
img = self.getLeagueTableImage('premier_league')
|
||
img.save('display_images/teams.ppm')
|
||
|
||
stock_ticker.scrollImageTransition(['display_images/teams.ppm', 'display_images/teams.ppm'], stocks = False)
|
||
|
||
elif msg == 'A': #everything
|
||
|
||
self.functions = {'stocks': self.getStockImage, 'crypto': self.getCryptoImage, 'forex': self.getForexImage,
|
||
'daily_weather':self.getDailyWeatherImage, 'today_weather': self.getDailyWeatherImage, 'league_table': self.getLeagueTableImage,
|
||
'league_games': self.getLeagueImage, 'news':self.getNewsImage, 'text': self.getUserText, 'display_image': self.getUserImage, 'display_gif':self.getUserGIF} #put this somewhere else
|
||
|
||
userSettings = ['display_gif', 'text', 'display_image', 'stocks', 'crypto', 'forex', 'today_weather', 'daily_weather', 'league_table', 'league_games', 'news'] # these wil be read from csv, just for demo
|
||
#userSettings = ['stocks', 'crypto'] # these wil be read from csv, just for demo
|
||
#userSettings = [ 'display_image', 'news'] # these wil be read from csv, just for demo
|
||
|
||
self.scrollFunctionsAnimated(userSettings, animation = 'down')
|
||
|
||
elif msg == 'K': # kill
|
||
self.resetMatrix()
|
||
|
||
if __name__ == '__main__':
|
||
|
||
with open('log.txt', "w") as log:
|
||
try:
|
||
stock_ticker = StockTicker()
|
||
|
||
#stock_ticker.process_msg(P')
|
||
#stock_ticker.process_msg('G')
|
||
#stock_ticker.process_msg('f')
|
||
#stock_ticker.process_msg('A')
|
||
|
||
|
||
|
||
|
||
while True:
|
||
msg = getInput()
|
||
stock_ticker.process_msg(msg)
|
||
except Exception as e:
|
||
|
||
exc_type, exc_obj, exc_tb = sys.exc_info()
|
||
fname = os.path.split(exc_tb.tb_frame.f_code.co_filename)[1]
|
||
log.write(str(e))
|
||
log.write('. file: ' + fname)
|
||
log.write('. line: ' + str(exc_tb.tb_lineno))
|
||
log.write('. type: ' + str(exc_type))
|
||
log.write('\n ' + "".join(traceback.format_exception(sys.exc_info()[0], sys.exc_info()[1], sys.exc_info()[2])))
|
||
raise(e)
|
||
|
||
|