Commit 21923da5 authored by BO ZHANG's avatar BO ZHANG 🏀
Browse files

parse task

parent 9932a9ca

csst_common/pipeline.py

+11 −7
Original line number Diff line number Diff line
@@ -66,13 +66,6 @@ class Pipeline: 
    """



    def __init__(self, **env_vars: Any):

        # record start time

        self.t_start = Time.now()



        # set message

        self.msg = sys.argv[1]

        self.msg_dict = self.json2dict(self.msg)



        # set environment variables

        for k, v in env_vars.items():

            os.environ[k] = str(v)
@@ -119,6 +112,17 @@ class Pipeline: 
        # exit code

        self.EXIT_CODES = EXIT_CODES



        # record start time

        self.t_start = Time.now()

        self.logger.info(f"t_start = {self.t_start.isot}")

        # set message

        self.logger.info(f"sys.argv[1] = {sys.argv[1]}")

        if len(sys.argv) == 2:

            self.task = self.json2dict(sys.argv[1])

        else:

            self.task = {}

        self.logger.info(f"task = {self.task}")



    @staticmethod

    def dict2json(d: dict):

        """Convert `dict` to JSON format string."""