B h @sddlZddlmZddlmZmZmZddlmZm Z m Z ddl m Z ddl mZddlmZdd lmZd ed feeeed d dZdedfeeeed ddZeedddZdS)N)split)AnyDictSequence)unquoteurlparse parse_qsl)random)HOTP)OTP)TOTP Z ABCDEFGHIJKLMNOPQRSTUVWXYZ234567)lengthcharsreturncs,|dkrtddfddt|DS)Nrz#Secrets should be at least 160 bitsc3s|]}tVqdS)N)r choice).0_)rKC:\Users\sanjo\AppData\Local\Qlobot\Launcher\ext_packages\pyotp\__init__.py sz random_base32..) ValueErrorjoinrange)rrr)rr random_base32 s  r(ZABCDEF0123456789cCs|dkrtdt||dS)Nrz#Secrets should be at least 160 bits)rr)rr)rrrrr random_hexsr)urircCsd}i}tt|}|jdkr&tdtd|jdddd}t|dkrX|d|d<n|d|d <|d|d<x(t|jD]\}}|d kr|}q~|d krd |kr|d dk r|d |krtd ||d <q~|d kr$|d krt j |d<n4|dkrt j |d<n|dkrt j |d<ntdq~|dkrRt |}|dkrHtd||d<q~|dkrjt ||d<q~|dkrt ||d<q~|dkr~td|q~W|std|jdkrt|f|S|jdkrt|f|StddS)z Parses the provisioning URI for the OTP; works for either TOTP or HOTP. See also: https://github.com/google/google-authenticator/wiki/Key-Uri-Format :param uri: the hotp/totp URI to parse :returns: OTP object NZotpauthzNot an otpauth URIz:|%3Ar )maxsplitrnameZissuersecretzHIf issuer is specified in both label and parameters, it should be equal. algorithmSHA1digestSHA256SHA512z;Invalid value for algorithm, must be SHA1, SHA256 or SHA512digits)zDigits may only be 6, 7, or 8ZperiodintervalcounterZ initial_countimagez{} is not a valid parameterzNo secret found in URItotphotpzNot a supported OTP type)rrschemerrpathlenrqueryhashlibsha1sha256sha512intformatnetlocr r )rr"Zotp_dataZ parsed_uriZaccountinfo_partskeyvaluer(rrr parse_urisV                        r>)r5rertypingrrr urllib.parserrrcompatr r0r Zotpr r/r listr9strrrr>rrrrs