U «ë flkã@sšddlZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z e  dd¡Z e  dd¡ZdZd ZGd d „d eƒZd d „ZGdd„dƒZdS)éN)Úprompt)ÚInMemoryHistoryZWIT_URLzhttps://api.wit.aiÚWIT_API_VERSIONZ20200513z> z,Learn more at https://wit.ai/docs/quickstartc@s eZdZdS)ÚWitErrorN)Ú__name__Ú __module__Ú __qualname__©r r ú+/tmp/pip-unpacked-wheel-2ezanbwc/wit/wit.pyrsrc Ks¸t|}| d|||¡d|dtddœ}| | di¡¡tj||f||dœ|—Ž}|jdkr„td t |jƒd |j d ƒ‚|  ¡} d | kr¤td | d ƒ‚| d||| ¡| S)Nz%s %s %szBearer zapplication/vnd.wit.z+json)Ú authorizationÚacceptÚheaders)r ÚparamséÈzWit responded with status: z (ú)ÚerrorzWit responded with an error: ) Ú WIT_API_HOSTÚdebugrÚupdateÚpopÚrequestsÚrequestÚ status_coderÚstrÚreasonÚjson) ÚloggerÚ access_tokenÚmethÚpathrÚkwargsÚfull_urlr Zrsprr r r Úreqs0 þ ÿþýüÿr"c@s¤eZdZdZiZdRdd„ZdSdd„ZdTdd„ZdUdd „ZdVd d „Z dWd d „Z dXdd„Z dYdd„Z dZdd„Z d[dd„Zd\dd„Zd]dd„Zd^dd„Zd_dd„Zd`dd„Zdad d!„Zdbd"d#„Zdcd$d%„Zddd&d'„Zded(d)„Zdfd*d+„Zdgd,d-„Zdhd.d/„Zdid0d1„Zdjd2d3„Zdkd4d5„Zdld6d7„Zdmd8d9„Z dnd:d;„Z!dodd?„Z#dqd@dA„Z$drdBdC„Z%dsdDdE„Z&dtdFdG„Z'dudHdI„Z(dvdJdK„Z)dwdLdM„Z*dxdNdO„Z+dydPdQ„Z,dS)zÚWitNcCs||_|pt t¡|_dS)N)rÚloggingÚ getLoggerrr)Úselfrrr r r Ú__init__5sz Wit.__init__cCsVi}|dk r||d<|r ||d<|r2t |¡|d<|r>||d<t|j|jdd|ƒ}|S)NÚnÚqÚcontextÚverboseÚGETz/message)rÚdumpsr"rr)r&Úmsgr*r(r+rÚrespr r r Úmessage9sz Wit.messagec Cs6i}|p i}|rd|d<t|j|jdd|||d}|S)aESends an audio file to the /speech API. Uses the streaming feature of requests (see `req`), so opening the file in binary mode is strongly recommended (see http://docs.python-requests.org/en/master/user/advanced/#streaming-uploads). Add Content-Type header as specified here: https://wit.ai/docs/http/20200513#post--speech-link :param audio_file: an open handler to an audio file :param headers: an optional dictionary with request headers :param verbose: for legacy versions, get extra information :return: Tr+ÚPOSTz/speech)Údatar ©r"rr)r&Z audio_filer r+rr/r r r ÚspeechFs ù z Wit.speechc Csx|dkr i}tƒ}ztt|dd ¡}Wnttfk rBYdSX|dkr^t| ||¡ƒqt|| ||¡ƒƒqdS)a Runs interactive command line chat between user and bot. Runs indefinitely until EOF is entered to the prompt. handle_message -- optional function to customize your response. context -- optional initial context. Set to {} if omitted NT)ÚhistoryZ mouse_support)rrÚINTERACTIVE_PROMPTÚrstripÚKeyboardInterruptÚEOFErrorÚprintr0)r&Zhandle_messager*r5r0r r r Ú interactiveasÿzWit.interactivecCs4i}|p i}|rd|d<t|j|jdd||d}|S)zH Returns names of all intents associated with your app. Tr+r,ú/intents©r r3©r&r r+rr/r r r Ú intent_listxsÿzWit.intent_listcCsPi}|p i}|r||d<|r$d|d<|dk r4||d<t|j|jdd||d}|S) zT Returns the list of the top detected locales for the text message. r)Tr+Nr(r,z /languager=r3)r&r.r(r r+rr/r r r Údetect_language…s"ÿzWit.detect_languagecCsDi}|p i}|rd|d<dtj |¡}t|j|jd|||d}|S)zy Returns all available information about an intent. :param intent_name: name of existing intent Tr+ú /intents/r,r=©ÚurllibÚparseÚ quote_plusr"rr©r&Ú intent_namer r+rÚendpointr/r r r Ú intent_info–sÿzWit.intent_infocCs4i}|p i}|rd|d<t|j|jdd||d}|S)zH Returns list of all entities associated with your app. Tr+r,ú /entitiesr=r3r>r r r Ú entity_list¦sÿzWit.entity_listcCsDi}|p i}|rd|d<dtj |¡}t|j|jd|||d}|S)zy Returns all available information about an entity. :param entity_name: name of existing entity Tr+ú /entities/r,r=rB©r&Ú entity_namer r+rrHr/r r r Ú entity_info³sÿzWit.entity_infocCs4i}|p i}|rd|d<t|j|jdd||d}|S)zF Returns list of all traits associated with your app. Tr+r,ú/traitsr=r3r>r r r Ú trait_listÃsÿzWit.trait_listcCsDi}|p i}|rd|d<dtj |¡}t|j|jd|||d}|S)zu Returns all available information about a trait. :param trait_name: name of existing trait Tr+ú/traits/r,r=rB©r&Ú trait_namer r+rrHr/r r r Ú trait_infoÐsÿzWit.trait_infocCsDi}|p i}|rd|d<dtj |¡}t|j|jd|||d}|S)zv Delete an intent associated with your app. :param intent_name: name of intent to be deleted Tr+rAÚDELETEr=rBrFr r r Ú delete_intentàsÿzWit.delete_intentcCsDi}|p i}|rd|d<dtj |¡}t|j|jd|||d}|S)zv Delete an entity associated with your app. :param entity_name: name of entity to be deleted Tr+rLrVr=rBrMr r r Ú delete_entityðsÿzWit.delete_entitycCsTi}|p i}|rd|d<dtj |¡dtj |¡}t|j|jd|||d}|S)zÌ Deletes a role associated with the entity. :param entity_name: name of entity whose particular role is to be deleted :param role_name: name of role to be deleted Tr+rLú:rVr=rB)r&rNZ role_namer r+rrHr/r r r Ú delete_roles* ÿþ ýÿÿzWit.delete_rolecCsTi}|p i}|rd|d<dtj |¡dtj |¡}t|j|jd|||d}|S)zØ Deletes a keyword associated with the entity. :param entity_name: name of entity whose particular keyword is to be deleted :param keyword_name: name of keyword to be deleted Tr+rLú /keywords/rVr=rB)r&rNÚ keyword_namer r+rrHr/r r r Údelete_keywords* ÿþ ýÿÿzWit.delete_keywordc Csdi}|p i}|rd|d<dtj |¡dtj |¡dtj |¡}t|j|jd|||d}|S)zÙ Delete a synonym of the keyword of the entity. :param entity_name: name of entity whose particular keyword is to be deleted :param keyword_name: name of keyword to be deleted Tr+rLr[z /synonyms/rVr=rB) r&rNr\Z synonym_namer r+rrHr/r r r Údelete_synonym,s2  ÿþ ýü ûÿÿzWit.delete_synonymcCsDi}|p i}|rd|d<dtj |¡}t|j|jd|||d}|S)zt Delete a trait associated with your app. :param intent_name: name of intent to be deleted Tr+rRrVr=rBrSr r r Ú delete_traitFsÿzWit.delete_traitcCsTi}|p i}|rd|d<dtj |¡dtj |¡}t|j|jd|||d}|S)zÍ Deletes a value associated with the trait. :param trait_name: name of trait whose particular value is to be deleted :param value_name: name of value to be deleted Tr+rRz/values/rVr=rB)r&rTZ value_namer r+rrHr/r r r Údelete_trait_valueVs* ÿþ ýÿÿzWit.delete_trait_valuecCsXi}|p i}|dk r||d<|r(||d<|r4||d<|r@||d<t|j|jdd|ƒ}|S)zè Returns a JSON array of utterances. :param limit: number of utterances to return :param offset: number of utterances to skip :param intents: list of intents to filter the utterances NÚlimitÚoffsetÚintentsr+r,ú /utterancesr3)r&rarbrcr r+rr/r r r Úget_utterancesls zWit.get_utterancesc CsRi}|p i}g}|D]}| d|i¡q|r4||d<t|j|jdd|||d}|S)zw Delete utterances from your app. :param utterances: list of utterances to be deleted Útextr+rVrd©rr )Úappendr"rr)r&Z utterancesr r+rr2Z utterancer/r r r Údelete_utterancesƒs"ù zWit.delete_utterancescCsLi}|p i}|dk r||d<|r(||d<|r4||d<t|j|jdd|ƒ}|S)z  Returns an array of all your apps. :param limit: number of apps to return :param offset: number of utterances to skip Nrarbr+r,ú/appsr3)r&rarbr r+rr/r r r Úget_apps›sz Wit.get_appscCsDi}|p i}|rd|d<dtj |¡}t|j|jd|||d}|S)ús Returns an object representation of the specified app. :param app_id: ID of existing app Tr+ú/apps/r,r=rB©r&Úapp_idr r+rrHr/r r r Úapp_info­sÿz Wit.app_infocCsDi}|p i}|rd|d<dtj |¡}t|j|jd|||d}|S)rlTr+rmrVr=rBrnr r r Ú delete_app½sÿzWit.delete_appcCsHi}|p i}|rd|d<dtj |¡d}t|j|jd|||d}|S)zk Returns an array of all tag groups for an app. :param app_id: ID of existing app Tr+rmz/tagsr,r=rBrnr r r Ú app_versionsÍsÿzWit.app_versionscCsTi}|p i}|rd|d<dtj |¡dtj |¡}t|j|jd|||d}|S)z– Returns an object representation of the specified app. :param app_id: ID of existing app :param tag_id: name of tag Tr+rmú/tags/r,r=rB)r&roZtag_idr r+rrHr/r r r Úapp_version_infoÝsÿÿzWit.app_version_infoc CsJi}|p i}d|i}d|d}|r,||d<t|j|jd||||d}|S)z“ Create a new version of your app. :param app_id: ID of existing app :param tag_name: name of tag Útagrmrsr+r1rgr3) r&roÚtag_namer r+rr2rHr/r r r Úcreate_app_versionðs  ù zWit.create_app_versioncCsTi}|p i}dtj |¡dtj |¡}|r8||d<t|j|jd|||d}|S)z˜ Delete a specific version of your app. :param app_id: ID of existing app :param tag_name: name of tag rmrsr+rVr=rB)r&rorvr r+rrHr/r r r Údelete_app_versions* ÿþ ýÿÿzWit.delete_app_versioncCs4i}|p i}|rd|d<t|j|jdd||d}|S)z^ Get a URL where you can download a ZIP file containing all of your app data. Tr+r,z/exportr=r3r>r r r Úexportsÿz Wit.exportcCsPi}|p i}|dk r||d<|r(||d<|r4||d<t|j|jdd||d}|S)z© Create a new app with all the app data from the exported app. :param name: name of the new app :param private: private if true NÚnameÚprivater+r1z/import)r2r3)r&rzr{Úzip_filer r+rr/r r r Ú import_app+s"ÿzWit.import_appc CsBi}|p i}d|i}d}|r$||d<t|j|jd||||d}|S)zƒ Creates a new intent with the given attributes. :param intent_name: name of intent to be created rzr<r+r1rgr3)r&rGr r+rr2rHr/r r r Ú create_intent?s ù zWit.create_intentc CsPi}|p i}||dœ}d}|r&||d<|r2||d<t|j|jd||||d} | S)a  Creates a new intent with the given attributes. :param entity_name: name of entity to be created :param roles: list of roles you want to create for the entity :param lookups: list of lookup strategies ©rzÚrolesrJÚlookupsr+r1rgr3) r&rNr€rr r+rr2rHr/r r r Ú create_entityVs$  ù zWit.create_entityc Cs\i}|p i}||dœ}dtj |¡} |r2||d<|r>||d<t|j|jd| |||d} | S)zï Updates the attributes of an entity. :param entity_name: name of entity to be updated :param roles: updated list of roles :param lookups: updated list of lookup strategies rrLrr+ÚPUTrgrB) r&Zcurrent_entity_nameZnew_entity_namer€rr r+rr2rHr/r r r Ú update_entityss$ ù zWit.update_entityc CsJi}|p i}dtj |¡d}|r,||d<t|j|jd||||d}|S)z­ Add a possible value into the list of keywords for the keywords entity. :param entity_name: name of entity to which keyword is to be added rLz /keywordsr+r1rgrB)r&rNr2r r+rrHr/r r r Úadd_keyword_value–sù zWit.add_keyword_valuec Csbi}|p i}dtj |¡dtj |¡d}d|i}|rD||d<t|j|jd||||d} | S)a< Create a new synonym of the canonical value of the keywords entity. :param entity_name: name of entity to which synonym is to be added :param keyword_name: name of keyword to which synonym is to be added :param synonym: name of synonym to be created rLr[z /synonymsÚsynonymr+r1rgrB) r&rNr\r†r r+rrHr2r/r r r Úcreate_synonym¬s2  ÿþ ýüÿù zWit.create_synonymc CsDi}|p i}||dœ}d}|r&||d<t|j|jd||||d}|S)z¼ Creates a new trait with the given attributes. :param trait_name: name of trait to be created :param values: list of values for the trait )rzÚvaluesrPr+r1rgr3) r&rTrˆr r+rr2rHr/r r r Ú create_traitÍs  ù zWit.create_traitc CsRi}|p i}d|i}dtj |¡d}|r4||d<t|j|jd||||d}|S)zÎ Creates a new trait with the given attributes. :param trait_name: name of trait to which new value is to be added :param new_value: name of new trait value ÚvaluerRz/valuesr+r1rgrB) r&rTÚ new_valuer r+rr2rHr/r r r Úcreate_trait_valueås ù zWit.create_trait_valuec Cs:i}|p i}d}|r||d<t|j|jd||||d}|S)zr Train your utterances. :param data: array of utterances with required arguments rdr+r1rgr3)r&r2r r+rrHr/r r r Útrainýsù z Wit.trainc CsRi}|p i}|||dœ}d} |r(||d<|r4||d<t|j|jd| |||d} | S)a Creates a new app for an existing user. :param app_name: name of new app :param lang: language code in ISO 639-1 format :param private: private if true :param timezone: default timezone of the app )rzÚlangr{rjÚtimezoner+r1rgr3) r&Úapp_namerŽr{rr r+rr2rHr/r r r Ú create_apps$  ù zWit.create_appc Cszi}|p i}i} dtj |¡} |r,|| d<|r8|| d<|rD|| d<|rP|| d<|r\||d<t|j|jd| || |d} | S) a Updates existing app with given attributes. :param app_name: new_name :param lang: language code in ISO 639-1 format :param private: private if true :param timezone: default timezone of the app rmrzrŽr{rr+rƒrgrB) r&rorrŽr{rr r+rr2rHr/r r r Ú update_app1s0ù zWit.update_appc Cs~i}|p i}i} dtj |¡dtj |¡} |r<|| d<|rH|| d<|rT|| d<|r`||d<t|j|jd| || |d} | S) a^ Update the tag's name or description, or move the tag to point to another tag. :param app_id: ID of existing app :param tag_name: name of existing tag :param new_name: name of new tag :param desc: new description of tag :param move_to: new name of tag rmrsruÚdescÚmove_tor+rƒrgrB) r&rorvÚnew_namer“r”r r+rr2rHr/r r r Úupdate_app_version\s: ÿþ ýÿù zWit.update_app_version)N)NNN)NN)NN)NN)NNN)NN)NN)NN)NN)NN)NN)NN)NN)NN)NN)NN)NN)NNNN)NN)NNN)NN)NN)NN)NN)NN)NN)NN)NN)NN)NNN)NNN)NN)NN)NN)NN)NN)NNN)NNNNNN)NNNNN)-rrrrZ _sessionsr'r0r4r;r?r@rIrKrOrQrUrWrXrZr]r^r_r`rerirkrprqrrrtrwrxryr}r~r‚r„r…r‡r‰rŒrr‘r’r–r r r r r#1sŒ           ÿ   ÿ           ÿ "ù # ÿ !   ÿ !ø /ør#)rr$ÚosÚurllib.requestrCÚ urllib.parseÚ urllib.errorrZprompt_toolkitrZprompt_toolkit.historyrÚgetenvrrr6Z LEARN_MOREÚ Exceptionrr"r#r r r r Ús