U n/e@sZddlmZddlmZmZmZmZddlZddl m Z ddl m Z mZGdddZdS))Client) create_table insert_ddf insert_df drop_tableN)check clickhousec@sNeZdZddifddZdddZdddZdd d Zdd d Zdd dZdS) ClickHouseNTc Cs|dkst|tkr&tdt|||_tf|}z|jWn&tk rj}z|W5d}~XYnX||_||_| dd|_ | dd|_ dS)a_summary_ Args: connection (dict, required): _description_. Defaults to None. verbose (bool, optional): _description_. Defaults to True. config (config, optional): _description_. Defaults to {}. Raises: Exception: _description_ Exception: _description_ Exception: _description_ Exception: _description_ e: _description_ Nz/plase input `connection` or but got connection n_row_per_loopin_partition_per_block ) typedict Exception_verboser connectionconnect_client _connectionget_n_row_per_loop_n_partition_per_block)selfrverboseconfigclienter9/tmp/pip-unpacked-wheel-t2z2wkof/dsmlibrary/clickhouse.py__init__ s zClickHouse.__init__cCst|tjtjfkr&tdt||dkr6td|dkrFtd||jkrftd|d|jt|j|||\}}|dkr|jdkr|j rt d |d n||S) a_summary_ Args: df (_type_, optional): _description_. Defaults to None. tableName (_type_, optional): _description_. Defaults to None. partition_by (_type_, optional): _description_. Defaults to None. Raises: Exception: _description_ Exception: _description_ Exception: _description_ Exception: _description_ e: _description_ Returns: _type_: _description_ @Please input `df=`, but got NzPlease input `tableName`zPlease input `partition_by`zkey `z$` not found in columns, columns are F9ztable z already exists!) rpd DataFrameddrcolumnsrrcoderprint)rdf tableNameZ partition_bystatusrrrrget_or_createTable5s zClickHouse.get_or_createTablecCst|tjtjfkr&tdt||dkrBtdt|dt|tjkrjt|j|||j|jdn(t|tjkrt |j|||jdndSdS)a?_summary_ Args: df (_type_, optional): _description_. Defaults to None. tableName (_type_, optional): _description_. Defaults to None. Raises: Exception: _description_ Exception: _description_ Returns: _type_: _description_ r!Nz%Expect `tableName` type str, but got z please input `tableName` str)rZddfr*r r )rr)r*r zSome thing wrong!) rr#r$r%rrrrrr)rr)r*rrrwriteXs zClickHouse.writecCs&|dkrtdt||j|S)z_summary_ Args: sqlQuery (_type_, optional): _description_. Defaults to None. Raises: Exception: _description_ Returns: _type_: _description_ Nz#plese input `sqlQuery` str but got )rrrZquery_dataframe)rZsqlQueryrrrreadvs zClickHouse.readcCs,|dkrtdt|t|j||jdS)z_summary_ Args: tableName (_type_, optional): _description_. Defaults to None. Raises: Exception: _description_ Returns: _type_: _description_ Nz$plese input `tableName` str but got )r table_namer)rrrrrrr*rrr dropTables zClickHouse.dropTablecCs tj|dtdtj|j|dS)zl_summary_ Args: tableName (_type_, optional): _description_. Defaults to None. r*)variableZ variableNameZdtype)rr/)rZ check_typestrr truncaterr0rrr truncateTableszClickHouse.truncateTable)NNN)NN)N)N)N) __name__ __module__ __qualname__r r,r-r.r1r5rrrrr s , #   r )Zclickhouse_driverrZutils.clickhouserrrrZpandasr#Zdask.dataframeZ dataframer%utilsrr r rrrrs