o ȳgM @sdZddlZddlZddlmZmZmZddlmZm Z m Z m Z m Z m Z mZmZddlZddlmZddlmZmZmZdZeGdd d ZeGd d d eZeGd d d eZeGdddeZeGdddeZeGdddeZ d4dedeedede fddZ dedeedefddZ! d4de"dede e e e e"ffddZ#ded e"de e$fd!d"Z%d5d#e"d$e"de edefd%d&Z&d$e"de e$fd'd(Z'd)e$d$e"de e$fd*d+Z(d,e$d$e"de e$fd-d.Z)d,e$d/e$d$e"de e$fd0d1Z*Gd2d3d3eZ+dS)6zUtil that calls clickup.N)asdict dataclassfields)AnyDictListMappingOptionalTupleTypeUnion)get_from_dict_or_env) BaseModel ConfigDictmodel_validatorzhttps://api.clickup.com/api/v2c@s.eZdZdZedeeefddfddZdS) ComponentzBase class for all components.datareturncCstN)NotImplementedErrorclsrra/var/www/html/chatdoc2/venv/lib/python3.10/site-packages/langchain_community/utilities/clickup.py from_dataszComponent.from_dataN) __name__ __module__ __qualname____doc__ classmethodrstrrrrrrrrs rc@seZdZUdZeed<eed<eed<eed<eed<eed<eed<eed <eeee fed <eeee fed <e eed <e eed <e eed<eed<eed<eed<e deee fddfddZ dS)TaskzClass for a task.idname text_content descriptionstatus creator_idcreator_username creator_email assigneeswatchersprioritydue_date start_datepointsteam_id project_idrrcCs|ddurdn|dd}|did|dd|dd|dd|dd|ddd|ddd |dd d |dd d |d d|dd|d|dd|dd|dd|dd|ddS)Nr,r"r#r$r%r&r'creatorr(usernamer)emailr*r+r-r.r/r0r1projectrr)rrr,rrrr-sD          zTask.from_dataN) rrrrint__annotations__r rrrr rrrrrrr!s(     r!c@seZdZUdZeed<eed<dZeeed<dZ ee ed<dZ ee ed<dZ ee ed<dZee ed <dZeeed <ed ed dfd dZdS)CUListzComponent class for a list. folder_idr#Ncontentr- due_date_timer,assigneer&rrc CsD||d|d|d|d|d|d|d|dd S) Nr9r#r:r-r;r,r<r&)r9r#r:r-r;r,r<r&)getrrrrrQszCUList.from_data)rrrrfloatr7r r:r r-r6r;boolr,r<r&rdictrrrrrr8Ds r8c@sHeZdZUdZeed<eed<eed<eed<ededdfdd Z d S) MemberzComponent class for a member.r"r3r4initialsrrcCs0||dd|dd|dd|dddS)Nuserr"r3r4rB)r"r3r4rBrrrrrrhs     zMember.from_dataN) rrrrr6r7r rrrrrrrrA_s rAc@sDeZdZUdZeed<eed<eeed<e de ddfddZ d S) TeamzComponent class for a team.r"r#membersrrcCs(dd|dD}||d|d|dS)NcSsg|]}t|qSr)rAr).0 member_datarrr |sz"Team.from_data..rEr"r#)r"r#rEr)rrrErrrrzszTeam.from_dataN) rrrrr6r7r rrArrrrrrrrDrs  rDc@sXeZdZUdZeed<eed<eed<eee fed<e deee fddfdd Z d S) SpacezComponent class for a space.r"r#privateenabled_featuresrrcCs>|dd}dd|dD}||d|d|d|d S) NspacesrcSsi|] \}}|dr||qS)enabledr)rFfeaturevaluerrr s z#Space.from_data..featuresr"r#rJ)r"r#rJrK)items)rr space_datarKrrrrs  zSpace.from_dataN) rrrrr6r7r r?rrrrrrrrrIs  rIFr componentfault_tolerantrc Csfzt||WSty2}z|r,dt|dt|d}t||WYd}~S|d}~ww)aParse a dictionary by creating a component and then turning it back into a dictionary. This helps with two things 1. Extract and format data from a dictionary according to schema 2. Provide a central place to do this in a fault-tolerant way z(Error encountered while trying to parse : z' Falling back to returning input data.N)rr Exceptionr warningswarn)rrTrUe warning_strrrrparse_dict_through_components  r\cCs0i}t|D]}|j|vr||j||j<q|S)a"Extract elements from a dictionary. Args: data: The dictionary to extract elements from. component: The component to extract elements from. Returns: A dictionary containing the elements from the input dictionary that are also in the component. )rr#)rrToutput attributerrr+extract_dict_elements_from_component_fieldss   r_queryc CsTzt|dfWStjy)}z|r#ddt|dfWYd}~S|d}~ww)aDParse a JSON string and return the parsed object. If parsing fails, returns an error message. :param query: The JSON string to parse. :return: A tuple containing the parsed object or None and an error message or None. Exceptions: json.JSONDecodeError: If the input is not a valid JSON string. Nz5Input must be a valid JSON. Got the following error: z". "Please reformat and try again.)jsonloadsJSONDecodeErrorr )r`rUrZrrr load_querys rdkeycCsX||vr*t||dkr*t||dkr"td|d||d||ddSdS)z%Fetch the first id from a dictionary.rzFound multiple rVz. Defaulting to first.r"N)lenrXrY)rrerrrfetch_first_ids rhurl access_tokencCs(d|i}tj|||d}||S)zFetch data from a URL. Authorizationheadersparams)requestsr=raise_for_statusra)rirjr`rmresponserrr fetch_datasrrcCstd}t||}t|dS)zFetch the team id./teamteams DEFAULT_URLrrrh)rjrirrrr fetch_team_ids   rwr0cC,td|d}t||ddid}t|dS)zFetch the space id./team//spacearchivedfalser`rLru)r0rjrirrrrfetch_space_id r~space_idcCrx)zFetch the folder id./space//folderr{r|r}foldersru)rrjrirrrrfetch_folder_idrrr9cCsV|r td|d}ntd|d}t||ddid}|r&d|vr&|dSt|dS) zFetch the list id./folder//listrr{r|r}r"listsru)rr9rjrirrrr fetch_list_ids  rc @seZdZUdZdZeeed<dZeeed<dZ eeed<dZ eeed<dZ eeed<e dd Z e d>d ed ed efddZed ededed eefddZeddeded efddZded eeeeffddZd eeeeefffddZd efddZd eeeffd d!Zd efd"d#Zd?d%ed&ed efd'd(Z d efd)d*Z!d%ed efd+d,Z"d efd-d.Z#d%ed efd/d0Z$d%ed efd1d2Z%d%ed efd3d4Z&d%ed efd5d6Z'd%ed efd7d8Z(d%ed efd9d:Z)d;ed%ed efd Add or remove assignees of a specified task. TrNrusersz3All users must be integers, not strings! "Got user z if type rr operationadd)rremrzInvalid operation (z). zValid options ['add', 'rem'].rrr*rrq) rdrr6typervr ValueErrorr0ror) rr`rrrCrirmassigne_payloadrnrrqrrrupdate_task_assigneess:    z'ClickupAPIWrapper.update_task_assigneesc Cs|t|dd\}}|durd|iS|j}td|d}d|jd}t|t}|}tj||||d } | } t | tddS) z% Creates a new task. TrNrrrr)rr0)rarmrn) rdrrvr0r_r!rrorrar\) rr`rrrrirnrrmrqrrrr create_tasks  zClickupAPIWrapper.create_taskc Cst|dd\}}|durd|iS|jr|jn|j}td|d}t|t}|}tj|||d}| } t | t dd} d| vrF| d|_ | S) z% Creates a new list. TrNrrrrarmr") rdr9rrvr_r!rrorrar\r8r) rr`rrlocationrirrmrqr parsed_listrrr create_list$s  zClickupAPIWrapper.create_listc Csxt|dd\}}|durd|iS|j}td|d}d|di}|}tj|||d}|} d | vr:| d |_| S) z' Creates a new folder. TrNrrrr#rr")rdrrvrrorrar) rr`rrrrirrmrqrrrr create_folder<s zClickupAPIWrapper.create_folderrcCs |dkr ||}nh|dkr||}n^|dkr|}nU|dkr'||}nK|dkr1||}nA|dkr;||}n7|dkrD|}n.|dkrM|}n%|d krV|}n|d kr`| |}n|d krj| |}nd d |di}zt |WSt yt|YSw)z Run the API.rr get_teamsrrrrrrrr ModeErrorzGot unexpected mode .)rrrrrrrrrrrradumpsrWr )rrr`r]rrrrunTs8              zClickupAPIWrapper.run)r)T)+rrrrrjr r r7r0rr9rr model_configrrrrrrrr@rrrr bytesrrrrr?rrrrrrrrrrrrrrrrsd          'r)Fr),rrarX dataclassesrrrtypingrrrrr r r r rolangchain_core.utilsr pydanticrrrrvrr!r8rArDrIr@r?r\r_r rdr6rhrrrwr~rrrrrrrsj( +