Maki/helpers.py

# 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')