main.py

# ! /usr/bin/env python3

import asyncio
import aiohttp
from discord import Game
from discord.ext import tasks
from discord.ext.commands import Bot
from selenium import webdriver
from selenium.webdriver.chrome.options import Options
from PIL import Image
from functools import reduce
import math
import operator
import discord
from imgurpython import *
import json
import os
import sys
import atexit
import praw
from praw.models import Message
import requests
import time

pid = str(os.getpid())
pidfile = "/tmp/mydaemon.pid"

if os.path.isfile(pidfile):
    print("%s already exists, exiting" % pidfile)
    sys.exit()
open(pidfile, 'w').write(pid)

BOT_PREFIX = ("?", "!")
client = Bot(command_prefix=BOT_PREFIX)
with open('tokens.json') as f:
    token_json = json.load(f)
discord_token = token_json['DISCORD_TOKEN']
imgur_client_id = token_json['IMGUR_CLIENT_ID']
imgur_client_secret = token_json['IMGUR_CLIENT_SECRET']
imgur_access_token = token_json['IMGUR_ACCESS_TOKEN']
imgur_refresh_token = token_json['IMGUR_REFRESH_TOKEN']
reddit_client_id = token_json['REDDIT_CLIENT_ID']
reddit_client_secret = token_json['REDDIT_CLIENT_SECRET']
reddit_password = token_json['REDDIT_PASSWORD']
reddit_user_agent = 'python:AprilFoolsMessengerBot:v1.0.1 (by /u/owenisbae)'
reddit_username = 'owenisbae'

options = Options()
options.add_argument('--headless')
options.add_argument('--no-sandbox')
driver = webdriver.Chrome(chrome_options=options)

imgClient = ImgurClient(imgur_client_id, imgur_client_secret, imgur_access_token, imgur_refresh_token)
album = 'b6ZIp1m'
uploadConfig = {
    'album': album,
    'name': 'Updated Summer fools pic',
    'title': 'Update Photo',
    'description': 'Grabbed automatically, hopefully this is important'
}


reddit = praw.Reddit(client_id=reddit_client_id,
                     client_secret=reddit_client_secret,
                     password=reddit_password,
                     user_agent=reddit_user_agent,
                     username=reddit_username)


def pidclose():
    os.unlink(pidfile)


atexit.register(pidclose)


