FilesExpand file tree

 master

/

tweet.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

executable file

·

149 lines (113 loc) · 4.35 KB

 master

/

tweet.py

Copy path

Top

File metadata and controls

• Code
• Blame

executable file

·

149 lines (113 loc) · 4.35 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

#!/usr/bin/env python

'''Post a message to twitter'''

__author__ = 'dewitt@google.com'

import ConfigParser

import getopt

import os

import sys

import twitter

USAGE = '''Usage: tweet [options] message

This script posts a message to Twitter.

Options:

-h --help : print this help

--consumer-key : the twitter consumer key

--consumer-secret : the twitter consumer secret

--access-key : the twitter access token key

--access-secret : the twitter access token secret

--encoding : the character set encoding used in input strings, e.g. "utf-8". [optional]

Documentation:

If either of the command line flags are not present, the environment

variables TWEETUSERNAME and TWEETPASSWORD will then be checked for your

consumer_key or consumer_secret, respectively.

If neither the command line flags nor the environment variables are

present, the .tweetrc file, if it exists, can be used to set the

default consumer_key and consumer_secret. The file should contain the

following three lines, replacing *consumer_key* with your consumer key, and

*consumer_secret* with your consumer secret:

A skeletal .tweetrc file:

[Tweet]

consumer_key: *consumer_key*

consumer_secret: *consumer_password*

access_key: *access_key*

access_secret: *access_password*

'''

def PrintUsageAndExit():

print USAGE

sys.exit(2)

def GetConsumerKeyEnv():

return os.environ.get("TWEETUSERNAME", None)

def GetConsumerSecretEnv():

return os.environ.get("TWEETPASSWORD", None)

def GetAccessKeyEnv():

return os.environ.get("TWEETACCESSKEY", None)

def GetAccessSecretEnv():

return os.environ.get("TWEETACCESSSECRET", None)

class TweetRc(object):

def __init__(self):

self._config = None

def GetConsumerKey(self):

return self._GetOption('consumer_key')

def GetConsumerSecret(self):

return self._GetOption('consumer_secret')

def GetAccessKey(self):

return self._GetOption('access_key')

def GetAccessSecret(self):

return self._GetOption('access_secret')

def _GetOption(self, option):

try:

return self._GetConfig().get('Tweet', option)

except:

return None

def _GetConfig(self):

if not self._config:

self._config = ConfigParser.ConfigParser()

self._config.read(os.path.expanduser('~/.tweetrc'))

return self._config

def main():

try:

shortflags = 'h'

longflags = ['help', 'consumer-key=', 'consumer-secret=',

'access-key=', 'access-secret=', 'encoding=']

opts, args = getopt.gnu_getopt(sys.argv[1:], shortflags, longflags)

except getopt.GetoptError:

PrintUsageAndExit()

consumer_keyflag = None

consumer_secretflag = None

access_keyflag = None

access_secretflag = None

encoding = None

for o, a in opts:

if o in ("-h", "--help"):

PrintUsageAndExit()

if o in ("--consumer-key"):

consumer_keyflag = a

if o in ("--consumer-secret"):

consumer_secretflag = a

if o in ("--access-key"):

access_keyflag = a

if o in ("--access-secret"):

access_secretflag = a

if o in ("--encoding"):

encoding = a

message = ' '.join(args)

if not message:

PrintUsageAndExit()

rc = TweetRc()

consumer_key = consumer_keyflag or GetConsumerKeyEnv() or rc.GetConsumerKey()

consumer_secret = consumer_secretflag or GetConsumerSecretEnv() or rc.GetConsumerSecret()

access_key = access_keyflag or GetAccessKeyEnv() or rc.GetAccessKey()

access_secret = access_secretflag or GetAccessSecretEnv() or rc.GetAccessSecret()

if not consumer_key or not consumer_secret or not access_key or not access_secret:

PrintUsageAndExit()

api = twitter.Api(consumer_key=consumer_key, consumer_secret=consumer_secret,

access_token_key=access_key, access_token_secret=access_secret,

input_encoding=encoding)

try:

status = api.PostUpdate(message)

except UnicodeDecodeError:

print "Your message could not be encoded. Perhaps it contains non-ASCII characters? "

print "Try explicitly specifying the encoding with the --encoding flag"

sys.exit(2)

print "%s just posted: %s" % (status.user.name, status.text)

if __name__ == "__main__":

main()