added a Description base class
This commit is contained in:
@ -1,4 +1,5 @@
|
||||
|
||||
from .desc import Description
|
||||
from .imgdesc import ImageDescription
|
||||
from .plotdesc import PlotDescription
|
||||
|
||||
|
8
grum/descs/desc.py
Normal file
8
grum/descs/desc.py
Normal 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}
|
||||
|
||||
|
||||
|
@ -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}
|
||||
|
||||
|
||||
|
||||
|
@ -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}
|
||||
|
||||
|
||||
|
||||
|
Reference in New Issue
Block a user