Skip to content

Commit c84bebc

Browse files
fix: apply ruff formatting and fix import sorting
Co-Authored-By: Aaron <AJ> Steers <[email protected]>
1 parent 906124e commit c84bebc

File tree

1 file changed

+13
-9
lines changed

1 file changed

+13
-9
lines changed

airbyte_cdk/sources/file_based/file_types/unstructured_parser.py

Lines changed: 13 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
# Copyright (c) 2023 Airbyte, Inc., all rights reserved.
33
#
44
import logging
5+
import mimetypes
56
import os
67
import traceback
78
from datetime import datetime
@@ -10,7 +11,6 @@
1011

1112
import backoff
1213
import dpath
13-
import mimetypes
1414
import nltk
1515
import requests
1616
from unstructured.file_utils.filetype import FileType, detect_filetype
@@ -330,15 +330,16 @@ def _read_file_remotely(
330330

331331
data = self._params_to_dict(format.parameters, strategy)
332332

333-
mime_type = mimetypes.guess_type(f"file.{filetype.name.lower()}")[0] if filetype else "application/octet-stream"
334-
333+
mime_type = (
334+
mimetypes.guess_type(f"file.{filetype.name.lower()}")[0]
335+
if filetype
336+
else "application/octet-stream"
337+
)
338+
335339
files = cast(Any, {"files": ("filename", file_handle, mime_type)})
336-
340+
337341
response = requests.post(
338-
f"{format.api_url}/general/v0/general",
339-
headers=headers,
340-
data=data,
341-
files=files
342+
f"{format.api_url}/general/v0/general", headers=headers, data=data, files=files
342343
)
343344

344345
if response.status_code == 422:
@@ -407,7 +408,10 @@ def _get_filetype(self, file: IOBase, remote_file: RemoteFile) -> Optional[FileT
407408
"""
408409
if remote_file.mime_type:
409410
for file_type in FileType:
410-
if mimetypes.guess_type(f"file.{file_type.name.lower()}")[0] == remote_file.mime_type:
411+
if (
412+
mimetypes.guess_type(f"file.{file_type.name.lower()}")[0]
413+
== remote_file.mime_type
414+
):
411415
return file_type
412416

413417
# set name to none, otherwise unstructured will try to get the modified date from the local file system

0 commit comments

Comments
 (0)