andis/intern2020
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

clean utils

Browse Source
This commit is contained in:
www
2020-12-27 19:19:50 +00:00
parent 40d9fac807
commit 77205f6b3d
1 changed files with 6 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
misc/utils.py
View File

@@ -2,27 +2,18 @@ import yaml
from collections.abc import Mapping from collections.abc import Mapping
from dataclasses import asdict from dataclasses import asdict
class SaveFileObject(): class SaveFileObject():
#def __init__(self,d={},**kwargs):
# self.data=d
# self.data.update(kwargs)
#def asdict(self):
# return self.data
def asdict(self): def asdict(self):
return asdict(self) return asdict(self)
def update(self, new): def update(self, new):
if new is None: if new is None:
new=dict() new=dict()
for key, value in new.items(): for key, value in new.items():
if hasattr(self, key): if hasattr(self, key):
setattr(self, key, value) setattr(self, key, value)
#def __getattr__(self, key):
# return self.data.get(key,None)
#def __setattr__(self, key,value):
# if not key in ["data"]:
# self.data[key]=value
# else:
# super().__setattr__(key,value)
class SaveFileMapping(Mapping): class SaveFileMapping(Mapping):
__file =None __file =None
@@ -54,14 +45,12 @@ class SaveFileMapping(Mapping):
def __getitem__(self,key): def __getitem__(self,key):
return self.data.get(key,None) return self.data.get(key,None)
def __setitem__(self,key,data):
def __setitem__(self,key,data):
if not isinstance(data,SaveFileObject): if not isinstance(data,SaveFileObject):
raise TypeError("data elements should be derived from SaveFileObject") raise TypeError("data elements should be derived from SaveFileObject")
if not callable(getattr(data,"asdict",None)): if not callable(getattr(data,"asdict",None)):
raise TypeError("Dataelements need a method asdict that returns a dict this should be out of the box if you use dataclasses") raise TypeError("Dataelements need a method asdict that returns a dict this should be out of the box if you use dataclasses")
#if not callable(getattr(data.__class__,"from_dict",None)):
# raise TypeError("Dataelements need a method to_dict that returns a dict")
self.data[key]= data self.data[key]= data
return self.data[key] return self.data[key]