source: server/lib/gutenbach/server/printer.py @ cad7502

#import alsaaudio as aa
from .exceptions import InvalidJobException, InvalidPrinterStateException
from gutenbach.ipp.attribute import Attribute
import gutenbach.ipp as ipp
import gutenbach.ipp.constants as const
import logging
import time

# initialize logger
logger = logging.getLogger(__name__)

class GutenbachPrinter(object):

    # for IPP
    attributes = [
        "printer-uri-supported",
        "uri-authentication-supported",
        "uri-security-supported",
        "printer-name",
        "printer-state",
        "printer-state-reasons",
        "ipp-versions-supported",
        "operations-supported",
        "charset-configured",
        "charset-supported",
        "natural-language-configured",
        "generated-natural-language-supported",
        "document-format-default",
        "document-format-supported",
        "printer-is-accepting-jobs",
        "queued-job-count",
        "pdl-override-supported",
        "printer-up-time",
        "compression-supported"
    ]

    #def __init__(self, name, card, mixer):
    def __init__(self, name):

        self.name = name
        self.uri = "ipp://localhost:8000/printers/" + self.name
        self.time_created = int(time.time())
        self.state = "idle"

        # if card >= len(aa.cards()):
        #     raise aa.ALSAAudioError(
        #       "Audio card at index %d does not exist!" % card)
        # elif mixer not in aa.mixers(card):
        #     raise aa.ALSAAudioError(
        #       "Audio mixer '%s' does not exist!" % mixer)
        
        # self.card = card
        # self.mixer = mixer

        self.finished_jobs = []
        self.active_jobs = []
        self.jobs = {}

        self._next_jobid = 0

    def __getattr__(self, attr):
        try:
            return self.__getattribute__(attr)
        except AttributeError:
            pass
        return self.__getattribute__(attr.replace("-", "_"))

    def __hasattr__(self, attr):
        try:
            getattr(self, attr)
            return True
        except AttributeError:
            return False

    ## Printer attributes

    @property
    def printer_uri_supported(self):
        return self.uri

    @property
    def uri_authentication_supported(self):
        return "none"

    @property
    def uri_security_supported(self):
        return "none"

    @property
    def printer_name(self):
        return self.name

    @property
    def printer_state(self):
        return 3 # idle

    @property
    def printer_state_reasons(self):
        return "none"

    @property
    def ipp_versions_supported(self):
        return ("1.0", "1.1")
    # XXX: We should query ourself for the supported operations

    @property
    def operations_supported(self):
        return 0xa # get-jobs

    @property
    def charset_configured(self):
        return "utf-8"

    @property
    def charset_supported(self):
        return "utf-8"

    @property
    def natural_language_configured(self):
        return "en-us"

    @property
    def generated_natural_language_supported(self):
        return "en-us"

    @property
    def document_format_default(self):
        return "application/octet-stream"

    @property
    def document_format_supported(self):
        return ("application/octet-stream", "audio/mp3")

    @property
    def printer_is_accepting_jobs(self):
        return True

    @property
    def queued_job_count(self):
        return len(self.active_jobs)

    @property
    def pdl_override_supported(self):
        return "not-attempted"

    @property
    def printer_up_time(self):
        return int(time.time()) - self.time_created

    @property
    def compression_supported(self):
        return "none"

    def get_printer_attributes(self, request):
        attributes = [(attr, getattr(self, attr)) for attr in self.attributes]
        attributes = map(lambda x: x if isinstance(x[1], (tuple, list)) else (x[0], [x[1]]),
                         attributes)
        return attributes

    ## Printer operations
    @property
    def next_jobid(self):
        self._next_jobid += 1
        return self._next_jobid

    def print_job(self, job):
        jobid = self.next_jobid
        self.active_jobs.append(jobid)
        self.jobs[jobid] = job
        job.enqueue(self, jobid)
        return jobid

    def complete_job(self, jobid):
        job = self.jobs[self.active_jobs.pop(0)]
        if job.jobid != jobid:
            raise InvalidJobException(
                "Completed job %d has unexpected job id %d!" % \
                (job.jobid, jobid))
        
        self.finished_jobs.append(job)
        job.finish()
        return job.jobid

    def start_job(self, jobid):
        job = self.jobs[self.active_jobs[0]]
        if job.jobid != jobid:
            raise InvalidJobException(
                "Completed job %d has unexpected job id %d!" % \
                (job.jobid, jobid))

        if job.status == 'playing':
            raise InvalidPrinterStateException(
                "Next job in queue (id %d) is " + \
                "already playing!" % jobid)

        job.play()

    def get_job(self, jobid):
        if jobid not in self.jobs:
            raise InvalidJobException(jobid)
        return self.jobs[jobid]

    def get_jobs(self):
        jobs = [self.jobs[job_id] for job_id in self.active_jobs]
        return jobs            

    def __repr__(self):
        return str(self)

    def __str__(self):
        return "<Printer '%s'>" % self.name