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 subprocess
|
|
|
|
|
|
import sys
|
|
|
|
|
|
|
|
|
|
|
|
# Init the config reader
|
|
|
|
|
|
config = configparser.ConfigParser()
|
|
|
|
|
|
config.read("config.ini")
|
|
|
|
|
|
|
|
|
|
|
|
logging.getLogger().setLevel(level=logging.ERROR)
|
|
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
logger.setLevel(level=logging.DEBUG)
|
|
|
|
|
|
|
|
|
|
|
|
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())
|
