update resume input
Browse files- maintenance_flow.py +4 -0
maintenance_flow.py
CHANGED
|
@@ -2,6 +2,10 @@ from prefect import flow,get_run_logger, pause_flow_run, settings
|
|
| 2 |
from prefect.blocks.notifications import SlackWebhook
|
| 3 |
from prefect.context import get_run_context
|
| 4 |
from DB_utils import set_maintenance_status,generate_empty_well,insert_maintenance_log
|
|
|
|
|
|
|
|
|
|
|
|
|
| 5 |
|
| 6 |
@flow
|
| 7 |
def request_wells_maintenance(maintenance_type):
|
|
|
|
| 2 |
from prefect.blocks.notifications import SlackWebhook
|
| 3 |
from prefect.context import get_run_context
|
| 4 |
from DB_utils import set_maintenance_status,generate_empty_well,insert_maintenance_log
|
| 5 |
+
from prefect.input import RunInput
|
| 6 |
+
|
| 7 |
+
class UserInput(RunInput):
|
| 8 |
+
name: str
|
| 9 |
|
| 10 |
@flow
|
| 11 |
def request_wells_maintenance(maintenance_type):
|