from prefect import task, flow, get_run_logger
from prefect.task_runners import ConcurrentTaskRunner
from random import seed, randint
from time import perf_counter
import logging

logging.basicConfig(level=<http://logging.INFO|logging.INFO>)
_logger = logging.getLogger(__name__)

# seed random number generator
seed(1)

task_runner = ConcurrentTaskRunner()


@task(name="a")
def a():
    i = randint(0, 100)
    logger = get_run_logger()
    <http://logger.info|logger.info>(f"We got {i}")
    return i


@task(name="b")
def b():
    i = randint(0, 100)
    logger = get_run_logger()
    <http://logger.info|logger.info>(f"We got {i}")
    return i


@task(name="c")
def c(_a, _b):
    logger = get_run_logger()
    v = _a * _b
    <http://logger.info|logger.info>(f"{_a} * {_b} = {v}")
    return v


@flow(name="some-flow", task_runner=task_runner)
def test_flow():
    _a = a()
    _b = b()
    _c = c(_a, _b)


def no_flow():
    _a = randint(0, 100)
    _<http://logger.info|logger.info>(f"We got {_a}")
    _b = randint(0, 100)
    _<http://logger.info|logger.info>(f"We got {_b}")
    _c = _a * _b
    _<http://logger.info|logger.info>(f"{_a} * {_b} = {_c}")


if __name__ == "__main__":
    start = perf_counter()
    for cycle in range(100):
        test_flow()

    print(f"Flow took {perf_counter() - start}")