DiscordCity.py

import asyncio
import pickle
import sys
import traceback

import discord
from discord.ext import commands


class DiscordCity(commands.Bot):
    def __init__(self):
        super().__init__(
            command_prefix=self._get_prefix,
            description="A bot for Discord Hack Week.\nGenerates a city out of the current server."
        )

        self.owner_ids = [
            270190067354435584,  # green
            334350969196380160,  # fish
            389814953562996736,  # kat
            286546315003953152  # Incog
        ]

        self.initial_extensions = ["cogs.info", "cogs.owner", "cogs.eh", "cogs.city"]
        self.load_exts()

    def _get_prefix(self, bot, message):
        if not message.guild:
            return ['asfajaskdaj']
        else:
            prefix = ['~']

        return commands.when_mentioned_or(*prefix)(bot, message)

    def load_exts(self):
        for extension in self.initial_extensions:
            try:
                self.load_extension(extension)
                print(f"Successfully loaded extension - {extension}")
            except Exception:
                print(
                    f"Failed to load extension - {extension}", file=sys.stderr)
                traceback.print_exc()

    async def is_owner(self, user):
        return user.id in self.owner_ids

    async def on_ready(self):
        await self.change_presence(activity=discord.Activity(name="Building Cities...", type=discord.ActivityType.playing))

        print(f"\n\nLogged in as: {self.user.name} - {self.user.id}")
        print(f"Version: {discord.__version__}\n")

    def run(self):
        print("Connecting to discordapp")
        with open("tooken.data", "rb") as f:
            tooken = pickle.load(f)
        super().run(tooken, bot=True, reconnect=True)


if __name__ == "__main__":
    if "win" in sys.platform:
        asyncio.set_event_loop(asyncio.ProactorEventLoop())

    DiscordCity().run()