thevar1able/ClickHouse
1
0
Fork 0
mirror of https://github.com/ClickHouse/ClickHouse.git synced 2024-11-21 23:21:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

Support requests parameters in ClickHouseHelper.insert_file

Browse Source
This commit is contained in:
Mikhail f. Shiryaev 2024-06-03 20:52:06 +02:00
parent 6d7c6a47cc
commit e64a033098
No known key found for this signature in database
GPG Key ID: 4B02ED204C7D93F4
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tests/ci/clickhouse_helper.py
View File

@ -4,9 +4,10 @@ import json
import logging
import time
from pathlib import Path
from typing import Dict, List, Optional
from typing import Any, Dict, List, Optional
import requests
from get_robot_token import get_parameter_from_ssm
from pr_info import PRInfo
from report import TestResults
@ -40,6 +41,7 @@ class ClickHouseHelper:
query: str,
file: Path,
additional_options: Optional[Dict[str, str]] = None,
**kwargs: Any,
) -> None:
params = {
"query": query,
@ -52,7 +54,7 @@ class ClickHouseHelper:
with open(file, "rb") as data_fd:
ClickHouseHelper._insert_post(
url, params=params, data=data_fd, headers=auth
url, params=params, data=data_fd, headers=auth, **kwargs
)
@staticmethod