devicehub-django/evidence/forms.py

140 lines
3.8 KiB
Python
Raw Normal View History

2024-07-30 17:38:04 +00:00
import json
2024-08-05 11:32:52 +00:00
import pandas as pd
2024-07-13 13:27:36 +00:00
2024-07-30 17:38:04 +00:00
from django import forms
2024-07-31 11:28:02 +00:00
from django.core.exceptions import ValidationError
2024-07-30 17:38:04 +00:00
from django.utils.translation import gettext_lazy as _
from utils.device import create_annotation, create_doc, create_index
2024-07-31 11:28:02 +00:00
from utils.forms import MultipleFileField
2024-08-05 11:32:52 +00:00
from device.models import Device
2024-07-30 17:38:04 +00:00
from evidence.parse import Build
2024-07-31 11:28:02 +00:00
from evidence.models import Annotation
2024-07-13 13:27:36 +00:00
class UploadForm(forms.Form):
2024-07-31 11:28:02 +00:00
evidence_file = MultipleFileField(label=_("File"))
2024-07-13 13:27:36 +00:00
def clean(self):
2024-07-31 11:28:02 +00:00
self.evidences = []
2024-07-30 17:38:04 +00:00
data = self.cleaned_data.get('evidence_file')
if not data:
return False
2024-07-31 11:28:02 +00:00
for f in data:
file_name = f.name
file_data = f.read()
if not file_name or not file_data:
return False
try:
file_json = json.loads(file_data)
Build(file_json, None, check=True)
exist_annotation = Annotation.objects.filter(
uuid=file_json['uuid']
).first()
if exist_annotation:
raise ValidationError("error: {} exist".format(file_name))
except Exception:
raise ValidationError("error in: {}".format(file_name))
self.evidences.append((file_name, file_json))
2024-07-30 17:38:04 +00:00
return True
def save(self, user, commit=True):
if not commit or not user:
return
2024-07-31 11:28:02 +00:00
for ev in self.evidences:
Build(ev[1], user)
class UserTagForm(forms.Form):
tag = forms.CharField(label=_("Tag"))
def __init__(self, *args, **kwargs):
self.uuid = kwargs.pop('uuid', None)
2024-09-19 16:22:17 +00:00
annotation = Annotation.objects.filter(
uuid=self.uuid,
type=Annotation.Type.SYSTEM,
key='CUSTOM_ID',
).first()
if annotation:
kwargs['initial'].update({'tag': annotation.value})
super().__init__(*args, **kwargs)
def clean(self):
data = self.cleaned_data.get('tag')
if not data:
return False
self.tag = data
return True
def save(self, user, commit=True):
if not commit:
return
Annotation.objects.create(
uuid=self.uuid,
2024-09-18 16:01:46 +00:00
owner=user.institution,
type=Annotation.Type.SYSTEM,
key='CUSTOM_ID',
value=self.tag
)
2024-08-05 11:32:52 +00:00
class ImportForm(forms.Form):
file_import = forms.FileField(label=_("File to import"))
def __init__(self, *args, **kwargs):
self.rows = []
self.properties = {}
self.user = kwargs.pop('user')
super().__init__(*args, **kwargs)
def clean_file_import(self):
data = self.cleaned_data["file_import"]
self.file_name = data.name
df = pd.read_excel(data)
df.fillna('', inplace=True)
data_pd = df.to_dict(orient='index')
if not data_pd or df.last_valid_index() is None:
self.exception(_("The file you try to import is empty!"))
for n in data_pd.keys():
if 'type' not in [x.lower() for x in data_pd[n]]:
raise ValidationError("You need a column with name 'type'")
for k, v in data_pd[n].items():
if k.lower() == "type":
if v not in Device.Types.values:
raise ValidationError("{} is not a valid device".format(v))
self.rows.append(data_pd[n])
return data
2024-08-05 11:32:52 +00:00
def save(self, commit=True):
table = []
for row in self.rows:
doc = create_doc(row)
annotation = create_annotation(doc, self.user)
table.append((doc, annotation))
2024-08-05 11:32:52 +00:00
if commit:
for doc, cred in table:
cred.save()
create_index(doc)
2024-08-05 11:32:52 +00:00
return table
return