def pull(variable):
driver = Driver()
driver.connect(mode='heroku')
query = 'select row_to_json({0}) from {0}'.format(variable)
result = driver.pull(query)
result = [x for t in result for x in t]
output_dct = {'content': result}
return jsonify(output_dct)
def overwrite(variable):
driver = Driver()
driver.connect(mode='heroku')
input_json = request.get_json(force=True)
input_json['table'] = variable
restart_query = render_template('restart.sql.jinja2', **input_json)
insert_query = render_template('jsondump.sql.jinja2', **input_json)
driver.push(restart_query)
driver.push(insert_query)
def push():
driver = Driver()
driver.connect(mode='heroku')
input_json = request.get_json(force=True)
query = render_template('jsondump.sql.jinja2', **input_json)
driver.push(query)
class Streamer(object):
def __init__(self,*args,**kwargs):
self.reddit = ''
self.account = ''
self.driver = Driver()
self.bot = ArchiveBot()
self.error = "There was an error processing your request."
def auth(self):
return praw.Reddit(client_id=self.account['CLIENT_ID'],
client_secret=self.account['CLIENT_SECRET'],
password=self.account['PASSWORD'],
user_agent=self.account['USER_AGENT'],
username=self.account['USERNAME'])
def connect(self,mode='local'):
self.driver.connect(mode='local')
query = 'SELECT * FROM accounts'
account = self.driver.pull(query)
account = account[0]
self.account = {
'id' : account[0],
'CLIENT_ID' : account[1],
'CLIENT_SECRET' : account[2],
'PASSWORD' : account[3],
'USER_AGENT' : account[4],
'USERNAME' : account[5]
}
self.reddit = self.auth()
def compile(self,**kwargs):
self.results = []
mentions = self.reddit.inbox.unread(**kwargs)
self.results.extend(mentions)
self.results.sort(key=lambda post: post.created_utc, reverse=True)
return self.results
def get_body(self,post):
try:
post_type = post.parent().__class__.__name__.lower()
except:
return self.error
if post_type == 'submission':
url = post.parent().url
title = post.parent().title
if 'reddit.com' in url:
try:
selftext = post.parent().selftext
body = " ".join([title, selftext])
except:
return self.error
else:
return " ".join([title, url])
elif post_type == 'comment' or post_type == 'message':
return post.parent().body
else:
return self.error
def get_url(self, body, length):
def __iter__(self):
username = self.account['USERNAME']
stream = praw.models.util.stream_generator(lambda **kwargs: self.compile(**kwargs))
for idx,post in enumerate(stream):
self.reddit.inbox.mark_read([post])
body = ''
if post.parent().author == username:
chain = []
while True:
try:
parent = post.parent()
chain.append(parent)
except:
break
chain_authors = [ x.author for x in chain ])
for idx,item in chain_authors[::-1]:
if item == username:
original_post = chain[idx-1]
body = self.get_body(original_post)
break
else:
body = self.get_body(post.parent())
chain_length = len([x for x in chain if x == username ])
post.reply(self.get_url(body, chain_length))
if __name__ == '__main__':
stream = Streamer()
stream.connect()
for post in stream:
print(post)
