File size: 2,825 Bytes
bcf367c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 |
# https://www.prefect.io/blog/unveiling-interactive-workflows
# https://prefecthq.github.io/prefect-slack/
# https://github.com/PrefectHQ/interactive_workflow_examples
import asyncio
from prefect import flow, get_run_logger, pause_flow_run, settings
from prefect.blocks.notifications import SlackWebhook
from prefect.context import get_run_context
from prefect.input import RunInput
slack_block = SlackWebhook.load("prefect-test")
class UserInput(RunInput):
github_username: str
comments: str = ""
flag_for_review: bool
@flow(name="sample-transfer") # persist result or not?
async def move_sample(
sample_link: str,
sample_name: str,
source_link: str,
source_name: str,
destination_link: str,
destination_name: str,
):
logger = get_run_logger()
message = f"Please move sample <{sample_link}|{sample_name}> from <{source_link}|{source_name}> to <{destination_link}|{destination_name}>." # noqa: E501
flow_run = get_run_context().flow_run
if flow_run and settings.PREFECT_UI_URL:
flow_run_url = (
f"{settings.PREFECT_UI_URL.value()}/flow-runs/flow-run/{flow_run.id}"
)
message += f"\n\nOpen the <{flow_run_url}|paused flow run>, click 'Resume ▶️', and follow the instructions." # noqa: E501
await slack_block.notify(message)
user_input = await pause_flow_run(
wait_for_input=UserInput.with_initial_data(
description="Please provide your GitHub username, any comments, and whether this sample transfer should be flagged for review.", # noqa: E501
# github_username="sgbaird",
# comments="",
flag_for_review=False,
),
timeout=300,
)
logger.info(f"🚀 Sample transfer initiated by {user_input.github_username}")
logger.info(f"💬 Comments: {user_input.comments}")
logger.info(f"🚩 Flagged for review: {user_input.flag_for_review}")
return user_input
if __name__ == "__main__":
sample_link = "https://pioreactor.com/en-ca/collections/accessories-and-parts/products/20ml-glass-vial-closed-cap-and-stir-bar?variant=42836211761208" # noqa: E501
sample_name = "sample_100d793c"
source_link = "https://ca.robotshop.com/products/dh-robotics-automated-screw-cap-decapper-intelligent-capping-machine" # noqa: E501
source_name = "Automated capper/decapper"
destination_link = "https://autotrickler.com/pages/autotrickler-v4"
destination_name = "Autotrickler v4"
user_input = asyncio.run(
move_sample(
sample_link=sample_link,
sample_name=sample_name,
source_link=source_link,
source_name=source_name,
destination_link=destination_link,
destination_name=destination_name,
) # type: ignore
)
1 + 1
|