mobile datawipe in csv report and pdf certificate

This commit is contained in:
Cayo Puigdefabregas 2023-06-12 17:38:32 +02:00
parent 2543f7f761
commit 6aa643c197
4 changed files with 116 additions and 7 deletions

View file

@ -47,6 +47,7 @@ from ereuse_devicehub.resources.device.models import (
Computer, Computer,
DataStorage, DataStorage,
Device, Device,
Mobile,
Placeholder, Placeholder,
) )
from ereuse_devicehub.resources.documents.device_row import ActionRow, DeviceRow from ereuse_devicehub.resources.documents.device_row import ActionRow, DeviceRow
@ -1198,6 +1199,10 @@ class ExportsView(View):
ac = device.last_erase_action ac = device.last_erase_action
if ac: if ac:
erasures.append(ac) erasures.append(ac)
elif isinstance(device, Mobile):
ac = device.last_erase_action
if ac:
erasures.append(ac)
return erasures return erasures
def get_costum_details(self, erasures): def get_costum_details(self, erasures):
@ -1241,15 +1246,19 @@ class ExportsView(View):
def get_server_erasure_hosts(self, erasures): def get_server_erasure_hosts(self, erasures):
erasures_host = [] erasures_host = []
erasures_mobile = []
erasures_on_server = [] erasures_on_server = []
for erase in erasures: for erase in erasures:
try: try:
if isinstance(erase.device, Mobile):
erasures_mobile.append(erase.device)
continue
if erase.parent.binding.kangaroo: if erase.parent.binding.kangaroo:
erasures_host.append(erase.parent) erasures_host.append(erase.parent)
erasures_on_server.append(erase) erasures_on_server.append(erase)
except Exception: except Exception:
pass pass
return erasures_host, erasures_on_server return erasures_host, erasures_on_server, erasures_mobile
def build_erasure_certificate(self): def build_erasure_certificate(self):
erasures = self.get_datastorages() erasures = self.get_datastorages()
@ -1261,9 +1270,10 @@ class ExportsView(View):
my_data, customer_details = self.get_costum_details(erasures) my_data, customer_details = self.get_costum_details(erasures)
a, b = self.get_server_erasure_hosts(erasures) a, b, c = self.get_server_erasure_hosts(erasures)
erasures_host, erasures_on_server = a, b erasures_host, erasures_on_server, erasures_mobile = a, b, c
erasures_host = set(erasures_host) erasures_host = set(erasures_host)
erasures_mobile = set(erasures_mobile)
result_success = 0 result_success = 0
result_failed = 0 result_failed = 0
@ -1278,7 +1288,8 @@ class ExportsView(View):
erasures_on_server = sorted(erasures_on_server, key=lambda x: x.end_time) erasures_on_server = sorted(erasures_on_server, key=lambda x: x.end_time)
erasures_normal = list(set(erasures) - set(erasures_on_server)) erasures_normal = list(set(erasures) - set(erasures_on_server))
erasures_normal = sorted(erasures_normal, key=lambda x: x.end_time) erasures_normal = sorted(erasures_normal, key=lambda x: x.end_time)
n_computers = len({x.parent for x in erasures} - erasures_host) n_computers = len({x.parent for x in erasures if x.parent} - erasures_host)
n_mobiles = len(erasures_mobile)
params = { params = {
'title': 'Device Sanitization', 'title': 'Device Sanitization',
@ -1289,10 +1300,12 @@ class ExportsView(View):
'software': software, 'software': software,
'my_data': my_data, 'my_data': my_data,
'n_computers': n_computers, 'n_computers': n_computers,
'n_mobiles': n_mobiles,
'result_success': result_success, 'result_success': result_success,
'result_failed': result_failed, 'result_failed': result_failed,
'customer_details': customer_details, 'customer_details': customer_details,
'erasure_hosts': erasures_host, 'erasure_hosts': erasures_host,
'erasure_mobiles': erasures_mobile,
'erasures_normal': erasures_normal, 'erasures_normal': erasures_normal,
} }
return flask.render_template('inventory/erasure.html', **params) return flask.render_template('inventory/erasure.html', **params)

View file

