from collections import namedtuple from dataclasses import dataclass import nanoid Metadata = namedtuple("Metadata", ["table", "fields"]) @dataclass class Field: """ Represents a single field in a Record. """ name: str value_type: type = str default: value_type | None = None unique: bool = False class Record(dict): """ Base type for a single database record. """ def __init__(self, raw_doc: dict = {}, doc_id: int = None, **params): # populate the metadata self._fields.append( # 1% collision rate at ~2M records Field("uid", default=nanoid.generate(size=8), unique=True) ) self._metadata = Metadata(table=self.__class__.__name__, fields={f.name: f for f in self._fields}) self.doc_id = doc_id super().__init__(dict({field.name: field.default for field in self._fields}, **raw_doc, **params)) def __setattr__(self, key, value): if key in self: self[key] = value super().__setattr__(key, value) def __getattr__(self, attr_name): if attr_name in self: return self.get(attr_name) return super().__getattr__(attr_name) def __repr__(self): return f"{self.__class__.__name__}[{self.doc_id}]: {self.items()}"