class MyClient(discord.Client):
    driver = webdriver.Chrome(chrome_options=options)

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)

        # create the background task and run it in the background
        # self.bg_task = self.loop.create_task(self.check_snekbait())
        self.bg_task = self.loop.create_task(self.check_kairos())
        self.bg_task = self.loop.create_task(self.check_layer())
        self.bg_task = self.loop.create_task(self.check_inbox())

    async def on_ready(self):
        print("Ready to record!")
        channel_id = 585303573613510680
        channel = self.get_channel(channel_id)
        message = "Is it getting hot in here? Or is that just Gryph :wink:"
        await channel.send(message)

    async def check_inbox(self):
        await self.wait_until_ready()
        msg_count = 0
        channel_id = 588704610391293961
        channel = self.get_channel(channel_id)
        while not self.is_closed():
            msg = ""
            for item in reddit.inbox.unread(limit=None):
                if isinstance(item, Message):
                    if str(item.author) == 'None':
                        author_print = 'Sub mods'
                    else:
                        author_print = str(item.author)
                    msg += ("From: " + author_print + "\nSubject: " + str(item.body) + "\n")
                    msg_count += 1
                    # await channel.send(msg)
                    item.mark_read()
                    print(msg)
            if msg != "":
                message = "Message reply received from mods: \n" + msg
                await channel.send(message)
            await asyncio.sleep(29)

    async def check_kairos(self):
        await self.wait_until_ready()
        channel_id = 588704610391293961
        channel = self.get_channel(channel_id)
        while not self.is_closed():
            screenshot_url2 = "https://old.reddit.com/r/ThePathOfKairos"
            try:
                driver.get(screenshot_url2)
                driver.save_screenshot('screenshot2.png')
            except:
                print("Owo what's this?")
                exit(-1)

                # driver.close()
            im1 = Image.open('screenshot2.png')
            im2 = Image.open('screenshot.png')
            h1 = im1.histogram()
            h2 = im2.histogram()
            rms = math.sqrt(reduce(operator.add,
                                   map(lambda a, b: (a - b) ** 2, h1, h2)) / len(h1))
            print(rms)
            if rms > 0.5:
                image_is_different = True
            else:
                image_is_different = False
            if image_is_different:
                # msg = "Check snekbait"
                imgur_image = imgClient.upload_from_path('screenshot2.png', config=uploadConfig, anon=False)
                imgur_link = str(imgur_image['link'])
                driver.save_screenshot('screenshot.png')
                msg = "New Path of Kairos message detected:" + imgur_link
                await channel.send(msg)
                # msg = "<@165688608190103552> and <@332245843983990786> and <@126011690419617792>"
                # await channel.send(msg)
            # else:
            # msg = "For debugging only"
            # await channel.send(msg)

            await asyncio.sleep(60)

    async def check_layer(self):
        await self.wait_until_ready()
        channel_id = 585303573613510680
        channel = self.get_channel(channel_id)
        while not self.is_closed():
            screenshot_url3 = "https://old.reddit.com/r/layer"
            try:
                driver.get(screenshot_url3)
                driver.save_screenshot('screenshot3.png')
            except:
                print("Owo what's this?")
                exit(-1)

                # driver.close()
            im1 = Image.open('screenshot3.png')
            im2 = Image.open('screenshot4.png')
            h1 = im1.histogram()
            h2 = im2.histogram()
            rms = math.sqrt(reduce(operator.add,
                                   map(lambda a, b: (a - b) ** 2, h1, h2)) / len(h1))
            print(rms)
            if rms > 0.5:
                image_is_different = True
            else:
                image_is_different = False
            if image_is_different:
                # msg = "Check snekbait"
                imgur_image = imgClient.upload_from_path('screenshot3.png', config=uploadConfig, anon=False)
                imgur_link = str(imgur_image['link'])
                driver.save_screenshot('screenshot4.png')
                msg = "OwO new Layer message detected: " + imgur_link
                await channel.send(msg)
                # msg = "<@165688608190103552> and <@332245843983990786> and <@126011690419617792>"
                # await channel.send(msg)
            # else:
            # msg = "For debugging only"
            # await channel.send(msg)

            await asyncio.sleep(60)
    # async def on_disconnect(self):
    #     driver.close()

    async def on_message(self, message):
        if message.author == client.user:
            return
        if '?update' in message.content.lower():
            split_message = message.content.split()
            if split_message[0].lower() != '?update':
                return
            screenshot_url2 = "https://www.reddit.com/r/ThePathOfKairos"
            driver.get(screenshot_url2)
            driver.save_screenshot('screenshot.png')
            imgur_image = imgClient.upload_from_path('screenshot2.png', config=uploadConfig, anon=False)
            imgur_link = str(imgur_image['link'])
            msg = "Path of Kairos reset to: " + imgur_link
            await message.channel.send(msg)
        # ModMsg Mostly provided by Satan#0001
        elif '?modmsg' in message.content.lower():
            split_message = message.content.split()
            if split_message[0].lower() != '?modmsg':
                return
            if len(split_message) > 1:
                solution = " ".join(split_message[1:])
                if "spam" in solution.lower():
                    await message.channel.send("The word 'spam' is blocked due to "
                                               "Reddit's auto help system picking it up. Sorry :(")
                else:
                    try:
                        reddit.subreddit('ghostisedoesnotsuck').message(solution, solution)
                        await message.channel.send("Solution sent!")
                    except:
                        await message.channel.send("Owo looks like yew made a fucky wucky. A weal FUCKO BOINGO")
            else:
                await message.channel.send("No message provided, I'm not sending blank lines....")

        elif '?inbox' in message.content.lower():
            split_message = message.content.split()
            if split_message[0].lower() != '?inbox':
                return
            msg_count = 0
            for item in reddit.inbox.unread(limit=None):
                if isinstance(item, Message):
                    if str(item.author) == 'None':
                        author_print = 'Sub mods'
                    else:
                        author_print = str(item.author)
                    msg = ("From: " + author_print + "\nSubject: " + str(item.body) + "\n")
                    msg_count += 1
                    await message.channel.send(msg)
                    item.mark_read()
            if msg_count == 0:
                msg = "Sorry, inbox is empty..."
                await message.channel.send(msg)
        # Overly complex function that uses regular requests for speed but defaults to asyncio requests when that fails
        elif '?tiny' in message.content.lower():
            split_message = message.content.split()
            if split_message[0].lower() != '?tiny':
                return
            if len(split_message) > 2:
                msg = "I'm sorry but it looks like you sent too much. Please use ``?link linkGuess``"
            else:
                try:
                    tiny_url = requests.head("https://tinyurl.com/"+str(split_message[1]))
                    if tiny_url.ok:
                        msg = "Looks like " + str(tiny_url.url) + " is a valid link. Czech it out!"
                    else:
                        msg = "Doesn't look like " + str(tiny_url.url) + " is legit. 9/10 a little " \
                                                                         "something for everybody"
                except:
                    try:
                        async with aiohttp.ClientSession() as session:
                            async with session.get("https://tinyurl.com/"+str(split_message[1])) as tiny_url:
                                if tiny_url.real_url:
                                    msg = "Looks like " + str(tiny_url.url) + " is a valid link. Czech it out!"
                                else:
                                    msg = "Doesn't look like " + str(tiny_url.url) + " is legit. 9/10 a little " \
                                                                              "something for everybody"
                    except:
                        msg = "Grabbing the url didn't work? Try again and/or ping Yew"
            await message.channel.send(msg)
        elif "?help" == message.content.lower():
            msg = "Help for Sir Lopez 2 The Electric Boogaloo\n" \
                  "Sir Lopez's Automatic features:\n" \
                  "Inbox:\n" \
                  "\tMonitors the bot's inbox and sends the messages it receives\n" \
                  "Path of Kairos:\n" \
                  "\tMonitors the Path of Kairos and screenshots it upon the subreddit changing or the bot failing\n" \
                  "====================================\n" \
                  "Sir Lopez's Manual Features (~~$69~~ FREE Premium DLC)\n" \
                  "``?update``: \n" \
                  "\tManually checks Path of Kairos and resets the ref. image to what it takes a picture of then\n" \
                  "``?inbox``:\n" \
                  "\tChecks the inbox and reports any unread messages.\n" \
                  "``modmsg <Message to send to mods>``:\n" \
                  "\tSends the mods of Path of Kairos the message you specify.\n" \
                  "``?tiny <URLPART>``:\n" \
                  "\tChecks https://tinyurl.com/URLPART and sees if it is a valid link or not.\n" \
                  "====================================\n" \
                  "Sir Lopez's Restricted functions: Won't work for anyone who can't handle the neutron style:\n" \
                  "``?reboot``:\n" \
                  "\tReboots Sir Lopez, great for when he's acting rudely.\n" \
                  "``?sleep <XX>``:\n:" \
                  "\tPuts Sir Lopez into an absolute slumber for XX minutes (CANNOT BE WOKEN UP AT ALL). " \
                  "You can't kill a god, but you may make him slumber...\n"
            channel = message.author.dm_channel
            await channel.send(msg)
        elif "?reboot" == message.content.lower() and \
                (message.author.id == 165688608190103552 or message.channel.id == 430464509006577668):
            exit(-1)
        elif "?sleep" in message.content.lower() and \
                (message.author.id == 165688608190103552 or message.channel.id == 430464509006577668):
            split_content = message.content.split()
            if len(split_content) != 2:
                return
            else:
                try:
                    time_to_sleep_in_min = int(split_content[1])
                    msg = "Going to sleep for " + split_content[1] + " minutes... See you later alligators"
                    await message.channel.send(msg)
                    time.sleep(time_to_sleep_in_min*60)
                    msg = "I'm back y'all"
                    message.channel.send(msg)
                except:
                    return


client = MyClient()
client.run(discord_token)