Mirrored_Repos/InvenTree
1
0
Fork 0
mirror of https://github.com/inventree/InvenTree synced 2024-08-30 18:33:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

Catch potential file processing errors

Browse Source
This commit is contained in:
Oliver 2022-02-09 08:31:08 +11:00
parent 5376c5b022
commit 8fc2695873
1 changed files with 12 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
InvenTree/part/serializers.py
View File

@ -818,13 +818,22 @@ class BomExtractSerializer(serializers.Serializer):
raise serializers.ValidationError(_("File is too large")) raise serializers.ValidationError(_("File is too large"))
# Read file data into memory (bytes object) # Read file data into memory (bytes object)
try:
data = bom_file.read() data = bom_file.read()
except Exception as e:
raise serializers.ValidationError(str(e))
if ext in ['csv', 'tsv', 'xml']: if ext in ['csv', 'tsv', 'xml']:
try:
data = data.decode() data = data.decode()
except Exception as e:
raise serializers.ValidationError(str(e))
# Convert to a tablib dataset (we expect headers) # Convert to a tablib dataset (we expect headers)
try:
self.dataset = tablib.Dataset().load(data, ext, headers=True) self.dataset = tablib.Dataset().load(data, ext, headers=True)
except Exception as e:
raise serializers.ValidationError(str(e))
for header in self.REQUIRED_COLUMNS: for header in self.REQUIRED_COLUMNS: