Improved formatting
This commit is contained in:
aceisace
@@ -1,5 +1,5 @@
|
||||
#!/usr/bin/python3
|
||||
# -*- coding: utf-8 -*-
|
||||
#!python3
|
||||
|
||||
"""
|
||||
Stocks Module for Inkycal Project
|
||||
|
||||
@@ -20,249 +20,252 @@ from inkycal.custom import write, internet_available
|
||||
from PIL import Image
|
||||
|
||||
try:
|
||||
import yfinance as yf
|
||||
import yfinance as yf
|
||||
except ImportError:
|
||||
print('yfinance is not installed! Please install with:')
|
||||
print('pip3 install yfinance')
|
||||
print('yfinance is not installed! Please install with:')
|
||||
print('pip3 install yfinance')
|
||||
|
||||
try:
|
||||
import matplotlib.pyplot as plt
|
||||
import matplotlib.image as mpimg
|
||||
import matplotlib.pyplot as plt
|
||||
import matplotlib.image as mpimg
|
||||
except ImportError:
|
||||
print('matplotlib is not installed! Please install with:')
|
||||
print('pip3 install matplotlib')
|
||||
print('matplotlib is not installed! Please install with:')
|
||||
print('pip3 install matplotlib')
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
class Stocks(inkycal_module):
|
||||
name = "Stocks - Displays stock market infos from Yahoo finance"
|
||||
|
||||
name = "Stocks - Displays stock market infos from Yahoo finance"
|
||||
# required parameters
|
||||
requires = {
|
||||
|
||||
# required parameters
|
||||
requires = {
|
||||
"tickers": {
|
||||
|
||||
"tickers": {
|
||||
|
||||
"label": "You can display any information by using "
|
||||
"the respective symbols that are used by Yahoo! Finance. "
|
||||
"Separate multiple symbols with a comma sign e.g. "
|
||||
"TSLA, U, NVDA, EURUSD=X"
|
||||
}
|
||||
"label": "You can display any information by using "
|
||||
"the respective symbols that are used by Yahoo! Finance. "
|
||||
"Separate multiple symbols with a comma sign e.g. "
|
||||
"TSLA, U, NVDA, EURUSD=X"
|
||||
}
|
||||
}
|
||||
|
||||
def __init__(self, config):
|
||||
def __init__(self, config):
|
||||
|
||||
super().__init__(config)
|
||||
super().__init__(config)
|
||||
|
||||
config = config['config']
|
||||
config = config['config']
|
||||
|
||||
# If tickers is a string from web-ui, convert to a list, else use
|
||||
# tickers as-is i.e. for tests
|
||||
if config['tickers'] and isinstance(config['tickers'], str):
|
||||
self.tickers = config['tickers'].replace(" ", "").split(',') #returns list
|
||||
else:
|
||||
self.tickers = config['tickers']
|
||||
# If tickers is a string from web-ui, convert to a list, else use
|
||||
# tickers as-is i.e. for tests
|
||||
if config['tickers'] and isinstance(config['tickers'], str):
|
||||
self.tickers = config['tickers'].replace(" ", "").split(',') # returns list
|
||||
else:
|
||||
self.tickers = config['tickers']
|
||||
|
||||
# give an OK message
|
||||
print(f'{__name__} loaded')
|
||||
# give an OK message
|
||||
print(f'{__name__} loaded')
|
||||
|
||||
def generate_image(self):
|
||||
"""Generate image for this module"""
|
||||
def generate_image(self):
|
||||
"""Generate image for this module"""
|
||||
|
||||
# Define new image size with respect to padding
|
||||
im_width = int(self.width - (2 * self.padding_left))
|
||||
im_height = int(self.height - (2 * self.padding_top))
|
||||
im_size = im_width, im_height
|
||||
logger.info(f'image size: {im_width} x {im_height} px')
|
||||
# Define new image size with respect to padding
|
||||
im_width = int(self.width - (2 * self.padding_left))
|
||||
im_height = int(self.height - (2 * self.padding_top))
|
||||
im_size = im_width, im_height
|
||||
logger.info(f'image size: {im_width} x {im_height} px')
|
||||
|
||||
# Create an image for black pixels and one for coloured pixels (required)
|
||||
im_black = Image.new('RGB', size = im_size, color = 'white')
|
||||
im_colour = Image.new('RGB', size = im_size, color = 'white')
|
||||
# Create an image for black pixels and one for coloured pixels (required)
|
||||
im_black = Image.new('RGB', size=im_size, color='white')
|
||||
im_colour = Image.new('RGB', size=im_size, color='white')
|
||||
|
||||
# Create tmp path
|
||||
tmpPath = '/tmp/inkycal_stocks/'
|
||||
# Create tmp path
|
||||
tmpPath = '/tmp/inkycal_stocks/'
|
||||
|
||||
try:
|
||||
os.mkdir(tmpPath)
|
||||
except OSError:
|
||||
print (f"Creation of tmp directory {tmpPath} failed")
|
||||
else:
|
||||
print (f"Successfully created tmp directory {tmpPath} ")
|
||||
try:
|
||||
os.mkdir(tmpPath)
|
||||
except OSError:
|
||||
print(f"Creation of tmp directory {tmpPath} failed")
|
||||
else:
|
||||
print(f"Successfully created tmp directory {tmpPath} ")
|
||||
|
||||
# Check if internet is available
|
||||
if internet_available() == True:
|
||||
logger.info('Connection test passed')
|
||||
else:
|
||||
raise Exception('Network could not be reached :/')
|
||||
# Check if internet is available
|
||||
if internet_available() == True:
|
||||
logger.info('Connection test passed')
|
||||
else:
|
||||
raise Exception('Network could not be reached :/')
|
||||
|
||||
# Set some parameters for formatting feeds
|
||||
line_spacing = 1
|
||||
line_height = self.font.getsize('hg')[1] + line_spacing
|
||||
line_width = im_width
|
||||
max_lines = (im_height // (self.font.getsize('hg')[1] + line_spacing))
|
||||
# Set some parameters for formatting feeds
|
||||
line_spacing = 1
|
||||
line_height = self.font.getsize('hg')[1] + line_spacing
|
||||
line_width = im_width
|
||||
max_lines = (im_height // (self.font.getsize('hg')[1] + line_spacing))
|
||||
|
||||
logger.debug(f"max_lines: {max_lines}")
|
||||
logger.debug(f"max_lines: {max_lines}")
|
||||
|
||||
# Calculate padding from top so the lines look centralised
|
||||
spacing_top = int( im_height % line_height / 2 )
|
||||
# Calculate padding from top so the lines look centralised
|
||||
spacing_top = int(im_height % line_height / 2)
|
||||
|
||||
# Calculate line_positions
|
||||
line_positions = [
|
||||
(0, spacing_top + _ * line_height ) for _ in range(max_lines)]
|
||||
# Calculate line_positions
|
||||
line_positions = [
|
||||
(0, spacing_top + _ * line_height) for _ in range(max_lines)]
|
||||
|
||||
logger.debug(f'line positions: {line_positions}')
|
||||
logger.debug(f'line positions: {line_positions}')
|
||||
|
||||
parsed_tickers = []
|
||||
parsed_tickers_colour = []
|
||||
chartSpace = Image.new('RGBA', (im_width, im_height), "white")
|
||||
chartSpace_colour = Image.new('RGBA', (im_width, im_height), "white")
|
||||
parsed_tickers = []
|
||||
parsed_tickers_colour = []
|
||||
chartSpace = Image.new('RGBA', (im_width, im_height), "white")
|
||||
chartSpace_colour = Image.new('RGBA', (im_width, im_height), "white")
|
||||
|
||||
tickerCount = range(len(self.tickers))
|
||||
tickerCount = range(len(self.tickers))
|
||||
|
||||
for _ in tickerCount:
|
||||
ticker = self.tickers[_]
|
||||
logger.info(f'preparing data for {ticker}...')
|
||||
for _ in tickerCount:
|
||||
ticker = self.tickers[_]
|
||||
logger.info(f'preparing data for {ticker}...')
|
||||
|
||||
yfTicker = yf.Ticker(ticker)
|
||||
yfTicker = yf.Ticker(ticker)
|
||||
|
||||
try:
|
||||
stockInfo = yfTicker.info
|
||||
except Exception as exceptionMessage:
|
||||
logger.warning(f"Failed to get '{ticker}' ticker info: {exceptionMessage}")
|
||||
try:
|
||||
stockInfo = yfTicker.info
|
||||
except Exception as exceptionMessage:
|
||||
logger.warning(f"Failed to get '{ticker}' ticker info: {exceptionMessage}")
|
||||
|
||||
try:
|
||||
stockName = stockInfo['shortName']
|
||||
except Exception:
|
||||
stockName = ticker
|
||||
logger.warning(f"Failed to get '{stockName}' ticker name! Using "
|
||||
"the ticker symbol as name instead.")
|
||||
try:
|
||||
stockName = stockInfo['shortName']
|
||||
except Exception:
|
||||
stockName = ticker
|
||||
logger.warning(f"Failed to get '{stockName}' ticker name! Using "
|
||||
"the ticker symbol as name instead.")
|
||||
|
||||
try:
|
||||
stockCurrency = stockInfo['currency']
|
||||
if stockCurrency == 'USD':
|
||||
stockCurrency = '$'
|
||||
elif stockCurrency == 'EUR':
|
||||
stockCurrency = '€'
|
||||
except Exception:
|
||||
stockCurrency = ''
|
||||
logger.warning(f"Failed to get ticker currency!")
|
||||
|
||||
try:
|
||||
precision = stockInfo['priceHint']
|
||||
except Exception:
|
||||
precision = 2
|
||||
logger.warning(f"Failed to get '{stockName}' ticker price hint! Using "
|
||||
"default precision of 2 instead.")
|
||||
try:
|
||||
stockCurrency = stockInfo['currency']
|
||||
if stockCurrency == 'USD':
|
||||
stockCurrency = '$'
|
||||
elif stockCurrency == 'EUR':
|
||||
stockCurrency = '€'
|
||||
except Exception:
|
||||
stockCurrency = ''
|
||||
logger.warning(f"Failed to get ticker currency!")
|
||||
|
||||
stockHistory = yfTicker.history("30d")
|
||||
stockHistoryLen = len(stockHistory)
|
||||
logger.info(f'fetched {stockHistoryLen} datapoints ...')
|
||||
previousQuote = (stockHistory.tail(2)['Close'].iloc[0])
|
||||
currentQuote = (stockHistory.tail(1)['Close'].iloc[0])
|
||||
currentHigh = (stockHistory.tail(1)['High'].iloc[0])
|
||||
currentLow = (stockHistory.tail(1)['Low'].iloc[0])
|
||||
currentOpen = (stockHistory.tail(1)['Open'].iloc[0])
|
||||
currentGain = currentQuote-previousQuote
|
||||
currentGainPercentage = (1-currentQuote/previousQuote)*-100
|
||||
firstQuote = stockHistory.tail(stockHistoryLen)['Close'].iloc[0]
|
||||
logger.info(f'firstQuote {firstQuote} ...')
|
||||
|
||||
def floatStr(precision, number):
|
||||
return "%0.*f" % (precision, number)
|
||||
|
||||
def percentageStr(number):
|
||||
return '({:+.2f}%)'.format(number)
|
||||
|
||||
def gainStr(precision, number):
|
||||
return "%+.*f" % (precision, number)
|
||||
try:
|
||||
precision = stockInfo['priceHint']
|
||||
except Exception:
|
||||
precision = 2
|
||||
logger.warning(f"Failed to get '{stockName}' ticker price hint! Using "
|
||||
"default precision of 2 instead.")
|
||||
|
||||
stockNameLine = '{} ({})'.format(stockName, stockCurrency)
|
||||
stockCurrentValueLine = '{} {} {}'.format(
|
||||
floatStr(precision, currentQuote), gainStr(precision, currentGain), percentageStr(currentGainPercentage))
|
||||
stockDayValueLine = '1d OHL: {}/{}/{}'.format(
|
||||
floatStr(precision, currentOpen), floatStr(precision, currentHigh), floatStr(precision, currentLow))
|
||||
maxQuote = max(stockHistory.High)
|
||||
minQuote = min(stockHistory.Low)
|
||||
logger.info(f'high {maxQuote} low {minQuote} ...')
|
||||
stockMonthValueLine = '{}d OHL: {}/{}/{}'.format(
|
||||
stockHistoryLen,floatStr(precision, firstQuote),floatStr(precision, maxQuote),floatStr(precision, minQuote))
|
||||
stockHistory = yfTicker.history("30d")
|
||||
stockHistoryLen = len(stockHistory)
|
||||
logger.info(f'fetched {stockHistoryLen} datapoints ...')
|
||||
previousQuote = (stockHistory.tail(2)['Close'].iloc[0])
|
||||
currentQuote = (stockHistory.tail(1)['Close'].iloc[0])
|
||||
currentHigh = (stockHistory.tail(1)['High'].iloc[0])
|
||||
currentLow = (stockHistory.tail(1)['Low'].iloc[0])
|
||||
currentOpen = (stockHistory.tail(1)['Open'].iloc[0])
|
||||
currentGain = currentQuote - previousQuote
|
||||
currentGainPercentage = (1 - currentQuote / previousQuote) * -100
|
||||
firstQuote = stockHistory.tail(stockHistoryLen)['Close'].iloc[0]
|
||||
logger.info(f'firstQuote {firstQuote} ...')
|
||||
|
||||
logger.info(stockNameLine)
|
||||
logger.info(stockCurrentValueLine)
|
||||
logger.info(stockDayValueLine)
|
||||
logger.info(stockMonthValueLine)
|
||||
parsed_tickers.append(stockNameLine)
|
||||
parsed_tickers.append(stockCurrentValueLine)
|
||||
parsed_tickers.append(stockDayValueLine)
|
||||
parsed_tickers.append(stockMonthValueLine)
|
||||
def floatStr(precision, number):
|
||||
return "%0.*f" % (precision, number)
|
||||
|
||||
parsed_tickers_colour.append("")
|
||||
if currentGain < 0:
|
||||
parsed_tickers_colour.append(stockCurrentValueLine)
|
||||
else:
|
||||
parsed_tickers_colour.append("")
|
||||
if currentOpen > currentQuote:
|
||||
parsed_tickers_colour.append(stockDayValueLine)
|
||||
else:
|
||||
parsed_tickers_colour.append("")
|
||||
if firstQuote > currentQuote:
|
||||
parsed_tickers_colour.append(stockMonthValueLine)
|
||||
else:
|
||||
parsed_tickers_colour.append("")
|
||||
def percentageStr(number):
|
||||
return '({:+.2f}%)'.format(number)
|
||||
|
||||
if (_ < len(tickerCount)):
|
||||
parsed_tickers.append("")
|
||||
parsed_tickers_colour.append("")
|
||||
def gainStr(precision, number):
|
||||
return "%+.*f" % (precision, number)
|
||||
|
||||
logger.info(f'creating chart data...')
|
||||
chartData = stockHistory.reset_index()
|
||||
chartCloseData = chartData.loc[:,'Close']
|
||||
chartTimeData = chartData.loc[:,'Date']
|
||||
stockNameLine = '{} ({})'.format(stockName, stockCurrency)
|
||||
stockCurrentValueLine = '{} {} {}'.format(
|
||||
floatStr(precision, currentQuote), gainStr(precision, currentGain),
|
||||
percentageStr(currentGainPercentage))
|
||||
stockDayValueLine = '1d OHL: {}/{}/{}'.format(
|
||||
floatStr(precision, currentOpen), floatStr(precision, currentHigh), floatStr(precision, currentLow))
|
||||
maxQuote = max(stockHistory.High)
|
||||
minQuote = min(stockHistory.Low)
|
||||
logger.info(f'high {maxQuote} low {minQuote} ...')
|
||||
stockMonthValueLine = '{}d OHL: {}/{}/{}'.format(
|
||||
stockHistoryLen, floatStr(precision, firstQuote), floatStr(precision, maxQuote),
|
||||
floatStr(precision, minQuote))
|
||||
|
||||
logger.info(f'creating chart plot...')
|
||||
fig, ax = plt.subplots() # Create a figure containing a single axes.
|
||||
ax.plot(chartTimeData, chartCloseData, linewidth=8) # Plot some data on the axes.
|
||||
ax.set_xticklabels([])
|
||||
ax.set_yticklabels([])
|
||||
chartPath = tmpPath+ticker+'.png'
|
||||
logger.info(f'saving chart image to {chartPath}...')
|
||||
plt.savefig(chartPath)
|
||||
logger.info(stockNameLine)
|
||||
logger.info(stockCurrentValueLine)
|
||||
logger.info(stockDayValueLine)
|
||||
logger.info(stockMonthValueLine)
|
||||
parsed_tickers.append(stockNameLine)
|
||||
parsed_tickers.append(stockCurrentValueLine)
|
||||
parsed_tickers.append(stockDayValueLine)
|
||||
parsed_tickers.append(stockMonthValueLine)
|
||||
|
||||
logger.info(f'chartSpace is...{im_width} {im_height}')
|
||||
logger.info(f'open chart ...{chartPath}')
|
||||
chartImage = Image.open(chartPath)
|
||||
chartImage.thumbnail((im_width/4,line_height*4), Image.BICUBIC)
|
||||
parsed_tickers_colour.append("")
|
||||
if currentGain < 0:
|
||||
parsed_tickers_colour.append(stockCurrentValueLine)
|
||||
else:
|
||||
parsed_tickers_colour.append("")
|
||||
if currentOpen > currentQuote:
|
||||
parsed_tickers_colour.append(stockDayValueLine)
|
||||
else:
|
||||
parsed_tickers_colour.append("")
|
||||
if firstQuote > currentQuote:
|
||||
parsed_tickers_colour.append(stockMonthValueLine)
|
||||
else:
|
||||
parsed_tickers_colour.append("")
|
||||
|
||||
chartPasteX = im_width-(chartImage.width)
|
||||
chartPasteY = line_height*5*_
|
||||
logger.info(f'pasting chart image with index {_} to...{chartPasteX} {chartPasteY}')
|
||||
if (_ < len(tickerCount)):
|
||||
parsed_tickers.append("")
|
||||
parsed_tickers_colour.append("")
|
||||
|
||||
if firstQuote > currentQuote:
|
||||
chartSpace_colour.paste(chartImage, (chartPasteX, chartPasteY))
|
||||
else:
|
||||
chartSpace.paste(chartImage, (chartPasteX, chartPasteY))
|
||||
logger.info(f'creating chart data...')
|
||||
chartData = stockHistory.reset_index()
|
||||
chartCloseData = chartData.loc[:, 'Close']
|
||||
chartTimeData = chartData.loc[:, 'Date']
|
||||
|
||||
im_black.paste(chartSpace)
|
||||
im_colour.paste(chartSpace_colour)
|
||||
logger.info(f'creating chart plot...')
|
||||
fig, ax = plt.subplots() # Create a figure containing a single axes.
|
||||
ax.plot(chartTimeData, chartCloseData, linewidth=8) # Plot some data on the axes.
|
||||
ax.set_xticklabels([])
|
||||
ax.set_yticklabels([])
|
||||
chartPath = tmpPath + ticker + '.png'
|
||||
logger.info(f'saving chart image to {chartPath}...')
|
||||
plt.savefig(chartPath)
|
||||
|
||||
# Write/Draw something on the black image
|
||||
for _ in range(len(parsed_tickers)):
|
||||
if _+1 > max_lines:
|
||||
logger.error('Ran out of lines for parsed_ticker_colour')
|
||||
break
|
||||
write(im_black, line_positions[_], (line_width, line_height),
|
||||
parsed_tickers[_], font = self.font, alignment= 'left')
|
||||
logger.info(f'chartSpace is...{im_width} {im_height}')
|
||||
logger.info(f'open chart ...{chartPath}')
|
||||
chartImage = Image.open(chartPath)
|
||||
chartImage.thumbnail((im_width / 4, line_height * 4), Image.BICUBIC)
|
||||
|
||||
# Write/Draw something on the colour image
|
||||
for _ in range(len(parsed_tickers_colour)):
|
||||
if _+1 > max_lines:
|
||||
logger.error('Ran out of lines for parsed_tickers_colour')
|
||||
break
|
||||
write(im_colour, line_positions[_], (line_width, line_height),
|
||||
parsed_tickers_colour[_], font = self.font, alignment= 'left')
|
||||
chartPasteX = im_width - (chartImage.width)
|
||||
chartPasteY = line_height * 5 * _
|
||||
logger.info(f'pasting chart image with index {_} to...{chartPasteX} {chartPasteY}')
|
||||
|
||||
if firstQuote > currentQuote:
|
||||
chartSpace_colour.paste(chartImage, (chartPasteX, chartPasteY))
|
||||
else:
|
||||
chartSpace.paste(chartImage, (chartPasteX, chartPasteY))
|
||||
|
||||
im_black.paste(chartSpace)
|
||||
im_colour.paste(chartSpace_colour)
|
||||
|
||||
# Write/Draw something on the black image
|
||||
for _ in range(len(parsed_tickers)):
|
||||
if _ + 1 > max_lines:
|
||||
logger.error('Ran out of lines for parsed_ticker_colour')
|
||||
break
|
||||
write(im_black, line_positions[_], (line_width, line_height),
|
||||
parsed_tickers[_], font=self.font, alignment='left')
|
||||
|
||||
# Write/Draw something on the colour image
|
||||
for _ in range(len(parsed_tickers_colour)):
|
||||
if _ + 1 > max_lines:
|
||||
logger.error('Ran out of lines for parsed_tickers_colour')
|
||||
break
|
||||
write(im_colour, line_positions[_], (line_width, line_height),
|
||||
parsed_tickers_colour[_], font=self.font, alignment='left')
|
||||
|
||||
# Save image of black and colour channel in image-folder
|
||||
return im_black, im_colour
|
||||
|
||||
# Save image of black and colour channel in image-folder
|
||||
return im_black, im_colour
|
||||
|
||||
if __name__ == '__main__':
|
||||
print('running module in standalone/debug mode')
|
||||
print('running module in standalone/debug mode')
|
||||
|
||||
Reference in New Issue
Block a user