QDbNddlZddlZddlZddlmZdZGddZGddZGd d Z Gd d Z Gd dZ ddZ dS)N) parseDeviceIDctjjt |}t |SN)xmletree ElementTreeXMLopenreadlist)filenamepreferreddriverss >>rrcbeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdS) DriverTypez A type of driver. z exact-cmdexactclosegenericnonec||_d|_g|_g|_Gdd}||_d|_dS)NceZdZddZdS)'DriverType.__init__..AlwaysTrueNcdSNTr.)rkds rgetz+DriverType.__init__..AlwaysTrue.getXstrr)r*r+r,r<r.rr AlwaysTruer7Ws(      rr=)nameppd_name attributesr'_fit _packagehint)rr>r=s rrzDriverType.__init__Qsc           JMM  rctj|tj|_|jdS|dd}t |dkrv|d}|dkr\|d}|dr;|dd}|d}|d kr|d|}d |z|_dSdSdSd |z|_dSdS) z: An optional PPD name regular expression. N:rrdrvz////z/usr/share/cups/drv/%sz/usr/lib/cups/driver/%s) rrrr?rBsplitlen startswithrfind)rr!partsschemerestrEfs r add_ppd_namezDriverType.add_ppd_name^s GRT22    ( F sA&& ;;??1XFQx??E**Kqrr(C 3ABww!"1"g,Ds,J))) KKw%>$F!!! ?rcx|j|tj|tjfdS)zl An optional IPP attribute name and regular expression to match against its values. N)r@appendrrr)rr>r!s r add_attributezDriverType.add_attributews3 rz7BD'A'A BCCCCCrc:|j|dS)z8 An optional IEEE 1284 Device ID match. Nr'rSrdeviceid_matchs radd_deviceid_matchzDriverType.add_deviceid_match~  n-----rci|_|D]&}d|j|<||jkrd|j|j<'dSr9)rArI FIT_EXACT FIT_EXACT_CMD)rtextfittypes radd_fitzDriverType.add_fitsU zz|| 5 5G!%DIg $.((04 $,-  5 5rc||_dSr) _packagekit)rhints rset_packagehintzDriverType.set_packagehintsrc|jS)z7 Return the name for this driver type. )r>rs rget_namezDriverType.get_names yrc4d|jt|fzS)Nz )r>idrs r__repr__zDriverType.__repr__s1TYD 4JJJrc(|j|d}|r#|jr|j|sd}|r[|jD]S\}}||vrd}nG||}t |t s|g}d}|D]} || rd}n|sd}nT|rr|jrd|vrd}nd|jr]|d} t | t s| g} d} | D]4} t| } |jD]}|| rd} n5| sd}|S)z Return True if there is a match for all specified criteria. ppdname: string attributes: dict fit: string FTz ppd-device-id) rAr<r?r&r@ isinstancer r'r)rr?r@fitmatchesr>r&valuesany_value_matchesvalue deviceidlistany_id_matches deviceidstrr's rr&zDriverType.matchs)--e,,  t} T]-@-@(-K-K G  #   ez))#GE#D)!6400&$XF$)!#E{{E**,0))#GE  $} $ !B!B $ */: !<662$0>L!&#/""K,k::H!%"" ;;11"-1N!E"&$#GrcdSrr.rs rget_packagehintzDriverType.get_packagehintstrN)r*r+r,r-r]r\ FIT_CLOSE FIT_GENERICFIT_NONErrQrTrYr`rdrfrir&rur.rrr0r0Fs$MII!KH ! ! !GGG2DDD... 555    KKK:::xrr0c0eZdZdZdZdZdZdZdZdS) DriverTypesz! A list of driver types. cg|_dSr) drivertypesrs rrzDriverTypes.__init__srcg}t|D]G}t|jd}t|D]}|jdkr!||jd/|jdkr-||jd|jdg|jdkrmt }t|D]9}|jdkr,||jd|jd:|||jdkr| |j | |I||_ dS) zA Load the list of driver types from an XML file. r>ppdnamer& attributer'r rlN) r r0attribtagrQrTrr"rYr`r^rSr|)rr|types drivertypetchildrXr s rloadzDriverTypes.loadsy {++  JJ-f566Aj)) + +9 ))NNEL$9::::Y+--OOU\&%9%*\'%:<<<<Y*,,%2%5%5N!%eMM 9//*44el66J6;l76KMMM((.9999Y%''IIuz*** LL!     rcP|jD]}||||r|cSdS)z Return the first matching drivertype for a PPD, given its name, attributes, and fitness, or None if there is no match. N)r|r&)rr~ppddictrlrs rr&zDriverTypes.matchsH * " "J'377 "!!!! "trcJtjd|jD|S)z[ Return the subset of driver type names that match a glob pattern. c6g|]}|Sr.)rf).0xs r z&DriverTypes.filter..s HHH! HHHr)fnmatchfilterr|rr!s rrzDriverTypes.filter s/ ~HHt7GHHH&(( (rc g}i}tj}|D]y\}}||||||} | r| } nd} || g} | || || <z|D]7} || gD]}||vr|| |f8|S)a Given a list of driver type names, a dict of PPD attributes by PPD name, and a dict of driver fitness status codes by PPD name, return a list of tuples in the form (driver-type-name, PPD-name), representing PPDs that match the list of driver types. The returned tuples will have driver types in the same order as the driver types given, with the exception that any blacklisted driver types will be omitted from the returned result. r4)r0rvr%r&r<rfrS) rr|ppdsdictrlppdnamesppdtypes fit_defaultr?ppd_dictrr>mdrivertypenames rget_ordered_ppdnamesz DriverTypes.get_ordered_ppdnamess * "*.."3"3   HhXx(BM:O:OPPJ !**-- dB''A HHh   HTNN* = =N$LL."== = =x''.(!;<<<<  = rN) r*r+r,r-rrr&rrr.rrrzrzsi!!!<   ((('''''rrzcNeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d S) PrinterTypez A make-and-model pattern and/or set of IEEE 1284 Device ID patterns for matching a set of printers, together with an ordered list of driver type names. c|d|_g|_g|_t|_t|_dSr)make_and_modelr'drivertype_patternssetavoid blacklistrs rrzPrinterType.__init__Cs3" #% UU rcNtj|tj|_dS)zR Set a make-and-model regular expression. Only one is permitted. N)rrrrrs radd_make_and_modelzPrinterType.add_make_and_modelJs !j'2488rc:|j|dS)z( Add a Device ID match. NrVrWs rrYzPrinterType.add_deviceid_matchPrZrc^|j|dS)z/ Append a driver type pattern. N)rrSstriprr>s radd_drivertype_patternz"PrinterType.add_drivertype_patternVs)  ''77777rc|jS)z: Return the list of driver type patterns. )rrs rget_drivertype_patternsz#PrinterType.get_drivertype_patterns\s ''rc:|j|dS)z3 Add an avoid driver type pattern. N)raddrs radd_avoidtype_patternz!PrinterType.add_avoidtype_patternbs rc|jS)zB Return the set of driver type patterns to avoid. )rrs rget_avoidtype_patternsz"PrinterType.get_avoidtype_patternshs zrc:|j|dS)z8 Add a blacklisted driver type pattern. N)rrrs radd_blacklistedzPrinterType.add_blacklistedns  D!!!!!rc|jS)zE Return the set of blacklisted driver type patterns. )rrs r get_blacklistzPrinterType.get_blacklistts ~rc|jduo |jgk}|jr|j|rd}|s#|jD]}||rd}n|S)a} Return True if there are no constraints to match against; if the make-and-model pattern matches; or if all of the IEEE 1284 Device ID patterns match. The deviceid parameter must be a dict indexed by Device ID field key, of strings; except for the CMD field which must be a list of strings. Return False otherwise. NT)rr'r&)rrr'rmr&s rr&zPrinterType.matchzs&$.F4=B3F   "((.99    ;;))"GErN)r*r+r,r-rrrYrrrrrrr&r.rrrr<s 999 ... 888 (((   """  rrc$eZdZdZdZdZdZdS)PreferenceOrderzA A policy for choosing the preference order for drivers. cg|_dSr)ptypesrs rrzPreferenceOrder.__init__s  rc>t|D]}t}t|D]O}|jdkr!||jd/|jdkrmt }t|D]9}|jdkr,||jd|jd:|||jdkr-t|D]}||j |jdkr.t|D]}| |j |jdkr,t|D]}| |j Q|j |d S) z3 Load the policy from an XML file. zmake-and-modelr&r'r r>driversrrN)r rrrrrr"rYrr^rrrrS)rrprinterptyperrXr rs rrzPreferenceOrder.loads ,-- ' 'GNNEg @ @9 000,,el7.CDDDDY*,,%2%5%5N!%eMM 9//*44el66J6;l76KMMM,,n====Y)++&*5kkGG 44joFFFFGY'))&*5kkFF 33Z_EEEEFY+--&*5kk@@ --z???? K   & & & &3 ' 'rc2|i}|d}g}t}t}|jD]}|||r|D]3}||D]} | |vr|| 4|D]/}||D]} || 0|D]/}||D]} || 0|rZg} |D]@} | | } || =| | 1#t$rY=wxYw| | |r"g} |D]} | |vr| | | }|S)a Return an accumulated list of driver types from all printer types that match a given printer's device-make-and-model and IEEE 1284 Device ID. The deviceid parameter must be None or a dict indexed by short-form upper-case field keys. N) rrr&rrrSrrrindex IndexErrorextend)rr|rr' orderedtypesr avoidtypesrr!ravoidedri remainings rget_ordered_typesz!PreferenceOrder.get_ordered_typess-  H  !N EE UU [ 3 3E{{NH55 3$<<??==G&1&8&8'&B&B== %\99(//<<<= %;;>>44G&1&8&8'&B&B44 " 33334 %225533G&1&8&8'&B&B33 ! z22223  *G  $**A..A$QNNA&&&&!H    ) ) )  %I! ) )I%%$$a((($Ls -E EEN)r*r+r,r-rrrr.rrrrsL'''@=====rrFc ddl}ddl}ddlm}ddlm}ddlm}ddl} |rd} || ||j d|Q| j | j | j tdd }|| jd <| j |d }|} t|\} } t!}|| t%}|| || z }|rt'd ||fz|} |}n$#t,$r|}YnwxYw||}|s|r|r*|d |}n't7|}d||dd|ddi}|D]Z\}}|ddkr |dd}t?|t@r|d}|sV|!dst'|t7|}|"|d|d|d|d|}|dd}|#|||}i}|$D]}|%|||<|&|||}d}|D]&\}} t'd|| ||| fz|dz }'\dS|'D]}!|(|!D]}"|)|!|"}#|!dz|"z}|#||d}i}|#$D]}tTj+||<|&||#|}t'|dzd}|D]\}} t'd|| |fz|dz } t'dS)Nrr)ppds)pprint)timec$t|dSr)print)rs r debugprintztest..debugprints 1IIIIIrrz..rCUPSHELPERS_XMLDIRzpreferreddrivers.xmlzTime to load %s: %.3fsrootz xxx://yyyMFG MDL) device-iddevice-make-and-modelrDrHrzxxx:DESr$rz%d %s (%s, %s)z%d %s (%s)),cupslocalerrrros.pathset_debugprint_fn setlocaleLC_ALLpathjoindirname__file__environrrzrrr ConnectiongetPPDs2AttributeErrorgetPPDsPPDssetUser getDevicesrr%findr<rkr rKgetPPDNamesFromDeviceIDrkeysgetInfoFromPPDNamergetMakes getModelsgetInfoFromModelr0rv)$xml_pathattachedr'debugrrrrrosr loadstartxmldrivertypesxmlpreferenceorderr| preforderloadtimeccupsppds ppdfinderdevicesdevidurideviceid_dictrlmmrr~ orderedppdsrrppdmakemodelrs$ rtestrsCKKKMMMNNN ,     +++ fmR(((7<<rw/I/I/3"5"5!&(((0BJ#$w||X'=>>HI+;X+F+F(^'..Kn%%%""I NN&'''tww"H ? &(H)==>>> A ::==    99<<  8$$IE8E   LL& ! ! !llooGG!8,,E#+3BG,,,BG,,8P  G#==++# # KCxx~~##JJ R00E5$'' !a >>6** s #U++G33WU^5 F F '}}//88G#-#7CLL)>> @H#OO b3h)FAs,3{:;;;FAA%   s&E;;FF)NFNF) rrxml.etree.ElementTreer cupshelpersrrrr0rzrrrr.rrr sW. &&&&&&"""$$$$$$$$LPPPPPPPPdbbbbbbbbHVVVVVVVVpeeeeeeeePoooooor