-
Notifications
You must be signed in to change notification settings - Fork 2
/
app.py
111 lines (88 loc) · 3.33 KB
/
app.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
import os
import praw
import sqlite3
import datetime
from praw.models import MoreComments
client_id = os.environ.get('REDDIT_CLIENT_ID')
client_secret = os.environ.get('REDDIT_SECRET')
password = os.environ.get('REDDIT_PASSWORD')
user_agent = os.environ.get('REDDIT_USER_AGENT')
username = os.environ.get('REDDIT_USERNAME')
reddit = praw.Reddit(client_id=client_id, client_secret=client_secret,
password=password, user_agent=user_agent,
username=username)
header = '''
Table of Questions and Answers. Original answer linked - Please upvote the original questions and answers. (I'm a bot.)
***
'''
footer = '''
---
[Source](https://github.com/johnsliao/ama_compiler)
'''
def compile(submission):
"""
Parses comments and returns Q&A as a reddit-friendly table format.
:param submission:
:return:
"""
comments_exist = False
comment_text = ''
comment_text += header
comment_text += '\n'
comment_text += 'Question | Answer | Link'
comment_text += '\n'
comment_text += '---------|----------|----------|'
comment_text += '\n'
author = submission.author
for comment in submission.comments:
if isinstance(comment, MoreComments):
continue
if comment.score <= 0:
continue
for reply in comment._replies:
if isinstance(reply, MoreComments):
continue
if reply.author == author:
answer = ' '.join([l.strip() for l in reply.body.split('\n')])
question = ' '.join([l.strip() for l in comment.body.split('\n')])
if answer.endswith('?'):
break
if len(comment_text + question + '|' + answer + '|[Here](' + comment.permalink + ')' + footer) > 9500:
break
comment_text += question + '|' + answer + '|[Here](' + comment.permalink + ')'
comment_text += '\n'
comments_exist = True
break
comment_text += footer
if not comments_exist:
return None
return comment_text
if __name__ == '__main__':
conn = sqlite3.connect('db.db')
c = conn.cursor()
for submission in reddit.subreddit('ama').top(limit=256, time_filter='week'):
submission_age = datetime.datetime.now() - datetime.datetime.utcfromtimestamp(submission.created_utc) # seconds
if submission_age < datetime.timedelta(seconds=60 * 60 * 24):
# Post is less than 24 hours old
continue
if submission_age > datetime.timedelta(seconds=60 * 60 * 24 * 7):
# Post is older than 1 week old
continue
if '[Request]' in submission.title:
continue
if submission.score < 10:
continue
c.execute("select * from posts where post_id=:post_id", {"post_id": submission.id})
if c.fetchone():
continue
try:
comment_text = compile(submission)
if not comment_text:
continue
submission.reply(comment_text)
c.execute("insert into posts (date, post_id) values (?,?)", [datetime.date.today(), submission.id])
conn.commit()
print('Successfully added comment to {}'.format(submission.id))
except Exception as e:
print(e)
exit(1)