Commit c4938e08 authored by BO ZHANG's avatar BO ZHANG 🏀
Browse files

auto identify node name

parent 19dc7050

csst_common/data_manager.py

+6 −5
Original line number Diff line number Diff line
@@ -306,13 +306,11 @@ class CsstMbiDataManager: 
        return dm



    @staticmethod

    def quick_dataset(hostname="dandelion", ver_sim="C5.2", dir_l1=".", exposure_id=100):

        """ quick dataset generator

    def quick_dataset(ver_sim="C5.2", dir_l1=".", exposure_id=100):

        """ quick dataset generator for tests on dandelion or PMO



        Parameters

        ----------

        hostname:

            {dandelion, ubuntu}

        ver_sim:

            {"C5.2"}

        dir_l1:
@@ -325,6 +323,9 @@ class CsstMbiDataManager: 
        CsstMbiDataManager

            the MBI data manager instance

        """

        # auto identify node name

        hostname = os.uname()[1]



        if hostname == "dandelion":

            # dandelion node

            dir_l0 = "/nfsdata/share/csst_simulation_data/Cycle-5-SimuData/multipleBandsImaging/" \
@@ -338,7 +339,7 @@ class CsstMbiDataManager: 
            path_aux = "/data/sim_data/MSC_0000100/ref/MSC_{}_*_{:02d}_combine.fits"

            dir_pcref = "/home/user/L1Pipeline/msc/gaia_dr3/"

        else:

            raise ValueError("@DM: invalid hostname! valid values are {dandelion, ubuntu}")

            raise ValueError("@DM: invalid hostname {}!".format(hostname))



        return CsstMbiDataManager(ver_sim=ver_sim, dir_l0=dir_l0, dir_l1=dir_l1, dir_pcref=dir_pcref, path_aux=path_aux)