inventree/InvenTree
2
0
Fork 0
mirror of https://github.com/inventree/InvenTree.git synced 2025-06-17 04:25:42 +00:00
Code Activity

Validation of matching fields is working

Browse Source
This commit is contained in:
eeintech
2021-05-05 12:40:48 -04:00
parent b4342d6203
commit 9773fee50b
4 changed files with 121 additions and 62 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
InvenTree/common/files.py
View File

@ -43,27 +43,42 @@ class FileManager:
# Update headers
self.update_headers()
def process(self, file):
""" Process file """
@classmethod
def validate(cls, file):
""" Validate file extension and data """
self.data = None
cleaned_data = None
ext = os.path.splitext(file.name)[-1].lower()
if ext in ['.csv', '.tsv', ]:
# These file formats need string decoding
raw_data = file.read().decode('utf-8')
# Reset stream position to beginning of file
file.seek(0)
elif ext in ['.xls', '.xlsx']:
raw_data = file.read()
# Reset stream position to beginning of file
file.seek(0)
else:
raise ValidationError(_(f'Unsupported file format: {ext}'))
try:
self.data = tablib.Dataset().load(raw_data)
cleaned_data = tablib.Dataset().load(raw_data)
except tablib.UnsupportedFormat:
raise ValidationError(_(f'Error reading {self.name} file (invalid format)'))
raise ValidationError(_('Error reading file (invalid format)'))
except tablib.core.InvalidDimensions:
raise ValidationError(_(f'Error reading {self.name} file (incorrect dimension)'))
raise ValidationError(_('Error reading file (incorrect dimension)'))
except KeyError:
# TODO: Find fix for XLSX format as it keeps on returning a KeyError
raise ValidationError(_('Error reading file (data could be corrupted)'))
return cleaned_data
def process(self, file):
""" Process file """
self.data = self.__class__.validate(file)
def update_headers(self):
""" Update headers """
@ -74,7 +89,7 @@ class FileManager:
""" Setup headers depending on the file name """
if not self.name:
return False
return
if self.name == 'order':
self.REQUIRED_HEADERS = [
@ -94,8 +109,6 @@ class FileManager:
# Update headers
self.update_headers()
return True
def guess_header(self, header, threshold=80):
""" Try to match a header (from the file) to a list of known headers