115 lines
7.5 KiB
Python
115 lines
7.5 KiB
Python
# ticker_extractor.py
|
|
|
|
import re
|
|
|
|
# A set of common English words and acronyms that look like stock tickers.
|
|
# This helps reduce false positives.
|
|
COMMON_WORDS_BLACKLIST = {
|
|
"401K", "403B", "457B", "ABC", "ABOUT", "ABOVE", "ADAM", "ADX", "AEDT", "AEST",
|
|
"AF", "AFAIK", "AFTER", "AH", "AI", "AINT", "ALD", "ALGOS", "ALIVE", "ALL",
|
|
"ALPHA", "ALSO", "AM", "AMA", "AMEX", "AND", "ANY", "APES", "APL", "APPL",
|
|
"APPLE", "AR", "ARE", "AREA", "ARK", "AROUND", "ASAP", "ASK", "ASS", "ASSET",
|
|
"AT", "ATH", "ATL", "ATM", "AUD", "AUM", "AV", "AVG", "AWS", "BABY",
|
|
"BAG", "BAGS", "BALLS", "BANG", "BBB", "BE", "BEAR", "BEARS", "BECN", "BEER",
|
|
"BELL", "BELOW", "BETA", "BETS", "BF", "BID", "BIG", "BIS", "BITCH", "BLEND",
|
|
"BNPL", "BOE", "BOJ", "BOLL", "BOMB", "BOND", "BORN", "BOTH", "BOTS", "BOYS",
|
|
"BRB", "BRICS", "BRK", "BRKB", "BRL", "BROKE", "BS", "BST", "BSU", "BT",
|
|
"BTC", "BTS", "BTW", "BULL", "BULLS", "BUST", "BUT", "BUY", "BUZZ", "CAD",
|
|
"CALL", "CAN", "CAP", "CASE", "CBD", "CBGM", "CBS", "CCI", "CD", "CEO",
|
|
"CEST", "CET", "CEX", "CFD", "CFO", "CHART", "CHECK", "CHF", "CHIPS", "CIA",
|
|
"CLEAN", "CLICK", "CLOSE", "CNBC", "CNN", "CNY", "COCK", "COGS", "COIL", "COKE",
|
|
"COME", "COST", "COULD", "COVID", "CPAP", "CPI", "CRO", "CRV", "CSE", "CSS",
|
|
"CST", "CTB", "CTEP", "CTO", "CULT", "CYCLE", "CZK", "DA", "DAILY", "DAO",
|
|
"DATE", "DAX", "DAY", "DAYS", "DCA", "DCF", "DD", "DEBT", "DEEZ", "DEMO",
|
|
"DEX", "DIA", "DID", "DIDNT", "DIP", "DITM", "DIV", "DIY", "DJIA", "DKK",
|
|
"DL", "DM", "DMV", "DNI", "DO", "DOE", "DOES", "DOGE", "DOJ", "DOM",
|
|
"DONT", "DOOR", "DOWN", "DPI", "DR", "DUDE", "DUMP", "DUTY", "DYI", "DYNK",
|
|
"DYODD", "DYOR", "EACH", "EARLY", "EARN", "EAST", "EASY", "ECB", "EDGAR", "EDIT",
|
|
"EDT", "EJ", "EMA", "EMJ", "END", "ENRON", "ENV", "EO", "EOD", "EOW",
|
|
"EOY", "EPA", "EPS", "ER", "ESG", "ESPP", "EST", "ETA", "ETF", "ETFS",
|
|
"ETH", "EU", "EUR", "EV", "EVEN", "EVERY", "EVTOL", "EXTRA", "EYES", "EZ",
|
|
"FAANG", "FAFO", "FAQ", "FAR", "FAST", "FBI", "FCFF", "FD", "FDA", "FEE",
|
|
"FFH", "FFS", "FGMA", "FIG", "FIGMA", "FIHTX", "FING", "FINRA", "FINT", "FINTX",
|
|
"FINTY", "FIRST", "FKIN", "FLT", "FLY", "FML", "FOLO", "FOMC", "FOMO", "FOR",
|
|
"FOREX", "FRAUD", "FRG", "FROM", "FSELK", "FSPSX", "FTD", "FTSE", "FUCK", "FUCKS",
|
|
"FUD", "FULL", "FUND", "FUNNY", "FVG", "FX", "FXAIX", "FXIAX", "FXROX", "FY",
|
|
"FYI", "FZROX", "GAAP", "GAIN", "GAVE", "GBP", "GC", "GDP", "GET", "GG",
|
|
"GJ", "GL", "GLHF", "GMAT", "GMI", "GMT", "GO", "GOAL", "GOAT", "GOD",
|
|
"GOING", "GOLD", "GONE", "GONNA", "GPT", "GPU", "GRAB", "GREAT", "GREEN", "GST",
|
|
"GTA", "GTFO", "GTG", "GUH", "GUYS", "HAD", "HAHA", "HALF", "HANDS", "HAS",
|
|
"HATE", "HAVE", "HBAR", "HCOL", "HEAR", "HEDGE", "HEGE", "HELP", "HEY", "HFCS",
|
|
"HFT", "HIGH", "HIGHS", "HINT", "HIS", "HKD", "HODL", "HODOR", "HOF", "HOLD",
|
|
"HOLY", "HOME", "HOUR", "HS", "HSA", "HT", "HTF", "HTML", "HUF", "HUGE",
|
|
"HYPE", "IANAL", "IB", "IBS", "ICT", "ID", "IDF", "IDK", "IF", "II",
|
|
"IKKE", "IKZ", "IM", "IMHO", "IMO", "IN", "INR", "INTO", "IP", "IPO",
|
|
"IRA", "IRAS", "IRC", "IRISH", "IRS", "IS", "ISA", "ISIN", "ISM", "ISN",
|
|
"IST", "IT", "ITM", "ITS", "ITWN", "IUIT", "IV", "IVV", "IWM", "JAVA",
|
|
"JD", "JFC", "JK", "JLR", "JMO", "JOIN", "JOKE", "JP", "JPOW", "JPY",
|
|
"JS", "JST", "JUN", "JUST", "KARMA", "KEEP", "KNEW", "KNOW", "KO", "KOHLS",
|
|
"KPMG", "KRW", "LANGT", "LARGE", "LAST", "LATE", "LATER", "LBO", "LDL", "LEADS",
|
|
"LEAP", "LEAPS", "LEARN", "LEI", "LETS", "LFG", "LIFE", "LIG", "LIGMA", "LIKE",
|
|
"LIMIT", "LIST", "LLC", "LLM", "LMAO", "LMN", "LOKO", "LOL", "LOLOL", "LONG",
|
|
"LOOK", "LOSE", "LOSS", "LOST", "LOVE", "LOW", "LOWER", "LOWS", "LP", "LTCG",
|
|
"LUPD", "LYING", "M&A", "MA", "MACD", "MAKE", "MAKES", "MANY", "MAX", "MBA",
|
|
"MC", "MCP", "ME", "MEME", "MERGE", "MERK", "MES", "MEXC", "MF", "MFER",
|
|
"MID", "MIGHT", "MIN", "MIND", "ML", "MLB", "MM", "MNQ", "MOASS", "MOM",
|
|
"MONEY", "MONTH", "MONY", "MOON", "MORE", "MOU", "MSK", "MUCH", "MUSIC", "MUST",
|
|
"MXN", "MY", "MYMD", "NASA", "NATO", "NBA", "NCAN", "NCR", "NEAR", "NEAT",
|
|
"NEED", "NEVER", "NEW", "NEWS", "NEXT", "NFA", "NFC", "NFL", "NFT", "NGMI",
|
|
"NIGHT", "NIQ", "NK", "NO", "NOK", "NONE", "NOPE", "NORTH", "NOT", "NOVA",
|
|
"NOW", "NQ", "NSA", "NTVS", "NULL", "NUT", "NUTS", "NUTZ", "NVM", "NW",
|
|
"NY", "NYSE", "NZ", "NZD", "OBBB", "OBI", "OBV", "OCF", "OCO", "OEM",
|
|
"OF", "OFA", "OFF", "OG", "OH", "OK", "OKAY", "OLD", "OMFG", "OMG",
|
|
"ON", "ONE", "ONLY", "OP", "OPEC", "OPENQ", "OPEX", "OPRN", "OR", "ORB",
|
|
"OS", "OSCE", "OT", "OTC", "OTM", "OUCH", "OUGHT", "OUT", "OVER", "OWN",
|
|
"PA", "PANIC", "PC", "PDT", "PE", "PEAK", "PEG", "PETA", "PEW", "PFC",
|
|
"PGHL", "PITA", "PLAN", "PLAYS", "PLN", "PM", "PMI", "PNL", "POC", "POMO",
|
|
"POP", "POS", "POSCO", "POV", "POW", "PPI", "PR", "PRICE", "PROFIT", "PS",
|
|
"PSA", "PST", "PT", "PTD", "PUSSY", "PUT", "PWC", "Q1", "Q2", "Q3",
|
|
"Q4", "QE", "QED", "QIMC", "QQQ", "QR", "RAM", "RATM", "RBA", "RBNZ",
|
|
"RE", "REACH", "READY", "REAL", "RED", "REIT", "REITS", "REKT", "RFK", "RH",
|
|
"RICO", "RIDE", "RIGHT", "RIP", "RISK", "RISKY", "ROCE", "ROCK", "ROE", "ROFL",
|
|
"ROI", "ROIC", "ROTH", "RRSP", "RSD", "RSI", "RT", "RTD", "RUB", "RUG",
|
|
"RULE", "RUST", "RVOL", "SAGA", "SALES", "SAME", "SAVE", "SAYS", "SBF", "SBLOC",
|
|
"SCALP", "SCAM", "SCHB", "SCIF", "SEC", "SEE", "SEK", "SELL", "SELLL", "SEP",
|
|
"SET", "SGD", "SHALL", "SHARE", "SHELL", "SHIT", "SHORT", "SHTF", "SI", "SIGN",
|
|
"SL", "SLIM", "SLOW", "SMA", "SMALL", "SO", "SOLIS", "SOME", "SOON", "SOUTH",
|
|
"SP", "SPAC", "SPDR", "SPEND", "SPLG", "SPX", "SPY", "SS", "START", "STAY",
|
|
"STEEL", "STILL", "STOCK", "STOOQ", "STOP", "STOR", "STQQQ", "STUCK", "STUDY", "SUS",
|
|
"SUV", "SWIFT", "SWING", "TA", "TAG", "TAKE", "TAM", "TBTH", "TEAMS", "TERM",
|
|
"TF", "TFSA", "THANK", "THAT", "THATS", "THE", "THEIR", "THEN", "THERE", "THESE",
|
|
"THEY", "THING", "THINK", "THIS", "TIA", "TIKR", "TIME", "TITS", "TJR", "TL",
|
|
"TL;DR", "TLDR", "TO", "TODAY", "TOLD", "TOS", "TOT", "TOTAL", "TP", "TRADE",
|
|
"TREND", "TRUE", "TRUMP", "TRUST", "TRY", "TSA", "TSP", "TSX", "TSXV", "TTM",
|
|
"TTYL", "TWO", "UCITS", "UGH", "UI", "UK", "UNDER", "UNTIL", "UP", "US",
|
|
"USA", "USD", "USSR", "UTC", "VALID", "VALUE", "VERY", "VFMXX", "VFV", "VIX",
|
|
"VLI", "VOO", "VP", "VR", "VRVP", "VSUS", "VTI", "VUAG", "VW", "VWAP",
|
|
"VXN", "VXUX", "WAGMI", "WAIT", "WALL", "WANT", "WATCH", "WAY", "WE", "WEB3",
|
|
"WEEK", "WEST", "WHALE", "WHAT", "WHEN", "WHICH", "WHO", "WHOS", "WHY", "WIDE",
|
|
"WILL", "WIRE", "WIRED", "WITH", "WL", "WON", "WOOPS", "WORDS", "WORTH", "WOULD",
|
|
"WP", "WRONG", "WSB", "WSJ", "WTF", "WV", "WWII", "WWIII", "XCUSE", "XD",
|
|
"XEQT", "XMR", "XO", "XRP", "XX", "YEAH", "YEET", "YES", "YET", "YIELD",
|
|
"YM", "YMMV", "YOLO", "YOU", "YOUR", "YOY", "YT", "YTD", "YUGE", "ZAR",
|
|
"ZEN", "ZERO"
|
|
}
|
|
|
|
def extract_tickers(text):
|
|
"""
|
|
Extracts potential stock tickers from a given piece of text.
|
|
A ticker is identified as a 1-5 character uppercase word, or a word prefixed with $.
|
|
"""
|
|
# Regex to find potential tickers:
|
|
# 1. Words prefixed with $: $AAPL, $TSLA
|
|
# 2. All-caps words between 1 and 5 characters: GME, AMC
|
|
ticker_regex = r"\$[A-Z]{1,5}\b|\b[A-Z]{2,5}\b"
|
|
|
|
potential_tickers = re.findall(ticker_regex, text)
|
|
|
|
# Filter out common words and remove the '$' prefix
|
|
tickers = []
|
|
for ticker in potential_tickers:
|
|
cleaned_ticker = ticker.replace("$", "").upper()
|
|
if cleaned_ticker not in COMMON_WORDS_BLACKLIST:
|
|
tickers.append(cleaned_ticker)
|
|
|
|
return tickers |