PUBGTwitterBot/main.py at master · zuve/PUBGTwitterBot · 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
83
84
85
import tweepy, praw, os, requests
from bs4 import BeautifulSoup
from bs4.element import Comment
from HTMLParser import HTMLParser
from markdownify import markdownify as md

def substring_indexes(substring, string):
    last_found = -1
    while True:
        last_found = string.find(substring, last_found + 1)
        if last_found == -1:
            break
        yield last_found

def body_from_post(html):
    soup = BeautifulSoup(html, 'html.parser')
    texts = soup.find("div", {"class": "announcement_body group_body_links"})
    return texts

h = HTMLParser()

#Logging variables
writeLinkLog = open("linkLog.txt", "a")
readLinkLog = open("linkLog.txt", "r")

postedLinks = [link.rstrip("\n") for link in readLinkLog]

readLinkLog.close()

#Twitter variables
TWITTER_CONSUMER_KEY = os.environ["PUBGBOT_CONSUMER_KEY"]
TWITTER_CONSUMER_SECRET = os.environ["PUBGBOT_CONSUMER_SECRET"]
TWITTER_ACCESS_KEY = os.environ["PUBGBOT_ACCESS_KEY"]
TWITTER_ACCESS_SECRET = os.environ["PUBGBOT_ACCESS_SECRET"]
auth = tweepy.OAuthHandler(TWITTER_CONSUMER_KEY, TWITTER_CONSUMER_SECRET)
auth.set_access_token(TWITTER_ACCESS_KEY, TWITTER_ACCESS_SECRET)
api = tweepy.API(auth)
pubg_twitter = 882248998547070981
recent_tweets = api.user_timeline(user_id=pubg_twitter, page=1, tweet_mode="extended")

#Reddit variables
REDDIT_CLIENT_ID = os.environ["PUBGBOT_REDDIT_CLIENT_ID"]
REDDIT_CLIENT_SECRET = os.environ["PUBGBOT_REDDIT_CLIENT_SECRET"]
REDDIT_USERNAME = os.environ["REDDIT_USERNAME"]
REDDIT_PASSWORD = os.environ["REDDIT_PASSWORD"]
reddit = praw.Reddit(client_id=REDDIT_CLIENT_ID, client_secret=REDDIT_CLIENT_SECRET, password=REDDIT_PASSWORD, user_agent="PUBG Twitter Bot 0.1", username=REDDIT_USERNAME)
subreddit = reddit.subreddit("PUBATTLEGROUNDS")

for t in recent_tweets:
    recent_tweet = h.unescape(t.full_text)
    recent_tweet_url = "https://twitter.com/statuses/" + str(t.id)
    recent_tweet_id = str(t.id)
    reply_status = t.in_reply_to_status_id
    steamLinks = list(url["expanded_url"] for url in t.entities["urls"] if "steamcommunity" in url["expanded_url"])
    if reply_status is None or int(api.get_status(reply_status).user.id_str) == pubg_twitter:
        if len(steamLinks) > 0:
            for url in steamLinks:
                r = requests.get(url).content
                titleBaseIndex = r.index('<h2 class="large_title"')
                titleStartIndex = r.index('>', titleBaseIndex) + 1
                titleEndIndex = r.index('<', titleStartIndex)
                title = r[titleStartIndex:titleEndIndex]
                if ("PC 1.0 Update" in title or "Dev Letter" in title or "Dev Blog" in title or "Patch Notes" in title) and not any(url in l for l in postedLinks):
                    writeLinkLog.write(url + "\n")
                    video_indexes = list(substring_indexes("dynamiclink_box", r))
                    transcription_comment = "Transcription of post for those unable to view - \n" + md(str(body_from_post(r))).replace("![]", "\n\n[Image]").replace(" *", "*").replace("***", "**\n*")
                    for i in video_indexes:
                        linkStartIndex = r.index("&quot;", i) + 6
                        linkEndIndex = r.index("&quot;", linkStartIndex)
                        video_id = r[linkStartIndex:linkEndIndex]
                        link = "https://www.youtube.com/watch?v=" + video_id
                        linkTitleStartIndex = r.index('<\/span>', linkEndIndex) + 8
                        linkTitleEndIndex = r.index("&nbsp;", linkTitleStartIndex)
                        linkTitle = r[linkTitleStartIndex:linkTitleEndIndex]
                        transcription_comment = transcription_comment.replace("<" + "https://youtu.be/" + video_id + ">", "VIDEO - [" + linkTitle + "]" + "(" + link +")")
                    post = subreddit.submit(title, url=url).flair.select("ba18c8f4-14b8-11e7-a6c6-0e11d8c4f614", text=None)
                    chars = len(transcription_comment)
                    comment = post
                    while chars > 10000:
                        cutoffIndex = transcription_comment.rfind("\n", 0, 10001)
                        comment = post.reply(transcription_comment[:cutoffIndex])
                        transcription_comment = transcription_comment[cutoffIndex:]
                        chars = len(transcription_comment)
                    comment.reply(transcription_comment)
writeLinkLog.close()