Skip to content

Commit 476583d

Browse files
committed
- Addressed Review comments.
1 parent 10ddf2f commit 476583d

7 files changed

Lines changed: 14 additions & 16 deletions

File tree

okta/api_client.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -599,9 +599,10 @@ def files_parameters(self, files: Dict[str, Union[str, bytes]]):
599599
try:
600600
from PIL import Image # Lazy import — Pillow is optional
601601
img = Image.open(io.BytesIO(filedata))
602+
img_format = img.format # Read format before verify() resets the object
602603
img.verify() # Verify it's actually a valid image
603604
format_to_ext = {'PNG': 'png', 'JPEG': 'jpg', 'GIF': 'gif'}
604-
ext = format_to_ext.get(img.format, 'png')
605+
ext = format_to_ext.get(img_format, 'png')
605606
filename = f"{k}.{ext}"
606607
mimetype = f"image/{ext if ext != 'jpg' else 'jpeg'}"
607608
pil_validated = True
@@ -626,7 +627,7 @@ def files_parameters(self, files: Dict[str, Union[str, bytes]]):
626627
mimetype = "application/octet-stream"
627628
else:
628629
raise ValueError("Unsupported file value")
629-
params.append(tuple([k, tuple([filename, filedata, mimetype])]))
630+
params.append((k, (filename, filedata, mimetype)))
630631
return params
631632

632633
def select_header_accept(self, accepts: List[str]) -> Optional[str]:

okta/http_client.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -135,11 +135,10 @@ async def send_request(self, request):
135135
filename = ""
136136
if isinstance(request["form"]["file"], str):
137137
filename = request["form"]["file"].split("/")[-1]
138-
data = aiohttp.FormData()
139-
with open(request["form"]["file"], "rb") as f:
140-
file_data = f.read()
141138
mimetype = mimetypes.guess_type(filename)[0] or "application/octet-stream"
142-
data.add_field("file", file_data, filename=filename, content_type=mimetype)
139+
data = aiohttp.FormData()
140+
file_handle = open(request["form"]["file"], "rb")
141+
data.add_field("file", file_handle, filename=filename, content_type=mimetype)
143142
params["data"] = data
144143
# Remove Content-Type header - aiohttp.FormData will set it with boundary
145144
params["headers"] = _remove_content_type_header(request_headers)

openapi/templates/api_client.mustache

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -607,9 +607,10 @@ class ApiClient:
607607
try:
608608
from PIL import Image # Lazy import — Pillow is optional
609609
img = Image.open(io.BytesIO(filedata))
610+
img_format = img.format # Read format before verify() resets the object
610611
img.verify() # Verify it's actually a valid image
611612
format_to_ext = {'PNG': 'png', 'JPEG': 'jpg', 'GIF': 'gif'}
612-
ext = format_to_ext.get(img.format, 'png')
613+
ext = format_to_ext.get(img_format, 'png')
613614
filename = f"{k}.{ext}"
614615
mimetype = f"image/{ext if ext != 'jpg' else 'jpeg'}"
615616
pil_validated = True
@@ -634,7 +635,7 @@ class ApiClient:
634635
mimetype = "application/octet-stream"
635636
else:
636637
raise ValueError("Unsupported file value")
637-
params.append(tuple([k, tuple([filename, filedata, mimetype])]))
638+
params.append((k, (filename, filedata, mimetype)))
638639
return params
639640

640641
def select_header_accept(self, accepts: List[str]) -> Optional[str]:

openapi/templates/okta/http_client.mustache

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -123,11 +123,10 @@ class HTTPClient:
123123
filename = ""
124124
if isinstance(request["form"]["file"], str):
125125
filename = request["form"]["file"].split("/")[-1]
126-
data = aiohttp.FormData()
127-
with open(request["form"]["file"], "rb") as f:
128-
file_data = f.read()
129126
mimetype = mimetypes.guess_type(filename)[0] or "application/octet-stream"
130-
data.add_field("file", file_data, filename=filename, content_type=mimetype)
127+
data = aiohttp.FormData()
128+
file_handle = open(request["form"]["file"], "rb")
129+
data.add_field("file", file_handle, filename=filename, content_type=mimetype)
131130
params["data"] = data
132131
# Remove Content-Type header - aiohttp.FormData will set it with boundary
133132
params["headers"] = _remove_content_type_header(request_headers)

openapi/templates/okta/request_executor.mustache

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -138,8 +138,6 @@ class RequestExecutor:
138138
oauth=False,
139139
keep_empty_params=False,
140140
):
141-
async def create_request(self, method: str, url: str, body: dict = None,
142-
headers: dict = None, form=None, oauth=False, keep_empty_params=False):
143141
"""
144142
Creates request for request executor's HTTP client.
145143

openapi/templates/setup.mustache

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,7 @@ setup(
7878
keywords=["OpenAPI", "OpenAPI-Generator", "Okta Admin Management"],
7979
install_requires=REQUIRES,
8080
extras_require={
81-
"images": ["pillow >= 9.0.0"],
81+
"images": ["pillow >= 9.0.0, < 12"],
8282
},
8383
packages=find_packages(exclude=["test", "tests"]),
8484
include_package_data=True,

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,7 @@ def get_version():
7878
keywords=["OpenAPI", "OpenAPI-Generator", "Okta Admin Management"],
7979
install_requires=REQUIRES,
8080
extras_require={
81-
"images": ["pillow >= 9.0.0"],
81+
"images": ["pillow >= 9.0.0, < 12"],
8282
},
8383
packages=find_packages(exclude=["test", "tests"]),
8484
include_package_data=True,

0 commit comments

Comments
 (0)