Python ループ処理させたい関数が複数あるときのテクニック

以下のように第一引数にループさせたい関数を、第二引数に前処理したい関数を指定するwrapper関数を作成して、マルチスレッドでtargetを指定する前にwrapして使ったりしています。

無限ループさせる例

import time

def wrap_loop_func(main_func: any, preprocessing_func: any = None):
    """関数をループさせる関数を返す"""
    def loop_func():
        if preprocessing_func:
            preprocessing_func()
        while True:
            main_func()
    return loop_func

def task():
    print('task')
    time.sleep(1)

def preprocessing():
    print('pre-processing has done.')
    

# loop_task = wrap_loop_func(task)
loop_task = wrap_loop_func(task, preprocessing)
loop_task()

実行結果

pre-processing has done.
task
task
task
.
..
...

終了条件と後処理をさせる例

上の処理だと味気ないので、終了条件と後処理も追加

import time

begin = time.time()

def wrap_loop_thread_task(main_func: any, exit_condition_func: any, preprocessing_func: any = None, postprocessing_func: any = None):
    """関数をループさせる関数を返す"""
    def loop_func():
        if preprocessing_func:
            preprocessing_func()
        while exit_condition_func():
            main_func()
        if postprocessing_func:
            postprocessing_func()
    return loop_func

def preprocessing():
    print('pre-processing has done.')

def postporcessing():
    print('post-processing has done.')

def exit_condition():
    """経過時間が３秒未満ならTrue"""
    return True if time.time() - begin < 3 else False

def task():
    print('task')
    time.sleep(1)

# loop_func = wrap_loop_thread_task(task,exit_condition)
loop_func = wrap_loop_thread_task(task,exit_condition,preprocessing,postporcessing)
loop_func()

実行結果

pre-processing has done.
task
task
task
post-processing has done.

終了条件が必要とは限らないので

第２引数がNoneの場合、lambda : Trueとする。（常にTrueを返す関数)

def wrap_loop_thread_task(main_func: any, exit_condition_func: any = None, preprocessing_func: any = None, postprocessing_func: any = None):
    """関数をループさせる関数を返す"""
    def loop_func():
        if preprocessing_func:
            preprocessing_func()
        if exit_condition_func:
            should_continue = exit_condition_func
        else:
            should_continue = lambda : True
        while should_continue():
            main_func()
        if postprocessing_func:
            postprocessing_func()
    return loop_func