Commit 93910a14 authored by BO ZHANG's avatar BO ZHANG 🏀
Browse files

verbose in query

parent e7aab276

csst_dag/dag/_dispatcher.py

+15 −6
Original line number Diff line number Diff line
@@ -159,9 +159,12 @@ class Dispatcher: 
        Find plan records.

        """

        # query

        qr = plan.find(**override_common_keys(PLAN_PARAMS, kwargs))

        prompt = "plan"

        qr_kwargs = override_common_keys(PLAN_PARAMS, kwargs)

        qr = plan.find(**qr_kwargs)

        assert qr.success, qr

        print(f"Found {len(qr.data)} plan records.")

        print(f">>> [{prompt}] query kwargs: {qr_kwargs}")

        print(f">>> [{prompt}] {len(qr.data)} records found.")

        # plan basis / obsid basis

        try:

            for _ in qr.data:
@@ -188,9 +191,12 @@ class Dispatcher: 
        Find level0 records.

        """

        # query

        qr = level0.find(**override_common_keys(LEVEL0_PARAMS, kwargs))

        prompt = "level0"

        qr_kwargs = override_common_keys(LEVEL0_PARAMS, kwargs)

        qr = level0.find(**qr_kwargs)

        assert qr.success, qr

        print(f"Found {len(qr.data)} level0 records.")

        print(f">>> [{prompt}] query kwargs: {qr_kwargs}")

        print(f">>> [{prompt}] {len(qr.data)} records found.")

        # data basis

        data_basis = extract_basis_table(

            qr.data,
@@ -204,9 +210,12 @@ class Dispatcher: 
        Find level1 records.

        """

        # query

        qr = level1.find(**override_common_keys(LEVEL1_PARAMS, kwargs))

        prompt = "level1"

        qr_kwargs = override_common_keys(LEVEL1_PARAMS, kwargs)

        qr = level1.find(**qr_kwargs)

        assert qr.success, qr

        print(f"Found {len(qr.data)} level1 records.")

        print(f">>> [{prompt}] query kwargs: {qr_kwargs}")

        print(f">>> [{prompt}] {len(qr.data)} records found.")

        # data basis

        data_basis = extract_basis_table(

            qr.data,