1
Fork 0
mirror of https://github.com/RYGhub/royalnet.git synced 2024-11-23 19:44:20 +00:00
royalnet/redditbot.py

72 lines
2.7 KiB
Python
Raw Normal View History

2018-09-05 17:48:34 +00:00
import praw
import configparser
import db
import logging
import telegram
import time
import raven
import os
import sys
2018-09-17 22:26:01 +00:00
import coloredlogs
2018-09-05 17:48:34 +00:00
# Init the config reader
config = configparser.ConfigParser()
config.read("config.ini")
2018-09-18 22:02:39 +00:00
logging.getLogger().disabled = True
2018-09-05 17:48:34 +00:00
logger = logging.getLogger(__name__)
2018-09-17 22:26:01 +00:00
coloredlogs.install(level="DEBUG", logger=logger)
2018-09-05 17:48:34 +00:00
sentry = raven.Client(config["Sentry"]["token"],
2018-09-13 16:38:21 +00:00
release=raven.fetch_git_sha(os.path.dirname(__file__)),
install_logging_hook=False,
2018-09-05 17:48:34 +00:00
hook_libraries=[])
def process():
logger.info("Retrieving parsed posts...")
session = db.Session()
results = session.query(db.ParsedRedditPost).all()
parsed_post_ids = [x.id for x in results]
session.close()
logger.info("Posts retrieved successfully.")
logger.info("Logging in to reddit...")
reddit = praw.Reddit(client_id=config["reddit"]["client_id"],
client_secret=config["reddit"]["client_secret"],
password=config["reddit"]["password"],
user_agent='Royal-Bot/4.1',
username=config["reddit"]["username"])
logger.info(f"Login as {reddit.user.me()} successful!")
logger.info(f"Logging in to Telegram...")
telegram_bot = telegram.Bot(config["Telegram"]["bot_token"])
logger.info(f"Login successful.")
r_royalgames = reddit.subreddit("royalgames")
for submission in r_royalgames.stream.submissions():
try:
if submission.id not in parsed_post_ids:
logger.info(f"New post found: {submission.id}")
logger.debug(f"Creating new db session...")
session = db.Session()
new_post = db.ParsedRedditPost(id=submission.id)
session.add(new_post)
logger.debug(f"Committing...")
session.commit()
session.close()
logger.debug("Sending Telegram notification...")
while True:
try:
telegram_bot.send_message(config["Telegram"]["main_group"],
2018-09-08 00:12:05 +00:00
f' Nuovo post su r/RoyalGames:\n'
2018-09-05 17:48:34 +00:00
f'<a href="https://reddit.com{submission.permalink}">{submission.title}</a>\n'
2018-09-08 00:12:05 +00:00
f'da <b>u/{submission.author}</b>',
2018-09-05 17:48:34 +00:00
parse_mode="HTML", disable_notification=True)
except telegram.error.TimedOut:
time.sleep(1)
else:
break
except Exception:
sentry.captureException(exc_info=sys.exc_info())