added a Description base class

This commit is contained in:
2023-04-14 23:17:02 +02:00
parent 00b12d800f
commit 4e95c1f8fd
4 changed files with 14 additions and 10 deletions

View File

@ -1,4 +1,5 @@
from .desc import Description
from .imgdesc import ImageDescription
from .plotdesc import PlotDescription

8
grum/descs/desc.py Normal file
View File

@ -0,0 +1,8 @@
class Description:
def to_dict(self):
return {k: v for k, v in self.__dict__.items() if not k.startswith("_") and k != "name" and v is not None}

View File

@ -1,7 +1,8 @@
import numpy as np
from .desc import Description
class ImageDescription:
class ImageDescription(Description):
def __init__(self, name, title=None, xlabel=None, ylabel=None, image=None):
self.name = name
@ -41,8 +42,4 @@ class ImageDescription:
return res
def to_dict(self):
return {k: v for k, v in self.__dict__.items() if not k.startswith("_") and k != "name" and v is not None}

View File

@ -1,5 +1,7 @@
from .desc import Description
class PlotDescription:
class PlotDescription(Description):
def __init__(self, name, title=None, xlabel=None, ylabel=None, xs=None, ys=None):
self.name = name
@ -45,8 +47,4 @@ class PlotDescription:
return res
def to_dict(self):
return {k: v for k, v in self.__dict__.items() if not k.startswith("_") and k != "name" and v is not None}