7jR/ dZddlmZddlZddlZddlZddlZddlZddlZddl m Z m Z m Z m Z mZmZddlmZmZddlmZddlmZddlmZdd lmZmZdd lmZmZdd lm Z m!Z!eGd d Z"GddZ#Gdde#Z$Gdde#Z%Gdde#Z&Gdde#Z'de(de(de(de e#fdZ)de(fdZ*dZ+ejXdZ-de e"fd Z.d!Z/d"e"de"fd#Z0d$e ee(ee(e e(e(fffde e"fd%Z1d1d"e"d&e e(de e"fd'Z2d$e e"de e"fd(Z3d)e"fd*Z4gfd+e e"d,e e(de e"fd-Z5d.ggfd$e eee(e(fee(e(e e(e(fffd,e e(de e!fd/Z6gfd$e ee(ee(e e(e(fffd,e e(de e!fd0Z7y)2z;Functions for searching through QMK keyboards and keymaps. ) dataclassN)CallableDictListOptionalTupleUnion)dottyDotty)cli) parallel_map) keymap_json)list_keyboardskeyboard_folder) list_keymaps locate_keymap)KeyboardKeymapBuildTarget BuildTargetc~eZdZUeed<eed<dZeed<dZeed<defdZ de fdZ d Z e defd Zdefd Zy) KeyboardKeymapDesckeyboardkeymapNdata extra_argsreturnc|jj|jjz tj|j djz SNT) sort_keys)r__hash__rjsondumpsrselfs C/home/hanthebot/chat/keyboard/qmk_firmware/lib/python/qmk/search.pyrzKeyboardKeymapDesc.__hash__sE}}%%'$++*>*>*@@4::dooimCnCwCwCyyyc|j|jtj|jdf|j|jtj|jdfkSr)rrr r!r)r#others r$__lt__zKeyboardKeymapDesc.__lt__s} t{{DJJtRV,WX\a\j\jlqlxlxz~{E{EFKFVFVbf{g\hh hr%ct|j|j}t|tr|j |_y||_yN)rrr isinstancer to_dictr)r#rs r$ load_datazKeyboardKeymapDesc.load_data"s24==$++6&0u&=DLLN 4 r%cH|jt|jSdSr*)rr r"s r$r zKeyboardKeymapDesc.dotty&s#'99#8uTYYBdBr%ct|j|j|j}|j|_|S)N)rrr )rrrrr)r#targets r$to_build_targetz"KeyboardKeymapDesc.to_build_target*s0*DMM$++\`\e\ef OO r%)__name__ __module__ __qualname__str__annotations__rdictrintrboolr(r-propertyr r rr1r%r$rrsmM KD$Jz#zhthHCuCC!:r%rcReZdZUdZeed<eeed<eed<eege fed<dZ y)FilterFunctiona\Base class for filters. It provides: - __init__: capture key and value Each subclass should provide: - func_name: how it will be specified on CLI >>> qmk find -f ... - apply: function that actually applies the filter ie: return whether the input kb/km satisfies the condition keyvalue func_nameapplyc ||_||_yr*)r>r?)r#r>r?s r$__init__zFilterFunction.__init__Ds r%N) r2r3r4__doc__r5r6rrrr9rCr;r%r$r=r=2s6  H C=N '($. //r%r=c eZdZdZdedefdZy)Existsexists target_inforc2|j|jvSr*r>r r#rHs r$rAz Exists.applyLsxx;,,,,r%Nr2r3r4r@rr9rAr;r%r$rFrFIsI-!3--r%rFc eZdZdZdedefdZy)AbsentabsentrHrc2|j|jvSr*rJrKs r$rAz Absent.applySsxx{0000r%NrLr;r%r$rNrNPsI1!311r%rNc eZdZdZdedefdZy)LengthlengthrHrc|j}|j|vxr.t||jt|jk(Sr*)r r>lenr8r?r#rH info_dottys r$rAz Length.applyZs< && J&W3z$((/C+DDJJ+WXr%NrLr;r%r$rRrRWsIY!3YYr%rRc eZdZdZdedefdZy)ContainscontainsrHrcp|j}|j|vxr|j||jvSr*)r r>r?rVs r$rAzContains.applybs3 && J&M4::DHH9M+MNr%NrLr;r%r$rYrY_sIO!3OOr%rYr@r>r?rcjtjD]}||jk(s|||cSy)zInitialize a filter subclass based on regex findings and return it. None if no there's no filter with the name queried. N)r=__subclasses__r@)r@r>r?subclasss r$_get_filter_classr_gs< #113( ** *C' '( r%ctjDcgc]}d|jd}}dj|ddd|dzScc}w)N'z, z and )r=r]r@join)fnamess r$ filter_helprfssV)7)F)F)H IAq Q  IE I 99U3BZ U59+#6 66 JsA c\tj tj}|t_tjj |tjj |tj|S#t$rtjj }YywxYwr*) r acquire_lock log_levelAttributeErrorloglevelsetLevelloggingroot release_lock)rlolds r$_set_log_levelrrxsxmm GGU LL%  J ggmmsB#B+*B+c#\Kttj}dt|ywr*)rrrnCRITICAL)rqs r$ignore_loggingrus! )) *C 3s*,ct5t|}t|Dcgc]}t||c}cdddScc}w#1swYyxYw)zQReturns a list of KeyboardKeymapDesc for all keymaps for the given keyboard. N)rurrrrrs r$ _all_keymapsrxsR  ["8,COPXCYZ"8V4Z[[Z[[sAAAAAcbt5t|||ndcdddS#1swYyxYw)zYReturns the keyboard name if the keyboard+keymap combination exists, otherwise None. N)rurrws r$_keymap_existsrzs4  Q(6:FxDQQQs%.r0cft5|j|cdddS#1swYyxYw)z6Ensures a KeyboardKeymapDesc has its data loaded. N)rur-)r0s r$_load_keymap_infor|s.  s'0targetscRg}|D]}d}t|tr|djd}|d}n|jd}t|dk7r&tj j d|gcS|jt|d|d|t|S)zExpand a list of make targets into a list of KeyboardKeymapDesc. Caters for 'all' in either keyboard or keymap, or both. Nr:zInvalid build target: r) r+tuplesplitrUr rkerrorappendrexpand_keymap_targets)r} split_targetsr0r split_targets r$expand_make_targetsrs M j fe $!!9??3/LJ!<<,L |  ! GGMM26(; <I/ Qa]ghi j ! //r% all_keyboardsc.| t}|jdk(r|jdk(ratjj dg}t t|D]}|j||D]}|j|_ |Stjj d|jdtjt|j}tdt ||Dcgc]$}t||j|j&c}S|jdk(r\tjj d|jdt|j}|D]}|j|_ |S|gScc}w) zExpand a keyboard input and keymap input into a list of KeyboardKeymapDesc. Caters for 'all' in either keyboard or keymap, or both. allz/Retrieving list of all keyboards and keymaps...z*Retrieving list of keyboards with keymap "z"...)rc |duSr*r;es r$z'_expand_keymap_target..sst}AtAr%rz)Retrieving list of keymaps for keyboard ")rrrr rkinfor rxextendr functoolspartialrzfilterr)r0rr}kbtkeyboard_filters r$_expand_keymap_targetrsy &(  % ==E ! GGLLJ KG"<? #r" # 1%00  1N GGLLEfmm_TXY Z'//v}}UObhjACOP_anCocpq\^&r6==VEVEVWq q ==E ! GGLLDV__DUUYZ ["6??3G 1%00  1N8Oqs8)Fcg}t}|D]}|jt||tt t |S)zhExpand a list of KeyboardKeymapDesc inclusive of 'all', into a list of explicit KeyboardKeymapDesc. )rrrlistsortedset)r}overall_targetsrr0s r$rrsMO"$MM4V]KLM s?+, --r%rc"|jSr*)r1rs r$_construct_build_targetrs   r% target_listfiltersc t|dk(r#tjjd|}|Stjjdt t |}t jd}t jd}|D]}|j|}|j|}||jdj} |jd} |jd} t| | | } | 2tjjd |jdt| j|}| d | d nd } tjjd | d| d | d|u|jd} |jd}|jd} tjjd| d|d| dd}t|| || |}ktjjd |tjjdtt!t#|}|S)zFilter a list of KeyboardKeymapDesc based on the supplied filters. Optionally includes the values of the queried info.json keys. rzPreparing target list...z=Parsing data for all matching keyboard/keymap combinations...zL^(?P[a-zA-Z]+)\((?P[a-zA-Z0-9_\.]+)(,\s*(?P[^#]+))?\)$zH^(?P[a-zA-Z0-9_\.]+)\s*(?P[\<\>\!=]=|\<|\>)\s*(?P[^#]+)$functionr>r?z Unrecognized filter expression: z , {fg_cyan}z {fg_reset}z"Filtering on condition: {fg_green}z{fg_reset}({fg_cyan}z)...opz!Filtering on condition: {fg_cyan}z {fg_reset} z {fg_cyan}z {fg_reset}...ctj}tjd|dtjdt ffd }|S)N^$rcL|jj}}dvrCt|dn|}t|}dk(r||kSdk(r||kDSdk(r||kSdk(r||k\Syt|dn|}dk(rj |duSdk(rj |duSy) N)<><=>=Frrrrz!=z==)r getr8r5search)rlhsrhskorulevs r$rdz7_filter_keymap_targets.._make_filter..fsggkk!n 66"%s{e"DC"%c(C Cx'*Sy 0!"c'*Sy 0!"d'*cz 1!"d'*cz 1"+#&s{e"DC Dy'+{{3'74'? ?!"d'+{{3'7t'C C"+r%)fnmatch translaterecompile IGNORECASEr)rrrexprrdrs``` @r$ _make_filterz,_filter_keymap_targets.._make_filter sK",,Q/D::$qk2==ADD/DD0Hr%)rUr rkrr r|rrmatchgrouplowerr_warningrrArrr)rrr} valid_targets function_re comparison_re filter_exprfunction_matchcomparison_matchr@r>r? filter_class value_strrrs r$_filter_keymap_targetsrsF  7|q  /0L NI  TU$%6 D jj!pq  #no "9 K(..{;N,22;? )*00<BBD $**51&,,W50CG 'GGOO&F~G[G[\]G^F_$`a &|'9'9= I CHCTmE7,?Z\  CI;NfgjfkkwyBxCCGHI!-&,,U3%++D1(..w7 B3%}UWTXXdejdkkz{|<!'|CU'C] S "B;- PQs9 v  /0vc-012 Nr%)rdefaultc d}t||}tt||}tt t t t|}t|S)=Search for build targets matching the supplied criteria. crt|dk(rt|d|d|dSt|d|dS)Nrrr)rrrrw)rUrrs r$ _make_descz)search_keymap_targets.._make_desc8s= q6Q;%qtAaDQqTR R%qtAaDA Ar%)maprrrrr rr)r}rrs r$search_keymap_targetsr5sMB *g&G$%:7%CWMG3|$;T']KLMG '?r%c tt||}ttt t t|}t |S)r)rrrrr rr)r}rs r$search_make_targetsrDs;%%8%A7KG3|$;T']KLMG '?r%r*)8rD dataclassesr contextlibrrr rnrtypingrrrrrr dotty_dictr r milcr qmk.utilr qmk.infor qmk.keyboardrr qmk.keymaprrqmk.build_targetsrrrr=rFrNrRrYr5r_rfrrcontextmanagerrurxrzr|rrrrrrrr;r%r$rs!  ??#! 82D  8.-^-1^1Y^YO~O  3 s x?W 7S7   [d#56[Q05G0eCsDcN7J1K,K&L!M0RVWiRj0("4T#YZ^_qZr@.4(:#;.EW@X.1XZM-?(@M4PS9M^bcu^vM`ewcwOQ 4eCHouS#tTWY\T\~E]?^.^(_#` CGHKCL VZ[fVg egeCsDcN7J1K,K&L!MX\]`Xakop{k|r%