Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- #!/usr/bin/env python
- # -*- coding: utf-8 -*-
- """
- SleekXMPP: The Sleek XMPP Library
- Copyright (C) 2010 Nathanael C. Fritz
- This file is part of SleekXMPP.
- See the file LICENSE for copying permission.
- """
- import sys
- import logging
- import getpass
- from optparse import OptionParser
- from sleekxmpp.exceptions import IqError, IqTimeout
- import sqlite3
- import time
- import datetime
- import re
- import urllib2
- import chardet
- import feedparser
- import sleekxmpp
- # Forcing UTF8
- if sys.version_info < (3, 0):
- from sleekxmpp.util.misc_ops import setdefaultencoding
- setdefaultencoding('utf8')
- else:
- raw_input = input
- class MUCBot(sleekxmpp.ClientXMPP):
- regex_url = re.compile(r"(https?://\S+)\.(\S+)")
- regex_title = re.compile(r"<title>([^<]+)</title>")
- url_ext_blacklist = [
- 'png', 'jpg', 'jpeg', 'gif', 'png', 'pdf', 'doc', 'xls',
- 'docx', 'djvu', 'ppt', 'pptx', 'avi', 'mp4', 'mp3', 'flac', 'pps',
- 'mp3', 'ogg', 'webm'
- ]
- def __init__(self, jid, password, room, nick):
- sleekxmpp.ClientXMPP.__init__(self, jid, password)
- self.room = room
- self.nick = nick
- self.add_event_handler("session_start", self.start)
- self.add_event_handler("groupchat_message", self.muc_message)
- self.schedule('RSS update', # Unique name for the timer
- 600, # Seconds to delay before firing
- self.rss_update, # Callback to execute
- args=(), # A tuple of positional argument values
- kwargs={}, # A dictionary of keyword argument values
- repeat=True) # Make the event happen every X seconds
- def rss_update(self):
- url_rss = 'http://rulinux.net/rss'
- rss_depth_time = 600
- current_time = datetime.datetime.utcnow()
- try:
- rss = feedparser.parse(url_rss)
- for r in rss.entries:
- time_rss = datetime.datetime.fromtimestamp(time.mktime(r.published_parsed))
- print current_time
- t = current_time - time_rss
- #TODO: normal time pars not like {-(3600*3)} for remove 3 hour
- time_delta = t.total_seconds()-(3600*3)
- if (0 < time_delta) and (time_delta < rss_depth_time):
- news_title = unicode(r.title)
- news_link = unicode(r.link)
- self.send_message(mto="rulinux@conference.jabber.ru",
- mbody="RULIN: %s %s" % (news_title, news_link),
- mtype='groupchat')
- else:
- break
- except:
- logging.info("DEBUG: RSS Feed problem, %s" , sys.exc_info()[0])
- def write_log(self,query):
- #TODO: Сделать раоту базы в тредах иначе приходится на каждый тред открывать и закрывать
- #SQLite objects created in a thread can only be used in that same thread
- connect_db = sqlite3.connect('./rulinux_xmpp_chat_logs.db')
- cursor_db = connect_db.cursor()
- cursor_db.execute("""INSERT INTO chat_log (id, time, jit, name, message)
- VALUES (NULL,?,?,?,?)""", query)
- connect_db.commit()
- connect_db.close()
- def start(self, event):
- self.get_roster()
- self.send_presence()
- self.plugin['xep_0045'].joinMUC(self.room,
- self.nick,
- # If a room password is needed, use:
- # password=the_room_password,
- wait=True)
- def muc_message(self, msg):
- # Cmopose data and write message to SQLite DB
- nick_name = unicode(msg['mucnick']) # User nick sowed in chat room
- full_jit = unicode(msg['from']) # Like user@jabb.en/UserName
- message = unicode(msg['body']) # Message body
- query = (time.time(), full_jit, nick_name, message,)
- self.write_log(query)
- # Disable self-interaction
- if msg['mucnick'] == self.nick:
- return
- # Try to run command
- if msg['body'].startswith(self.nick):
- tokens = msg['body'].split()
- if len(tokens) > 1:
- command = tokens[1]
- if command == 'ping':
- self.try_ping(msg['from'], msg['mucnick'])
- elif "http" in msg['body']:
- self.try_say_url_info(msg['body'], msg['from'])
- def try_say_url_info(self, text, mucjid):
- try:
- parse_result = re.search(self.regex_url, text)
- if not parse_result:
- logging.info("DEBUG: cant find URL")
- raise
- url_ext = parse_result.group(2).lower()
- if url_ext in self.url_ext_blacklist:
- logging.info("DEBUG: url extension blocked")
- raise
- url = parse_result.group(1) + "." + parse_result.group(2)
- req = urllib2.Request(url)
- #TODO: добавить ограничение на очередь запросов
- response = urllib2.urlopen(req, timeout = 2)
- #6000 MAGIC number for header in bytes, working on youtube
- data = response.read(6000)
- enc = chardet.detect(data)
- data = data.decode(enc['encoding'], errors='ignore')
- title = re.search(self.regex_title, data).group(1)
- self.send_message(mto=mucjid.bare,
- mbody="TITLE: %s" % (title),
- mtype='groupchat')
- except: #IOError: IndexError, urllib2.URLError
- logging.info("DEBUG URL Rarsing error: %s", sys.exc_info()[0])
- def try_ping(self, pingjid, nick):
- try:
- rtt = self['xep_0199'].ping(pingjid,
- timeout=10)
- self.send_message(mto=pingjid.bare,
- mbody="%s, pong is: %s" % (nick, rtt),
- mtype='groupchat')
- #logging.info("Success! RTT: %s", rtt)
- except IqError as e:
- logging.info("Error pinging %s: %s",
- pingjid,
- e.iq['error']['condition'])
- except IqTimeout:
- logging.info("No response from %s", pingjid)
- if __name__ == '__main__':
- optp = OptionParser()
- # Output verbosity options.
- optp.add_option('-q', '--quiet', help='set logging to ERROR',
- action='store_const', dest='loglevel',
- const=logging.ERROR, default=logging.INFO)
- optp.add_option('-d', '--debug', help='set logging to DEBUG',
- action='store_const', dest='loglevel',
- const=logging.DEBUG, default=logging.INFO)
- optp.add_option('-v', '--verbose', help='set logging to COMM',
- action='store_const', dest='loglevel',
- const=5, default=logging.INFO)
- # JID and password options.
- optp.add_option("-j", "--jid", dest="jid",
- help="JID to use")
- optp.add_option("-p", "--password", dest="password",
- help="password to use")
- optp.add_option("-r", "--room", dest="room",
- help="MUC room to join")
- optp.add_option("-n", "--nick", dest="nick",
- help="MUC nickname")
- opts, args = optp.parse_args()
- # Setup logging.
- logging.basicConfig(level=opts.loglevel,
- format='%(levelname)-8s %(message)s')
- if opts.jid is None:
- opts.jid = raw_input("Username: ")
- if opts.password is None:
- opts.password = getpass.getpass("Password: ")
- if opts.room is None:
- opts.room = raw_input("MUC room: ")
- if opts.nick is None:
- opts.nick = raw_input("MUC nickname: ")
- # Setup the MUCBot and register plugins.
- xmpp = MUCBot(opts.jid, opts.password, opts.room, opts.nick)
- xmpp.register_plugin('xep_0030') # Service Discovery
- xmpp.register_plugin('xep_0045') # Multi-User Chat
- xmpp.register_plugin('xep_0199') # XMPP Ping
- xmpp.register_plugin('xep_0004') # Data Forms
- xmpp.register_plugin('xep_0060') # PubSub
- # Connect to the XMPP server and start processing XMPP stanzas.
- if xmpp.connect():
- xmpp.process(block=True)
- print("Done")
- else:
- print("Unable to connect.")
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement