创建蓝图布局脚本 - AWS Glue

创建蓝图布局脚本

蓝图布局脚本必须包含在工作流中生成实体的函数。您可以随意命名此函数。AWS Glue 使用配置文件确定函数的完全限定名称。

您的布局函数执行以下操作:

  • (可选)实例化 Job 类来创建 Job 对象,并传递诸如 CommandRole 的参数。如果您使用 AWS Glue 控制台或 API 来创建任务,这些是您要指定的任务属性。

  • (可选)实例化 Crawler 类来创建 Crawler 对象,并传递名称、角色和目标参数。

  • 要指示对象(工作流实体)之间的依赖项,请将 DependsOnWaitForDependencies 等其他参数传递到 Job()Crawler()。此部分的后面将说明这些参数。

  • 实例化 Workflow 类以创建返回到 AWS Glue 的工作流对象,传递 Name 参数、Entities 参数和可选的 OnSchedule 参数。Entities 参数指定要包含在工作流中的所有任务和爬网程序。如需了解如何构建 Entities 对象,请参阅此部分后面的示例项目。

  • 返回 Workflow 对象。

有关 JobCrawlerWorkflow 类的定义,请参阅AWS Glue 蓝图类参考

该布局函数接受以下输入参数。

参数 描述
user_params 蓝图参数名称和值的 Python 词典。有关更多信息,请参阅 指定蓝图参数
system_params Python 词典包含两个属性:regionaccountId

这里是一个文件名为 Layout.py 的简单布局生成器脚本:

import argparse import sys import os import json from awsglue.blueprint.workflow import * from awsglue.blueprint.job import * from awsglue.blueprint.crawler import * def generate_layout(user_params, system_params): etl_job = Job(Name="{}_etl_job".format(user_params['WorkflowName']), Command={ "Name": "glueetl", "ScriptLocation": user_params['ScriptLocation'], "PythonVersion": "2" }, Role=user_params['PassRole']) post_process_job = Job(Name="{}_post_process".format(user_params['WorkflowName']), Command={ "Name": "pythonshell", "ScriptLocation": user_params['ScriptLocation'], "PythonVersion": "2" }, Role=user_params['PassRole'], DependsOn={ etl_job: "SUCCEEDED" }, WaitForDependencies="AND") sample_workflow = Workflow(Name=user_params['WorkflowName'], Entities=Entities(Jobs=[etl_job, post_process_job])) return sample_workflow

示例脚本会导入所需的蓝图库,并包含 generate_layout 函数,用于生成包含两个任务的工作流。这是一个非常简单的脚本。更复杂的脚本可以使用额外的逻辑和参数来生成具有许多任务和爬网程序(甚至可变数量的任务和爬网程序)的工作流。

使用 DependsOn 参数

DependsOn 参数是此实体对工作流中其他实体所具有的依赖关系的字典表示形式。格式如下。

DependsOn = {dependency1 : state, dependency2 : state, ...}

此字典中的密钥表示实体的对象引用,而不是名称,而值是对应于要监视的状态的字符串。AWS Glue 推断合适的触发器。有关有效状态,请参阅条件结构

例如,任务可能依赖于爬网程序的成功完成。如果您定义了名为 crawler2 的爬网程序对象,如下所示:

crawler2 = Crawler(Name="my_crawler", ...)

然后一个依赖于 crawler2 的对象将包含一个构造函数参数,例如:

DependsOn = {crawler2 : "SUCCEEDED"}

例如:

job1 = Job(Name="Job1", ..., DependsOn = {crawler2 : "SUCCEEDED", ...})

如果实体省略了 DependsOn,则该实体依赖于工作流启动触发器。

使用 WaitForDependencies 参数

WaitForDependencies 参数定义任务或爬网程序实体是应该等待它依赖的全部实体完成,还是等待任意实体完成。

允许的值为“AND”或”ANY“。

使用 OnSchedule 参数

Workflow 类构造函数的 OnSchedule 参数是 cron 表达式,用于定义工作流的启动触发器。

如果指定此参数,AWS Glue 创建一个带有相应计划的计划触发器。如果未指定启动触发器,则工作流的启动触发器是按需触发器。