@ -1066,6 +1066,9 @@ class Device(Thing):
return lot return lot
return None return None
def is_mobile(self):
return isinstance(self, Mobile)
def __lt__(self, other): def __lt__(self, other):
return self.id < other.id return self.id < other.id
@ -1375,11 +1378,11 @@ class Computer(Device):
@property @property
def external_document_erasure(self): def external_document_erasure(self):
"""Returns the external ``DataStorage`` proof of erasure.""" """Returns the external ``DataStorage`` proof of erasure."""
from ereuse_devicehub.resources.action.models import DataWipe from ereuse_devicehub.resources.action.models import EraseDataWipe
urls = set() urls = set()
try: try:
ev = self.last_action_of(DataWipe) ev = self.last_action_of(EraseDataWipe)
urls.add(ev.document.url.to_text()) urls.add(ev.document.url.to_text())
except LookupError: except LookupError:
pass pass
@ -1490,6 +1493,48 @@ class Mobile(Device):
raise ValidationError('{} is not a valid meid.'.format(value)) raise ValidationError('{} is not a valid meid.'.format(value))
return value return value
@property
def last_erase_action(self):
erase_auto = None
erase_manual = None
if self.binding:
erase_auto = self.privacy
erase_manual = self.binding.device.privacy
if self.placeholder:
erase_manual = self.privacy
if self.placeholder.binding:
erase_auto = self.placeholder.binding.privacy
if erase_auto and erase_manual:
return (
erase_auto
if erase_auto.created > erase_manual.created
else erase_manual
)
if erase_manual:
return erase_manual
if erase_auto:
return erase_auto
return None
@property
def privacy(self):
"""Returns the privacy compliance state of the data storage.
This is, the last erasure performed to the data storage.
"""
from ereuse_devicehub.resources.action.models import EraseBasic
try:
ev = self.last_action_of(EraseBasic)
except LookupError:
ev = None
return ev
def get_size(self):
return self.data_storage_size
class Smartphone(Mobile): class Smartphone(Mobile):
pass pass

View file

@ -343,6 +343,7 @@ class DeviceRow(BaseDeviceRow):
if isinstance(device, d.Mobile): if isinstance(device, d.Mobile):
self['IMEI'] = device.imei or '' self['IMEI'] = device.imei or ''
self.get_erasure_datawipe_mobile(device)
def components(self): def components(self):
"""Function to get all components information of a device.""" """Function to get all components information of a device."""
@ -417,6 +418,24 @@ class DeviceRow(BaseDeviceRow):
self['{} {} Size (MB)'.format(ctype, i)] = none2str(component.size) self['{} {} Size (MB)'.format(ctype, i)] = none2str(component.size)
self['{} {} Speed (MHz)'.format(ctype, i)] = none2str(component.speed) self['{} {} Speed (MHz)'.format(ctype, i)] = none2str(component.speed)
def get_erasure_datawipe_mobile(self, device):
actions = sorted(device.actions)
erasures = [a for a in actions if a.type == 'EraseDataWipe']
erasure = erasures[-1] if erasures else None
if erasure:
self['Erasure DataStorage 1'] = none2str(device.chid)
serial_number = none2str(device.imei)
storage_size = none2str(device.data_storage_size)
self['Erasure DataStorage 1 Serial Number'] = serial_number
self['Erasure DataStorage 1 Size (MB)'] = storage_size
self['Erasure DataStorage 1 Software'] = erasure.document.software
self['Erasure DataStorage 1 Result'] = get_result(erasure)
self['Erasure DataStorage 1 Type'] = erasure.type
self['Erasure DataStorage 1 Date'] = format(erasure.document.date or '')
self['Erasure DataStorage 1 Certificate URL'] = (
erasure.document.url and erasure.document.url.to_text() or ''
)
def get_datastorage(self, ctype, i, component): def get_datastorage(self, ctype, i, component):
"""Particular fields for component DataStorage. """Particular fields for component DataStorage.
A DataStorage can be HardDrive or SolidStateDrive. A DataStorage can be HardDrive or SolidStateDrive.
@ -455,6 +474,10 @@ class DeviceRow(BaseDeviceRow):
self['Erasure {} {} Size (MB)'.format(ctype, i)] = none2str(component.size) self['Erasure {} {} Size (MB)'.format(ctype, i)] = none2str(component.size)
self['Erasure {} {} Software'.format(ctype, i)] = erasure.document.software self['Erasure {} {} Software'.format(ctype, i)] = erasure.document.software
self['Erasure {} {} Result'.format(ctype, i)] = get_result(erasure) self['Erasure {} {} Result'.format(ctype, i)] = get_result(erasure)
self['Erasure {} {} Type'.format(ctype, i)] = erasure.type
self['Erasure {} {} Date'.format(ctype, i)] = format(
erasure.document.date or ''
)
self['Erasure {} {} Certificate URL'.format(ctype, i)] = ( self['Erasure {} {} Certificate URL'.format(ctype, i)] = (
erasure.document.url and erasure.document.url.to_text() or '' erasure.document.url and erasure.document.url.to_text() or ''
) )

