3 'Y@sddlZddlmZddlmZddlZddlZddlZddlZddl Z ddl Z ddl Z ddlm Z mZddl mZddgZd e jd fd e jd fd e jdfde jdfde jdfde jdfde jdfde jdfde jdfde jdfde jdfde jd fg ZiZxeD]\ZZeeej<qWd!d"d#d$d%d&d'Ze jd(Ze jd)Ze jd*Ze jd+e jZd,dZd-d.Zd/d0ZGd1ddZ d2d3Z!dS)4N) parseDeviceID)xmldriverprefs) _debugprintset_debugprint_fn)reduceppdMakeModelSplitPPDsHPzhdeskjet|dj[ 0-9]?|laserjet|lj|color laserjet|color lj|designjet|officejet|oj|photosmart|ps |psc|edgelineZEpsonzstylus|aculaserZApplez.stylewriter|imagewriter|deskwriter|laserwriterZCanonz%pixus|pixma|selphy|imagerunner|bj|lbpZBrotherz hl|dcp|mfcZXeroxz/docuprint|docupage|phaser|workcentre|homecentreLexmarkzoptra|(:color )?jetprinterzKONICA MINOLTAzmagicolor|pageworks|pageproKyocerazfs-|km-|taskalfaZRicohZaficioZOceZ varioprintZOkizokipage|microlineZDeskJetZLaserJetZ OfficeJetzColor LaserJet PhotoSmart)ZdjZljZojzcolor ljzps zhp Z turboprintz! v(?:er\.)?\d(?:\d*\.\d+)?(?: |$)zn,| hpijs| foomatic/| - | w/| \(| postscript| ps| pdf| pxl| zjs| zxs| pcl3| printer|_bt| pcl| ufr ii| br-scriptz series| all-in-onec Cs<|jd}d}|j}x$tD]\}}|j|r|}|}PqW|dkoNtj|r|jd}|d'kr|jd}||kr||d|}n |d|}y|jdd\}}Wn|}d}YnXt j dd |}t j d d |}t j dd |}t j d d |}t j d d |}t j d d|}d}n|j dr6d}|dd}n|j drTd}|dd}n|j drrd}|dd}nj|j drd}|dd}nL|j drd}|dd}n.d}y|jd d\}}Wn|}d}YnX|j} |rB| j dr | j dr d }d!} n6| j d"r.| j d#r.d}d$} nt j| }|rB|}|j} | jd%} | d(krtj| } | r| j} | d| } |d| }tj| }|r|j}| d|} |d|}tjd|dd&\}}|r|j} | d!kr,x@tjD]4\}}| j |r||t|d}|j} PqW|j}||fS))z Split a ppd-make-and-model string into a canonical make and model pair. @type ppd_make_and_model: string @param ppd_make_and_model: IPP ppd-make-and-model attribute @return: a string pair representing the make and the model NFz TurboPrintr _rz(?<=[a-z])(?=[0-9]) z(?<=[a-z])(?=[A-Z])z JetZJetz Photo Smartr Tzkonica minolta zKONICA MINOLTAzlexmark international r z kyocera mita r zkyocera z fuji xerox z Fuji Xerox ZhewlettZpackardr hpZkonicaZminoltazkonica minoltaz v)countr)striplower _MFR_BY_RANGEmatch_RE_turboprintsearchfindrfindsplitresub startswithendswith_MFR_NAMES_BY_LOWERget_RE_version_numbersstart_RE_ignore_suffix_RE_ignore_seriessubn_HP_MODEL_BY_NAMEitemslen)ppd_make_and_modelmakeZ cleanup_makelmfrregexpmodeltZt2ZmakelZmodellvZvmatchZvstartsuffixZ suffixstartnnamefullnamer=2/usr/lib/python3/dist-packages/cupshelpers/ppds.pyrws                            c Cs|jj}d}d}d}d}|}d}xtt|D]p}||jr\||krV|rV|d7}|}n*||jr||kr||r||d7}|}n|}||jr2|||7}d}q2W|S)NrrrFrT)rrranger0isalphaisdigitisalnum) ZstrinZlstrinZ normalizedZBLANKZALPHAZDIGITZlastcharZ alnumfoundir=r=r> normalizes*       rEcCst|tr|dS|S)z{If we don't know whether getPPDs() or getPPDs2() was used, this function can unwrap an item from a list in either case.r) isinstancelist)xr=r=r> _singleton)s rIc @seZdZdZdZdZdZdZej j Z ej j Z ej j Z ej j Z ej jZe ee ee ee eeeiZd"ddZd d Zd d Zd dZddZddZd#ddZd$ddZd%ddZddZd&ddZddZd d!ZdS)'r a\ This class is for handling the list of PPDs returned by CUPS. It indexes by PPD name and device ID, filters by natural language so that foreign-language PPDs are not included, and sorts by driver type. If an exactly-matching PPD is not available, it can substitute with a PPD for a similar model or for a generic driver. rrr?NcCs|j|_d|_d|_tj|_tj|_|dkrbt j j d}|dkrbddl m }t jj|jd}y8t jj|d}tj|\}}|jj||jj|Wn>tk r}z"td||fd|_d|_WYdd}~XnX|dks|dks|d krd }|jd } | dkr|d| } n|} g} xx|jjD]j\} } yt| d }Wntk r^w,YnX|d krnq,||kr|q,|| krq,| j| q,Wx| D]} |j| =qWd|jkrt|jdd}|jdsd||jdd<dS)a% @type ppds: dict @param ppds: dict of PPDs as returned by cups.Connection.getPPDs() or cups.Connection.getPPDs2() @type language: string @param language: language name, as given by the first element of the pair returned by locale.getlocale() NZCUPSHELPERS_XMLDIRr)config cupshelperszpreferreddrivers.xmlzError loading %s: %sCZPOSIXZen_USrzppd-natural-languageZenrawzppd-make-and-modelzGeneric r)copyppdsmakesidsrZ DriverTypes drivertypesZPreferenceOrder preforderosenvironr(rrKpathjoinZ sysconfdirZPreferredDriversload Exceptionprintr r/rIKeyErrorappendr%)selfrPZlanguageZxml_dirrKZxmlfilerSZpreferenceordereuZshort_languageZ to_removeppdnameppddictZnatural_languageZ makemodelr=r=r>__init__Ks\               z PPDs.__init__c CsX|jt|jj}|jtjdy|jd|jddWnt k rRYnX|S)zb @returns: a list of strings representing makes, sorted according to the current locale )keyGenericr) _init_makesrGrQkeyssortlocalestrxfrmremoveinsert ValueError)r^Z makes_listr=r=r>getMakess z PPDs.getMakesc CsR|jyt|j|j}Wntk r2gSXdd}|jtj|d|S)zS @returns: a list of strings representing models, sorted using cups.modelSort() cSst|}t|}tj||S)N)rEcups modelSort)abfirstsecondr=r=r>compare_modelssz&PPDs.getModels..compare_models)rd)rfrGrQrgr\rh functools cmp_to_key)r^r2Z models_listrur=r=r> getModelsszPPDs.getModelsc Cs0|jy|j||Stk r*iSXdS)z Obtain a list of PPDs that are suitable for use with a particular printer model, given its make and model name. @returns: a dict, indexed by ppd-name, of dicts representing PPDs (as given by cups.Connection.getPPDs) N)rfrQr\)r^r2r6r=r=r>getInfoFromModels zPPDs.getInfoFromModelcCs |j|S)zM @returns: a dict representing a PPD, as given by cups.Connection.getPPDs )rP)r^rar=r=r>getInfoFromPPDNameszPPDs.getInfoFromPPDNamecCs|jj|tjjS)N)_fit_to_statusr(r DriverTypeFIT_NONE)r^fitr=r=r>getStatusFromFitszPPDs.getStatusFromFitcCsJ|dkr g}|dkrg}|dkr$i}|jr|jri}x|D]}|j|||<q:W|jj|j||}tdt||jj|||} tdt| dd| D}tdt|t} x&|D]} | jd\} } }| j |qW| rFg}x.|D]&}|jd\} } }|| kr|j |qW|rFx"|D]}||kr$|j |q$W|}|S)a` Sort a list of PPD names by preferred driver type. @param ppdnamelist: PPD names @type ppdnamelist: string list @param downloadedfiles: Filenames from packages downloaded @type downloadedfiles: string list @param make_and_model: device-make-and-model name @type make_and_model: string @param devid: Device ID dict @type devid: dict indexed by Device ID field name, of strings; except for CMD field which must be a string list @param fit: Driver fit string for each PPD name @type fit: dict of PPD name:fit @returns: string list Nz9Valid driver types for this printer in priority order: %sz5PPDs with assigned driver types in priority order: %scSsg|] }|dqS)rr=).0Ztyp_namer=r=r> sz2PPDs.orderPPDNamesByPreference..z(Resulting PPD list in priority order: %s/) rSrTrPZget_ordered_typesrreprZget_ordered_ppdnamesset rpartitionaddr])r^ ppdnamelistdownloadedfilesmake_and_modeldevidr~rPraZ orderedtypesZ orderedppdsZdownloadedfnamesZdownloadedfilerWZslashfnameZdownloadedppdnamesZppdfnamer=r=r>orderPPDNamesByPreferencesF       zPPDs.orderPPDNamesByPreferencerc.Cstd||f|}|}|j|dkr,g}i} |j} |j} d} y*x |j| | D]} |j| | <qVWd} Wntk rYnX| dkryXx |jd| D]} |j| | <qWtd|jd| td||ftd d} Wntk rYnXtd d}|jd}| d kr8t|\}}t|} t|} td | td | dddd}| |j krt|j | }n<| |kr|| }||j kr|}|} td| |j | }td||dk r|j |}|j t|}xL| ddgD]>}| j |dr|t |dd}t|} td| qW| |j | kr||| }x||jD]$} |j| | <td| | | fqRWnzt|d|\}}t|}td|||j | kr||}x8t||jD]$} |j| | <td| | | fqW| rL|rL|j||\}}||jkrLx*|D]"} || | <td| | | fq&W|rt|tkrj|jd}td|j|}|rx,|D]$}|j| |<td| ||fqW| s|j d rR|j d rR|j d sd|krRt}x(| jD]}|j dr|j|qWt |dkrLtd |x|D] } | | =q|j| | <t|jd'| q>Wt |t d(d)| jDkrtd*|x&|D] } | | =qWntd+|d,| svd-d.g}&d}'xt|&D]l}(td/|(|(d0})x<|jjD].}*|*j|(s |*j|)r|j| |*<d}'PqW|'r>Ptd1|(qW|'svtd2|j| t|jjd<| st j!d3d4t"|d5}+yt#d6d7|},Wnt$k rd },YnXd8||f}-|,r|-d9|,7}-|r|-d:|7}-td;|+t|-| S)=a9 Obtain a best-effort PPD match for an IEEE 1284 Device ID. @param mfg: MFG or MANUFACTURER field @type mfg: string @param mdl: MDL or MODEL field @type mdl: string @param description: DES or DESCRIPTION field, optional @type description: string @param commandsets: CMD or COMMANDSET field, optional @type commandsets: string @param uri: device URI, optional (only needed for debugging) @type uri: string @param make_and_model: device-make-and-model string @type make_and_model: string @returns: a dict of fit (string) indexed by PPD name z %s %sNFTzhewlett-packardrz&**** Incorrect IEEE 1284 Device ID: %sz **** Actual ID is MFG:%s;MDL:%s;z4**** Please report a bug against the HPLIP componentzTrying make/model namesrzmfgl: %szmdll: %sZlexmarkz kyocera mita)zhewlett-packardzlexmark internationalZkyocerazremapped mfgl: %szmake: %srrzunprefixed mdll: %sz%s: %szre-split mdll: %s,zChecking CMD fieldzipp:zipps:Zdnssdz._ippz driverless:rz$Removed %s due to non-IPP connectionz ppd-device-idCMD:zppd-type postscriptZ POSTSCRIPTz: %scSsg|]\}}|dkr|qS)genericr=)rdmr=r=r>rsz0PPDs.getPPDNamesFromDeviceID..zRemoved %s due to CMD mis-matchzNot removing %s z3due to CMD mis-match as it would leave nothing goodz textonly.ppdzpostscript.ppdz '%s' fallbackz.gzzFallback '%s' not availablez#No fallback available; choosing anyz //[^@]*@/?z//)patternreplstringcSs |d|S)Nrr=)rHyr=r=r>sz.PPDs.getPPDNamesFromDeviceID..zMFG:%s;MDL:%s;zCMD:%s;zDES:%s;zNo ID match for device %s:r)%r _init_idsrrR FIT_EXACTr\rfrrElmakesrQlmodelsr%r0rgrG_findBestMatchPPDsr}typer"_getPPDNameFromCommandSet FIT_GENERICrrrPrIr(rr FIT_EXACT_CMDr/r&r#r$strr TypeError).r^mfgmdl description commandsetsurirZorig_mfgZorig_mdlr~ZmfglmdllZ id_matchedZeachmdlsr2ZmfgreplZrmfgZmdlslprefixr6Zmfg2Zmdl2Zmdl2lsrPrZdriverZfailedraZ exact_cmdZ ppd_cmd_fieldZppdZ ppd_device_idZppd_device_id_dictZppd_typeZusableZpdlZ fallbacksfoundZfallbackZ fallbackgzZppdpathZ sanitised_uricmdidr=r=r>getPPDNamesFromDeviceIDsP                                                zPPDs.getPPDNamesFromDeviceIDc Cs|dkr g}|dkrg}|j||||||}||||d} |jt|j||| |} tdt| |j|| d} td| d| f| | dfS)a Obtain a best-effort PPD match for an IEEE 1284 Device ID. The status is one of: - L{STATUS_SUCCESS}: the match was successful, and an exact match was found - L{STATUS_MODEL_MISMATCH}: a similar match was found, but the model name does not exactly match - L{STATUS_GENERIC_DRIVER}: no match was found, but a generic driver is available that can drive this device according to its command set list - L{STATUS_NO_DRIVER}: no match was found at all, and the returned PPD name is a last resort @param mfg: MFG or MANUFACTURER field @type mfg: string @param mdl: MDL or MODEL field @type mdl: string @param description: DES or DESCRIPTION field, optional @type description: string @param commandsets: CMD or COMMANDSET field, optional @type commandsets: string @param uri: device URI, optional (only needed for debugging) @type uri: string @param downloadedfiles: filenames from downloaded packages @type downloadedfiles: string list @param make_and_model: device-make-and-model string @type make_and_model: string @returns: an integer,string pair of (status,ppd-name) N)MFGMDLZDESrzFound PPDs: %srzUsing %s (status: %d))rrrGrgrrr) r^rrrrrrrr~rrZstatusr=r=r>getPPDNameFromDeviceID#s$& zPPDs.getPPDNameFromDeviceIDcstd|j}|jdr2|dd}|dd}d}d}tj}dd|D}|j||f|jtjdd d |j ||f}||d g} |d t |kr| j||d t| ddd |d | d dnt| ddd |xX| D]P\} } t j j | |g} t | |krt| j}t | }td | |fqW|r|t |dkr|} |t |kr|j}n|j}n |j}d} |jtjtjd fdd|D}d}x^|jdD]P}|dkr|}d}x*tt |D]}||jrd}PqW|r|}PqWd}d}d}xLtt |D]<}||jrj|dkr\|}|}|d 7}n |dkrrusz+PPDs._findBestMatchPPDs..cSstj|d|dS)Nr)rorp)rHrr=r=r>rwsz)PPDs._findBestMatchPPDs..)rdrz <= z%s: match length %dr?csg|]}|j|fqSr=)r)rrH)rr=r>rsrFTz%dz'Searching for model ID '%s', '%s' %% %d z#Ignoring %d of %d digits, trying %siirrrr)rrr&rGrgr]rhrvrwindexr0rUrW commonprefixr FIT_CLOSEr}rorpr"r@rBintpow)r^rrrZbest_mdlZ best_matchlenZmdlnamesZ mdlnameslrD candidates candidateZ candidatelrrr~ZmdlitemsZmodelidZwordZ have_digitsZdigitsZ digits_startZ digits_endZ modelnumberZ modelpatternZ ignore_digitsrZdivr;rPr=)rr>rds              zPPDs._findBestMatchPPDsc s|dkr g}yjjdWntk r6dSXfdd}dd|D}d|ksld|ksld |krt|d Sd |ksd |ksd |ksd|kr|ddSd|kr|ddSd|kr|ddSd|ksd|kr|ddSd|kr|ddSd|ks d|ks d|kr|dSdS)zfReturn ppd-name list or None, given a list of strings representing the command sets supported.Nrecs0x*|D]"}j|\}}|jkr|SqWdS)N)rr)rr6rrP)modelsr^r=r>r(s   z+PPDs._getPPDNameFromCommandSet..getcSsg|] }|jqSr=)r)rrHr=r=r>rsz2PPDs._getPPDNameFromCommandSet..rZ postscript2zpostscript level 2 emulationZ PostScriptZpclxlzpcl-xlZpcl6zpcl 6 emulationz PCL 6/PCL XLz PCL LaserZpcl5ezPCL 5eZpcl5czPCL 5cZpcl5zpcl 5 emulationzPCL 5ZpclzPCL 3Zescpl2zesc/p2Zescp2ezESC/P Dot Matrix)rfrQr\)r^rr(Zcmdsetsr=)rr^r>rs6      zPPDs._getPPDNameFromCommandSetcCs|jr dStj}i}i}i}i}x |jjD]\}}t|d}t|} t| g} |jdg} t| t st| g} tdd| D} | r.t | dkrt} t|jddj } | r| d7} t | } xf| D]^}|j d r|jd r|dt |d}|sqt |}|j | s| |}| jt|qWx| D]\} }t | } t |}| |krp| || <i|| <i|| <n|| } ||| kr||| |<i|| |<n || |}||| ||<q4W| | kr| j| | r0| \} }| |kr|| j|t}ni|| <t}|jd d| D}||| |<q0Wx|jD]\} }t | } || }xv|jD]j\}}|| jt |}|sq^|||}x6|D].}|| jt |}|||}|j|qWq^Wqrsz$PPDs._init_makes..rzppd-makerrr)cSsg|] }|dqS)rr=)rrHr=r=r>rYszinit_makes: %.3fs)rQtimerPr/rIrrr(rFrGr0rstriprEr%r&rrkunionupdaterrr)r^ZtstartrQrraliasesrarbr1Z ppd_mm_splitZppd_makes_and_modelsZ ppd_productsr2ZlmakeZ ppd_productZlprodr6ZlmodelrZ main_makeZ modelnamesZ main_modelZ main_ppdsZ eachmodelZ this_modelrPr=r=r>rfs                  zPPDs._init_makesc Cs|jr dSi}x|jjD]\}}t|jd}|s6qt|}|dj}|dj}d}t|dkrjd}t|dkrzd}|rq||kri||<|||krg|||<|||j|qW||_dS)Nz ppd-device-idrrFrT) rRrPr/rIr(rrr0r]) r^rRrarbrZid_dictZlmfgZlmdlZbadr=r=r>rrs.      zPPDs._init_ids)NN)NNNNN)rNNN)rNNNN)N)__name__ __module__ __qualname____doc__ZSTATUS_SUCCESSZSTATUS_MODEL_MISMATCHZSTATUS_GENERIC_DRIVERZSTATUS_NO_DRIVERrr|rrrrr}r{rcrnrxryrzrrrrrrrfrr=r=r=r>r 0sL K C   >y (mcCs tddS)NzBusage: ppds.py [--deviceid] [--list-models] [--list-ids] [--debug])r[r=r=r=r> _show_helpsr)"rorLrrr itertoolsrrrios.pathrUrvr#rrr__all__compilerr'r4r5rr.rr)r+Ir,rrErIr rr=r=r=r>sd                .h