refactor CsstResult (8ef667d1) · Commits · Wu Maochun / csst_common

csst_common/status.py

+13 −5

Original line number	Original line	Diff line number	Diff line
	@@ -10,6 +10,7 @@ Modified-History:
	2022-10-07, Bo Zhang, added Numpydoc docstring		2022-10-07, Bo Zhang, added Numpydoc docstring
	"""		"""
	from enum import IntEnum		from enum import IntEnum
			from typing import Optional


	class CsstStatus(IntEnum):		class CsstStatus(IntEnum):
	@@ -26,6 +27,7 @@ class CsstStatus(IntEnum):
	>>> return CsstStatus.PERFECT		>>> return CsstStatus.PERFECT
	>>> return CsstStatus.ERROR		>>> return CsstStatus.ERROR
	"""		"""

	PERFECT = 0		PERFECT = 0
	WARNING = 1		WARNING = 1
	ERROR = 2		ERROR = 2
	@@ -38,17 +40,23 @@ class CsstResult:
	--------		--------
	>>> CsstResult(CsstStatus.PERFECT, ["file1.fits", "file2.fits"])		>>> CsstResult(CsstStatus.PERFECT, ["file1.fits", "file2.fits"])
	"""		"""
	def __init__(self, status: CsstStatus = CsstStatus.ERROR, file_list=[], **additional_output):
			def __init__(
			self,
			status: CsstStatus = CsstStatus.ERROR,
			file_list: Optional = None,
			**output,
			):
			if file_list is None:
			file_list = []
	assert isinstance(status, CsstStatus)		assert isinstance(status, CsstStatus)
	assert type(file_list) is list		assert type(file_list) is list
	self.status = status		self.status = status
	self.file_list = file_list		self.file_list = file_list
	self.additional_output = additional_output		self.output = output

	def __repr__(self):		def __repr__(self):
	return f""" <CsstResult status={self.status}>		return f""" <CsstResult status={self.status}>
	- file_list: {self.file_list}		- file_list: {self.file_list}
	- addtional_output: {self.additional_output} \n		- output: {self.output} \n
	"""		"""