def capture_file()

in artifacts/count_people/infer.py [0:0]

• 16 lines of code
• 7 McCabe index (conditional complexity)

def capture_file(src_file, timeout=1):
    start = time.time()

    # wait for timeout, then check for source file availability
    while True:
        time.sleep(1/max_frame_rate)
        if os.path.exists(src_file):
            break
        if time.time() > start + timeout:
            raise Exception(f"source {src_file} doesn't exist within timeout") 

    # capture the source file by renaming to a new file. Using ms as suffix to file to minimize any collisions
    ms_count = int((time.time() %1)*1000)
    path_parts = list(os.path.split(src_file))
    file_parts = path_parts[-1].split('.')

    file_parts[-2] += str(ms_count)
    path_parts[-1] = ".".join(file_parts)

    new_file = os.path.join(*path_parts)
    # rename is atomic, so if this call succeeds, you can use the new file
    os.rename(src_file, new_file)

    return new_file