80 lines
2.1 KiB
Python
80 lines
2.1 KiB
Python
|
|
||
|
|
# define a object type enum
|
||
|
|
from abc import ABC, abstractmethod
|
||
|
|
from enum import Enum
|
||
|
|
from .object_id import ObjectID
|
||
|
|
import hashlib
|
||
|
|
import json
|
||
|
|
import pickle
|
||
|
|
|
||
|
|
class KnowledgeObject(ABC):
|
||
|
|
def __init__(self, object_type: int, desc: dict = {}, body: dict = {}):
|
||
|
|
self.desc = desc
|
||
|
|
self.body = body
|
||
|
|
self.object_type = object_type
|
||
|
|
|
||
|
|
def get_object_type(self):
|
||
|
|
return self.object_type
|
||
|
|
|
||
|
|
def object_id(self) -> ObjectID:
|
||
|
|
return self.calculate_id()
|
||
|
|
|
||
|
|
def set_desc_with_key_value(self, key, value):
|
||
|
|
self.desc[key] = value
|
||
|
|
|
||
|
|
def get_desc_with_key(self, key):
|
||
|
|
return self.desc.get(key)
|
||
|
|
|
||
|
|
def get_desc(self) -> dict:
|
||
|
|
return self.desc
|
||
|
|
|
||
|
|
def set_body_with_key_value(self, key, value):
|
||
|
|
self.body[key] = value
|
||
|
|
|
||
|
|
def get_body_with_key(self, key):
|
||
|
|
return self.body.get(key)
|
||
|
|
|
||
|
|
def get_body(self) -> dict:
|
||
|
|
return self.body
|
||
|
|
|
||
|
|
def calculate_id(self):
|
||
|
|
# Convert the object_type and desc to string and compute the SHA256 hash
|
||
|
|
data = json.dumps({"object_type": self.object_type, "desc": self.desc})
|
||
|
|
sha256 = hashlib.sha256()
|
||
|
|
sha256.update(data.encode())
|
||
|
|
return ObjectID(sha256.digest())
|
||
|
|
|
||
|
|
def encode(self) -> bytes:
|
||
|
|
return pickle.dumps(self)
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def decode(data: bytes):
|
||
|
|
return pickle.loads(data)
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
# define a text chunk class
|
||
|
|
class TextChunkObject(KnowledgeObject): # pylint: disable=too-few-public-methods
|
||
|
|
def __init__(self, text: str):
|
||
|
|
super().__init__(ObjectType.TextChunk)
|
||
|
|
self.text = text
|
||
|
|
|
||
|
|
|
||
|
|
# define a image class
|
||
|
|
class ImageObject(KnowledgeObject): # pylint: disable=too-few-public-methods
|
||
|
|
def __init__(self, meta, path):
|
||
|
|
super().__init__(ObjectType.Image)
|
||
|
|
self.meta = meta
|
||
|
|
self.path = path
|
||
|
|
|
||
|
|
|
||
|
|
# define a email class
|
||
|
|
class EmailObject(KnowledgeObject): # pylint: disable=too-few-public-methods
|
||
|
|
def __init__(self, meta):
|
||
|
|
super().__init__(ObjectType.Email)
|
||
|
|
self.meta = meta
|
||
|
|
self.text = [ObjectID]
|
||
|
|
self.images = [ObjectID]
|
||
|
|
|
||
|
|
|