Port to newer praw

Obviously, everything changed in various ways and the doc was useless
This commit is contained in:
moneromooo 2017-10-19 21:44:03 +01:00
parent 55905122fb
commit 08691aadbb
1 changed files with 42 additions and 25 deletions

View File

@ -15,6 +15,7 @@ import time
import threading import threading
import re import re
import praw import praw
import logging
import tipbot.config as config import tipbot.config as config
from tipbot.log import log_error, log_warn, log_info, log_log from tipbot.log import log_error, log_warn, log_info, log_log
from tipbot.user import User from tipbot.user import User
@ -44,7 +45,7 @@ class RedditNetwork(Network):
try: try:
cfg=config.network_config[self.name] cfg=config.network_config[self.name]
self.login=cfg['login'] self.login=cfg['login']
password=GetPassword(self.name) password=GetPassword(self.name+'/password')
self.subreddits=cfg['subreddits'] self.subreddits=cfg['subreddits']
user_agent=cfg['user_agent'] user_agent=cfg['user_agent']
self.update_period=cfg['update_period'] self.update_period=cfg['update_period']
@ -52,9 +53,19 @@ class RedditNetwork(Network):
self.keyword=cfg['keyword'] self.keyword=cfg['keyword']
self.use_unread_api=cfg['use_unread_api'] self.use_unread_api=cfg['use_unread_api']
self.cache_timeout=cfg['cache_timeout'] self.cache_timeout=cfg['cache_timeout']
client_id=GetPassword(self.name+'/client_id')
client_secret=GetPassword(self.name+'/client_secret')
username=GetPassword(self.name+'/username')
self.reddit=praw.Reddit(user_agent=user_agent,cache_timeout=self.cache_timeout) if False:
self.reddit.login(self.login,password) handler = logging.StreamHandler()
handler.setLevel(logging.DEBUG)
logger = logging.getLogger('prawcore')
logger.setLevel(logging.DEBUG)
logger.addHandler(handler)
self.reddit=praw.Reddit(client_id=client_id,client_secret=client_secret,password=password,user_agent=user_agent,username=username)
log_info("Logged in reddit as " + str(self.reddit.user.me()))
self.items_cache=dict() self.items_cache=dict()
self.stop = False self.stop = False
@ -107,18 +118,23 @@ class RedditNetwork(Network):
return return
if not hasattr(item.author,'name'): if not hasattr(item.author,'name'):
log_warn('author of %s has no name field, ignored' % str(item.id)) log_warn('author of %s has no name field, ignored' % str(item.id))
try: if True:
item.mark_as_read() try:
except Exception,e: item.mark_read()
log_warning('Failed to mark %s as read: %s' % (item.id,str(e))) except Exception,e:
log_warn('Failed to mark %s as read: %s' % (item.id,str(e)))
return return
author=self.canonicalize(item.author.name) author=self.canonicalize(item.author.name)
if author==self.canonicalize(self.login): if author and author==self.canonicalize(self.login):
return return
if item.id in self.last_seen_ids: if item.id in self.last_seen_ids:
#log_log('Already seen %s %.1f hours ago by %s: %s (%s), skipping' % (item.id,age/3600,str(author),repr(title),repr(item.body))) log_log('Already seen %s %.1f hours ago by %s: %s (%s), skipping' % (item.id,age/3600,str(author),repr(title),repr(item.body)))
try:
item.mark_read()
except Exception,e:
log_warn('Failed to mark %s as read: %s' % (item.id,str(e)))
return return
age=time.time()-item.created_utc age=time.time()-item.created_utc
@ -155,7 +171,7 @@ class RedditNetwork(Network):
line=line.replace(self.keyword,'').strip() line=line.replace(self.keyword,'').strip()
if self.on_command: if self.on_command:
try: try:
parent_item=self.reddit.get_info(thing_id=item.parent_id) parent_item=next(self.reddit.info([item.parent_id]))
if not hasattr(parent_item,'author'): if not hasattr(parent_item,'author'):
raise RuntimeError('Parent item has no author') raise RuntimeError('Parent item has no author')
author=parent_item.author.name author=parent_item.author.name
@ -167,10 +183,11 @@ class RedditNetwork(Network):
self.on_command(link,synthetic_cmd) self.on_command(link,synthetic_cmd)
except Exception,e: except Exception,e:
log_error('Failed to tip %s\'s parent: %s' % (item.id,str(e))) log_error('Failed to tip %s\'s parent: %s' % (item.id,str(e)))
try: if True:
item.mark_as_read() try:
except Exception,e: item.mark_read()
log_warning('Failed to mark %s as read: %s' % (item.id,str(e))) except Exception,e:
log_warn('Failed to mark %s as read: %s' % (item.id,str(e)))
def _schedule_reply(self,item,recipient,text): def _schedule_reply(self,item,recipient,text):
log_log('scheduling reply to %s:%s: %s' % (item.id if item else '""',recipient or '""',text)) log_log('scheduling reply to %s:%s: %s' % (item.id if item else '""',recipient or '""',text))
@ -214,7 +231,10 @@ class RedditNetwork(Network):
if fullname in self.items_cache: if fullname in self.items_cache:
item=self.items_cache[fullname] item=self.items_cache[fullname]
if not item: if not item:
item=self.reddit.get_info(thing_id=fullname) item = self.reddit.mesage(fullname)
if not item:
gen=self.reddit.info([fullname])
item=next(gen, None)
if not item: if not item:
log_error('Failed to find item %s to post %s' % (fullname,text)) log_error('Failed to find item %s to post %s' % (fullname,text))
redis_lpop('reddit:replies') redis_lpop('reddit:replies')
@ -226,9 +246,6 @@ class RedditNetwork(Network):
redis_lpop('reddit:replies') redis_lpop('reddit:replies')
except praw.errors.RateLimitExceeded,e:
log_info('Rate limited trying to send %s, will retry: %s' % (data,str(e)))
return False
except Exception,e: except Exception,e:
log_error('Error sending %s, will retry: %s' % (data,str(e))) log_error('Error sending %s, will retry: %s' % (data,str(e)))
return False return False
@ -256,15 +273,15 @@ class RedditNetwork(Network):
self._parse(message,not message.was_comment) self._parse(message,not message.was_comment)
else: else:
messages=self.reddit.get_inbox() for message in self.reddit.inbox.unread(limit=self.load_limit):
for message in messages: #if not message.was_comment:
if not message.was_comment:
self._parse(message,True) self._parse(message,True)
sr=self.reddit.get_subreddit("+".join(self.subreddits)) #print "Submissions from %s" % ("+".join(self.subreddits))
comments=sr.get_comments(limit=self.load_limit) #sr=self.reddit.subreddit("+".join(self.subreddits))
for comment in comments: #for s in sr.new(limit=self.load_limit):
self._parse(comment,False) # for comment in s.comments:
# self._parse(comment,False)
while self._post_next_reply(): while self._post_next_reply():
pass pass