polygres-cli 0.1.0__py3-none-any.whl → 0.1.1__py3-none-any.whl

polygres_cli/cli.py

@@ -45,7 +45,6 @@ UUID_LIKE_RE = re.compile(
 )
 SQL_IDENTIFIER_RE = re.compile(r"^[A-Za-z_][A-Za-z0-9_]*$")
 MIGRATION_NAME_RE = SQL_IDENTIFIER_RE
-MAX_CSV_UPLOAD_BYTES = 1024**3
 GRAPH_CONFIGURATION_KEYS = {
     "registered_tables",
     "registered_relationships",
@@ -992,19 +991,6 @@ def handle_text_configs_delete(ctx: Context, args: argparse.Namespace) -> int:
 
 def handle_import_csv(ctx: Context, args: argparse.Namespace) -> int:
     file_path = _readable_file(args.file)
-    try:
-        file_size = file_path.stat().st_size
-    except OSError as exc:
-        raise CliError(
-            "VALIDATION_ERROR", f"Unable to read file: {file_path}", exit_code=USAGE
-        ) from exc
-    if file_size > MAX_CSV_UPLOAD_BYTES:
-        raise CliError(
-            "IMPORT_LIMIT_EXCEEDED",
-            "CSV file exceeds the 1 GiB local upload limit.",
-            exit_code=USAGE,
-            details={"limit_bytes": MAX_CSV_UPLOAD_BYTES, "file_size_bytes": file_size},
-        )
     _validate_identifiers(args.schema, args.table)
     preview_fields = {
         "target_schema": args.schema,
@@ -1032,22 +1018,20 @@ def handle_import_csv(ctx: Context, args: argparse.Namespace) -> int:
     columns = preview_payload.get("columns")
     if not isinstance(columns, list):
         raise CliError("IMPORT_INVALID", "CSV preview response did not include columns.")
-    import_fields = {
+    import_fields: dict[str, object] = {
         "target_schema": args.schema,
         "target_table": args.table,
         "mode": args.mode,
         "encoding": str(preview_payload.get("encoding", args.encoding)),
         "delimiter": str(preview_payload.get("delimiter", args.delimiter or "")),
         "quote_char": str(preview_payload.get("quote_char", args.quote_char or '"')),
-        "has_header": (
-            "true" if preview_payload.get("has_header", not args.no_header) else "false"
-        ),
+        "has_header": bool(preview_payload.get("has_header", not args.no_header)),
     }
     effective_escape = preview_payload.get("escape_char", args.escape_char)
     if effective_escape is not None:
         import_fields["escape_char"] = str(effective_escape)
-    import_fields.update({"job_id": job_id, "columns": json.dumps(columns)})
-    started = ctx.client.start_csv_import(project_id, file_path, import_fields)
+    import_fields.update({"job_id": job_id, "columns": columns})
+    started = ctx.client.start_csv_import(project_id, import_fields)
     output = _import_output(started)
     if args.wait and output["import"].get("status") not in {"succeeded", "failed", "cancelled"}:
         import_id = output["import"].get("id")

polygres_cli/cli_client.py

@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-import json
+import json as jsonlib
 import time
 from collections.abc import Callable
 from email.utils import parsedate_to_datetime
@@ -13,7 +13,7 @@ import httpx
 from polygres_cli.cli_errors import AUTH, UNAVAILABLE, CliError, api_error_from_response
 from polygres_cli.cli_secrets import redact_string
 
-VERSION = "0.1.0"
+VERSION = "0.1.1"
 RETRY_STATUSES = {408, 429, 500, 502, 503, 504}
 HEAVY_REQUEST_TIMEOUT = 120.0
 
@@ -121,16 +121,13 @@ class CliControlPlaneClient:
             )
 
     def start_csv_import(
-        self, project_id: str, file: Path, fields: dict[str, str]
+        self, project_id: str, fields: dict[str, Any]
     ) -> dict[str, Any]:
-        with file.open("rb") as handle:
-            return self._multipart(
-                "POST",
-                f"/projects/{project_id}/imports/csv",
-                handle,
-                file.name,
-                fields,
-            )
+        return self._post(
+            f"/projects/{project_id}/imports/csv",
+            fields,
+            timeout=HEAVY_REQUEST_TIMEOUT,
+        )
 
     def list_imports(self, project_id: str) -> dict[str, Any]:
         return self._get(f"/projects/{project_id}/imports")
@@ -292,10 +289,12 @@ class CliControlPlaneClient:
             try:
                 request_kwargs: dict[str, Any] = {
                     "headers": headers,
-                    "json": json,
                     "data": data,
                     "files": files,
                 }
+                if json is not None:
+                    headers["Content-Type"] = "application/json"
+                    request_kwargs["content"] = jsonlib.dumps(json)
                 request_timeout = timeout
                 if remaining is not None:
                     request_timeout = min(request_timeout or self._timeout, remaining)
@@ -392,7 +391,7 @@ def _json_payload(response: httpx.Response) -> dict[str, Any]:
         return {}
     try:
         payload = response.json()
-    except json.JSONDecodeError:
+    except jsonlib.JSONDecodeError:
         return {}
     return payload if isinstance(payload, dict) else {}
 

polygres_cli/cli_errors.py

@@ -16,6 +16,7 @@ LOCAL_DEPENDENCY = 9
 
 HTTP_EXIT_CODES = {
     400: USAGE,
+    413: USAGE,
     401: AUTH,
     403: PERMISSION,
     404: NOT_FOUND,

{polygres_cli-0.1.0.dist-info → polygres_cli-0.1.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: polygres-cli
-Version: 0.1.0
+Version: 0.1.1
 Summary: Command-line interface for Polygres
 Project-URL: Homepage, https://polygres.com
 Project-URL: Documentation, https://docs.evokoa.com/polygres

polygres_cli-0.1.1.dist-info/RECORD

@@ -0,0 +1,13 @@
+polygres_cli/__init__.py,sha256=8Rp0AQILGzy9FXAxyB6P9xshB8dKxfYMjY8Iw0W7jlY,44
+polygres_cli/cli.py,sha256=xRQBz8cAujfmTebwxRKNboTjyi_dN7do95BUM4vd5HI,73912
+polygres_cli/cli_auth.py,sha256=2AXXNvBb3P1Nt96KVHwGAnLEo_IV4cfSkRx6tvRsVNU,3070
+polygres_cli/cli_client.py,sha256=kOAIqln4yRxIiT7CiIWjF_Q92WyYaoYZux0AkjHh4rM,15636
+polygres_cli/cli_config.py,sha256=cLxyXNhY07nrjxTsZt1ZH_L1H-HyRV6li4UIHMduR5o,2985
+polygres_cli/cli_errors.py,sha256=SL-y4pMqFH0Lx-6MkElQymAQYW-xuDlCoCI4qki61ZQ,2349
+polygres_cli/cli_output.py,sha256=8efqKYqKy4Uf1C_Uzkf1zVT-KVtdJ4SkCwvoise9wG0,1680
+polygres_cli/cli_secrets.py,sha256=q_NiebEIHcc09BZxNw6sVyJ86fF3dYkzrhgmbfviz5M,1962
+polygres_cli-0.1.1.dist-info/METADATA,sha256=jN4RvBenDzdQ8yqvKZqCSJYNBZZYiplesEBRv5JU2mI,3275
+polygres_cli-0.1.1.dist-info/WHEEL,sha256=lCkmxWfQsSc9CfIClYeavTdQeEX2toPqufh9gI35EQA,87
+polygres_cli-0.1.1.dist-info/entry_points.txt,sha256=j_tqtMTwo6q54nc_7ZAVArSHEbKAIcJXEr6VUptRcHE,51
+polygres_cli-0.1.1.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
+polygres_cli-0.1.1.dist-info/RECORD,,

polygres_cli-0.1.0.dist-info/RECORD

@@ -1,13 +0,0 @@
-polygres_cli/__init__.py,sha256=8Rp0AQILGzy9FXAxyB6P9xshB8dKxfYMjY8Iw0W7jlY,44
-polygres_cli/cli.py,sha256=TlKF8Epnu7d8lU8D4FqDrdQrw9nh-BgSNxyhEbIO1X0,74489
-polygres_cli/cli_auth.py,sha256=2AXXNvBb3P1Nt96KVHwGAnLEo_IV4cfSkRx6tvRsVNU,3070
-polygres_cli/cli_client.py,sha256=vFFShCRcfu1ilQKieA0gl1mPSpJhWwau2OE-8O1GsBk,15591
-polygres_cli/cli_config.py,sha256=cLxyXNhY07nrjxTsZt1ZH_L1H-HyRV6li4UIHMduR5o,2985
-polygres_cli/cli_errors.py,sha256=llCplR7X9blAWHrWOXUntHS3KqCcToJOTSR-U1076Cw,2333
-polygres_cli/cli_output.py,sha256=8efqKYqKy4Uf1C_Uzkf1zVT-KVtdJ4SkCwvoise9wG0,1680
-polygres_cli/cli_secrets.py,sha256=q_NiebEIHcc09BZxNw6sVyJ86fF3dYkzrhgmbfviz5M,1962
-polygres_cli-0.1.0.dist-info/METADATA,sha256=TyUr6DSbSfMrHbp2wzCwM1auVL1AjDA53BtNFPcObk4,3275
-polygres_cli-0.1.0.dist-info/WHEEL,sha256=lCkmxWfQsSc9CfIClYeavTdQeEX2toPqufh9gI35EQA,87
-polygres_cli-0.1.0.dist-info/entry_points.txt,sha256=j_tqtMTwo6q54nc_7ZAVArSHEbKAIcJXEr6VUptRcHE,51
-polygres_cli-0.1.0.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
-polygres_cli-0.1.0.dist-info/RECORD,,