discordbot.py

import discord
import json
from json.decoder import JSONDecodeError
import logging
import sys
import os

'''

	This application should run on boot and stay running at all times. The 
	purpose of it is to take in a message on discord, save it in the 
	configuration file, and refresh the inky phat display.

'''

intents = discord.Intents.default()
intents.message_content = True
client = discord.Client(intents=intents)

#logging.basicConfig(filename = 'log.txt', format='%(asctime)s [%(name)s]: %(message)s', level=logging.INFO) #log.txt: time [discordbot]: message
#logging.getLogger().addHandler(logging.StreamHandler()) #print to console
#logger = logging.getLogger('DiscordBot')

config_location = "config/config.json"

try:
	with open(config_location) as json_data_file:
		config = json.load(json_data_file)
except IOError:
	#logger.error("Configuration file does not exist in config/config.json. Pull a new one from https://github.com/corey-schneider/inky-phat-messenger/blob/main/config/config.json")
	sys.exit("Configuration file does not exist in config/config.json. Pull a new one from https://github.com/corey-schneider/inky-phat-messenger/blob/main/config/config.json")

try:
	TOKEN = config["discord"]["token"]
	ALLOWED_SENDER = config["discord"]["allowed_sender"]
	RECIPIENT = config["discord"]["recipient"]
except (JSONDecodeError, KeyError):
	#logger.error("Discord token not found in config.json")
	sys.exit("Discord token not found in config.json")

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

@client.event
async def on_message(message):
	if message.author == client.user:
		return

	if str(message.author) == ALLOWED_SENDER or ALLOWED_SENDER == "":
		if message.content.startswith('') and not message.content.startswith('!'):
			if config["message"] != message.content: # if the pi is restarted and the message is the same, don't rewrite it
				config["message"] = message.content
				with open(config_location, "w") as f:
					json.dump(config, f)
					print("Successfully wrote new message to config.json")
			else:
				print("Messages were the same; not rewriting.")
				#logger.info("same messages")
			await message.channel.send("Sent to inky display: \""+message.content+"\"")
			#print("Rewritten: "+str(display.show_rewritten_message(message.content)))
			os.system("python3 display.py")

			#user_id = client.fetch_user(client.get_all_members(), name="test", discriminator="123").id
			
			'''
			not yet working
			
			user = await client.fetch_user(RECIPIENT)
			print(user.name)
			try:
				await user.send("New message on inky display!")
			except discord.Forbidden:
				pass'''
		elif message.content.startswith('!shutdown'):
			os.system("sudo shutdown now")
			await message.channel.send("Shutting down...")
		elif message.content.startswith('!reboot') or message.content.startswith('!restart'):
			os.system("sudo reboot")
			await message.channel.send("Restarting...")


		

	else:
		await message.channel.send('You are not authorized to send a command.')
		#logger.info("User does not match. Ignoring command. Expected: \""+ALLOWED_SENDER+"\", got \""+str(message.author)+"\".")
		print("User does not match. Ignoring command. Expected: \""+ALLOWED_SENDER+"\", got \""+str(message.author)+"\".")

client.run(TOKEN)