twweet-cli/main.py at master · pposca/twweet-cli · 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
from Twweeter import Twweeter
import os
import sys
from os.path import expanduser
from Listener import Listener

# Twitter API credentials
home = expanduser("~")


def _decorator(f):
    f()
    print( "DONE \n")
    return f


class TwweetCLI():

    def __init__(self):
        self.twweeterObj = None
        self.listenerObj = None

    def _decorator(self, f):
        self.f()
        print( "DONE \n")
        return self.f

    def check_data_dir_exists(self):
        try:
            original_umask = os.umask(0)
            os.makedirs(home+'/.twweet-cli/data')
        except OSError:
            pass
        finally:
            os.umask(original_umask)


    def home_select_action(self):

        option = input(
            '1.Get tweets of any user \n2.Get tweets of particular hashtag \n3.Get trending topics\n4.Read your timeline\n5.Get your followers list \n6.Get your tweets\nPress 99 to exit or press 66 to go back to main menu :: \n')
        if option == '99':
            sys.exit(0)
        if option == '66':
            print(('\n\n'))
            return False
        if option == '1':
            self.twweeterObj.get_all_tweets(input('Enter the username whose twweet\'s you want to grab '))
        elif option == '2':
            words=input('Enter the hashtag or word\nyou may enter multiple words/hashtags separated by a "," : ')
            self.listenerObj.streamWordOrHashtag(wordsList=words)
            # get_tweets_of_hashtag(input('Enter the hashtag : '))
        elif option == '3':
            self.twweeterObj.get_trending_topics()
        elif option == '4':
            print(('\nStreaming tweets from your TimeLine...'))
            self.listenerObj.streamYourTL()
            return False
            # readTimeLine(api)
        elif option == '5':
            self.twweeterObj.getFollowersList()
        elif option == '6':
            self.twweeterObj.getTweets()
        else:
            print(('please choose any of the above options\n \n'))


    def main(self):
        self.twweeterObj = Twweeter()
        self.listenerObj = Listener(self.twweeterObj)
        self.check_data_dir_exists()
        print(('Press 99 to quit the Application'))
        while True:
            option = input('Enter \'twweet\' or \'get\' or \'edit\': ')
            if option =='99':
                break
            if option == 'twweet':
                tweet = input('Enter your twweet\n')
                self.twweeterObj.api.update_status(status=tweet)
                # Yes, tweet is called 'status' rather confusing
            elif option == 'get':
                check = True
                while check:
                    check = self.home_select_action()
            elif option == 'edit':
                self.twweeterObj.createCreds()
            else :
                print(('Please choose any of the above options \n \n'))

def cli():
    Tww = TwweetCLI()
    Tww.main()

if __name__ == "__main__":
    cli()