twitter-sort/main.py at master · exPHAT/twitter-sort · 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
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
# Twitter-Sort
# main.py
# Aaron Taylor
#
# This sorting technique allows a program to pass this script a list of
# numbers, and be returned the sorted version of them. This works by tweeting
# the list to Twitter asking for them to be sorted, and when there is a
# reply, we check to ensure they're sorted. If they are, we return.

import tweepy
import sys
import os
from webbrowser import open as site  # For the auth token
import settings

keysPath = "keys.txt"
if len(sys.argv) == 1:
    print("Arguments required!")
    sys.exit()

numbers = eval("".join(sys.argv[1:]))

auth = tweepy.OAuthHandler(settings.API_KEY, settings.API_SECRET)

if os.path.exists(keysPath):
    keysFile = open(keysPath, "r+")
else:
    keysFile = open(keysPath, "a+")
    keysFile.seek(0)

if len(keysFile.read()) == 0:
    authUrl = auth.get_authorization_url()
    site(authUrl)
    authPIN = input("Enter your auth PIN: ")
    try:
        auth.get_access_token(str(authPIN))
    except tweepy.TweepErrori as e:
        print("Error: Getting access token failed: " + e.message)
        sys.exit()
    # Save for next call
    keysFile.write(auth.access_token + " " + auth.access_token_secret)
else:
    keysFile.seek(0)
    keys = keysFile.read().split(" ")
    if len(keys) == 2:
        auth.set_access_token(keys[0], keys[1])
    else:
        # The file is set up incorrectly
        print("Error: Something is wrong with", keysPath)
        sys.exit()

api = tweepy.API(auth)

tweetID = api.update_status(status="Can you sort these numbers? " + str(numbers)).id


class ReplyListener(tweepy.StreamListener):

    def on_status(self, status):
        # There is a status tweeted tagging the user

        if status.in_reply_to_status_id == tweetID:
            # There was a reply to the aaron-sort tweet
            parsedNumbers = "".join(status.text.split(" ")[1:])\
                .replace(" ", "")\
                .replace("[", "")\
                .replace("]", "")\
                .replace("(", "")\
                .replace(")", "").split(",")
            givenNumbers = list(map(int, parsedNumbers))

            # Ensure the given numbers are sorted

            areSorted = True
            if len(givenNumbers) != len(numbers):
                areSorted = False
            for n in givenNumbers:
                if givenNumbers.count(n) != numbers.count(n):
                    areSorted = False
                    break
            for i in range(len(givenNumbers)):
                if i > 0:
                    if not givenNumbers[i] >= givenNumbers[i - 1]:
                        areSorted = False
                        break

            if areSorted:
                print(givenNumbers)  # Print the sorted numbers to the console
                api.update_status(
                    "@" + status.author.screen_name + " Awesome! Thanks!",
                    in_reply_to_status_id=status.id)
                return False
            else:
                api.update_status(
                    "@" + status.author.screen_name +
                    " Those numbers aren't sorted!",
                    in_reply_to_status_id=status.id)
                return True
        else:
            # They did not reply to the aaron-sort tweet
            return True

    def on_error(self, statusCode):
        # There was a listener error

        print("There was a listener error with the code", statusCode)
        return True

    def on_timeout(self):
        # There was a listener timeout

        print("There was a listener timeout")
        return True

listener = tweepy.streaming.Stream(auth, ReplyListener())
# Listen for a response to the user
listener.filter(track=["@" + api.me().screen_name])