Print images from dev too

This commit is contained in:
2025-12-31 04:42:05 +00:00
parent 461844ef70
commit e3cfd0c5ee

20
main.py
View File

@@ -16,19 +16,25 @@ from aiomqtt import Client
VENDOR_ID = 0x0416 VENDOR_ID = 0x0416
PRODUCT_ID = 0x5011 PRODUCT_ID = 0x5011
STATIC_URL = 'https://static.my.protospace.ca/' PROD_STATIC_URL = 'https://static.my.protospace.ca/'
DEV_STATIC_URL = 'https://static.spaceport.dns.t0.vc/'
PRINTER_WIDTH = 384 PRINTER_WIDTH = 384
def print_picture(filename, p): def print_picture(topic, filename, p):
logging.info('New picture submitted: %s', filename) logging.info('New picture submitted: %s', filename)
if topic.startswith('dev_'):
url = DEV_STATIC_URL
else:
url = PROD_STATIC_URL
try: try:
p.hw('INIT') #p.hw('INIT')
response = requests.get(STATIC_URL + filename, timeout=5) response = requests.get(url + filename, timeout=5)
response.raise_for_status() response.raise_for_status()
print('New image detected, printing...') logging.info('New image detected, printing...')
last_image_content = response.content last_image_content = response.content
img = Image.open(io.BytesIO(response.content)) img = Image.open(io.BytesIO(response.content))
@@ -41,7 +47,7 @@ def print_picture(filename, p):
p.cut() p.cut()
except requests.exceptions.RequestException as e: except requests.exceptions.RequestException as e:
print(f'Error downloading image: {e}') logging.info(f'Error downloading image: {e}')
async def process_mqtt(message, p): async def process_mqtt(message, p):
@@ -50,7 +56,7 @@ async def process_mqtt(message, p):
logging.debug('MQTT topic: %s, message: %s', topic, text) logging.debug('MQTT topic: %s, message: %s', topic, text)
if 'spaceport/drawing/new' in topic: if 'spaceport/drawing/new' in topic:
print_picture(text, p) print_picture(topic, text, p)
else: else:
logging.debug('Invalid topic, returning') logging.debug('Invalid topic, returning')
return return