Пишем бота для twitter на python

Нашел все-таки время разобраться с OAuth авторизацией. Этот позволяет социальным сервисам интегрироваться между собой обеспечивая безопасный метод обмена информацией. Для интересующихся еще ссылка.

Приступим к написанию бота. Первое, что необходимо сделать - зарегистрировать приложение на сервисе twitter. В результате получаем Consumer key и Consumer secret.



Теперь разберем сам скрипт. Он написан на основе библиотеки oauth-python-twitter2, задача: фолловить и анфолловить случайных пользователей. Скрипт можно скачать здесь.

1. #!/usr/bin/env python
2. # -*- coding: utf-8 -*-
4. from oauth import oauth
5. from oauthtwitter import OAuthApi
6. import sys
7. import os
9. twitter = None
10. consumerKey = ''
11. consumerSecret = ''
13. def auth():
14.   authToken = None
15.   authSecret = None
16.   if os.path.exists('/tmp/twitter.tmp'):
17.     f = open('/tmp/twitter.tmp', 'r')
18.     authToken = f.readline().strip()
19.     authSecret = f.readline().strip()
20.     print "oauth_token: " + authToken
21.     print "oauth_token_secret: " + authSecret
22.     f.close()
23.   needAuth = True
24.   if authToken!=None and authSecret!=None:
25.     twitter = OAuthApi(consumerKey, consumerSecret, authToken, authSecret)
26.     if twitter.autorized():
27.       needAuth = False
29.   if needAuth:
30.     twitter = OAuthApi(consumerKey, consumerSecret)
32.     temp_credentials = twitter.getRequestToken()
33.     print temp_credentials
35.     print twitter.getAuthorizationURL(temp_credentials)
37.     oauth_verifier = raw_input('What is the PIN? ')
38.     access_token = twitter.getAccessToken(temp_credentials, oauth_verifier)
39.     print access_token
41.     print("oauth_token: " + access_token['oauth_token'])
42.     print("oauth_token_secret: " + access_token['oauth_token_secret'])
44.     f = open('/tmp/twitter.tmp', 'w')
45.     f.write('%s\n%s'%(access_token['oauth_token'], access_token['oauth_token_secret']))
46.     f.close()
48.     twitter = OAuthApi(consumerKey, consumerSecret, access_token['oauth_token'], access_token['oauth_token_secret'])
49.   return twitter
52. def unfollow():
53.   friends = twitter.GetFriendsIDs()
54.   for i in range(min(len(friends),30)):
55.     twitter.UnfollowUser(friends[i])
58. def follow(query):
59.   users = twitter.searchByQuery(query)
60.   count = min(30, len(users))
61.   for user in users:
62.     print user['id']
63.     twitter.FollowUser(user['id'])
66. if __name__=='__main__':
67.   if len(sys.argv)<2:
68.     print "Usage: %s <mode>"%sys.argv[0]
69.     print "Mode:"
70.     print "  follow   - find and follow by twitter users"
71.     print "  unfollow - unfollow twitter users"
72.     print ""
73.     sys.exit(0)
75.   if sys.argv[1]=='unfollow':
76.     twitter = auth()
77.     unfollow()
78.   elif sys.argv[1]=='follow':
79.     twitter = auth()
80.     follow('twitter')

самое интересное в скрипте, метод auth. В файле /tmp/twitter.tmp я храню токен авторизации пользователя. Если этот токен не существует или не принимается пользователем, то вывожу на экран урл авторизации и жду пока пользователь авторизуеться в твиттере и введет PIN в приложение. По этому пину и получаем токен авторизации который и сохраняем в файл /tmp/twitter.tmp до следующего запуска скрипта.

Методы follow вызывает метод search твиттера и фолловит найденых пользователей. Метод unfollow берет список фолловеров и отписываеться от некоторых из них.