botworm/botworm.py at develop · seallard/botworm · 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
from utils.reddit import Reddit
from utils.goodreads import Goodreads
from utils.book_title_parser import BookTitleParser
from utils.recommendation_lister import RecommendationLister
from utils.recommendation_tracker import RecommendationTracker


def main():

    reddit = Reddit()
    goodreads = Goodreads()
    title_parser = BookTitleParser()
    tracker = RecommendationTracker()

    filtered_posts = tracker.filter_posts(reddit.get_posts())

    for post in filtered_posts:

        tracker.track_post(post)
        lister = RecommendationLister()

        filtered_comments = tracker.filter_comments(reddit.get_comments(post))

        for comment in filtered_comments:
            title_strings = title_parser.extract_titles(comment.text)

            for title in title_strings:
                book = goodreads.get_book(title)

                if book:
                    comment.books.append(book)
                    lister.add(book, comment)

            tracker.track_comment(comment)

        recommendations = lister.get()

        if recommendations != []:

            if tracker.bot_has_commented(post.id):
                bot_comment = tracker.most_recent_comment_by_bot(post.id)
                rec_index = reddit.edit_table(bot_comment, recommendations)
                remaining_recommendations = recommendations[rec_index:]

                if remaining_recommendations != []:

                    bot_comments = reddit.create_tables(remaining_recommendations, False)

                    praw_bot_comment = reddit.get_comment(bot_comment.id)
                    reddit.post_comments(praw_bot_comment, bot_comments)
                    [tracker.track_comment(comment) for comment in bot_comments]

            else:
                bot_comments = reddit.create_tables(recommendations, True)
                reddit.post_comments(post, bot_comments)
                [tracker.track_comment(comment) for comment in bot_comments]


if __name__ == "__main__":
    main()