48 lines
1.3 KiB
Python
48 lines
1.3 KiB
Python
|
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()}"
|