mirror of
https://github.com/MrDetonia/Maki.git
synced 2024-11-14 08:49:35 -05:00
68 lines
1.5 KiB
Python
Executable File
68 lines
1.5 KiB
Python
Executable File
# Maki
|
|
# ----
|
|
# Discord bot by MrDetonia
|
|
#
|
|
# Copyright 2017 Zac Herd
|
|
# Licensed under BSD 3-clause License, see LICENSE.md for more info
|
|
|
|
|
|
# IMPORTS
|
|
import asyncio
|
|
import discord
|
|
import logging
|
|
import datetime
|
|
|
|
# LOCAL IMPORTS
|
|
from common import *
|
|
|
|
|
|
# clamps an integer
|
|
def clamp(n, small, large): return max(small, min(n, large))
|
|
|
|
|
|
# converts a datetime to a string
|
|
def strfromdt(dt):
|
|
return dt.strftime('%Y-%m-%d %H:%M:%S')
|
|
|
|
|
|
# converts a timestamp to a datetime
|
|
def dtfromts(ts):
|
|
return datetime.datetime.fromtimestamp(ts)
|
|
|
|
|
|
# logging setup
|
|
def logger():
|
|
logger = logging.getLogger('discord')
|
|
logger.setLevel(logging.DEBUG)
|
|
handler = logging.FileHandler(filename='discord.log', encoding='utf-8', mode='w')
|
|
handler.setFormatter(logging.Formatter('%(asctime)s:%(levelname)s:%(name)s: %(message)s'))
|
|
logger.addHandler(handler)
|
|
|
|
|
|
# send_message wrapper (deals with Discord's shit API)
|
|
@asyncio.coroutine
|
|
def discord_send(client, message, response):
|
|
if response is not '' and message.server.id not in quiet:
|
|
for attempt in range(5):
|
|
try:
|
|
yield from client.send_message(message.channel, response)
|
|
except discord.errors.HTTPException:
|
|
continue
|
|
else:
|
|
break
|
|
else:
|
|
print('ERROR: Failed to send message to discord after 5 attempts')
|
|
|
|
|
|
# send typing signal to Discord
|
|
@asyncio.coroutine
|
|
def discord_typing(client, message):
|
|
for attempt in range(5):
|
|
try:
|
|
yield from client.send_typing(message.channel)
|
|
except discord.errors.HTTPException:
|
|
continue
|
|
else:
|
|
break
|
|
else:
|
|
print('ERROR: Failed to send typing signal to discord after 5 attempts') |