import praw import configparser import db import logging # python-telegram-bot has a different name # noinspection PyPackageRequirements import telegram import time import raven import os import sys import coloredlogs # Init the config reader config = configparser.ConfigParser() config.read("config.ini") logging.getLogger().disabled = True logger = logging.getLogger(__name__) os.environ["COLOREDLOGS_LOG_FORMAT"] = "%(asctime)s %(levelname)s %(name)s %(message)s" coloredlogs.install(level="DEBUG", logger=logger) sentry = raven.Client(config["Sentry"]["token"], release=raven.fetch_git_sha(os.path.dirname(__file__)), install_logging_hook=False, 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"], f'ℹ️ Nuovo post su r/RoyalGames:\n' f'' f'{submission.title}\n' f'da u/{submission.author}', parse_mode="HTML", disable_notification=True) except telegram.error.TimedOut: time.sleep(1) else: break except Exception: sentry.captureException(exc_info=sys.exc_info())