ICQ Python bot

Hi % habrauser% ! This is my first habrotopik. In which I will tell you how to make an ICQ bot on pythone. The bot will sit in the network and send, at the user's request, a list of the latest articles of the habr.
To implement the bot, we will use the Twisted library. Of course, there are already specialized libraries for working with ICQ on python. Such as py-icq or nanoicq . But since I was already familiar with Twisted, the choice fell on him.

In order not to reinvent the wheel, take the lion's part of the program from the example :

# UIN UIN = "" # PASS = "" # host = ( "login.icq.com" , 5238) # / icqMode = 1 # Status message AMSG = "I'm here +)" * This source code was highlighted with Source Code Highlighter .

1. # UIN UIN = "" # PASS = "" # host = ( "login.icq.com" , 5238) # / icqMode = 1 # Status message AMSG = "I'm here +)" * This source code was highlighted with Source Code Highlighter .
2. # UIN UIN = "" # PASS = "" # host = ( "login.icq.com" , 5238) # / icqMode = 1 # Status message AMSG = "I'm here +)" * This source code was highlighted with Source Code Highlighter .
3. # UIN UIN = "" # PASS = "" # host = ( "login.icq.com" , 5238) # / icqMode = 1 # Status message AMSG = "I'm here +)" * This source code was highlighted with Source Code Highlighter .
4. # UIN UIN = "" # PASS = "" # host = ( "login.icq.com" , 5238) # / icqMode = 1 # Status message AMSG = "I'm here +)" * This source code was highlighted with Source Code Highlighter .
5. # UIN UIN = "" # PASS = "" # host = ( "login.icq.com" , 5238) # / icqMode = 1 # Status message AMSG = "I'm here +)" * This source code was highlighted with Source Code Highlighter .
6. # UIN UIN = "" # PASS = "" # host = ( "login.icq.com" , 5238) # / icqMode = 1 # Status message AMSG = "I'm here +)" * This source code was highlighted with Source Code Highlighter .
7. # UIN UIN = "" # PASS = "" # host = ( "login.icq.com" , 5238) # / icqMode = 1 # Status message AMSG = "I'm here +)" * This source code was highlighted with Source Code Highlighter .

# UIN UIN = "" # PASS = "" # host = ( "login.icq.com" , 5238) # / icqMode = 1 # Status message AMSG = "I'm here +)" * This source code was highlighted with Source Code Highlighter .

Here the initial part of us and ends. Download the necessary libraries.
')

1. from twisted.words.protocols import oscar
2. from twisted.internet import protocol, reactor
3. import feedparser
4. import urllib2 # Please do not bother with this. For what it is, I'll tell you later

* This source code was highlighted with Source Code Highlighter .

Feedparser we need to pull off and otparsit rss tape habr. Now the bot itself. Here, almost all of the example, with the exception of minor changes ...

1. # Class Bot
2. class Bot (oscar.BOSConnection):
3. capabilities = [oscar.CAP_CHAT]
4. def initDone (self):
5. print "Connect" , UIN, "to server" , host [0], host [1]
6. self.requestSelfInfo (). addCallback (self.gotSelfInfo)
7. self.requestSSI (). addCallback (self.gotBuddyList)
8. self.setAway (AMSG)
9. def gotSelfInfo (self, user):
10. print user .__ dict__
11. self.name = user.name
12. def gotBuddyList (self, l):
13. print l
14. self.activateSSI ()
15. self.setProfile ( "" "ICQBot" "" )
16. self.setIdleTime (0)
17. self.clientReady ()
18. def gotAway (self, away, user):
19. if away:
20. print "User" , user, ":" , away

* This source code was highlighted with Source Code Highlighter .

This is where we have the tastiest part. Inbound processing:

1. def receiveMessage (self, user, multiparts, flags):
2. print "\ n <From:" , user.name
3. print "<Message:" , multiparts [0] [0] .decode ( 'cp1251' )
4. command = multiparts [0] [0] .lower (). split ( '' ) # split the message by spaces
5. PREF_C = "!" # teams will start with us!

* This source code was highlighted with Source Code Highlighter .

We will have the first team! Habr, but there are some minor exceptions. If the command syntax! Habr n, then the last n posts will be displayed. If the syntax command is! Habr n1 n2, then from n1 to n2 posts (my little whim). Of course, the output is limited to 20 posts, because In their xml page it is exactly 20.

1. if command [0] == (PREF_C + "habr" ):
2. rss = feedparser.parse ( 'http://habrahabr.ru/rss' ) # pick up the xml page and parsim
3. feeds = [0.20] # This is for additional functionality here)
4. if len (command) == 2:
5. try :
6. feeds [1] = int (command [1])
7. except:
8. feeds [1] = 0
9. if feeds [1] <1: feeds [1] = 1
10. if feeds [1]> 20: feeds [1] = 20
11. elif len (command) == 3:
12. try :
13. feeds [0] = int (command [1]) - 1
14. feeds [1] = int (command [2])
15. except:
16. feeds [0] = 0
17. feeds [1] = 19
18. if feeds [0] <0: feeds [0] = 0
19. if feeds [0]> 19: feeds [0] = 19
20. if feeds [1] <1: feeds [1] = 1
21. if feeds [1]> 20: feeds [1] = 20
22. mes = ''
23. for feed in range (feeds [0], feeds [1]):
24. mes + = rss.entries [feed] .title + '\ n'
25. self.sendMessage (user.name, mes)

