Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@

import asyncio
import logging
import mimetypes
from pathlib import Path
from typing import TYPE_CHECKING, Any

Expand All @@ -16,6 +17,10 @@
# Configure logging
logger = logging.getLogger(__name__)

# Parquet MIME types are not included in mimetypes by default
mimetypes.add_type("application/vnd.apache.parquet", ".pqt")
mimetypes.add_type("application/vnd.apache.parquet", ".parquet")


class UploadLowLevelClient(LowLevelClientBase, WithRestClient):
"""Low-level client for file upload operations.
Expand Down Expand Up @@ -182,7 +187,6 @@ def _mime_and_content_type_from_path(path: Path) -> tuple[str, str | None, str |
Returns:
A tuple of (file_name, mime_type, content_encoding).
"""
import mimetypes

file_name = path.name
mime, encoding = mimetypes.guess_type(path)
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,16 @@
"""Tests for UploadLowLevelClient functionality."""

from pathlib import Path

from sift_client._internal.low_level_wrappers.upload import UploadLowLevelClient


class TestUploadLowLevelClient:
class TestMimeAndContentTypeFromPath:
def test_parquet_file_extension(self):
_, mime, _ = UploadLowLevelClient._mime_and_content_type_from_path(Path("data.parquet"))
assert mime == "application/vnd.apache.parquet"

def test_pqt_file_extension(self):
_, mime, _ = UploadLowLevelClient._mime_and_content_type_from_path(Path("data.pqt"))
assert mime == "application/vnd.apache.parquet"
Loading