7j'bdZddlZddlmZddlmZddlmZddlm Z m Z m Z m Z ddl mZmZmZmZdd lmZmZmZmZmZmZmZmZdd lmZdd lmZmZm Z d Z!d Z"dZ#dZ$dZ%ejLdddddejLdddddejNddZ(y)zVQMK Doctor Check out the user's QMK environment and make sure it's ready to compile. N)cli)yesno) submodules) QMK_FIRMWAREQMK_FIRMWARE_UPSTREAM QMK_USERSPACEHAS_QMK_USERSPACE) CheckStatuscheck_binariescheck_binary_versionscheck_submodules)git_check_repogit_get_branch git_get_taggit_get_last_log_entrygit_get_common_ancestor git_is_dirtygit_get_remotesgit_check_deviation) in_virtualenv)qmk_userspace_pathsqmk_userspace_validateUserspaceValidationErrorcpdfd}fd} ddlm}|dz jr7tjj d||dz j dDcgc] }|| }}t|dkDr0tjj d d j|ntjjd |dz j d Dcgc] }|| }}t|dkDr?tjj d d j|tjStjjdtjScc}wcc}w#t$r1tjj dYtjSwxYw)Nci}t|d5}|D]=}d|vs|jdd\}}|j||j<? ddd|S#1swY|SxYw)z7Load a simple key=value file into a dictionary r=r N)opensplitstrip)filevarsflinekeyvalues L/home/hanthebot/chat/keyboard/qmk_firmware/lib/python/qmk/cli/doctor/main.py_load_kvp_filez%distrib_tests.._load_kvp_filesr $_ 6 6$;!%C!3JC(- D% 6 6   6  s A8AA'c |}|jddd|jddd|jddS#t$r/}tjj d|d|cYd}~Sd}~wwxYw) z2Parse the QMK toolchain release info file TOOLCHAIN_HOSTunknown:TOOLCHAIN_TARGET COMMIT_HASHz1Error reading QMK toolchain release info file: %sz%Unknown toolchain release info file: Nget Exceptionrlogwarningr"r#er)s r(_parse_toolchain_release_filez4distrib_tests.._parse_toolchain_release_files B!$'Dhh/;e=ffghlhphpq~AJiKhLM M B GGOOOQR S:4&A A BsAA A=$A82A=8A=c |}|jddd|jddS#t$r/}tjj d|d|cYd}~Sd}~wwxYw)z3Parse the QMK flashutils release info file FLASHUTILS_HOSTr,r-r/z2Error reading QMK flashutils release info file: %sz&Unknown flashutils release info file: Nr0r5s r(_parse_flashutils_release_filez5distrib_tests.._parse_flashutils_release_file(sq C!$'Dhh0)<=Qtxx W`?a>bc c C GGOOPRS T;D6B B Cs.2 A*$A%A*%A*r)QMK_DISTRIB_DIRetcz3Found QMK tools distribution directory: {fg_cyan}%sztoolchain_release_*z!Found QMK toolchains: {fg_cyan}%sz, z!No QMK toolchains manifest found.zflashutils_release_*z!Found QMK flashutils: {fg_cyan}%sz!No QMK flashutils manifest found.z!QMK tools distribution not found.) qmk.clir;existsrr3infogloblenjoinr4 ImportErrorr OK)r7r:r;r" toolchains flashutilsr)s @r( distrib_testsrGsu BC:+ e # + + - GGLLNP_ `KZ]bKbJhJhi~JA$7=AJA:" @$))JBWX CDL[^cLcKiKikALBC48>CJC:" @$))JBWX >>  CD >>A C :  89 >>:s7AE; E1/A5E;$E63?E;E;1 E;;(F54F5ctjj}d|vsd|vr ddlm}|Sd|vr ddlm}|Sd|vr ddlm}|Stjjd |tjS) z5Determine our OS and run platform specific tests darwinmacosr ) os_test_macoslinux) os_test_linuxwindows)os_test_windowszUnsupported OS detected: %s) platformlowerrJrKrLrMrNrOrr3r4r WARNING) platform_idrKrMrOs r(os_testsrTHs{##%++-K;'["8( K ( k !,   5{C"""c tj}t}|s1tjj dtj }|St}|rtjjd|t}|r tjjd|t}|r/tjj dtj }t}d|jvst|djddvr1tjj dtj }|St|}|d vr2|r0tjj d |tj }|d d fD],}tjjd |t!|.dD]-}tjjd|t#|d/|S)zRun Git-related checks zL{fg_yellow}QMK home does not appear to be a Git repository! (no .git folder)zGit branch: %szRepo version: %sz1{fg_yellow}Git has unstashed/uncommitted changes.upstreamurlz[{fg_yellow}The official repository does not seem to be configured as git remote "upstream".)masterdevelopzS{fg_yellow}The local "%s" branch contains commits not found in the upstream branch.upstream/masterupstream/developz- Latest %s: %s)r\r]z- Common ancestor with %s: %sHEAD)r rDrrr3r4rRrr?rrrkeysrr1rrr)statusgit_ok git_branch repo_version git_dirty git_remotes git_deviationbranchs r( git_testsrh[s^^F F  fg$$8 M5$%  GGLL): 6&=L />$I ST$,,)+K!1!1!337LT_`jTkToTopuwyTz7z }~$,, M!4J ? !66=GGOO$y|FG(00F)+<>PQ\FGGLL!2F>>N)rr3r?rr`r_error) sub_statusssub_infosub_name sub_shorthash sub_describesub_last_log_timestamps r(output_submodule_statusrwsGGLL$%""$J __  Ka= X 'H5@H5LH[1RTM3=3I8J/rLG[_gGgX.B%Cmo "%+ r(2.D-ET,WYZgYhhijk 8*,HIJ KrUc|r"tjjd|tD]1} t |tjjd|d3t&tjjdttjjd ty#t $r&tjj d|dYt$rS}tjj d|dtjj d|d|Yd}~ d}~wwxYw) NzQMK home: {fg_cyan}z&Testing userspace candidate: {fg_cyan}z({fg_reset} -- {fg_green}Valid `qmk.json`z({fg_reset} -- {fg_red}Missing `qmk.json`z({fg_reset} -- {fg_red}Invalid `qmk.json`z -- {fg_cyan}z&/qmk.json{fg_reset} validation error: zQMK userspace: {fg_cyan}zUserspace enabled: {fg_cyan}) rr3r?rrFileNotFoundErrorr4rrr ) qmk_firmwarepatherrs r(userspace_testsr}s   ,\N;<#%c c "4 ( GGLLCD6Iuv wc   1-ABGGLL12C1DEF! { GGOOFtfLxy z' c GGOOFtfLxy z GGOOodV3[\_[`a b b cs.B66,D?$D?,AD::D?z-yz--yes store_trueTzAnswer yes to all questions.)actionarg_onlyhelpz-nz--nozAnswer no to all questions.zBasic QMK environment checksc|jjd|jjdtj|jjd|j|jjdt t x}}ttdt}|tjk(s&|tjk(r|tjk(r|}tr|jjdt}|tjk(r|jjdn?|tjk(r|jj!dntj}t#}tj|vrtj}n5tj|vr#|tjk(rtj}t%}|tjk(r|jjd nt'r+t)d d rt+j,t%}|tjk(rtj}n6|tjk(r#|tjk(rtj}t/|tjk(r|jjd y|tjk(r|jjdy|jjd|jjd|jjdy)zBasic QMK environment checks. This is currently very simple, it just checks that all the expected binaries are on your system. TODO(unclaimed): * [ ] Compile a trivial program with each compiler z(QMK Doctor is checking your environment.zPython version: %szCLI version: %szQMK home: {fg_cyan}%sNzCLI installed in virtualenv.zAll dependencies are installed.z/Issues encountered while checking dependencies.zSubmodules are up to date.z'Would you like to clone the submodules?T)defaultz{fg_green}QMK is ready to gorz<{fg_yellow}QMK is ready to go, but minor problems were foundr zW{fg_red}Major problems detected, please fix these problems before proceeding.{fg_reset}z{fg_blue}If you're missing dependencies, try following the instructions on: https://docs.qmk.fm/newbs_getting_started{fg_reset}z{fg_blue}Additionally, check out the FAQ (https://docs.qmk.fm/#/faq_build) or join the QMK Discord (https://discord.gg/qmk) for help.{fg_reset})r3r?rPpython_versionversionrrTrGr}rhr ERRORrDrRrr r4r rrrrupdaterw)rr` os_status git_statusbin_okver_oksub_oks r(doctorrsjGGLL;<GGLL%x'>'>'@AGGLL"CKK0GGLL(,7!#FYODJ[&&&9 +F:YdYlYlKl  34 F   67 ;&& & IJ""# $FF"""    &6[^^+C$$ F   12  &OY] ^    %'F [&& & &&F {** *v/G ((F  34 ;&& &  ST  no  X Y  g hrU))__doc__rPmilcrmilc.questionsrqmkr qmk.constantsrrrr checkr r r rqmk.gitrrrrrrrr qmk.commandsr qmk.userspacerrrrGrTrhrwr}argument subcommandrrUr(rs __WWeee&__3l#&%PK$G&dGL4Fde dF<$Ebc./E0dfErU