Skip to content

Commit 4dfa03d

Browse files
rootalvarolopez
authored andcommitted
add original_filename in UploadedFile(), set defaults to None for compatibility
1 parent f6f6d30 commit 4dfa03d

File tree

1 file changed

+11
-2
lines changed

1 file changed

+11
-2
lines changed

deepaas/model/v2/wrapper.py

Lines changed: 11 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,8 @@
4040

4141
UploadedFile = collections.namedtuple("UploadedFile", ("name",
4242
"filename",
43-
"content_type"))
43+
"content_type",
44+
"original_filename"))
4445
"""Class to hold uploaded field metadata when passed to model's methods
4546
4647
.. py:attribute:: name
@@ -53,9 +54,16 @@
5354
5455
.. py:attribute:: content_type
5556
56-
Content-type of the uploaded file.
57+
Content-type of the uploaded file
58+
59+
.. py:attribute:: original_filename
60+
61+
Filename of the original file being uploaded.
5762
"""
5863

64+
# set defaults to None, mainly for compatibility (vkoz)
65+
UploadedFile.__new__.__defaults__ = (None, None, None, None)
66+
5967

6068
class ModelWrapper(object):
6169
"""Class that will wrap the loaded models before exposing them.
@@ -266,6 +274,7 @@ def predict(self, *args, **kwargs):
266274
name=val.name,
267275
filename=name,
268276
content_type=val.content_type,
277+
original_filename=val.filename
269278
)
270279
kwargs[key] = aux
271280
# FIXME(aloga); cleanup of tmpfile here

0 commit comments

Comments
 (0)