Skip to content

Commit 7bdf29b

Browse files
committed
Filter files with extension in list_tar_file
1 parent 1f5f98c commit 7bdf29b

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

src/server/oasisapi/files/models.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,9 @@ def list_tar_file(RelatedFile):
3131
return None
3232
# Need to verify this is a tar file
3333
tarf = tarfile.open(fileobj=BytesIO(RelatedFile.read()), mode='r')
34-
return tarf.getnames()
34+
# check if file by checking extension
35+
files = [f for f in tarf.getnames() if os.path.splitext(f)[-1]]
36+
return files
3537

3638

3739
def random_file_name(instance, filename):

src/server/oasisapi/files/v1_api/views.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -91,8 +91,9 @@ def _handle_get_related_file(parent, field, request):
9191

9292
if list_files:
9393
files = list_tar_file(f)
94+
9495
# todo: change this to a proper response
95-
return JsonResponse({'files': files})
96+
return Response(files)
9697

9798
# Original Fallback method - Reutrn data 'as is'
9899
response = StreamingHttpResponse(_get_chunked_content(file_obj), content_type=f.content_type)

0 commit comments

Comments
 (0)