hojicha/bot.py at main · GLOSHSEP/hojicha · 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
#bot.py

import os
import discord
import datetime as dt
import subprocess
import hojicha_functions as hf
import requests
from dotenv import load_dotenv


load_dotenv()
TOKEN = os.getenv('DISCORD_TOKEN')
fortune_file=os.getenv('FORTUNE_DB_PATH')
encoding='utf-8'
wakeup_time=dt.datetime.now()

client = discord.Client()

@client.event
async def on_ready():
	print('We have logged in as {0.user}'.format(client))

async def send_big_message(message,string):
	filename=str(dt.datetime.now())+'.txt'
	save_prefix='/tmp'
	filepath=save_prefix+'/'+filename
	response_file = open(filepath,'w')
	max_len=80
	chunks = [string[i:i+max_len] for i in range(0,len(string),max_len)]
	for chunk in chunks:
		response_file.write(chunk)
		response_file.write('\n')
	response_file.close()
	response_file = open(filepath,'r')

	discord_file=discord.File(response_file,filename)
	await message.channel.send(file=discord_file)
	response_file.close()

@client.event
async def on_message(message):
	msg=message.content
	if message.author == client.user:
		return
	if msg.startswith('!hojicha'):
		msg_args=msg.split('!hojicha')
		if msg_args[1]=='' or msg_args[1].upper()==' HI' or msg_args[1].upper()==' HELLO':
			await message.channel.send('Hello!')
		elif msg_args[1].upper()==' SPEAK':
			res = subprocess.check_output(['fortune', fortune_file])
			await message.channel.send(res.decode(encoding))
		elif msg_args[1].upper().startswith(' Q'):
			await message.channel.send(hf.yes_no(msg_args[1]))
		elif msg_args[1].upper().startswith(' UPTIME'):
			await message.channel.send("I have been awake for "+ hf.hojicha_uptime(wakeup_time))
		elif msg_args[1].upper().startswith(' DRAW CARD'):
			await message.channel.send(hf.draw_a_tarot_card(msg_args[1]))
		elif msg_args[1].upper().startswith(' PICK'):
			await message.channel.send(hf.pick_one(msg_args[1][6:]))
		elif msg_args[1].upper().startswith(' ROLL'):
			await message.channel.send(hf.roll_dice(msg_args[1].upper() ))
		elif msg_args[1].upper().startswith(' T2B'):
			if len(message.attachments)==1:
				text_to_parse=requests.get(message.attachments[0].url).content.decode("utf-8")
			else:
				text_to_parse=msg_args[1][5:]

			response_string=hf.text_to_binary(text_to_parse)
			if len(response_string) > 2000:
				await send_big_message(message,response_string)
			else:
				await message.channel.send(response_string)
		elif msg_args[1].upper().startswith(' B2T'):
			if len(message.attachments)==1:
				text_to_parse=requests.get(message.attachments[0].url).content.decode("utf-8")
			else:
				text_to_parse=msg_args[1][5:]

			response_string=hf.binary_to_text(text_to_parse)
			if len(response_string) > 2000:
				await send_big_message(message,response_string)
			else:
				await message.channel.send(response_string)
		elif msg_args[1].upper().startswith(' CTOF'):
			text_to_parse=msg_args[1][5:]
			response_string=hf.c_to_f(text_to_parse)
			await message.channel.send(response_string)
		elif msg_args[1].upper().startswith(' FTOC'):
			text_to_parse=msg_args[1][5:]
			response_string=hf.f_to_c(text_to_parse)
			await message.channel.send(response_string)
		elif msg_args[1].upper().startswith(' HELP'):
			await message.channel.send(hf.help_text())
		else:
			await message.channel.send('Hojicha slowly looks in your direction')


client.run(TOKEN)