View file

@ -209,6 +209,16 @@
</td> </td>
</tr> </tr>
{% endif %} {% endif %}
{% if n_mobiles %}
<tr style="padding-top:5px;">
<td style="width:20%;">
<span>N&deg; of mobiles:</span>
</td>
<td style="width:80%;">
<span>{{ n_mobiles }}</span>
</td>
</tr>
{% endif %}
<tr style="padding-top:5px;"> <tr style="padding-top:5px;">
<td style="width:20%;"> <td style="width:20%;">
<span>N&deg; of data storage unit(s):</span> <span>N&deg; of data storage unit(s):</span>
@ -322,12 +332,20 @@
<tbody> <tbody>
{% for erasure in erasures %} {% for erasure in erasures %}
<tr style="border-bottom: 1px dashed #000;"> <tr style="border-bottom: 1px dashed #000;">
{% if erasure.device.is_mobile() %}
<td>
{{ (erasure.device.imei or '') }}
</td>
<td>
</td>
{% else %}
<td> <td>
{{ (erasure.device.serial_number or '').upper() }} {{ (erasure.device.serial_number or '').upper() }}
</td> </td>
<td> <td>
{{ (erasure.parent.serial_number or '').upper() }} {{ (erasure.parent.serial_number or '').upper() }}
</td> </td>
{% endif %}
<td> <td>
{{ erasure.get_public_name() }} {{ erasure.get_public_name() }}
</td> </td>
@ -351,12 +369,21 @@
<h3>Technical Details</h3> <h3>Technical Details</h3>
</div> </div>
{% endif %} {% endif %}
{% if erasure.device.is_mobile() %}
<h4>{{ (erasure.device.imei or '') }}</h4>
<dl>
<dt>Mobile Drive:</dt>
<dd>Model: {{ erasure.device.model }}</dd>
<dd>IMEI: {{ (erasure.device.imei or '') }}</dd>
<dd>DHID: {{ erasure.device.dhid }}</dd>
<dd>Size: {{ erasure.device.get_size() or '' }}</dd>
{% else %}
<h4>{{ (erasure.device.serial_number or '').upper() }}</h4> <h4>{{ (erasure.device.serial_number or '').upper() }}</h4>
<dl> <dl>
<dt>Storage Drive:</dt> <dt>Storage Drive:</dt>
<dd>Model: {{ erasure.device.model }}</dd> <dd>Model: {{ erasure.device.model }}</dd>
<dd>SN: {{ (erasure.device.serial_number or '').upper() }}</dd> <dd>SN: {{ (erasure.device.serial_number or '').upper() }}</dd>
<dd>Size: {{ erasure.device.get_size()}}</dd> <dd>Size: {{ erasure.device.get_size() or '' }}</dd>
{% if erasure.parent %} {% if erasure.parent %}
<br /> <br />
<dt>Computer Host:</dt> <dt>Computer Host:</dt>
@ -364,6 +391,7 @@
<dd>SN: {{ (erasure.parent.serial_number or '').upper() }}</dd> <dd>SN: {{ (erasure.parent.serial_number or '').upper() }}</dd>
<dd>DHID: {{ erasure.parent.dhid }}</dd> <dd>DHID: {{ erasure.parent.dhid }}</dd>
{% endif %} {% endif %}
{% endif %}
<br /> <br />
<dt>Erasure:</dt> <dt>Erasure:</dt>