fintic-tracker/stockTicker.py

# 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
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"):
            raise Exception("Currently, only RGB mode is supported for SetImage(). Please create images with mode 'RGB' or convert first with image = image.convert('RGB'). Pull requests to support more modes natively are also welcome :)")

        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)
            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)
        
        print('done update')
        kill = False 
        i = 0 # keep track of which image we are displaying
        while True:
            
            update_process = Process(target = self.updateMultiple, args = (options,))
            update_process.start()
            print('display_images/' + options[i % len(options)] +'.ppm')
        
            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 == 'traditional':
                offset_x = 128
            
        
            offset_y = 0
            #first scroll image in from bottom
            
            if animation == 'up':
                offset_y = 32
                while offset_y > 0:
                    
                    self.setImage(image, offset_x = offset_x, offset_y = offset_y)
                    offset_y -= 1
                    
                    time.sleep(self.delay)
                    kill = self.checkKilled()
                    if kill: break
                    
            elif animation == 'down':
                offset_y = -32
                while offset_y < 0:
                    
                    self.setImage(image, offset_x = offset_x, offset_y = offset_y)
                    offset_y += 1
                    
                    time.sleep(self.delay)
                    kill = self.checkKilled()
                    if kill: break
                    
                    
            
            
            while offset_x > -img_width:
                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, offset_x = offset_x, offset_y = offset_y)
                
                if animation == 'continuous':
                    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)
                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, buff = 0):
        '''
        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 + buff + 3, height+3))
        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()

 
        font = ImageFont.load("./fonts/texgyre-27.pil")
        
        img = self.textImage(text, font, int(r), int(g), int(b), True, buff = 50)
        return img
        
    
    def getNewsImage(self):
        headline_font = ImageFont.load("./fonts/10x20.pil")
        source_font = ImageFont.load("./fonts/10x20.pil")
        
        image_list = []
 
        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_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 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):
        font = ImageFont.load("./fonts/texgyre-27.pil")
        title_img = self.textImage('CRYPTO', font, matrix_height = True)
        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):
        font = ImageFont.load("./fonts/texgyre-27.pil")
        title_img = self.textImage('FOREX', font, matrix_height = True)
        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):
        font = ImageFont.load("./fonts/texgyre-27.pil")
        title_img = self.textImage('STOCKS', font, matrix_height = True)
        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 getLeagueImage(self, league, time):
        
        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)
            
           
            
            
            #away_timage = self.textImage(sports_info[away_team]['code'], small_font, r = 255, g = 255, b = 255)
            
            date_timage = self.textImage(date, small_font, r = 255, g = 255, b = 255)
            #round_timage = self.textImage('round:' + rond, 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))
                #vs_timage = self.textImage(sports_info[home_team]['code'] + 'vs' + sports_info[away_team]['code'], med_font, r = 255, g = 255, b = 255)
                
                #img.paste(vs_timage, (x_offset, 12))
                
                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, 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))
            
        return img
            
    def getTeamsImage(self, league):
        
        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))
            
        return img
           

    def getTodayWeatherImage(self):
        img = Image.new('RGB', (200, 32))
        
        f = open( "csv/weather_location.txt", 'r' )
        location = f.read()
        f.close()
        
        
        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))
        
        return img
        
    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' )
        location = f.read()
        f.close()
        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))
        
        
        
        return img1
    
    
    #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 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'))
            #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': #teams
            img = self.getTeamsImage('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, 
                'weather':self.getDailyWeatherImage} #put this somewhere else
                
            userSettings = ['stocks', 'crypto', 'forex'] # these wil be read from csv, just for demo
            #userSettings = ['stocks', 'crypto'] # these wil be read from csv, just for demo
            #userSettings = [ 'forex', 'weather'] # 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('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)