Anson/Simple-Stock-Bot
1
0
Fork 0
mirror of https://gitlab.com/simple-stock-bots/simple-stock-bot.git synced 2025-06-16 15:17:28 +00:00
Code Releases Activity

caching symbol search results

Browse Source
This commit is contained in:
Anson 2020-04-15 06:47:05 -07:00
parent 3b96fdf4ed
commit 1a7cfb71b6
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
functions.py
View File

@ -15,6 +15,7 @@ class Symbol:
SYMBOL_REGEX = "[$]([a-zA-Z]{1,4})" SYMBOL_REGEX = "[$]([a-zA-Z]{1,4})"
LIST_URL = "http://oatsreportable.finra.org/OATSReportableSecurities-SOD.txt" LIST_URL = "http://oatsreportable.finra.org/OATSReportableSecurities-SOD.txt"
searched_symbols = {}
def __init__(self, IEX_TOKEN: str): def __init__(self, IEX_TOKEN: str):
self.IEX_TOKEN = IEX_TOKEN self.IEX_TOKEN = IEX_TOKEN
@ -48,6 +49,10 @@ class Symbol:
Returns: Returns:
List of Tuples -- A list tuples of every stock sorted in order of how well they match. Each tuple contains: (Symbol, Issue Name). List of Tuples -- A list tuples of every stock sorted in order of how well they match. Each tuple contains: (Symbol, Issue Name).
""" """
try: # https://stackoverflow.com/a/3845776/8774114
return self.searched_symbols[search]
except KeyError:
pass
if self.symbol_ts - datetime.now() > timedelta(hours=3): if self.symbol_ts - datetime.now() > timedelta(hours=3):
self.symbol_list, self.symbol_ts = self.get_symbol_list() self.symbol_list, self.symbol_ts = self.get_symbol_list()
@ -65,7 +70,9 @@ class Symbol:
symbols.sort_values(by="Match", ascending=False, inplace=True) symbols.sort_values(by="Match", ascending=False, inplace=True)
symbols = symbols.head(10) symbols = symbols.head(10)
return list(zip(list(symbols["Symbol"]), list(symbols["Description"]))) symbol_list = list(zip(list(symbols["Symbol"]), list(symbols["Description"])))
self.searched_symbols[search] = symbol_list
return symbol_list
def find_symbols(self, text: str): def find_symbols(self, text: str):
""" """