o 0i&@sdZddlZddlZddlmZmZmZmZddlZddl m Z ddl m Z ddl mZmZmZmZmZmZmZmZeee dZejd d d eDed Zee Zejd dd eDed Zdeedee fddZeddZejdgddej ddfdedededdfddZ!ejddgd dRd eeej"d!dfdeddfd"d#Z#ejd$gd%d     dSd&eeeej$d'd(dfd)eeeej$d*dfd+eeeej$d,d-dfd.eeeeej$d/d0dfd1eeeej$d2dfdeddfd3d4Z%ejd5gd6d     dSd eeej"d7dfd)eeeej$d8dfd&eeeeej$d'd9dfd.eeeeej$d/d:dfd1eeeej$d;dfdeddfdd?gd dRd eeej"d@dfdeddfdAdBZ'ejdCdDgd dRd eeej"dEdfdeddfdFdGZ(ejdHdIdJgd  dTd eeej"dKdfdLee)ej$dMdNdOdfdeddfdPdQZ*dS)UaContains commands to manage webhooks on the Hugging Face Hub. Usage: # list all webhooks hf webhooks ls # show details of a single webhook hf webhooks info # create a new webhook hf webhooks create --url https://example.com/hook --watch model:bert-base-uncased # create a webhook watching multiple items and domains hf webhooks create --url https://example.com/hook --watch org:HuggingFace --watch model:gpt2 --domain repo # update a webhook hf webhooks update --url https://new-url.com/hook # enable / disable a webhook hf webhooks enable hf webhooks disable # delete a webhook hf webhooks delete N) AnnotatedOptionalget_argsget_type_hints)WEBHOOK_DOMAIN_T)WebhookWatchedItem) FormatOpt OutputFormatQuietOptTokenOptapi_object_to_dict get_hf_apiprint_list_output typer_factorytypeWatchedItemTypecCi|]}||qSr).0trrV/home/kim/smarthome/.venv/lib/python3.10/site-packages/huggingface_hub/cli/webhooks.py ?r)r WebhookDomaincCrrrrdrrrrBrvaluesreturnc Csg}tt}|D]9}d|vrtd|dd|d|dd\}}||vr8td|dd|d|t||dq|S) aYParse 'type:name' strings into WebhookWatchedItem objects. Args: values: List of strings in the format 'type:name' (e.g., 'model:bert-base-uncased', 'org:HuggingFace'). Returns: List of WebhookWatchedItem objects. Raises: typer.BadParameter: If any value doesn't match the expected format. :zCExpected format 'type:name' (e.g. 'model:bert-base-uncased'), got 'z'. Valid types: , .rzInvalid type ')rname)tuple_WATCHED_TYPEStyper BadParameterjoinsplitappendr)ritemsZ valid_typesvkindr"rrr _parse_watchEs r-zManage webhooks on the Hub.)helpz list | ls)zhf webhooks lszhf webhooks ls --format jsonzhf webhooks ls -q)ZexamplesFformatquiettokencCs:t|d}dd|D}t|||gdddddS) z'List all webhooks for the current user.r1cSsg|]}t|qSr)r rwrrr sszwebhooks_ls..)idurldisableddomainswatchedc SsV|dd|dp dt|ddd|dpgdd d |d p&gDgS) Nr6r7z(job)r8Fr r9css8|]}t|tr|dd|dnt|VqdS)rrr"N) isinstancedictstrr3rrr ~s( z0webhooks_ls....r:)getr>r')itemrrrys    zwebhooks_ls..)r/r0headersZrow_fnN)rZ list_webhooksr)r/r0r1apiresultsrrr webhooks_lsds  rFinfozhf webhooks info abc123 webhook_idzThe ID of the webhook.cCs.t|d}||}ttjt|dddS)z/Show full details for a single webhook as JSON.r2indentN)rZ get_webhookprintjsondumpsr rHr1rDwebhookrrr webhooks_infos rQcreate)zQhf webhooks create --url https://example.com/hook --watch model:bert-base-uncasedzjhf webhooks create --url https://example.com/hook --watch org:HuggingFace --watch model:gpt2 --domain repozJhf webhooks create --job-id 687f911eaea852de79c4a50a --watch user:julien-cwatchz--watchzRItem to watch, in 'type:name' format (e.g. 'model:bert-base-uncased'). Repeatable.r7zBURL to send webhook payloads to. Mutually exclusive with --job-id.job_idz--job-idz]ID of a Job to trigger (from job.id) instead of pinging a URL. Mutually exclusive with --url.domainz--domainzNDomain to watch: 'repo' or 'discussions'. Repeatable. Defaults to all domains.secretz.Optional secret used to sign webhook payloads.c Cs|dur |dur td|dur|durtdt|d}t|}|r,dd|Dnd}|j|||||d} td| jttjt | d d dS) z|Create a new webhook. Provide either --url (to ping a remote server) or --job-id (to trigger a Job), but not both. Nz+Provide either --url or --job-id, not both.z!Provide either --url or --job-id.r2cSg|]}|jqSrvaluerrrrr5rz#webhooks_create..)r7rTr:r9rVzWebhook created: rIrJ) r%r&rr-Zcreate_webhookrLr6rMrNr ) rSr7rTrUrVr1rD watched_itemsr9rPrrrwebhooks_creates,   r[update)z8hf webhooks update abc123 --url https://new-url.com/hookz:hf webhooks update abc123 --watch model:gpt2 --domain repoz,hf webhooks update abc123 --secret newsecretz The ID of the webhook to update.z$New URL to send webhook payloads to.ziNew list of items to watch, in 'type:name' format. Repeatable. Replaces the entire existing watched list.zBNew list of domains to watch: 'repo' or 'discussions'. Repeatable.z)New secret used to sign webhook payloads.c Csnt|d}|r t|nd}|rdd|Dnd}|j|||||d} td| jttjt| dddS) z>Update an existing webhook. Only provided options are changed.r2NcSrWrrXrrrrr5rz#webhooks_update..)r7r:r9rVzWebhook updated: rIrJ)rr-Zupdate_webhookrLr6rMrNr ) rHr7rSrUrVr1rDrZr9rPrrrwebhooks_updates &r]enablezhf webhooks enable abc123z The ID of the webhook to enable.cC(t|d}||}td|jdS)zEnable a disabled webhook.r2zWebhook enabled: N)rZenable_webhookrLr6rOrrrwebhooks_enable r`disablezhf webhooks disable abc123z!The ID of the webhook to disable.cCr_)zDisable an active webhook.r2zWebhook disabled: N)rZdisable_webhookrLr6rOrrrwebhooks_disable rarcdeletezhf webhooks delete abc123zhf webhooks delete abc123 --yesz The ID of the webhook to delete.yesz--yesz-yzSkip confirmation prompt.cCsP|std|d}|stdtt|d}||td|dS)zDelete a webhook permanently.z)Are you sure you want to delete webhook 'z'?zAborted.r2zWebhook deleted: N)r%confirmrLZAbortrZdelete_webhook)rHrer1rfrDrrrwebhooks_deletes  rg)N)NNNNN)FN)+__doc__enumrMtypingrrrrr%Zhuggingface_hub.constantsrZhuggingface_hub.hf_apirZ _cli_utilsr r r r r rrrr$Enumr>rZ _DOMAIN_TYPESrlistr-Z webhooks_clicommandtablerFZArgumentrQOptionr[r]r`rcboolrgrrrrs   (           0       &