Files
OpenBB/openbb_terminal/common/behavioural_analysis/reddit_helpers.py
Colin Delahunty ee34a1f278 Feature/add isort linter (#4114)
* Fixed isort

* Fixed issues

* FIxed linters

---------

Co-authored-by: James Maslek <jmaslek11@gmail.com>
2023-02-04 21:24:47 -05:00

34 lines
821 B
Python

"""Reddit Helpers"""
__docformat__ = "numpy"
import re
from typing import List
import praw
def find_tickers(submission: praw.models.reddit.submission.Submission) -> List[str]:
"""Extracts potential tickers from reddit submission.
Parameters
----------
submission : praw.models.reddit.submission.Submission
Reddit post to scan
Returns
-------
List[str]
List of potential tickers
"""
ls_text = [submission.selftext, submission.title]
submission.comments.replace_more(limit=0)
for comment in submission.comments.list():
ls_text.append(comment.body)
l_tickers_found = []
for s_text in ls_text:
for s_ticker in set(re.findall(r"([A-Z]{3,5} )", s_text)):
l_tickers_found.append(s_ticker.strip())
return l_tickers_found