0ptr/nullptr/models/base.py

from copy import deepcopy
from dataclasses import dataclass
from nullptr.util import sg

@dataclass
class Base:
  identifier = 'symbol'
  symbol: str
  store: object
  
  def __init__(self, symbol, store):
    self.store = store
    self.symbol = symbol
  
  def __hash__(self):
    return hash((str(type(self)), self.symbol))
  
  def __eq__(self, other):
    return self.symbol == other.symbol and type(self) == type(other)
    
  def seta(self, attr,  d, name=None, interp=None):
    if name is None:
      name = attr
    val = sg(d, name)
    if val is not None:
      if interp is not None:
        val = interp(val)
      setattr(self, attr, val)
  
  def setlst(self, attr, d, name, member):
    val = sg(d, name)
    if val is not None:
      lst = [sg(x, member) for x in val]
      setattr(self, attr, lst)
    
  def __setattr__(self, name, value):
    if name not in ['symbol','store','__dict__']:
      self.store.dirty(self)
    super().__setattr__(name, value)
  
  def update(self, d):
    pass
  
  def load(self, d):
    self.__dict__ = d
    
  def dict(self):
    r = {}
    for k,v in self.__dict__.items():
      if k in ['store']:
        continue
      r[k] = deepcopy(v)
    return r
    
  def path(self):
    raise NotImplementedError('path')
  
  @classmethod
  def ext(self):
    raise NotImplementedError('extension')
  
  def type(self):
    return self.__class__.__name__
  
  def __str__(self):
    return self.f()
    
  def f(self, detail=1):
    r = self.symbol
    if detail > 1:
      r += '.' + self.ext()
    return r