home *** CD-ROM | disk | FTP | other *** search
/ PC Welt 2006 November (DVD) / PCWELT_11_2006.ISO / casper / filesystem.squashfs / usr / lib / hplip / print < prev    next >
Encoding:
Text File  |  2006-08-30  |  8.2 KB  |  285 lines
  1. #!/usr/bin/env python
  2. # -*- coding: utf-8 -*-
  3. #
  4. # (c) Copyright 2003-2006 Hewlett-Packard Development Company, L.P.
  5. #
  6. # This program is free software; you can redistribute it and/or modify
  7. # it under the terms of the GNU General Public License as published by
  8. # the Free Software Foundation; either version 2 of the License, or
  9. # (at your option) any later version.
  10. #
  11. # This program is distributed in the hope that it will be useful,
  12. # but WITHOUT ANY WARRANTY; without even the implied warranty of
  13. # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14. # GNU General Public License for more details.
  15. #
  16. # You should have received a copy of the GNU General Public License
  17. # along with this program; if not, write to the Free Software
  18. # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  19. #
  20. # Author: Don Welch
  21. #
  22.  
  23. __version__ = '2.0'
  24. __title__ = 'Print Utility'
  25. __doc__ = "A simple front end to 'lpr'. Provides a print UI from the Device Manager if kprinter, gtklp, or xpp are not installed."
  26.  
  27. # Std Lib
  28. import sys, os, getopt, re, socket
  29.  
  30. # Local
  31. from base.g import *
  32. from base.msg import *
  33. from base import utils, device
  34. import base.async_qt as async
  35. from prnt import cups
  36.  
  37. # PyQt
  38. if not utils.checkPyQtImport():
  39.     log.error("PyQt/Qt initialization error. Please check install of PyQt/Qt and try again.")
  40.     sys.exit(1)
  41.  
  42. from qt import *
  43. from ui.printerform import PrinterForm
  44.  
  45. app = None
  46. printdlg = None
  47. client = None
  48.  
  49. USAGE = [(__doc__, "", "name", True),
  50.          ("Usage: hp-print [PRINTER|DEVICE-URI] [OPTIONS] [FILE LIST]", "", "summary", True),
  51.          utils.USAGE_ARGS,
  52.          utils.USAGE_DEVICE,
  53.          ("To specify a CUPS printer:", "-P<printer>, -p<printer> or --printer=<printer>", "option", False),
  54.          utils.USAGE_SPACE,
  55.          utils.USAGE_OPTIONS,
  56.          utils.USAGE_LOGGING1, utils.USAGE_LOGGING2, utils.USAGE_LOGGING3,
  57.          utils.USAGE_HELP,
  58.          ("[FILELIST]", "", "heading", False),
  59.          ("Optional list of files:", """Space delimited list of files to print. Files can also be selected for print by adding them to the file list in the UI.""", "option", False),
  60.          utils.USAGE_SPACE,
  61.          utils.USAGE_NOTES,
  62.          utils.USAGE_STD_NOTES1, utils.USAGE_STD_NOTES2, 
  63.          ]
  64.                  
  65.  
  66. def usage(typ='text'):
  67.     if typ == 'text':
  68.         utils.log_title(__title__, __version__)
  69.         
  70.     utils.format_text(USAGE, typ, __title__, 'hp-print', __version__)
  71.     sys.exit(0)
  72.  
  73.  
  74.  
  75. class print_client(async.dispatcher):
  76.  
  77.     def __init__(self):
  78.         async.dispatcher.__init__(self)
  79.         self.create_socket(socket.AF_INET, socket.SOCK_STREAM)
  80.         self.connect((prop.hpssd_host, prop.hpssd_port)) 
  81.         self.in_buffer = ""
  82.         self.out_buffer = ""
  83.         self.fields = {}
  84.         self.data = ''
  85.         self.error_dialog = None
  86.         self.signal_exit = False
  87.  
  88.         # handlers for all the messages we expect to receive
  89.         self.handlers = {
  90.                         'eventgui' : self.handle_eventgui,
  91.                         'unknown' : self.handle_unknown,
  92.                         'exitguievent' : self.handle_exitguievent,
  93.                         }
  94.  
  95.         self.register_gui()
  96.  
  97.     def handle_read(self):
  98.         log.debug("Reading data on channel (%d)" % self._fileno)
  99.         log.debug(repr(self.in_buffer))
  100.  
  101.         self.in_buffer = self.recv(prop.max_message_len)
  102.  
  103.         if self.in_buffer == '':
  104.             return False
  105.  
  106.         remaining_msg = self.in_buffer
  107.  
  108.         while True:
  109.             try:
  110.                 self.fields, self.data, remaining_msg = parseMessage(remaining_msg)
  111.             except Error, e:
  112.                 log.debug(repr(self.in_buffer))
  113.                 log.warn("Message parsing error: %s (%d)" % (e.opt, e.msg))
  114.                 self.out_buffer = self.handle_unknown()
  115.                 log.debug(self.out_buffer)
  116.                 return True
  117.  
  118.             msg_type = self.fields.get('msg', 'unknown')
  119.             log.debug("%s %s %s" % ("*"*40, msg_type, "*"*40))
  120.             log.debug(repr(self.in_buffer))
  121.  
  122.             try:
  123.                 self.out_buffer = self.handlers.get(msg_type, self.handle_unknown)()
  124.             except Error:
  125.                 log.error("Unhandled exception during processing")
  126.  
  127.             if len(self.out_buffer): # data is ready for send
  128.                 self.sock_write_notifier.setEnabled(True)
  129.  
  130.             if not remaining_msg:
  131.                 break
  132.  
  133.         return True
  134.  
  135.     def handle_write(self):
  136.         if not len(self.out_buffer):
  137.             return
  138.  
  139.         log.debug("Sending data on channel (%d)" % self._fileno)
  140.         log.debug(repr(self.out_buffer))
  141.         
  142.         try:
  143.             sent = self.send(self.out_buffer)
  144.         except:
  145.             log.error("send() failed.")
  146.  
  147.         self.out_buffer = self.out_buffer[sent:]
  148.  
  149.  
  150.     def writable(self):
  151.         return not ((len(self.out_buffer) == 0)
  152.                      and self.connected)
  153.  
  154.     def handle_exitguievent(self):
  155.         self.signal_exit = True
  156.         if self.signal_exit:
  157.             if printdlg is not None:
  158.                 printdlg.close()
  159.             qApp.quit()
  160.  
  161.         return ''
  162.  
  163.     # EVENT
  164.     def handle_eventgui(self):
  165.         global printdlg
  166.         try:
  167.             job_id = self.fields['job-id']
  168.             event_code = self.fields['event-code']
  169.             event_type = self.fields['event-type']
  170.             retry_timeout = self.fields['retry-timeout']
  171.             lines = self.data.splitlines()
  172.             error_string_short, error_string_long = lines[0], lines[1]
  173.             device_uri = self.fields['device-uri']
  174.  
  175.             log.debug("Event: %d '%s'" % (event_code, event_type))
  176.  
  177.             printdlg.EventUI(event_code, event_type, error_string_short,
  178.                              error_string_long, retry_timeout, job_id,
  179.                              device_uri)
  180.  
  181.         except:
  182.             log.exception()
  183.  
  184.         return ''
  185.  
  186.     def handle_unknown(self):
  187.         #return buildResultMessage('MessageError', None, ERROR_INVALID_MSG_TYPE)
  188.         return ''
  189.  
  190.     def handle_messageerror(self):
  191.         return ''
  192.  
  193.     def handle_close(self):
  194.         log.debug("closing channel (%d)" % self._fileno)
  195.         self.connected = False
  196.         async.dispatcher.close(self)
  197.  
  198.     def register_gui(self):
  199.         out_buffer = buildMessage("RegisterGUIEvent", None, 
  200.                                   {'type': 'print', 
  201.                                    'username': prop.username})
  202.         self.send(out_buffer)
  203.  
  204.  
  205. def main(args):
  206.     try:
  207.         opts, args = getopt.getopt(sys.argv[1:], 'P:p:d:hl:g',
  208.                                    ['printer=', 'device=', 'help', 
  209.                                     'help-rest', 'help-man', 'logging='])
  210.     except getopt.GetoptError:
  211.         usage()
  212.  
  213.     printer_name = None
  214.     device_uri = None
  215.     log_level = logger.DEFAULT_LOG_LEVEL
  216.     bus = 'cups'
  217.  
  218.     if os.getenv("HPLIP_DEBUG"):
  219.         log.set_level('debug')
  220.     
  221.     for o, a in opts:
  222.         if o in ('-h', '--help'):
  223.             usage()
  224.  
  225.         elif o == '--help-rest':
  226.             usage('rest')
  227.             
  228.         elif o == '--help-man':
  229.             usage('man')
  230.  
  231.         elif o in ('-p', '-P', '--printer'):
  232.             printer_name = a
  233.  
  234.         elif o in ('-d', '--device'):
  235.             device_uri = a
  236.  
  237.         elif o in ('-l', '--logging'):
  238.             log_level = a.lower().strip()
  239.             if not log.set_level(log_level):
  240.                 usage()
  241.                 
  242.         elif o == '-g':
  243.             log.set_level('debug')
  244.  
  245.  
  246.     # Security: Do *not* create files that other users can muck around with
  247.     os.umask (0077)
  248.     
  249.     utils.log_title(__title__, __version__)
  250.  
  251.     log.set_module('hp-print')
  252.  
  253.     global client
  254.     try:
  255.         client = print_client()
  256.     except Error:
  257.         log.error("Unable to create client object.")
  258.         sys.exit(0)
  259.  
  260.     # create the main application object
  261.     global app
  262.     app = QApplication(sys.argv)
  263.  
  264.     global printdlg
  265.     printdlg = PrinterForm(client.socket, bus, device_uri, printer_name, args)
  266.     printdlg.show()
  267.     app.setMainWidget(printdlg)
  268.  
  269.     user_config = os.path.expanduser('~/.hplip.conf')
  270.     loc = utils.loadTranslators(app, user_config)
  271.  
  272.     try:
  273.         log.debug("Starting GUI loop...")
  274.         app.exec_loop()
  275.     except KeyboardInterrupt:
  276.         pass
  277.     except:
  278.         log.exception()
  279.     
  280.     return 0
  281.  
  282. if __name__ == "__main__":
  283.     sys.exit(main(sys.argv[1:]))
  284.  
  285.