o j?àhDã@s\dZddlZddlmZmZddlZddlZddlZdZGdd„de ƒZ Gdd„de ƒZ dS) z* Detection tool for nVidia graphics cards éN)ÚPopenÚPIPEz)/usr/share/ubuntu-drivers-common/obsoletec@seZdZdZdS)ÚNoDatadirErrorz4Exception thrown when no modaliases dir can be foundN)Ú__name__Ú __module__Ú __qualname__Ú__doc__©r r ú?/usr/lib/python3/dist-packages/NvidiaDetector/nvidiadetector.pyr"src@sxeZdZdZddefdd„Zdd„Zdd„Zd d „Zd d „Z d d„Z dd„Z dd„Z dd„Z dd„Zdd„Zdd„ZdS)ÚNvidiaDetectiona A simple class to: * Detect the available graphics cards * See what drivers support them (If they are NVIDIA cards). If more than one card is available, try to find the highest common driver version which supports them all. (READ the comments in the code for further details) * Return the recommended driver version NcCsbddi|_||_||_| |¡|_| ¡| ¡| ¡| ¡|r)|  ¡dSt |  ¡ƒdS)a… printonly = if set to None will make an instance of this class return the selected driver. If set to True it won't return anything. It will simply and print the choice. verbose = if set to True will make the methods print what is happening. ÚcurrentièN) Ú _NvidiaDetection__driver_aliasesÚ printonlyÚverboseÚgetObsoletePackagesÚ oldPackagesÚ detectionÚgetDataÚgetCardsÚremoveUnsupportedÚprintSelectionÚprintÚ selectDriver)ÚselfrrÚobsoleter r r Ú__init__3s   zNvidiaDetection.__init__cCs(|j ¡D] \}}||kr|SqdS)z7Get the name of a driver from its corresponding integerN)r Úitems)rÚvalueÚkÚvr r r Ú__get_name_from_valueQs ÿz%NvidiaDetection.__get_name_from_valuecCs:|j |¡}|dur| d¡rt| dd¡ƒ}t|ƒ}|S)z2Get the integer associated to the name of a driverNz-serverÚ)r ÚgetÚendswithÚintÚreplace)rÚnamerr r r Ú__get_value_from_nameXs  z%NvidiaDetection.__get_value_from_namecCsDg}zdd„t|ƒ ¡Dƒ}dd„|Dƒ}W|Sty!Y|Sw)z.Read the list of obsolete packages from a filecSsg|]}| ¡‘qSr )Ústrip©Ú.0Úxr r r Ú esz7NvidiaDetection.getObsoletePackages..cSsg|]}|dkr|‘qS)r!r r)r r r r,fs)ÚopenÚ readlinesÚIOError)rrÚtempListr r r ras þþz#NvidiaDetection.getObsoletePackagesc Csâg|_tddgtdd}| ¡d d¡}t d¡}t d¡}|D]M}| |¡}| |¡}|rN| d ¡  ¡  ¡}| d ¡  ¡  ¡} |d | } |j  | ¡q!|rn| d ¡  ¡  ¡}| d ¡  ¡  ¡} |d | } |j  | ¡q!d S) z^ Detect the models of the graphics cards and store them in self.cards Úlspciz-nT©ÚstdoutÚuniversal_newlinesrÚ z6.*03(80|0[0-2]): *(?P.+):(?P.+) \(.+\)z/.*03(80|0[0-2]): *(?P.+):(?P.+)ÚvendorÚdeviceú:N) ÚcardsrrÚ communicateÚsplitÚreÚcompileÚmatchÚgroupr(ÚlowerÚappend) rÚp1ÚpÚ indentifier1Ú indentifier2ÚlineÚm1Úm2Úid1Úid2Úidr r r rks(      €ôzNvidiaDetection.detectionc Csœi|_t d¡}t d¡}t ¡D]¦}|j d¡r(d|jvs(d|jvs(d|jvr)qz|jjd}Wn t t fy<Yqw|  |j¡}|rK|  d¡}n t  d |j¡q| |¡}zK|r£|| d ¡dd … d d ¡}| d¡D]2}|  | ¡¡} | s†t  d|j|f¡qp|   d¡ ¡} |   d¡ ¡} |j |g¡ | d| ¡qpWqty·t  d|j|f¡Yqwt|j ¡ƒdkrÌtj ¡tdƒd Sd S)zi Get the data from the modaliases for each driver and store them in self.drivers zpci:v0000(.+)d0000(.+)svz5nvidia(?:\-driver|)\-([0-9]+)(:?\-server|)(:?\:i386|)únvidia-ÚupdatesÚ experimentalr Ú Modaliaseséz/%s package has unexpected name scheme. Skippingú(Nú)r!ú,z&Package %s has unexpected modalias: %sér8z*Package %s has invalid modalias header: %srÚnone)Údriversr<r=ÚaptÚCacher&Ú startswithÚ candidateÚrecordÚKeyErrorÚAttributeErrorr>r?ÚloggingÚerrorÚ%_NvidiaDetection__get_value_from_nameÚfindr%r;Úlstripr@Ú setdefaultrAÚ ValueErrorÚlenÚkeysÚsysr3Úflushr) rÚvendor_product_reÚ package_reÚpackageÚmÚ package_matchÚstripped_package_nameÚdriver_versionÚaliasÚvpr6Úproductr r r rŒs^       ü  ÿ ÿ ÿ€  ÿÿ  þzNvidiaDetection.getDatacCsØi|_g|_ |jD]}|d| d¡…dkr(|jr"td|dƒ|j |¡q t|jdd|_ |jD]4}d}|j D] }||j|vr\d}|jrRtd ||fƒ|j  |g¡ |¡q<|si|j  |g¡ d ¡q5d S) zˆ See if the detected graphics cards are NVIDIA cards. If they are NVIDIA cards, append them to self.nvidiaCards rr8Ú10dezNVIDIA card found (rRT©ÚreverseFzCard %s supported by driver %sN) ÚdriversForCardsÚ nvidiaCardsr9rarrrAÚsortedrVÚ orderedListrc)rÚcardÚ supportedÚdriverr r r rÈs0  €  €€÷zNvidiaDetection.getCardscCs\g}|jD]}d|j|vr| |¡q|D]}|jr!td|ƒ|j |¡|j|=qdS)zf Remove unsupported cards from self.nvidiaCards and from self.driversForCards NzRemoving unsupported card )rvrArrrwÚremove)rÚunsupportedCardsrzÚ unsupportedr r r rös  €   üz!NvidiaDetection.removeUnsupportedc Cs„t|jƒ}|dkr¶|dkru i}|jD]}|j|D]}| |d¡||d7<qqt|dd} g}|D] }|||krD| |¡q7t|ƒdkrc |jdd|d}|jrb|jsbt d|ƒn- |d}|jrt|jstt d|ƒn |jt |j  ¡ƒdd}|jr|jst d|ƒ |  |¡} | dur¨|dkr d p¡d t | ƒ}|S|dkr®d p¯d t |ƒ}|S |jr¾t d ƒd}|S) zZ If more than one card is available, try to get the highest common driver rrPTrtzRecommended NVIDIA driver: zRecommended NVIDIA driver: %dNi†znvidia-driver-rLzNo NVIDIA package to install)rerwrvrcrxrAÚsortrrrÚlistrfÚ%_NvidiaDetection__get_name_from_valueÚstr) rÚ cardsNumberÚ occurrencerzÚdrvÚ occurrencesÚ candidatesr|ÚchoiceÚ driver_namer r r rsZ   þ   €    €  €    ÷ ùzNvidiaDetection.selectDriverc CsÚ g}g}tddgtdd}| ¡d}| d¡}|D]}| d¡dkr.| | d ¡d¡qt|tƒrPz | |¡r> W|SW|St yO| |¡Y|Sw|D]}z| |¡r[ WqRt yj| |¡YqRw|S) zð USAGE: * pkglist is the list of packages you want to check * use lists for one or more packages * use a string if it is only one package * lists will work well in both cases Údpkgz--get-selectionsTr2rr5z installéÿÿÿÿú ) rrr:r;rarAÚ isinstancerƒÚindexrd) rÚpkglistÚlinesÚ notinstalledrBrCÚcrFÚpkgr r r Úcheckpkgds<  €   ô ö  ö € ÿzNvidiaDetection.checkpkgcCs| |j¡}t|ƒt|jƒkS)z° oldPackages = a list of the names of the obsolete drivers notInstalled = a list of the obsolete drivers which are not installed )r•rre)rÚ notInstalledr r r Ú getDriversˆs zNvidiaDetection.getDriverscCsV| ¡}| ¡r |rtj ¡t|ƒdStj ¡tdƒdStj ¡t|ƒdS)z6 Part for the kernel postinst.d/ hook rUN)rr—rgr3rhr)rr|r r r r˜s      zNvidiaDetection.printSelection)rrrrÚobsoletePackagesPathrr‚r`rrrrrrr•r—rr r r r r &s   !<.^$ r ) rr<Ú subprocessrrrgr^rWr˜Ú ExceptionrÚobjectr r r r r Ús