295 lines
11 KiB
Python
295 lines
11 KiB
Python
import csv
|
|
import datetime
|
|
|
|
from io import StringIO
|
|
from collections import OrderedDict
|
|
|
|
import marshmallow
|
|
from flask import current_app as app, render_template, request, make_response
|
|
from flask.json import jsonify
|
|
from flask_sqlalchemy import Pagination
|
|
from marshmallow import fields, fields as f, validate as v
|
|
from sqlalchemy.orm import aliased
|
|
from teal import query
|
|
from teal.cache import cache
|
|
from teal.resource import View
|
|
|
|
from ereuse_devicehub import auth
|
|
from ereuse_devicehub.db import db
|
|
from ereuse_devicehub.resources import search
|
|
# from ereuse_devicehub.resources.device.definitions import ComponentDef
|
|
from ereuse_devicehub.resources.device.models import Component, Computer, Device, Manufacturer, \
|
|
RamModule, Processor, DataStorage
|
|
from ereuse_devicehub.resources.device.search import DeviceSearch
|
|
from ereuse_devicehub.resources.event.models import Rate, Event
|
|
from ereuse_devicehub.resources.lot.models import Lot, LotDevice
|
|
from ereuse_devicehub.resources.tag.model import Tag
|
|
|
|
|
|
class OfType(f.Str):
|
|
def __init__(self, column: db.Column, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.column = column
|
|
|
|
def _deserialize(self, value, attr, data):
|
|
v = super()._deserialize(value, attr, data)
|
|
return self.column.in_(app.resources[v].subresources_types)
|
|
|
|
|
|
class RateQ(query.Query):
|
|
rating = query.Between(Rate.rating, f.Float())
|
|
appearance = query.Between(Rate.appearance, f.Float())
|
|
functionality = query.Between(Rate.functionality, f.Float())
|
|
|
|
|
|
class TagQ(query.Query):
|
|
id = query.Or(query.ILike(Tag.id), required=True)
|
|
org = query.ILike(Tag.org)
|
|
|
|
|
|
class LotQ(query.Query):
|
|
id = query.Or(query.QueryField(Lot.descendantsq, fields.UUID()))
|
|
|
|
|
|
class Filters(query.Query):
|
|
_parent = aliased(Computer)
|
|
_device_inside_lot = (Device.id == LotDevice.device_id) & (Lot.id == LotDevice.lot_id)
|
|
_component_inside_lot_through_parent = (Device.id == Component.id) \
|
|
& (Component.parent_id == _parent.id) \
|
|
& (_parent.id == LotDevice.device_id) \
|
|
& (Lot.id == LotDevice.lot_id)
|
|
|
|
type = query.Or(OfType(Device.type))
|
|
model = query.ILike(Device.model)
|
|
manufacturer = query.ILike(Device.manufacturer)
|
|
serialNumber = query.ILike(Device.serial_number)
|
|
rating = query.Join(Device.id == Rate.device_id, RateQ)
|
|
tag = query.Join(Device.id == Tag.device_id, TagQ)
|
|
lot = query.Join(_device_inside_lot | _component_inside_lot_through_parent, LotQ)
|
|
|
|
|
|
class Sorting(query.Sort):
|
|
id = query.SortField(Device.id)
|
|
created = query.SortField(Device.created)
|
|
|
|
|
|
class DeviceView(View):
|
|
class FindArgs(marshmallow.Schema):
|
|
search = f.Str()
|
|
filter = f.Nested(Filters, missing=[])
|
|
sort = f.Nested(Sorting, missing=[])
|
|
page = f.Integer(validate=v.Range(min=1), missing=1)
|
|
|
|
def get(self, id):
|
|
"""
|
|
Devices view
|
|
---
|
|
description: Gets a device or multiple devices.
|
|
parameters:
|
|
- name: id
|
|
type: integer
|
|
in: path
|
|
description: The identifier of the device.
|
|
responses:
|
|
200:
|
|
description: The device or devices.
|
|
"""
|
|
return super().get(id)
|
|
|
|
def one(self, id: int):
|
|
"""Gets one device."""
|
|
if not request.authorization:
|
|
return self.one_public(id)
|
|
else:
|
|
return self.one_private(id)
|
|
|
|
def one_public(self, id: int):
|
|
device = Device.query.filter_by(id=id).one()
|
|
return render_template('devices/layout.html', device=device)
|
|
|
|
@auth.Auth.requires_auth
|
|
def one_private(self, id: int):
|
|
device = Device.query.filter_by(id=id).one()
|
|
return self.schema.jsonify(device)
|
|
|
|
@auth.Auth.requires_auth
|
|
def find(self, args: dict):
|
|
"""Gets many devices."""
|
|
search_p = args.get('search', None)
|
|
query = Device.query
|
|
if search_p:
|
|
properties = DeviceSearch.properties
|
|
tags = DeviceSearch.tags
|
|
query = query.join(DeviceSearch).filter(
|
|
search.Search.match(properties, search_p) | search.Search.match(tags, search_p)
|
|
).order_by(
|
|
search.Search.rank(properties, search_p) + search.Search.rank(tags, search_p)
|
|
)
|
|
query = query.filter(*args['filter']).order_by(*args['sort'])
|
|
if 'text/csv' in request.accept_mimetypes:
|
|
return self.generate_post_csv(query)
|
|
else:
|
|
devices = query.paginate(page=args['page'], per_page=30) # type: Pagination
|
|
ret = {
|
|
'items': self.schema.dump(devices.items, many=True, nested=1),
|
|
# todo pagination should be in Header like github
|
|
# https://developer.github.com/v3/guides/traversing-with-pagination/
|
|
'pagination': {
|
|
'page': devices.page,
|
|
'perPage': devices.per_page,
|
|
'total': devices.total,
|
|
'previous': devices.prev_num,
|
|
'next': devices.next_num
|
|
},
|
|
'url': request.path
|
|
}
|
|
return jsonify(ret)
|
|
|
|
def generate_post_csv(self, query):
|
|
"""
|
|
Get device query and put information in csv format
|
|
:param query:
|
|
:return:
|
|
"""
|
|
data = StringIO()
|
|
cw = csv.writer(data)
|
|
first = True
|
|
for device in query:
|
|
d = DeviceRow(device)
|
|
if first:
|
|
cw.writerow(name for name in d.keys())
|
|
first = False
|
|
cw.writerow(v for v in d.values())
|
|
output = make_response(data.getvalue())
|
|
output.headers['Content-Disposition'] = 'attachment; filename=export.csv'
|
|
output.headers['Content-type'] = 'text/csv'
|
|
return output
|
|
|
|
def generate_erase_certificate(self, query):
|
|
data = StringIO()
|
|
cw = csv.writer(data)
|
|
first = True
|
|
for device in query:
|
|
d = DeviceRow(device)
|
|
if first:
|
|
cw.writerow(name for name in d.keys())
|
|
first = False
|
|
cw.writerow(v for v in d.values())
|
|
# cw = csv.DictWriter(d, fieldnames=keys)
|
|
output = make_response(data.getvalue())
|
|
output.headers['Content-Disposition'] = 'attachment; filename=export.csv'
|
|
output.headers['Content-type'] = 'text/csv'
|
|
return output
|
|
|
|
|
|
class EraseDataStorage(OrderedDict):
|
|
def __init__(self, device: Device) -> None:
|
|
super().__init__()
|
|
self.device = device
|
|
|
|
|
|
# General Information
|
|
self['Organization'] = device.org
|
|
self['Date report'] = datetime.time()
|
|
self['Erase Information'] = device.org + 'ha borrado los siguientes discos acorde a ..' + eraseType
|
|
# Devices information for row {TABLE}
|
|
self['Computer Serial Number'] = device.serial_number
|
|
self['Computer Tag'] = device.tags
|
|
self['DataStorage Serial Number'] = device.components.data_storage.serial_number
|
|
self['Erase Date'] = device.event.erase.event_date
|
|
self['Erase Status'] = device.event.erase.privacy
|
|
self['Erase Type'] = device.event.erase.type
|
|
# For each DataStorage
|
|
self['DataStorage Serial Number'] = device.components.data_storage.serial_number
|
|
self['DataStorage Model'] = device.components.data_storage.model
|
|
self['DataStorage Manufacturer'] = device.components.data_storage.manufacturer
|
|
self['DataStorage Size (MB)'] = device.data_storage_size
|
|
self['Erase Date'] = device.event.erase.event_date
|
|
self['Erase Status'] = device.components.data_storage.privacy
|
|
# Erase information
|
|
self['Tool used to erase'] = device.erase_tool
|
|
self['Steps'] = device.events.erase.steps
|
|
self['Elapsed time'] = device.events.erase.erase_time
|
|
self['Final clean with zeros'] = 'Yes|No'
|
|
# Optional more computer info
|
|
self['Computer Serial Number'] = device.serial_number
|
|
self['Computer Model'] = device.model
|
|
self['Computer Manufacturer'] = device.manufacturer
|
|
self['Computer Tag'] = device.tags
|
|
|
|
class DeviceRow(OrderedDict):
|
|
NUMS = {
|
|
Processor.t: 1
|
|
}
|
|
|
|
def __init__(self, device: Device) -> None:
|
|
super().__init__()
|
|
self.device = device
|
|
self['Type'] = device.t
|
|
if isinstance(device, Computer):
|
|
self['Chassis'] = device.chassis.
|
|
self['Tag 1'] = self['Tag 2'] = self['Tag 3'] = ''
|
|
for i, tag in zip(range(1, 3), device.tags):
|
|
self['Tag {}'.format(i)] = format(tag)
|
|
self['Serial Number'] = device.serial_number
|
|
self['Price'] = device.price
|
|
self['Model'] = device.model
|
|
self['Manufacturer'] = device.manufacturer
|
|
self['Registered in '] = device.created
|
|
if isinstance(device, Computer):
|
|
self['Processor'] = device.processor_model
|
|
self['RAM (GB)'] = device.ram_size
|
|
self['Size (MB)'] = device.data_storage_size
|
|
rate = device.rate
|
|
if rate:
|
|
self['Rate'] = rate.rating
|
|
self['Range'] = rate.rating_range
|
|
self['Processor Rate'] = rate.processor
|
|
self['Processor Range'] = rate.workbench.processor_range
|
|
self['RAM Rate'] = rate.ram
|
|
self['RAM Range'] = rate.workbench.ram_range
|
|
self['Data Storage Rate'] = rate.data_storage
|
|
self['Data Storage Range'] = rate.workbench.data_storage_range
|
|
if isinstance(device, Computer):
|
|
self.components()
|
|
|
|
def components(self):
|
|
assert isinstance(self.device, Computer)
|
|
for type in app.resources[Component.t].subresources_types: # type: str
|
|
max = self.NUMS.get(type, 4)
|
|
i = 1
|
|
for component in (r for r in self.device.components if r.type == type):
|
|
self.fill_component(type, i, component)
|
|
i += 1
|
|
if i >= max:
|
|
break
|
|
while i < max:
|
|
self.fill_component(type, i)
|
|
i += 1
|
|
|
|
def fill_component(self, type, i, component = None):
|
|
self['{} {} Serial Number'.format(type, i)] = component.serial_number if component else ''
|
|
if isinstance(component, DataStorage):
|
|
self['{} {} Compliance'.format()] = component.compliance
|
|
|
|
|
|
class ManufacturerView(View):
|
|
class FindArgs(marshmallow.Schema):
|
|
search = marshmallow.fields.Str(required=True,
|
|
# Disallow like operators
|
|
validate=lambda x: '%' not in x and '_' not in x)
|
|
|
|
@cache(datetime.timedelta(days=1))
|
|
def find(self, args: dict):
|
|
search = args['search']
|
|
manufacturers = Manufacturer.query \
|
|
.filter(Manufacturer.name.ilike(search + '%')) \
|
|
.paginate(page=1, per_page=6) # type: Pagination
|
|
return jsonify(
|
|
items=app.resources[Manufacturer.t].schema.dump(
|
|
manufacturers.items,
|
|
many=True,
|
|
nested=1
|
|
)
|
|
)
|