common.py

import copy
from dataclasses import dataclass, field
import json

def from_proto_model_list(clazz, records):
    return [clazz().from_proto_model(r) for r in records]

def from_dict_list(clazz, records):
    return [clazz().from_dict(r) for r in records]

def to_proto_model_list(protoObjClazz, records):
    return [r.to_proto_model(protoObjClazz) for r in records]

def default_field(obj):
    return field(default_factory=lambda: copy.copy(obj))

class Dict(dict):
    __setattr__ = dict.__setitem__
    __getattr__ = dict.__getitem__    
    def __init__(self, **args):
        super(Dict, self).__init__()
        self.update(args)

def dict_to_object(dictObj):
    if not isinstance(dictObj, dict):
        return dictObj
    inst=Dict()
    for k,v in dictObj.items():
        inst[k] = dict_to_object(v)
    return inst

@dataclass
class BaseModel:
    def from_dict(self, data = {}):
        for k in self.__dataclass_fields__.keys():
            self.__setattr__(k, data.get(k, None))
        return self

    def from_proto_model(self, record):
        if record is None:
            return None
        for k in self.__dataclass_fields__.keys():
            if k == 'header':
                if record.__getattribute__(k):
                    self.__setattr__(k, json.loads(record.__getattribute__(k)))
            else:
                self.__setattr__(k, record.__getattribute__(k))
        return self
    
    def to_proto_model(self, protoObjClazz):
        obj = protoObjClazz()
        for k in self.__dataclass_fields__.keys():
            v = self.__getattribute__(k)
            if v is not None:
                if isinstance(v, list) or isinstance(v, tuple):
                    obj.__getattribute__(k).extend(v)
                elif isinstance(v, dict):
                    if k == 'header':
                        obj.__setattr__(k, json.dumps(v))
                    else:
                        obj.__getattribute__(k).update(v)                    
                else:
                    obj.__setattr__(k, v)
        return obj

@dataclass
class Gaia3Record(BaseModel):
    SolutionId: int = -1
    Designation: str = ''
    SourceId: int = -1