python-teos/pisa/appointment.py

import json

from pisa.encrypted_blob import EncryptedBlob


# Basic appointment structure
class Appointment:
    # DISCUSS: 35-appointment-checks
    def __init__(self, locator, start_time, end_time, dispute_delta, encrypted_blob, cipher, hash_function):
        self.locator = locator
        self.start_time = start_time    # ToDo: #4-standardize-appointment-fields
        self.end_time = end_time    # ToDo: #4-standardize-appointment-fields
        self.dispute_delta = dispute_delta
        self.encrypted_blob = EncryptedBlob(encrypted_blob)
        self.cipher = cipher
        self.hash_function = hash_function
        self.triggered = False

    @classmethod
    def from_dict(cls, appointment_data):
        locator = appointment_data.get("locator")
        start_time = appointment_data.get("start_time")    # ToDo: #4-standardize-appointment-fields
        end_time = appointment_data.get("end_time")    # ToDo: #4-standardize-appointment-fields
        dispute_delta = appointment_data.get("dispute_delta")
        encrypted_blob_data = appointment_data.get("encrypted_blob")
        cipher = appointment_data.get("cipher")
        hash_function = appointment_data.get("hash_function")

        if all([locator, start_time, end_time, dispute_delta, encrypted_blob_data, cipher, hash_function]) is not None:
            appointment = cls(locator, start_time, end_time, dispute_delta, EncryptedBlob(encrypted_blob_data), cipher,
                              hash_function)

        else:
            raise ValueError("Wrong appointment data, some fields are missing")

        return appointment

    def to_dict(self):
        # ToDO: #3-improve-appointment-structure
        appointment = {"locator": self.locator, "start_time": self.start_time, "end_time": self.end_time,
                       "dispute_delta": self.dispute_delta, "encrypted_blob": self.encrypted_blob.data,
                       "cipher": self.cipher, "hash_function": self.hash_function}

        return appointment

    def to_json(self):
        return json.dumps(self.to_dict(), sort_keys=True, separators=(',', ':'))