o Tj @sdZddlmZddlZddlZddlZddlmZmZddd Z d d Z dddZ d ddZ d!ddZ d"ddZd#ddZdS)$z?Pro subcommands: update, list, config (fingerprint management).) annotationsN)FingerprintManagerFingerprintUpdateErrorvaluestrreturncCs|ds td|S)Nimp_zAPI key must start with 'imp_'.) startswithargparseArgumentTypeErrorrr T/home/dinkstrade/pdmp-scanner/venv/lib/python3.10/site-packages/curl_cffi/cli/pro.py_validate_api_key s  rcCsH|durdSt|tr|rdSdSt|ttBr tj|dddSt|S)Ntruefalse),:F) separators ensure_ascii) isinstanceboollistdictjsondumpsrr r r r _format_cells  rheaders list[str]rowslist[list[str]]csdd|D|D]}t|D]\}}t|t||<qq d fdd }t||td d d D|D]}t||q;dS) NcSsg|]}t|qSr )len.0headerr r r sz _print_table.. row_valuesrrrcsdfddt|DS)Nz | c3s"|] \}}||VqdS)N)ljust)r$idxrwidthsr r #s z3_print_table..render_row..)join enumerate)r'r*r r render_row"sz _print_table..render_rowz-+-css|]}d|VqdS)-Nr )r$widthr r rr,(sz_print_table..)r'rrr)r.maxr"printr-)rr rowr)cellr/r r*r _print_tables r6fingerprint_rowslist[dict[str, object]]cs@gd}g}|D]fdd|D}||qt||dS)N)typenamebrowserversionos os_versionh3_fingerprintscsg|]}t|qSr )rr#itemr rr&9sz-_print_fingerprints_table..)appendr6)r7rr r4r r@r_print_fingerprints_table-s   rCcCsttj|ddddS)NF)indentr)r3rr)r r r r_print_fingerprints_json>srFNonecCsV|jdddd|jdddd}|jdd d d |jd d dd}|jddtdddS)z.Register update, list, and config subcommands.updatezUpdate local fingerprints cachez-Update fingerprints from the impersonate API.)help descriptionrz"List local and native fingerprintsz#List local and native fingerprints.z--json store_truezOutput fingerprints as JSON.)actionrIconfigzConfigure API accessz$Set the API key for impersonate.pro.z --api-keyTzAPI key for impersonate.pro)requiredr9rIN) add_parser add_argumentr) subparsers list_parser config_parserr r radd_pro_parsersBs4 rTrc Cs|jdkr=zt}Wnty'}ztt|tjdWYd}~d Sd}~ww|dkr.dnd}td|d|d d S|jd krUt}|j rOt |d St |d S|jd krjt |j td td SdS)zAHandle pro subcommands. Returns True if handled, False otherwise.rH)fileNrszTotal z fingerprintz in cache.TrrMzAPI key saved to F)commandrupdate_fingerprintsrr3rsysstderrlist_fingerprintsrrFrC set_api_keyapi_keyget_config_path)argsupdatedexcsuffixr r r rhandle_pro_commandes.     rd)rrrr)rrr r!)r7r8)r r8)rrG)rr)__doc__ __future__rr rrZcurl_cffi.fingerprintsrrrrr6rCrFrTrdr r r rs     #