* This source code was highlighted with Source Code Highlighter .

The command! Habrn n will display the contents of the description field of the n field.

1. elif command [0] == (PREF_C + "habrn" ):
2. rss = feedparser.parse ( 'http://habrahabr.ru/rss' )
3. try :
4. feedn = int (command [1]) - 1
5. except:
6. feedn = 0
7. if feedn <0: feedn = 0
8. if feedn> 19: feedn = 19
9. self.sendMessage (user.name, rss.entries [feedn] .title + '\ n' + rss.entries [feedn] .description)

* This source code was highlighted with Source Code Highlighter .

Here I want to mention import urllib2, which I entered at the beginning. Sometimes I need to know the ip of my home server in order to log in via ssh. And of course, I would not want anyone else to know his address.

1. elif command [0] == (PREF_C + "ip" ) and user.name == '368576236' :
2. file_s = urllib2.urlopen (urllib2.Request ( 'http://api.wipmania.com/' ))
3. response = file_s.read ()
4. self.sendMessage (user.name, response.split ( "<br>" ) [0])
5. f.close ()

* This source code was highlighted with Source Code Highlighter .

If you look at http://api.wipmania.com/ , then you can find a pretty simple html page which will have our external ip.

Well, the final part:

1. class BotAuth (oscar.OscarAuthenticator):
2. BOSClass = Bot
3. protocol.ClientCreator (reactor, BotAuth, UIN, PASS, icq = icqMode) .connectTCP (* host)
4. reactor.run ()

* This source code was highlighted with Source Code Highlighter .

That's all. I want to say thanks to the person who helped get the ip for the last command.

Here is the whole code:

1. # - * - coding: utf-8 - * -
2. # ICQ bot
4. #Uin
5. UIN = "123456"
6. PASS = "pass"
9. # Server
10. host = ( "login.icq.com" , 5238)
11. icqMode = 1
13. # Status message
14. AMSG = "I'm here +)"
16. from twisted.words.protocols import oscar
17. from twisted.internet import protocol, reactor
18. import feedparser
19. import urllib2
21. # Class Bot
22. class Bot (oscar.BOSConnection):
24. capabilities = [oscar.CAP_CHAT]
26. def initDone (self):
27. print "Connect" , UIN, "to server" , host [0], host [1]
29. self.requestSelfInfo (). addCallback (self.gotSelfInfo)
30. self.requestSSI (). addCallback (self.gotBuddyList)
32. self.setAway (AMSG)
34. def gotSelfInfo (self, user):
35. print user .__ dict__
36. self.name = user.name
39. def gotBuddyList (self, l):
40. print l
41. self.activateSSI ()
42. self.setProfile ( "" "ICQBot" "" )
43. self.setIdleTime (0)
44. self.clientReady ()
46. def gotAway (self, away, user):
47. if away:
48. print "User" , user, ":" , away
50. def receiveMessage (self, user, multiparts, flags):
51. print "\ n <From:" , user.name
52. print "<Message:" , multiparts [0] [0] .decode ( 'cp1251' )
53. command = multiparts [0] [0] .lower (). split ( '' )
56. PREF_C = "!"
57. if command [0] == (PREF_C + "habr" ):
58. rss = feedparser.parse ( 'http://habrahabr.ru/rss' )
59. feeds = [0.20]
60. if len (command) == 2:
61. try :
62. feeds [1] = int (command [1])
63. except:
64. feeds [1] = 0
66. if feeds [1] <1: feeds [1] = 1
67. if feeds [1]> 20: feeds [1] = 20
69. elif len (command) == 3:
70. try :
71. feeds [0] = int (command [1]) - 1
72. feeds [1] = int (command [2])
73. except:
74. feeds [0] = 0
75. feeds [1] = 19
77. if feeds [0] <0: feeds [0] = 0
78. if feeds [0]> 19: feeds [0] = 19
79. if feeds [1] <1: feeds [1] = 1
80. if feeds [1]> 20: feeds [1] = 20
82. mes = ''
83. for feed in range (feeds [0], feeds [1]):
84. mes + = rss.entries [feed] .title + '\ n'
86. self.sendMessage (user.name, mes)
89. elif command [0] == (PREF_C + "habrn" ):
90. rss = feedparser.parse ( 'http://habrahabr.ru/rss' )
91. try :
92. feedn = int (command [1]) - 1
93. except:
94. feedn = 0
95. if feedn <0: feedn = 0
96. if feedn> 19: feedn = 19
98. self.sendMessage (user.name, rss.entries [feedn] .title + '\ n' + rss.entries [feedn] .description)
101. elif command [0] == (PREF_C + "ip" ) and user.name == '123456' :
102. file_s = urllib2.urlopen (urllib2.Request ( 'http://api.wipmania.com/' ))
103. response = file_s.read ()
104. self.sendMessage (user.name, response.split ( " <br> " ) [0])
105. f.close ()
107. class BotAuth (oscar.OscarAuthenticator):
108. BOSClass = Bot
110. protocol.ClientCreator (reactor, BotAuth, UIN, PASS, icq = icqMode) .connectTCP (* host)
111. reactor.run ()

* This source code was highlighted with Source Code Highlighter .

Thanks for attention!