3 Z$x@sddlZddlZddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z GdddZ dd Z d d ZGd d d ZGdddeZddZddZddZddZddZddZddZedkredS)N) _debugprint)config)reducec@seZdZddgZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ d/ddZd0ddZddZddZd d!Zd"d#Zd$d%Zd1d'd(Zd2d)d*Zd3d+d,Zd-d.ZdS)4PrinteroptionslocalcKsX||_||_g|_t|dk}d}|rB|jf||jr>d}nd}|rN|jd|_dS)z @param name: printer name @type name: string @param connection: CUPS connection @type connection: CUPS.Connection object @param kw: printer attributes @type kw: dict indexed by string rTFN)name connection class_memberslenupdateis_class getAttributes_ppd)selfr r kwZhave_kwZ fetch_attrsr9/usr/lib/python3/dist-packages/cupshelpers/cupshelpers.py__init__s   zPrinter.__init__cCs|jdk rtj|jdS)N)rosunlink)rrrr__del__9s zPrinter.__del__cCs d|jS)Nz)r )rrrr__repr__=szPrinter.__repr__c Csvdd}d}t|}x\tjD]R}|j|r|||d}||jkrFq|dkrRd}t||t|jtt|@qWdS)NcSs|jS)N)lower)strrrr _ascii_lowerBsz+Printer._expand_flags.._ascii_lowerZ CUPS_PRINTER_classr) r cups__dict__ startswith_flags_blacklistsetattrbooltypegetattr)rrprefixZ prefix_lengthr Z attr_namerrr _expand_flags@s   zPrinter._expand_flagscKs|jdd|_|jtjk|_|jdd|_|jdd|_|jdd|_|jdd|_|jd d|_ |jd d|_ |jd d|_ t |j t kr|j g|_ |j |jdkr|j |_|`|jd g|_t |jt kr|jg|_|jj||_dS) z Update object from printer attributes. @param kw: printer attributes @type kw: dict indexed by string z printer-staterz device-uriz printer-infozprinter-is-sharedNzprinter-locationzprinter-make-and-modelz printer-typezprinter-uri-supportedz member-names)getstaterZIPP_PRINTER_STOPPEDenabledZ device_uriinfoZ is_sharedlocationmake_and_modelr$ uri_supportedlistr'Z not_sharedr sortother_attributes)rrrrrr Ss(     zPrinter.updatecCs|jj|j}i|_i|_dddgfddddddgfd|_x|jD]\}}|jd r|d td  }|d krtqF|j |dd p|jj |d pd}t |t rt dd|}||j|<|d|kr||d}||f|j|<qF|jd rF|dkrF|d krF|d krF|j d rF||j|<qFW|j dd!\|_|_|j ddg|_|j d d|_|j ddg|_|j d dppd|_|j ddg|_d|_g|_d|kr|d|_d|_nd|kr|d|_dj|j|_|jf|d S)"z Fetch further attributes for the printer. Normally only a small set of attributes is fetched. This method is for fetching more. FalseTruenoneZsinglez single-thickZdoublez double-thick)Z landscapez page-borderz-defaultN job-sheetsprinter-error-policyprinter-op-policy notify-eventsdocument-formatnotify-lease-durationz -supportedr(cSs |d|S)N,r)xyrrrsz'Printer.getAttributes..zjob-sheets-defaultzrequesting-user-name-zjob-sheets-supportedzprinter-error-policy-supporteddefaultzprinter-op-policy-supportedTzrequesting-user-name-allowedFzrequesting-user-name-deniedz, )r6r7r8r9r:r;)r5r5)r ZgetPrinterAttributesr Z attributesr2Zpossible_attributesitemsendswithr r) isinstancer0rr Zjob_sheet_startZ job_sheet_endZjob_sheets_supportedZ error_policyZerror_policy_supportedZ op_policyZop_policy_supportedZ default_allow except_usersjoinZexcept_users_stringr )rattrskeyvaluer Z supportedrrrros`              zPrinter.getAttributescCsR|jdjdsdS|jddd}|jdd}|jdd}|dkrNd}|S) zc Find out which server defines this printer. @returns: server URI or None rzipp://N/:zlocalhost.localdomainZ localhost)r/r split)rurirrr getServerszPrinter.getServercCsd}|jdkrpy |jj|j|_tj|j}Wn@tjk rn}z"|j\}}|tjkr\d}nWYdd}~XnX|dkr|jdk rtj|j}|S)z Obtain the printer's PPD. @returns: cups.PPD object, or False for raw queues @raise cups.IPPError: IPP error NF) rr getPPDr rZPPDIPPErrorargsZ IPP_NOT_FOUND)rresultZemargsemrrrrOs    zPrinter.getPPDcCsDt|tr.tjtj}|dkr.t|j|d}|jj|j ||dS)z Set a printer's option. @param name: option name @type name: string @param value: option value @type value: option-specific .N) rCfloatlocale nl_langinfo RADIXCHARrreplacer ZaddPrinterOptionDefaultr )rr rHZ radixcharrrr setOptions  zPrinter.setOptioncCs|jj|j|dS)zh Unset a printer's option. @param name: option name @type name: string N)r ZdeletePrinterOptionDefaultr )rr rrr unsetOptionszPrinter.unsetOptionNcCs>|r|jj|jn&|r,|jj|j|dn|jj|jdS)z Set the printer's enabled state. @param on: whether it will be enabled @type on: bool @param reason: reason for this state @type reason: string )reasonN)r enablePrinterr ZdisablePrinter)ronr]rrr setEnableds zPrinter.setEnabledcCs>|r|jj|jn&|r,|jj|j|dn|jj|jdS)z Set the printer's accepting state. @param on: whether it will be accepting @type on: bool @param reason: reason for this state @type reason: string )r]N)r acceptJobsr Z rejectJobs)rr_r]rrr setAcceptings zPrinter.setAcceptingcCs|jj|j|dS)zy Set the printer's shared state. @param on: whether it will be accepting @type on: bool N)r ZsetPrinterSharedr )rr_rrr setSharedszPrinter.setSharedcCs|jj|j|dS)zs Set the printer's error policy. @param policy: error policy @type policy: string N)r ZsetPrinterErrorPolicyr )rpolicyrrrsetErrorPolicyszPrinter.setErrorPolicycCs|jj|j|dS)z{ Set the printer's operation policy. @param policy: operation policy @type policy: string N)r ZsetPrinterOpPolicyr )rrdrrrsetOperationPolicy'szPrinter.setOperationPolicycCs|jj|j||dS)z Set the printer's job sheets. @param start: start sheet @type start: string @param end: end sheet @type end: string N)r ZsetPrinterJobSheetsr )rstartendrrr setJobSheets0s zPrinter.setJobSheetscCst|trX|j}dd|D}g}x|D]}|j|q*Wdd|D}dd|D}|rn|jj|j|n|jj|j|dS)z Set access control list. @param allow: whether to allow by default, otherwise deny @type allow: bool @param except_users: exception list @type except_users: string list cSsg|]}|jdqS)r<)rL).0urrr Fsz%Printer.setAccess..cSsg|] }|jqSr)strip)rjrkrrrrlJscSsg|] }|r|qSrr)rjZ_frrrrlKsN)rCrrLextendr ZsetPrinterUsersDeniedr ZsetPrinterUsersAllowed)rZallowrDZusersrkrrr setAccess;s  zPrinter.setAccessFc Csg}y@ydddg}|jj|d}Wntk r@|jj}YnXWntjk rZ|SXx|jD]\}}y"|d}||jddd}WnwfYnX||jkrqf| sd|krf|ddkrf|j||dk rft ||krfPqfW|S) z Find out whether jobs are queued for this printer. @param only_tests: whether to restrict search to test pages @type only_tests: bool @returns: list of job IDs zjob-idzjob-printer-urizjob-name)requested_attributesrJrNz Test Page) r getJobs TypeErrorrrPrArindexr appendr ) r only_testslimitretrjobsidrFrMrrr jobsQueuedRs0    zPrinter.jobsQueuedcCsg}yFydddg}|jjd|d}Wn"tk rF|jjdd}YnXWntjk r`|SXx|jD]|\}}y"|d}||jddd }WnwlYnX||jkrql|jdtj tj krql|j ||d k rlt ||krlPqlW|S) zo Find out whether there are preserved jobs for this printer. @return: list of job IDs zjob-idzjob-printer-uriz job-stateZ completed) which_jobsrp)r|rJrN) r rqrrrrPrArsr r)ZIPP_JOB_PENDINGZIPP_JOB_COMPLETEDrtr )rrvrwrxryrzrFrMrrr jobsPreservedws2    zPrinter.jobsPreservedcCs|jd|dS)zl Find out whether test jobs are queued for this printer. @returns: list of job IDs T)rurv)r{)rrvrrr testsQueuedszPrinter.testsQueuedc Cst|jj|jtjR}yd}|jj||jdWnJtjk r}z,|j \}|tj tj gkrfdStj|WYdd}~XnX|j ddd|j D}d}d}xN|D]F}|jdr|jd d } | |jkrd |d d||<d }|d 7}qW|rf|j d|jdd|D|j|j|j dy|jj||jdWntjk rddSX|SQRXdS)z9 Set this printer as the system default. z/admin/conf/lpoptions)fdFNrcSsg|]}|jdqS)zUTF-8)decode)rjlinerrrrlsz(Printer.setAsDefault..zDefault  rzDest TcSsg|]}|jdqS)zUTF-8)encode)rjrrrrrls)r setDefaultr tempfileZ TemporaryFileZgetFilefilenorZ HTTPErrorrQZHTTP_NOT_FOUNDZHTTP_AUTHORIZATION_CANCELEDseek readlinesr rL writelinestruncateflushZputFile) rfresourcerSslinesZchangedirr rrr setAsDefaults@        zPrinter.setAsDefault)N)N)FN)N)N)__name__ __module__ __qualname__r!rrrr'r rrNrOr[r\r`rbrcrerfriror{r}r~rrrrrrs,D       % # rcCs\|j}|j}xF|jD]:\}}t||f|}|||<||kr|||_|jjqW|S)z Obtain a list of printers. @param connection: CUPS connection @type connection: CUPS.Connection object @returns: L{Printer} list ) getPrintersZ getClassesrArr r1)r Zprintersclassesr Zprinterrrrrs rc Csi}|jd}x<|D]4}|jddkr(q|jdd\}}|j||j<qWd|krd|jd|dd|kr||jd|dd|kr|jd |dxdD]}|j|dqW|d dkrg|d <n|d jd|d <|S)z Parse an IEEE 1284 Device ID, so that it may be indexed by field name. @param id: IEEE 1284 Device ID, without the two leading length bytes @type id: string @returns: dict indexed by field name ;rKrZ MANUFACTURERMFGZMODELMDLz COMMAND SETCMDCLSDESSNSPJr(r<) rrrrrrrrr)rLfindrm setdefault)rzid_dictpiecesZpiecer rHrrr parseDeviceIDs&     rc@s(eZdZdZddZddZddZdS) Devicez. This class represents a CUPS device. cKs||_|jdd|_|jdd|_|jdd|_|jdd|_|jdd|_|jd}|d|_t |d k|_ t |j|_ |j d }|dkr|j jd d r||d d |j d <d S)z~ @param uri: device URI @type uri: string @param kw: device attributes @type kw: dict z device-classr(z device-infozdevice-make-and-modelz device-idzdevice-locationrKrrzserial=rNr)rMr)Z device_classr,r.rzr-rLr$r rrrr)rrMrZ uri_piecesrrrrrs    zDevice.__init__cCs d|jS)Nz)rM)rrrrrszDevice.__repr__cCs|dkr dS|j|jkr&|jr"dSdS|j}|dkrt|jjddkrJd}n*|jjddkr`d }n|jjd dkrtd }|d kr|jjjd dkrd}|j}|dkr|jjddkrd}n*|jjddkrd }n|jjd dkrd }|d kr|jjjd dkrd}|j r||kr|dkr(dS|dkr6dS|dkrDdS|dkrRdS|dkr`dS|dkrndS|dkr|dS|dkrdS|dkrdS|dkrdS|dkrdS|dkrdS|dkrdS|dkrdS|d krdS|d krdS|dkrdS|dkrdS|d kr$dS|d kr2dS|dkr@dS|dkrNdS|d kr\dS|d krjdS|dkrxdS|dkrdSt|jt|jk}|s|j|jk}|S)z9 Compare devices by order of preference. NTFZdnssdz._ipprZdnssdiz._pdl-datastreamZdnssdsz ._printerZdnssdlZusbZfaxZusbfaxserialZparallelZhpZhpfaxZippsZippZsocketZlpdrrrrrrrr)rr$rMrrr#rzr,)rotherstypeZotyperRrrr__lt__!s                             z Device.__lt__N)rrr__doc__rrrrrrrrsrc@seZdZddZddZdS)_GetDevicesCallcCsHd|kr*|jd|_|j|d<|jf|Sd|_|jf|}|j||S)NZ reply_handler)r)_client_reply_handler_reply_handler getDevices)rr kwdsrRrrrcalls    z_GetDevicesCall.callcCsbxD|jD]8\}}t|f|}|||<|jdkr |jdkr |j|_q W|jrZ|j||n|SdS)Nr()rArr,r.r)rr ZdevicesrMdatadevicerrrrs  z_GetDevicesCall._reply_handlerN)rrrrrrrrrrs rcKst}|j||S)z Obtain a list of available CUPS devices. @param connection: CUPS connection @type connection: cups.Connection object @returns: a list of L{Device} objects @raise cups.IPPError: IPP Error )rr)r roprrrrs rcCs.|j||j||jdkr*|j|dS)a Set a new printer enabled, accepting jobs, and (if necessary) the default printer. @param connection: CUPS connection @type connection: cups.Connection object @param name: printer name @type name: string @raise cups.IPPError: IPP error N)r^raZ getDefaultr)r r rrractivateNewPrinters   rcsfddfdd}x||D]v}|jdkr2q"|j|j}|r"|j|jkr"|j}xB|jD]8}|d|kr\|j|j|tdt|jt|fq\Wq"WdS) z Copy default options between PPDs. @param ppd1: source PPD @type ppd1: cups.PPD object @param ppd2: destination PPD @type ppd2: cups.PPD object cs0|jdd}x|jD]}|j|qW|S)N)rZ subgroupsrn)grouprg)getPPDGroupOptionsrrrs z*copyPPDOptions..getPPDGroupOptionsc3s,x&|jD]}x|D] }|VqWqWdS)N)Z optionGroups)ppdroption)rrriteratePPDOptionss z)copyPPDOptions..iteratePPDOptionsZ PageRegionchoicez set %s = %sN)keywordZ findOptionZuiZ defchoicechoices markOptionrrepr)Zppd1Zppd2rrZ new_optionrHrr)rrcopyPPDOptionss       rcCsd}ddddddg}x|D]}||krd}qWyltd }xV|D]N}|jd rNq>|jjjd rfd}q>|jjjd r~d}q>|jd kr>Pq>W|jWn YnXy|jd|td|Wntd|YnXdS)z Set the PPD page size according to locale. @param ppd: PPD @type ppd: cups.PPD object @param language: language, as given by the first element of locale.setlocale @type language: string ZA4CZPOSIXZenZen_USZen_CAZfr_CAZLetterz/etc/papersize#Za4letterr(ZPageSizezset PageSize = %sz*Failed to set PageSize (%s not available?)N)openr rmrcloserr)rZlanguagesizerZeachrrrrrsetPPDPageSizes0       rc s:ddd}gfdd}d}}|jd}|r`|jjdd }|jd d }|jd d }|jdd}|jdd ks|jdd!krd }|jd}x|D]}|jjd} x| D]} | jjd} | d}||}|s||qtjj|dkrt | } d} d}xT| | krP| | }|j |rD|t |d}||}|sB||P| d7} qWqW|sPqW|sn| r6t j dd\}}tj ||j|tj|dtjtj|d}d}x|D]z}|j |r|t |djjd }y|jd\}}}WnwYnX||tjd}|s|tjd|qWS)"z Check that all relevant executables for a PPD are installed. @param ppd: PPD @type ppd: cups.PPD object @returns: string list, representing missing executables /usr/bin:/binc(Ss|dkr dS|jd}|d3kr*|d|}t|dkr:dS|ddkrttj|tjrdtd||Std |dS|jd d4krdS|d5krdSxJ|jd D]<}|jtjj tjj |}tj|tjrtd||SqWtd2||fdS)6N-Zbuiltin%rrtruerJz %s: foundz %s: NOT found=rKrU[aliasbindbreakcdcontinuedeclareechoelseevalexecexitexportfiifkillletrpopdprintfpushdpwdreadreadonlysetshiftshoptsourcetestthentrapr$ulimitumaskunaliasunsetwaitz%s: NOT found in %srr)'rKrUrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr$rrrrr) rr raccessX_OKrrLrstrippathsep)r rpZ componentfilerrr pathcheck s<       z%missingExecutables..pathcheckcs,|jd}|dkr|d|}j|dS)Nrrr)rrt)exer)exes_to_installrr add_missing/s  z'missingExecutables..add_missingNZFoomaticRIPCommandLinez&& r(z""z<(r&r|rrZgsz -sIjsServer=T)textZrtz *cupsFilter:z/filter:/usr/lib64/cups/filterz/filter/)rrr)ZfindAttrrHrZrrLrmrrbasenamer r rZmkstemprZwriteFdlseekSEEK_SETfdopenrZcupsserverbindir)rrrrZexepathattrZcmdlineZpipespipeZcmdscmdrQZargnZargisearchargZtmpfdZtmpfnamerrZmimetypeZcostr)rrmissingExecutablessr $                 rcCst|}g|fS)z Check that all relevant executables for a PPD are installed. @param ppd: PPD @type ppd: cups.PPD object @returns: string list pair, representing missing packages and missing executables )r)rZ executablesrrrmissingPackagesAndExecutabless rcCs0tj}x"t|jD]}t|j|jqWdS)N)rZ ConnectionrvaluesprintrMr)crrrr_mainsr__main__)rpprintrrrestringrWr(rr functoolsrrrrrobjectrrrrrrrrrrrrrs.0   7  (