initial commit
This commit is contained in:
@@ -0,0 +1,65 @@
|
||||
import praw
|
||||
from config import get_reddit, Config
|
||||
|
||||
import time
|
||||
|
||||
class ModBot:
|
||||
def __init__(self):
|
||||
self.reddit = get_reddit()
|
||||
self.subreddit = self.reddit.subreddit(Config.SUBREDDIT)
|
||||
self.wiki_page = Config.WIKI_PAGE
|
||||
self.triggers = []
|
||||
self.comments = []
|
||||
|
||||
def fetch_wiki_config(self):
|
||||
import yaml
|
||||
try:
|
||||
wiki_content = self.subreddit.wiki[self.wiki_page].content_md
|
||||
# Example Automoderator YAML format:
|
||||
# triggers:
|
||||
# - trigger: help
|
||||
# comment: |
|
||||
# Thank you for your report!
|
||||
# This post is now approved.
|
||||
# - trigger: question
|
||||
# comment: |
|
||||
# This post has been approved.
|
||||
config = yaml.safe_load(wiki_content)
|
||||
self.triggers = []
|
||||
self.comments = []
|
||||
for entry in config.get('triggers', []):
|
||||
self.triggers.append(entry.get('trigger', '').strip())
|
||||
self.comments.append(entry.get('comment', '').strip())
|
||||
except Exception as e:
|
||||
print(f"Error fetching wiki config: {e}")
|
||||
|
||||
def run(self):
|
||||
print("ModBot started. Watching for mod reports...")
|
||||
while True:
|
||||
self.fetch_wiki_config()
|
||||
for report in self.subreddit.mod.reports(limit=25):
|
||||
self.handle_report(report)
|
||||
time.sleep(30) # Poll every 30 seconds
|
||||
|
||||
def handle_report(self, report):
|
||||
if not hasattr(report, 'mod_reports') or not report.mod_reports:
|
||||
return
|
||||
for mod_report in report.mod_reports:
|
||||
report_text = mod_report[0].lower()
|
||||
for idx, trigger in enumerate(self.triggers):
|
||||
if trigger.lower() in report_text:
|
||||
self.approve_and_comment(report, self.comments[idx])
|
||||
break
|
||||
|
||||
def approve_and_comment(self, submission, comment_text):
|
||||
try:
|
||||
submission.mod.approve()
|
||||
comment = submission.reply(comment_text)
|
||||
comment.mod.distinguish(sticky=True)
|
||||
print(f"Approved and commented on: {submission.id}")
|
||||
except Exception as e:
|
||||
print(f"Error approving/commenting: {e}")
|
||||
|
||||
if __name__ == "__main__":
|
||||
bot = ModBot()
|
||||
bot.run()
|
||||
Reference in New Issue
Block a user