foxpy/main3.py at master · plusreed/foxpy · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
from discord.ext import commands
import discord
import datetime, re
import asyncio
import copy
import logging
import traceback
import sys
from collections import Counter
import config

description = """
I'm Fox, a multi-purpose and modular Discord bot.
"""


init_cogs = [
    'plugins.admin.eval',
    'plugins.admin.shutdown',
    'plugins.core.math',
    'plugins.core.ping',
    'plugins.core.lastfm',
    'plugins.music.voice',
]

dc_log = logging.getLogger(discord)
dc_log.setLevel(logging.DEBUG)
log = logging.getLogger()
log.setLevel(logging.INFO)
handler = logging.FileHandler(filename='fox.log', encoding='utf-8', mode='w')
log.addHandler(handler)

hattr = dict(hidden=True)

prefix = ['$', '^']
fox = commands.Bot(command_prefix=prefix, description=description, pm_help=None, help_attrs=hattr)


@fox.event
async def on_command_error(error, ctx):
    if isinstance(error, commands.NoPrivateMessage):
        await fox.send_message(ctx.message.author, "Sorry, you can't use this command in private messages.")
    elif isinstance(error, commands.DisabledCommand):
        await fox.send_message(ctx.message.author, 'Sorry, it looks like that command is disabled.')
    elif isinstance(error, commands.CommandInvokeError):
        print('In {0.command.qualified_name}:'.format(ctx))
        traceback.print_tb(error.original.__traceback__)
        print('{0.__class__.__name__}: {0}'.format(error.original))


@fox.event
async def on_ready():
    print('Fox is now ready!')
    print('Username: ' + fox.user.name)
    print('ID: ' + fox.user.id)
    print('------')
    if not hasattr(fox, 'uptime'):
        fox.uptime = datetime.datetime.utcnow()


@fox.event
async def on_resumed():
    print("Fox has resumed.")


@fox.event
async def on_command(command, ctx):
    fox.commands_used[command.name] += 1
    message = ctx.message
    if message.channel.is_private:
        destination = 'Private Message'
    else:
        destination = '#{0.channel.name} ({0.server.name})'.format(message)

    log.info('{0.timestamp}: {0.author.name} in {1}: {0.content}'.format(message, destination))


@fox.event
async def on_message(message):
    if message.author.bot:
        return

    await fox.process_commands(message)


# @bot.command()
# async def ping():
#     await bot.say("pong")


if __name__ == '__main__':
    if any('debug' in arg.lower() for arg in sys.argv):
        print("Fox is running in debug mode. The command prefix is now '^^'.")
        fox.command_prefix = '^^'

    fox.client_id = config.BOT_ID
    fox.commands_used = Counter()
    for plugin in init_cogs:
        try:
            fox.load_extention(plugin)
        except Exception as e:
            print('Error: failed to load plugin {}\n{}: {}'.format(plugin, type(e).__name__, e))

    fox.run(config.BOT_TOKEN)
    handlers = log.handlers[:]
    for hdlr in handlers:
        hdlr.close()
        log.removeHandler(hdlr)