7jr dZddlZddlZddlmZddlZddlZddlm Z ddl m Z ddl m Z ddlmZdZd Zd Zd Zej(d d dej*j,dej(dddddej(dej*j/dddej(ddej0j2ej0j4dddej(dddd d!"ej(d#d$dd%&ej6d'd(Zy))z0Generate a keymap.c from a configurator export. N)cli) info_json)KeymapJSONEncoder) dump_lines) generate_jsoncdt|dj}t|dk(r|dSdS)z1Assume layout macro when only 1 is available layoutsrN)listkeyslen) keyboard_datar s I/home/hanthebot/chat/keyboard/qmk_firmware/lib/python/qmk/cli/via2json.py_find_via_layout_macrors6=+0023GW*71:44c ttd|}t|dk(r tStjd}tjd}t}|D]"}|j |}ttd|}t}|D]}d|vsd|vr|j |}|D]} | jd} d } g} | dd k(rd } | j| d dnF| dd k(rd} | j| d dn'| jd| vr| j dn| gttd| } |j| | d|j||j|%|S)Nct|S)N)boolfs rz!_convert_macros..s tAwrrz (}\,)|(\,{)z({.+?})c|dvrdSdS)N)Nz},z,{FTrs rrz!_convert_macros.."sq4F/FeDr{}z {}tap+downr -up,c&|jddS)NKC_)replace)ss rrz!_convert_macros..9s %8Lr)actionkeycodes) r filterr recompilesplitfindallstripappendextendmap) via_macros split_regexmacro_group_regexmacros via_macromacro macro_datam macro_groups macro_group macro_actionmacro_keycodess r_convert_macrosr>sf. ;$Q3.'+ &--k!"o>&--{HZk.?.?.Dal`mn&*#.Ln*]%^N%%>&Z['\,!!!$5 %6  j!C!"F Mrc tjd}tdt|D]U}tdt||D]8}|||}|j |}|sd|j d|||<:W|S)NzMACRO0?\(([0-9]{1,2})\)rMC_r )r*r+ranger matchgroup) keymap_datamacro_noijkcr9s r_fix_macro_keysrIDszz45H 1c+& '7q#k!n-. 7AQ"Br"A&)!''!*$6 Aq!  77 rcf|dj|}|s?tjjd|dtjj dy|d}t }t|D]\}}|j||dg|jdd }td |d d D]t}td |d d D]]} |t|k\s||d d |k7s||d d | k7r|j|d|| ggn|jd|| gg|d z }_vt } |dD]n} d }t } | D](} ||d | j||d | g|d z }*| j| Dcgc]}|d  } }| j| p| Scc}w)Nr z LAYOUT macro z! is not a valid one for keyboard !layoutmatrixc"|dd|ddfS)Nr rr)ks rrz _via_to_keymap..[sAaDGQqT!W#5r)keyr matrix_sizerowscolsr layers) getrlogerrorargskeyboardr enumerater/sortrAr insert) via_backupr keymap_layout layout_data sorting_hatindexdataposrow_numcol_numrDlayer layer_datarPrHs r_via_to_keymaprhOs  *..}=K   m_4UVYV^V^VgVgUhhijkh'K&K -4 tE4>23456 CM-8@AQ m 1HC  &KH% 'V  C3".!!;s#3A#6"<= 1HC  &01be1 1:& ' 2s F.z-oz--outputTzFile to write to)arg_onlytypehelpz-qz--quiet store_truez&Quiet mode, only output error messages)rir'rkfilenamerzVIA Backup JSON file)rjrirkz-kbz --keyboardzThe keyboard's name)rj completerrirequiredrkz-kmz--keymapvia2jsonzThe keymap's name)ridefaultrkz-lz--layoutzThe keymap's layout)rirkz0Convert a VIA backup json to keymap.json format.c|jjjd5}tj|}dddt |jj }|jjr|jjn t|}|s4|jjd|jj dyt||}|s4|jjd|jj dyt}|jdrt|d}t|}t!|jj"|jj |||}tj$|t&d g}t)|jj*||jj,y#1swYxYw) zConvert a VIA backup json to keymap.json format. This command uses the `qmk.keymap` module to generate a keymap.json from a VIA backup json. The generated keymap is written to stdout, or to a file if -o is provided. rnNz(Couldn't find LAYOUT macro for keyboard z+. Please specify it with the '-l' argument.FzGCould not extract valid keycode data from VIA backup matching keyboard rKr5T)cls sort_keys)rXrmopenjsonloadrrYrLrrVrWrhr rUr>rIrkeymapdumpsrroutputquiet) rfdr]rr^rDr8 keymap_json keymap_liness rrqrqusp      $#YYr] #chh//0M(+xxCHHOOrIrhargumentpathnormpathFileTyperYkeyboard_folderkeyboard_completer subcommandrqrrrrsG /#$5*Z#LdJCHH4E4EL^_ dI\Hpq jsxx005Kab e\ (D(DPSP\P\PoPoz~JNUkl eZ$ I]^ dJ4JKBC#>DL_lcr`#>r