def rss_add_channel(t, s, p):
global RSS_GLOB
global RSS_CACHE_FILE
if not p:
reply(t, s, u'Укажите url-адрес нового канала!')
return
d = feedparser.parse(p)
if d['status']!=200:
reply(t, s, u'Указанный адрес не может быть прочитан. Статус '+str(d['status']))
return
i = [x.lower() for x in RSS_GLOB.keys()]
if p.lower() in i:
reply(t, s, u'Такой url уже есть в списке!')
return
level = int(user_level(s[1]+'/'+s[2], s[1]))
#if level<40:
# reply(t, s, u'Заявка на добавление вашего канала отправлена админам бота!')
# for x in [x for x in GLOBACCESS.keys() if GLOBACCESS[x]==100]:
# msg(s[3], x, u'Заявка на добавление RSS-канала от пользователя '+s[1]+'/'+s[2]+u'\n Наберите !rss+ '+p+u' для добавления!')
# return
title = ('' if not d.feed.get('title','') else '('+d.feed['title']+')')
RSS_GLOB[p] = {'sub':{}, 'ids':[], 'title':title, 'show':0, 'cljid':s[3]}
write_file(RSS_CACHE_FILE, str(RSS_GLOB))
reply(t, s, u'Источник " '+p+u' " '+title+u' успешно добавлен!')
def rss_reader():
global RSS_HASH_FILE
global RSS_GLOB
hm = eval(read_file(RSS_HASH_FILE))
page, bjid = '', ''
while RSS_THREAD:
time.sleep(120)
try:
list = [x for x in RSS_GLOB.keys() if RSS_GLOB[x]['sub']]
for x in list:
try: c = feedparser.parse(x)
except: continue
try:
if c['status']!=200:
continue
except: continue
time.sleep(2.5)
j = c.entries
random.shuffle(j)
for i in j:#2
tit = rss_md5(i.get('title','-'))
if not tit in hm:#3 RSS_GLOB[x]['ids']:
if len(hm)>1005:
hm = hm[-15:]
hm.append(tit)
for user in RSS_GLOB[x]['sub']:#4
bjid = RSS_GLOB[x]['sub'][user]
if user.count('@con') | user.count('@chat') | user.count('@muc'):#5
if not user in GROUPCHATS:
del RSS_GLOB[x]['sub'][user]
write_file(RSS_CACHE_FILE, str(RSS_GLOB))
continue
else:
if GROUPCHATS[user][get_bot_nick(user)]['ismoder']:
bjid = GROUPCHATS[user][get_bot_nick(user)]['jid']
bjid = bjid.split('/')[0]
if c['encoding'] not in ['UTF-8','utf8','utf-8','ascii']:
try:
page = '\n'.join([c.feed['title'],i.title, i.summary, i.link])
page = page.encode(c['encoding'])
page = page.decode(c['encoding'],'replace')
page = 'RSS '+page
except:
continue
if bjid in CLIENTS:
msg(bjid, user, ufnsc(page))
else:
page = '\n'.join(['RSS '+c.feed['title'],i.title, i.summary, i.link])
if bjid in CLIENTS: msg(bjid, user, ufnsc(page))
break
except Exception as err:
print 'error in RSS'
write_file(RSS_HASH_FILE, str(hm))
def addrss(path):
rssurl = path.group().replace('rss:','')
import rss.feedparser as feedparser
data = feedparser.parse(rssurl)
values = "<h2>%s</h2><br/>" % (data.feed.title)
values += "%s<br/>" % (data.channel.description)
for entry in data['entries']:
values += "<h3><a href='%s'> %s </a></h3><br/>" % (entry.link, entry.title)
values += "%s <br/>" % (entry.summary)
return values
def addrss(path):
rssurl = path.group().replace('rss:', '')
import rss.feedparser as feedparser
data = feedparser.parse(rssurl)
values = "<h2>%s</h2><br/>" % (data.feed.title)
values += "%s<br/>" % (data.channel.description)
for entry in data['entries']:
values += "<h3><a href='%s'> %s </a></h3><br/>" % (entry.link,
entry.title)
values += "%s <br/>" % (entry.summary)
return values
def rss_manager(t, s, p):
global RSS_THREAD
if not p:
rss_lists(t, s, p)
return
if p.split()>1 and p.split()[0].lower() in ['?',u'кратко',u'short'] and p.split()[1].isdigit():
if not len(RSS_GLOB)>=int(p.split()[1]):
reply(t, s, u'Канала с таким номером не найдено!')
return
rss = feedparser.parse(RSS_GLOB.keys()[int(p.split()[1])-1])
if rss.entries:
rss = '\n- '.join([x.get('title','-') for x in rss.entries])
reply(t, s, '- '+rss)
return
jid = get_true_jid(s)
if p.isdigit() and p!='0':
if not len(RSS_GLOB)>=int(p):
reply(t, s, u'Канала с таким номером не найдено!')
return
if s[1] in GROUPCHATS:
level = int(user_level(s[1]+'/'+s[2], s[1]))
if level<20:
reply(t, s, u'Для подписки в чат у вас должен быть доступ админа(20)!')
return
jid = s[1]
if not jid in RSS_GLOB[RSS_GLOB.keys()[int(p)-1]]['sub']:
RSS_GLOB[RSS_GLOB.keys()[int(p)-1]]['sub'][jid]=s[3]
reply(t, s, jid+u' успешно подписан на '+RSS_GLOB.keys()[int(p)-1])
if not RSS_THREAD:
RSS_THREAD = 1
threading.Thread(None, rss_reader, 'rss_reader'+str(INFO['thr'])).start()
else:
del RSS_GLOB[RSS_GLOB.keys()[int(p)-1]]['sub'][jid]
reply(t, s, jid+u' удален из рассылки '+RSS_GLOB.keys()[int(p)-1])
write_file(RSS_CACHE_FILE, str(RSS_GLOB))
else:
p = p.lower()
if len(p)<3:
reply(t, s, u'Минимум 3 символа для поиска!')
return
rep = ''
list = []
k = 0
for x in RSS_GLOB.keys():
rss = feedparser.parse(x)
j = rss.entries
for i in j:
try: a = i.title+i.summary
except: continue
try:
a = a.encode(rss['encoding']).decode(rss['encoding'],'replace')
a = a.lower()
except: pass
if a.count(p):
k = 1
mm='\n'.join(['- '+i.title, i.summary, ' '+i.link])
try:
mm = mm.encode(rss['encoding'])
mm = mm.decode(rss['encoding'],'replace')
except: pass
hsh = rss_md5(mm)
if not hsh in list:
list.append(hsh)
rep+= ufnsc(mm)+'\n***\n'
reply(t, s, (rep.strip() if k else u'Ничего не найдено!'))
