Commit 5d22de31 authored by BO ZHANG's avatar BO ZHANG 🏀
Browse files

add p.init_file and p.new_file

parent 5b03f95c

csst_common/pipeline.py

+7 −1
Original line number Original line Diff line number Diff line
@@ -5,7 +5,7 @@ import warnings 
from typing import Any

from typing import Any





from astropy import time

from astropy import time



from .file import File

from .ccds import CCDS

from .ccds import CCDS

from .dfs import DFS

from .dfs import DFS

from .logger import get_logger

from .logger import get_logger
@@ -105,6 +105,12 @@ class Pipeline: 
        # Reset warning filters

        # Reset warning filters

        warnings.resetwarnings()

        warnings.resetwarnings()





    def init_file(self, file_path):

        return File(file_path, new_dir=self.dir_output)



    def new_file(self, file_name="test.fits"):

        return os.path.join(self.dir_output, file_name)







# class ErrorTrace:

# class ErrorTrace:

#     """Write error trace to file."""

#     """Write error trace to file."""