main.py

import discord
import requests, string
from bs4 import BeautifulSoup

import paho.mqtt.subscribe as subscribe

import json, ssl
import config, discord_token

from printer_config import PRINTERS

intents = discord.Intents.default()
intents.message_content = True
intents.messages = True

client = discord.Client(intents=intents)

def get_printer_status():
    
    output_string = """\n"""

    for p in PRINTERS:
        printer = PRINTERS[p]

        auth = {"username":printer["username"],"password":printer["access_code"]}
        tls = ssl._create_unverified_context()

        msg = subscribe.simple(topics=printer["topic_name"],hostname=printer["ip"],port=printer["port"],auth=auth,tls=tls)
        x = json.loads(msg.payload)

        printer_name = printer["name"] + " (" + p + ")"
        job_name = x["print"]["subtask_name"]
        printer_state = x["print"]["gcode_state"]
        min_remain = x["print"]["mc_remaining_time"]

        if printer_state == "RUNNING":
            output_string = output_string + \
                """```Printer: {0}\nPrinter State: Active\nJob Name: {1}\nMinutes Remaining: {2}```""".format(printer_name,job_name,min_remain)
        else:
            output_string = output_string + \
                """```Printer: {0}\nPrinter State: Idle```""".format(printer_name)

    return output_string

def get_shop_hours():
    r = requests.get(config.SHOP_HOURS_URL)
    soup = BeautifulSoup(r.text, 'html.parser')
    shop_hours = soup.find(id="shophours")

    hours_dict = {}

    for tr in shop_hours.find_all('tr'):
        td = tr.find_all("td")
        hours_dict[td[0].text] = td[1].text

    markdown_string = f'```\nCurrent Shop Hours (fetched from https://synshop.org/hours) \n===\n'

    for k,v in hours_dict.items():
        spaces = ""
        s = 11 - (len(k) + 1)
        for x in range(s):
            spaces = spaces + " "

        markdown_string = markdown_string + f'{k}:{spaces}{v}\n'

    markdown_string = markdown_string + f'\n{config.SHOP_ADDRESS}\n'
    markdown_string = markdown_string + f'\n{config.MEMBERSHIP_NOTICE}\n```'

    return(markdown_string)


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

@client.event
async def on_message(message):
    if message.author == client.user:
        return
    
    m = message.content.upper()
    m = m.translate(str.maketrans('', '', string.punctuation))

    if m == "GETPRINTERSTATUS":
        await message.channel.send(get_printer_status())

    for phrase in config.PHRASES:
        if phrase in m:
            print(f'matched on: {phrase}')
            await message.channel.send(get_shop_hours())
            break

client.run(discord_token.TOKEN)