Genesis-Sentiment-Analyzer/info.py at main · Ahsan-Siddiqi/Genesis-Sentiment-Analyzer · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
import praw, os
from datetime import datetime
from dotenv import load_dotenv

from analyzer import analysis, extract_ticker

# Load variables from env
load_dotenv()

# create instance of reddit
reddit = praw.Reddit(
    client_id="OlMydM-kjWVxaKFeXBrNuA",
    client_secret=os.getenv('REDDIT_CLIENT_SECRET'),
    user_agent="windows:genesis:v1 (by /u/[user] )", # =========== add user
)


def grabSubInfo(subs):
    """Grabs basic info from given subreddits.

    Params:
    subs (list of subreddits)

    Returns:
    submissions (list of dicts/submissions)
    """

    submissions = []
    subreddits = ""

    # concatenate list of subs into one string (PRAW request format "sub1+sub2")
    for subreddit in subs:
        subreddits += subreddit + "+"
    subreddits = subreddits[:-1]

    try:
        for submission in reddit.subreddit(subreddits).hot(limit=10):
            date = datetime.fromtimestamp(submission.created_utc)
            date = date.strftime("%Y-%m-%d %H:%M")
            submissions.append({"title" : submission.title,
                                "body" : submission.selftext,
                                "vote_ratio": submission.upvote_ratio,
                                "num_comments": submission.num_comments,
                                "date": date,
                                "tickers": extract_ticker(submission),
                                "sentiment": analysis(submission),
                                "link": submission.permalink})
    except Exception as e:
        print(f"Error While fetching search and processing data: {e}")
        return -1

    return submissions

def searchSubInfo(text):
    """Grabs basic info from given search query.

    Params:
    text (string)

    Returns:
    submissions (list of dicts/submissions)
    """

    submissions = []
    subreddits = ""
    try:
        for submission in reddit.subreddit("Stocks+StockMarket+Investing+ValueInvesting+Economics+Technology+Finance").search(text, "hot", limit=10):
            date = datetime.fromtimestamp(submission.created_utc)
            date = date.strftime("%Y-%m-%d %H:%M")
            submissions.append({"title" : submission.title,
                                "body" : submission.selftext,
                                "vote_ratio": submission.upvote_ratio,
                                "num_comments": submission.num_comments,
                                "date": date,
                                "tickers": extract_ticker(submission),
                                "sentiment": analysis(submission),
                                "link": submission.permalink})
    except Exception as e:
        print(f"Error While fetching search and processing data: {e}")
        return -1

    return submissions