connectandprint.py

# -*- coding: utf-8 -*-
from __future__ import absolute_import, division, print_function, unicode_literals

import octoprint.plugin
import flask
import time


class ConnectAndPrintPlugin(octoprint.plugin.StartupPlugin,
                            octoprint.plugin.EventHandlerPlugin, 
                                octoprint.plugin.RestartNeedingPlugin):
    def __init__(self):
        self.psu_control_mode = False

    def on_startup(self, host, port):
        psucontrol_helpers = self._plugin_manager.get_helpers("psucontrol")
        if not psucontrol_helpers or 'register_plugin' not in psucontrol_helpers.keys():
            self._logger.debug("PSU-Control not available, skipping registration")
            return

        self._logger.debug("Registering plugin with PSUControl")
        self.psu_control_mode = True
        psucontrol_helpers['register_plugin'](self)

    def on_event(self, event, payload):
        if event == octoprint.events.Events.UPLOAD:
            if(payload['print']):
                self._logger.info(
                    "Uploaded file to print detected, connecting and printing...")
            
                self._connect_and_print(payload["path"])
            else:
                self._logger.info(
                    "Uploaded file detected, connecting and printing...")

    def _connect_and_print(self, file_path):
        printer = self._printer

        if self.psu_control_mode:
            psu_on = self._plugin_manager.get_helpers("psucontrol")["get_psu_state"]()
            self._logger.debug("PSU-Status: "+str(psu_on))
            if not psu_on:
                self._logger.debug("Trying to turn PSU on")
                self._plugin_manager.get_helpers("psucontrol")["turn_psu_on"]()

        if not printer.is_operational():
            printer.connect()

        self._logger.info("Waiting for printer to connect...")
        timeout = 120  # 2 minutes in seconds
        start_time = time.time()

        while not printer.is_operational() and time.time() - start_time < timeout:
            time.sleep(1)

        if printer.is_operational():
            printer.select_file(file_path, False)
            printer.start_print()
        else:
            self._logger.error("Printer connection timed out after 2 minutes.")

    def get_update_information(self):
        return dict(
            connectandprint=dict(
                displayName=self._plugin_name,
                displayVersion=self._plugin_version,
                type="httpheader",
                header_name="ETag",
                url="https://raw.githubusercontent.com/Maxinger15/connectandprint/master/connectandprint.py",
                method="single_file_plugin"
            )
        )

__plugin_name__ = "Connect And Print"
__plugin_pythoncompat__ = ">=2.7,<4"
__plugin_version__ = "1.2.0"
__plugin_description__='Automatically connect to your printer on file upload and start printing'
__plugin_author__="Max Grallinger"
__plugin_url__="https://github.com/Maxinger15/connectandprint"
__plugin_license__="AGPLv3"
__plugin_implementation__ = ConnectAndPrintPlugin()
__plugin_hooks__ = {
    "octoprint.plugin.softwareupdate.check_config": __plugin_implementation__.get_update_information
}