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

prevent duplicate keys due to caps

Browse Source
This commit is contained in:
Anson 2020-04-15 07:38:49 -07:00
parent 9552ef169f
commit 5fea3efee5
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
functions.py
View File

@ -52,6 +52,7 @@ class Symbol:
List of Tuples -- A list tuples of every stock sorted in order of how well they match. Each tuple contains: (Symbol, Issue Name).
"""
schedule.run_pending()
search = search.lower()
try: # https://stackoverflow.com/a/3845776/8774114
return self.searched_symbols[search]
except KeyError:
@ -59,13 +60,13 @@ class Symbol:
symbols = self.symbol_list
symbols["Match"] = symbols.apply(
lambda x: fuzz.ratio(search.lower(), f"{x['symbol']}".lower()), axis=1,
lambda x: fuzz.ratio(search, f"{x['symbol']}".lower()), axis=1,
)
symbols.sort_values(by="Match", ascending=False, inplace=True)
if symbols["Match"].head().sum() < 300:
symbols["Match"] = symbols.apply(
lambda x: fuzz.partial_ratio(search.lower(), x["name"].lower()), axis=1,
lambda x: fuzz.partial_ratio(search, x["name"].lower()), axis=1,
)
symbols.sort_values(by="Match", ascending=False, inplace=True)