o lq i\V@sBdZddlZddlZddlZddlmZddlmZddlm Z ddl m Z ddl m Z ddl m Z dd l mZdd l mZdd lmZdd lmZdd lmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlm Z ddlm!Z!e"e#Z$GdddeZ%dS)z. This module provides a client class for BTS. N) bce_v1_signer) BceBaseClient)BceClientError)bce_http_client)handler)http_content_types) http_headers) http_methods)bts)INVALID_ARGS_ERROR)create_instance_args_2_dict)CreateInstanceArgs)Row)Cell) QueryCell)BatchQueryRowArgs) QueryRowArgs)ScanArgs)create_table_args_2_dict)update_table_args_2_dict)_decode)_encodec@seZdZdZd)ddZd*ddZd)ddZd)d d Zd)d d Zd)d dZ d)ddZ d)ddZ d)ddZ d)ddZ d)ddZd)ddZd)ddZd)ddZd)dd Zd)d!d"Zd)d#d$Zd%d&Z   d+d'd(ZdS), BtsClientz BTS Client NcCst||dSN)r__init__)selfconfigrg/home/app/PaddleOCR-VL/.venv_paddleocr/lib/python3.10/site-packages/baidubce/services/bts/bts_client.pyr3szBtsClient.__init__cCsDtjd|}|durt}|jtj||tj|tdt j t j idS)ax create instance :param instance_name: instance name :type instance_name: string :param create_instance_args: arguments for create instance :type create_instance_args: CreateInstanceArgs :param config: None :type config: BceClientConfiguration :return: :rtype baidubce.bce_response.BceResponse /Ndefaultpathrbodyheaders) r URL_PREFIXr _send_requestr PUTjsondumpsr r CONTENT_TYPErJSON)r instance_nameZcreate_instance_argsrr#rrrcreate_instance7s   zBtsClient.create_instancecC tjd|}|jtj||dS)z drop instance :param instance_name: instance name :type instance_name: string :param config: None :type config: BceClientConfiguration :return: :rtype baidubce.bce_response.BceResponse rr#rr r&r'r DELETErr-rr#rrr drop_instanceL zBtsClient.drop_instancecCsd}|jtj||dS)z list instances :param config: None :type config: BceClientConfiguration :return: :rtype baidubce.bce_response.BceResponse s /v1/instancesr0)r'r GET)rrr#rrrlist_instances[s zBtsClient.list_instancescCr/)z show instance :param instance_name: instance name :type instance_name: string :param config: None :type config: BceClientConfiguration :return: :rtype baidubce.bce_response.BceResponse rr0r r&r'r r6r3rrr show_instancehr5zBtsClient.show_instancecC>tjd|d|}|jtj||tj|tdtj t j idS)a create table :param instance_name: instance name :type instance_name: string :param table_name: table name :type table_name: string :param create_table_args: arguments for create table :type create_table_args: CreateTableArgs :param config: None :type config: BceClientConfiguration :return: :rtype baidubce.bce_response.BceResponse r/table/r r") r r&r'r r(r)r*rrr+rr,)rr- table_nameZcreate_table_argsrr#rrr create_tablex    zBtsClient.create_tablecCr:)a update table :param instance_name: instance name :type instance_name: string :param table_name: table name :type table_name: string :param update_table_args: arguments for update table :type update_table_args: UpdateTableArgs :param config: None :type config: BceClientConfiguration :return: :rtype baidubce.bce_response.BceResponse rr;r r") r r&r'r r(r)r*rrr+rr,)rr-r<Zupdate_table_argsrr#rrr update_tabler>zBtsClient.update_tablecC(tjd|d|}|jtj||dS)a@ drop table :param instance_name: instance name :type instance_name: string :param table_name: table name :type table_name: string :param config: None :type config: BceClientConfiguration :return: :rtype baidubce.bce_response.BceResponse rr;r0r1rr-r<rr#rrr drop_tablezBtsClient.drop_tablecCr@)a@ show table :param instance_name: instance name :type instance_name: string :param table_name: table name :type table_name: string :param config: None :type config: BceClientConfiguration :return: :rtype baidubce.bce_response.BceResponse rr;r0r8rArrr show_tablerCzBtsClient.show_tablecCs$tjd|d}|jtj||dS)z list tables :param instance_name: instance name :type instance_name: string :param config: None :type config: BceClientConfiguration :return: :rtype baidubce.bce_response.BceResponse rs/tablesr0r8r3rrr list_tabless zBtsClient.list_tablesc Cs |dus |jdkrtt}t||t|jgd}z9|jD]3}t|tr;| }t|d|d<|d |q t|t rPt|d|d<|d |q ttWnt yc}z|d}~wwt |} tjd|d|d} |jtj| || tjtjid S) a put row :param instance_name: instance name :type instance_name: string :param table_name: table name :type table_name: string :param put_row_args: arguments for put row :type put_row_args: Row :param config: None :type config: BceClientConfiguration :return: :rtype baidubce.bce_response.BceResponse NrowkeycellsvaluerIrr;/rowr")rHrr _loggerdebugrrI isinstancerto_dictappenddict Exceptionr)r*r r&r'r r(rr+rr,) rr-r<Z put_row_argsrexrow_datacellZ cell_dictr$r#rrrput_rows8        zBtsClient.put_rowc Cs,|durtt}t||g}|jD]^}zNt|tr+|}|ddkr*ttnt|trAt |}| ddkr@ttnttt |d|d<| dgD] } t | d| d<qS| |Wqtyr}z|d}~wwtd|i} tjd|d|d } |jtj| || tjtjid S) a batch put row :param instance_name: instance name :type instance_name: string :param table_name: table name :type table_name: string :param batch_put_row_args: arguments for batch put row :type batch_put_row_args: BatchPutRowArgs :param config: None :type config: BceClientConfiguration :return: :rtype baidubce.bce_response.BceResponse NrHrFrIrJrowsrr;/rowsr")rr rLrMrWrNrrOrQcopydeepcopygetrrPrRr)r*r r&r'r r(rr+rr,) rr-r<Zbatch_put_row_argsrrS rows_datarowrTrUr$r#rrr batch_put_rows@        zBtsClient.batch_put_rowc Cs|dust|dr|jdkrtt}t||t|jgd}z/t|drO|jrO|jD]!}t|t r>|d | q-t|t rK|d |q-ttWnt y^}z|d}~wwt|}tjd|d|d} |jtj| ||tjtjid S) a delete row :param instance_name: instance name :type instance_name: string :param table_name: table name :type table_name: string :param delete_row_args: arguments for delete row :type delete_row_args: QueryRowArgs :param config: None :type config: BceClientConfiguration :return: :rtype baidubce.bce_response.BceResponse NrHrFrGrIrr;rKr")hasattrrHrr rLrMrrIrNrrPrOrQrRr)r*r r&r'r r2rr+rr,) rr-r<Zdelete_row_argsrrSZdelete_row_datarUr$r#rrr delete_row/s4       zBtsClient.delete_rowc Cs|dus t|ts ttg}z8|jD]2}t|tr|}n t|tr'|}nttd|vr<|drs8