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

update retry

parent b80a0ddd

csst_common/utils/_retry.py

+3 −1
Original line number Diff line number Diff line
@@ -17,6 +17,7 @@ def retry(func, *args, **kwargs): 
    else:

        n_try = 5

    # is DFS?

    # TODO: support DFS2

    is_dfs = func.__class__.__module__.startswith("csst_dfs_client")

    # start trials

    for attempt in range(n_try):
@@ -24,7 +25,8 @@ def retry(func, *args, **kwargs): 
        try:

            res = func(*args, **kwargs)

            if is_dfs:

                assert res["code"] == 0, res

                # TODO: support DFS2

                assert res.success, res

            return res

        except Exception as e:

            # print error messages