from prefect import task, Flow
from prefect.storage import Docker
from prefect.run_configs import ECSRun


@task(log_stdout=True)
def task_name():
    do task_logic
    print(value)
    return(value)

with Flow("Test-Rostering-Flow-3", storage=Docker(registry_url="url",
                                                  image_name="test",
                                                  base_image="image"
                                                  )) as flow:
    value = task_name()

networkConfiguration = {
    'awsvpcConfiguration': {
        'securityGroups': [
            'sg-xxxxxx',
        ]
    }
}
flow.run_config = ECSRun(
    task_role_arn="arn:aws:iam::xxxx:role/somethingTaskRole",
    run_task_kwargs=networkConfiguration,
    env={"SOME_VAR": "value"})
print(flow.run_config)
flow.run()
flow.register(project_name="STAGING/test-setup", labels=['staging'])