U hjM@sdZddlZddlZddlmZmZmZddlmZm Z m Z m Z m Z m Z mZmZddlZddlmZmZddlmZdZeGdd d ZeGd d d eZeGd d d eZeGdddeZeGdddeZeGdddeZd2eeeee dddZeeeedddZ d3e!ee e e e e!fdddZ"ee!e e#ddd Z$d4e!e!e eed!d"d#Z%e!e e#d$d%d&Z&e#e!e e#d'd(d)Z'e#e!e e#d*d+d,Z(e#e#e!e e#d-d.d/Z)Gd0d1d1eZ*dS)5zUtil that calls clickup.N)asdict dataclassfields)AnyDictListMappingOptionalTupleTypeUnion) BaseModelroot_validator)get_from_dict_or_envzhttps://api.clickup.com/api/v2c@s,eZdZdZeeeefddddZdS) ComponentzBase class for all components.datareturncCs tdS)N)NotImplementedErrorclsrrI/tmp/pip-unpacked-wheel-9gdii04g/langchain_community/utilities/clickup.py from_dataszComponent.from_dataN) __name__ __module__ __qualname____doc__ classmethodrstrrrrrrrrsrc@seZdZUdZeed<eed<eed<eed<eed<eed<eed<eed <eeee fed <eeee fed <e eed <e eed <e eed<eed<eed<eed<e eee fddddZ dS)TaskzClass for a task.idname text_content descriptionstatus creator_idcreator_username creator_email assigneeswatchersprioritydue_date start_datepointsteam_id project_idrcCs|ddkrdn |dd}||d|d|d|d|dd|dd|dd|dd |d |d ||d |d |d|d|dddS)Nr+r!r"r#r$r%Zcreatorusernameemailr)r*r,r-r.r/project)r!r"r#r$r%r&r'r(r)r*r+r,r-r.r/r0r)rrr+rrrr-s&     zTask.from_dataN) rrrrint__annotations__rrrrr rrrrrrr s&    r c@seZdZUdZeed<eed<dZeeed<dZ ee ed<dZ ee ed<dZ ee ed<dZee ed <dZeeed <eedd d d ZdS)CUListzComponent class for a list. folder_idr"Ncontentr, due_date_timer+assigneer%rc CsD||d|d|d|d|d|d|d|dd S) Nr7r"r8r,r9r+r:r%)r7r"r8r,r9r+r:r%)getrrrrrQszCUList.from_data)rrrrfloatr5rr8r r,r4r9boolr+r:r%rdictrrrrrr6Ds r6c@sFeZdZUdZeed<eed<eed<eed<eeddddZ d S) MemberzComponent class for a member.r!r1r2initialsrcCs0||dd|dd|dd|dddS)Nuserr!r1r2r@)r!r1r2r@rrrrrrhs     zMember.from_dataN) rrrrr4r5rrrrrrrrr?_s r?c@sBeZdZUdZeed<eed<eeed<e e ddddZ dS) TeamzComponent class for a team.r!r"membersrcCs(dd|dD}||d|d|dS)NcSsg|]}t|qSr)r?r).0Z member_datarrr |sz"Team.from_data..rCr!r")r!r"rCr)rrrCrrrrzszTeam.from_dataN) rrrrr4r5rrr?rrrrrrrrBrs  rBc@sVeZdZUdZeed<eed<eed<eee fed<e eee fddddZ d S) SpacezComponent class for a space.r!r"privateenabled_featuresrcCs>|dd}dd|dD}||d|d|d|d S) NspacesrcSsi|]\}}|dr||qS)Zenabledr)rDZfeaturevaluerrr sz#Space.from_data..featuresr!r"rG)r!r"rGrH)items)rrZ space_datarHrrrrs  zSpace.from_dataN) rrrrr4r5rr=rrrrrrrrrFs rFF)r componentfault_tolerantrc Cspzt||WStk rj}z<|rVdt|dt|d}t||WY S|W5d}~XYnXdS)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 Exceptionrwarningswarn)rrNrOeZ warning_strrrrparse_dict_through_components   rU)rrNrcCs0i}t|D]}|j|kr ||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")rrNoutput attributerrr+extract_dict_elements_from_component_fieldss   rXqueryrOrc Cs^zt|dfWStjk rX}z(|rDddt|dfWY S|W5d}~XYnXdS)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)rZrOrTrrr load_querys  r^)rkeyrcCsX||krTt||dkrTt||dkrDtd|d||d||ddSdS)z%Fetch the first id from a dictionary.rzFound multiple rPz. Defaulting to first.r!N)lenrRrS)rr_rrrfetch_first_ids rb)url access_tokenrZrcCs(d|i}tj|||d}||S)zFetch data from a URL. Authorizationheadersparams)requestsr;raise_for_statusr[)rcrdrZrgresponserrr fetch_datasrl)rdrcCstd}t||}t|dS)zFetch the team id./teamteams DEFAULT_URLrlrb)rdrcrrrr fetch_team_ids  rq)r/rdrcCs,td|d}t||ddid}t|dS)zFetch the space id./team//spacearchivedfalserZrIro)r/rdrcrrrrfetch_space_idsrw)space_idrdrcCs,td|d}t||ddid}t|dS)zFetch the folder id./space//folderrtrurvfoldersro)rxrdrcrrrrfetch_folder_idsr|)rxr7rdrcCsZ|rtd|d}ntd|d}t||ddid}|rLd|krL|dSt|dSd S) zFetch the list id./folder//listryrtrurvr!listsNro)rxr7rdrcrrrr fetch_list_ids rc@seZdZUdZdZeeed<dZeeed<dZ eeed<dZ eeed<dZ eeed<Gdd d Z e d;eeed d d Ze eeeeedddZeddeedddZeeeeefdddZeeeeeffdddZedddZeeefdddZedd d!Zdkeys)rrZZtaskrhrrrrget_task_attributes   z$ClickupAPIWrapper.get_task_attributec Csrt|dd\}}|dkr d|iStd|d}d|jdd}|}|d |d i}tj||||d }d |iS) rTrNrrrrrrrJrgrhr[rk)r^rpr/rriput) rrZ query_dictrrcrhrgpayloadrkrrr update_taskszClickupAPIWrapper.update_taskc Cst|dd\}}|dkr d|iS|dD]*}t|ts(dd|dt|iSq(td|d }|}|d d kr|dgd }n2|d d krg|dd }ntd|d ddd|jdd}d|i} tj |||| d} d| iS)z> Add or remove assignees of a specified task. TrNrZusersz3All users must be integers, not strings! "Got user z if type rrZ operationadd)rremrzInvalid operation (z). zValid options ['add', 'rem'].rrr)rrk) r^rr4typerpr ValueErrorr/rir) rrZrrrArcrgZassigne_payloadrhrrkrrrupdate_task_assigneess:     z'ClickupAPIWrapper.update_task_assigneesc Cs|t|dd\}}|dkr d|iS|j}td|d}d|jd}t|t}|}tj||||d } | } t | tddS) z% Creates a new task. TrNrrrr)rr/)r[rgrh) r^rrpr/rXr rrirr[rU) rrZrrrrcrhrrgrkrrrr create_tasks  zClickupAPIWrapper.create_taskc Cst|dd\}}|dkr d|iS|jr,|jn|j}td|d}t|t}|}tj|||d}| } t | t dd} d| kr| d|_ | S) z% Creates a new list. TrNrr}r~r[rgr!) r^r7rxrprXr rrirr[rUr6r) rrZrrlocationrcrrgrkrZ parsed_listrrr create_list"s  zClickupAPIWrapper.create_listc Csxt|dd\}}|dkr d|iS|j}td|d}d|di}|}tj|||d}|} d | krt| d |_| S) z' Creates a new folder. TrNrryrzr"rr!)r^rxrprrirr[r) rrZrrrxrcrrgrkrrrr create_folder:s zClickupAPIWrapper.create_folder)moderZrcCs|dkr||}n|dkr(||}n|dkr:|}n|dkrN||}n|dkrb||}n|dkrv||}nn|dkr|}n\|dkr|}nJ|d kr|}n8|d kr| |}n$|d kr| |}nd d |di}z t |WSt k rt|YSXdS)z Run the API.rrZ get_teamsrrrrrrrrZ ModeErrorzGot unexpected mode .N)rrrrrrrrrrrr[dumpsrQr)rrrZrVrrrrunRs6            zClickupAPIWrapper.run)r)T)*rrrrrdr rr5r/rxr7rrrrrrrrr>rrrr bytesrrrrrr=rrrrrrrrrrrrrrrrsN      'r)F)F)N)+rr[rRZ dataclassesrrrtypingrrrrr r r r riZlangchain_core.pydantic_v1r rZlangchain_core.utilsrrprr r6r?rBrFr>r=rUrXrr^r4rbrlrqrwr|rrrrrrsX( +