client.cpython-313.pyc•7.69 kB
�
�|OgX � �X � S r SSKrSSKJrJrJrJr SSKrSSKJ r J
r
J r " S S5 r g)z!Notion API client implementation.� N)�Any�Dict�List�Optional� )�Database�Page�
SearchResultsc �d � \ rS rSrSrS\4S jrS\\ 4S jr SS\S \
\ \\ 4 S
\
\\ \\ 4 S \
\ S \
S\ \\ 4 4 S
jjr SS\S\ \\ 4 S\
\\ \\ 4 S\4S jjr SS\S\ \\ 4 S\
\ S\4S jjr SS\S \
\ \\ 4 S\
\ \\ 4 S \
\ S \
S\4 S jjrSrg) � NotionClient� z+Client for interacting with the Notion API.�api_keyc �: � Xl SU l SU 3SSS.U l g)zAInitialize the Notion client.
Args:
api_key: Notion API key
zhttps://api.notion.com/v1zBearer zapplication/jsonz
2022-06-28)�
Authorizationz Content-TypezNotion-VersionN�r �base_url�headers)�selfr s �C/Users/chasecabanillas/Projects/notion-mcp/src/notion_mcp/client.py�__init__�NotionClient.__init__ s) � � � �3��
� &�w�i� 0� .� *�
�� � �returnc
� �� # � [ R " 5 ISh v�N nUR U R S3U R SSS.SSSS .S
.S 9I Sh v�N nUR 5 UR
5 nUR S 5 ( d / sSSS5 ISh v�N $ US Vs/ sH n[ S
0 UD6PM snsSSS5 ISh v�N $ N� N~ N8s snf N! , ISh v�N ( d f g= f7f)z1List all databases the integration has access to.N�/search�object�database)�property�value�d �
descending�last_edited_time)� direction� timestamp)�filter� page_size�sort�r �json�results� ) �httpx� AsyncClient�postr r �raise_for_statusr) �getr )r �client�response�data�dbs r �list_databases�NotionClient.list_databases s� � � ��$�$�&�&�&� #�[�[��=�=�/��)� � � � %-�!+� � "%�%1�%7� �
� )� � �H�
� %� %� '��=�=�?�D��8�8�I�&�&��' '�&�&�( .2�)�_�=�_�r�H�N�r�N�_�=�) '�&�&�� '��( >�) '�&�&�&�s� �C:�C�C:�6C �C�;C � C:� C� C:�$C �+C�?C � C:�
C �C:�C �C:�C � C:� C7�&C) �' C7�3C:N� database_idr% �sorts� start_cursorr& c � � # � SU0nU( a X&S' U( a X6S' U( a XFS' [ R " 5 ISh v�N nUR U R SU S3U R US9I Sh v�N nUR 5 UR
5 sSSS5 ISh v�N $ Nh N7 N ! , ISh v�N ( d f g= f7f) zQuery a database.r& r% r8 r9 Nz /databases/z/queryr( )r, r- r. r r r/ r) ) r r7 r% r8 r9 r& �bodyr1 r2 s r �query_database�NotionClient.query_database1 s� � � �
��
�� � #��N� � !��M� �#/� � ��$�$�&�&�&� #�[�[��=�=�/��[�M��@� � � �� )� � �H�
� %� %� '��=�=�?� '�&�&�� '�&�&�&�sY �?C
�B*�C
�/B0�4B,�5#B0� C
�$B.�%C
�,B0�.C
�0C�6B9 �7 C�C
� parent_id�
properties�childrenc � �~ # � SU0US.nU( a X4S' [ R " 5 ISh v�N nUR U R S3U R US9I Sh v�N nUR 5 [
S0 UR 5 D6sSSS5 ISh v�N $ Nn N@ N ! , ISh v�N ( d f g= f7f)zCreate a new page.r7 )�parentr? r@ Nz/pagesr( r+ )r, r- r. r r r/ r r) )r r> r? r@ r; r1 r2 s r � create_page�NotionClient.create_pageM s� � � � %�i�0�$�
�� � '�� ��$�$�&�&�&� #�[�[��=�=�/��(� � � �� )� � �H�
� %� %� '��*�(�-�-�/�*� '�&�&�� '�&�&�&�sV �,B=�B �B=�,B#� B � ,B#� B=�B!�B=� B#�!B=�#B:�)B, �* B:�6B=�page_id�archivedc � �t # � SU0nUb X4S' [ R " 5 ISh v�N nUR U R SU 3U R US9I Sh v�N nUR 5 [
S0 UR 5 D6sSSS5 ISh v�N $ Np N@ N ! , ISh v�N ( d f g= f7f)zUpdate a page.r? NrF z/pages/r( r+ )r, r- �patchr r r/ r r) )r rE r? rF r; r1 r2 s r � update_page�NotionClient.update_paged s� � � � �j�)�� � � '�� ��$�$�&�&�&� #�\�\��=�=�/��� �2� � � �� *� � �H�
� %� %� '��*�(�-�-�/�*� '�&�&�� '�&�&�&�sV �%B8�B�B8�.B �B�,B � B8�B �B8�B � B8� B5�$B' �% B5�1B8�queryr' c
� �0 # � UUS.nU( a X&S' U( a X6S' U( a XFS' [ R " 5 ISh v�N nUR U R S3U R US9I Sh v�N nUR 5 UR
5 n / n
U R S/ 5 H� n U S S
:X a U
R [ S0 U D65 M( U S S :X d M3 0 n U R S 0 5 R 5 H u p�[ S0 UD6X�' M X�S ' U
R [ S0 U D65 M� [ S
U
U R S5 U R SS5 S9sSSS5 ISh v�N $ GN2 GN N ! , ISh v�N ( d f g= f7f)zSearch Notion.)rK r& r% r' r9 Nr r( r* r r �pager? �list� next_cursor�has_moreF)r r* rO rP r+ )r, r- r. r r r/ r) r0 �appendr �items�
PropertyValuer r
)r rK r% r' r9 r& r; r1 r2 r3 r* �itemr? �keyr s r �search�NotionClient.searchx sn � � � �"�
�� � #��N� � ��L� �#/� � ��$�$�&�&�&� #�[�[��=�=�/��)� � � �� )� � �H�
� %� %� '��=�=�?�D� �G� ����B�/����>�Z�/��N�N�8�#3�d�#3�4��(�^�v�-�!#�J�&*�h�h�|�R�&@�&F�&F�&H�
��*7�*@�%�*@�
�� 'I�)3� �&��N�N�4�<�$�<�0� 0� !� � � �H�H�]� 3� ���*�e�4� �- '�&�&�� '�&�&�&�sb �A F�E4�F�,E<�2E7�3A'E<� BE<�" F�.E:�/F�7E<�:F�<F�F � F�Fr )NNNr )N)� NNNr )�__name__�
__module__� __qualname__�__firstlineno__�__doc__�strr r r r5 r r r �intr< r rC �boolrI r
rV �__static_attributes__r+ r r r r s� � �5�
��
� >�d�8�n� >�6 ,0�04�&*� �
#��#� ��c�3�h� �(�#� ��T�#�s�(�^�,�-� #�
�s�m� #� �
#�
�c�3�h� �#�@ 48� +��+� ��c��N�+� �4��S�#�X��/�0� +�
� +�6 $(� +��+� ��c��N�+� �4�.� +�
� +�, �+/�)-�&*� �
/��/� ��c�3�h� �(�/� �t�C��H�~�&� /�
�s�m� /� �
/�
�/� /r r )
r] �os�typingr r r r r, �
models.notionr r r
r r+ r r �<module>re s& �� '� � ,� ,� � 8� 8�_� _r