devicehub-django/snapshot/parse.py

import os
import json
import shutil
import xapian
import hashlib

from datetime import datetime
from snapshot.xapian import search, index
from snapshot.models import Snapshot, Annotation
from device.models import Device
from utils.constants import ALGOS


class Build:
    def __init__(self, snapshot_json, user):
        self.json = snapshot_json
        self.uuid = self.json['uuid']
        self.user = user
        self.hid = None

        self.index()
        self.create_annotations()

    def index(self):
        snap = json.dumps(self.json)
        index(self.uuid, snap)

    def get_hid_14(self):
        device = self.json['device']
        manufacturer = device.get("manufacturer", '')
        model = device.get("model", '')
        chassis = device.get("chassis", '')
        serial_number = device.get("serialNumber", '')
        sku = device.get("sku", '')
        hid = f"{manufacturer}{model}{chassis}{serial_number}{sku}"
        return hashlib.sha3_256(hid.encode()).hexdigest()

    def create_annotations(self):
        algorithms = {
            'hidalgo1': self.get_hid_14(),
        }

        annotation = Annotation.objects.filter(
            owner=self.user,
            type=Annotation.Type.SYSTEM,
            key='hidalgo1',
            value = algorithms['hidalgo1']
        ).first()

        if annotation:
            device = annotation.device
        else:
            device = Device.objects.create(
                type=self.json["device"]["type"],
                manufacturer=self.json["device"]["manufacturer"],
                model=self.json["device"]["model"],
                owner=self.user
            )

        for k, v in algorithms.items():
            Annotation.objects.create(
                uuid=self.uuid,
                owner=self.user,
                device=device,
                type=Annotation.Type.SYSTEM,
                key=k,
                value=v
            )