from log_flow import log_flow
from prefect.client import OrionClient
from prefect.deployments import Deployment, run_deployment
from prefect.filesystems import RemoteFileSystem
from prefect.infrastructure import DockerContainer


def main():
    # Storage Block
    storage = RemoteFileSystem(
        basepath='<s3://yaaam>',
        settings={
            'key': 'minioadmin',
            'secret': 'minioadmin',
            'client_kwargs': {'endpoint_url': '<minio_ip>:9000'}
        },
    )
    storage.save('minio', overwrite=True)

    # Infrastructure Block
    infrastructure = DockerContainer(
        name='log-demo',
        image='prefect-demo',
        image_pull_policy='NEVER',
    )
    infrastructure.save('log-demo', overwrite=True)

    # Deployment
    deployment = Deployment.build_from_flow(
        flow=log_flow,
        name='log-simple-docker',
        parameters={'name': 'Marvin'},
        infra_overrides={
            'env': {
                'PREFECT_LOGGING_LEVEL': 'DEBUG',
            }, 
        },
        work_queue_name='test',
        infrastructure=infrastructure,
        storage=storage,
    )
    deployment.apply()
    
    print(run_deployment(
        'log-simple-docker',
        client=OrionClient(api='<http://localhost:4200>'),
    ))

if __name__ == "__main__":
    main()