blob: 79646ff88edf2431fb47ac3468ae3ba7ce00fc84 [file] [log] [blame]
<!DOCTYPE html>
<!--[if IE 8]><html class="no-js lt-ie9" lang="en" > <![endif]-->
<!--[if gt IE 8]><!--> <html class="no-js" lang="en" > <!--<![endif]-->
<html xmlns="http://www.w3.org/1999/xhtml" lang="en">
<head>
<meta charset="utf-8" />
<meta name="viewport" content="width=device-width, initial-scale=1">
<title>airflow.providers.google.cloud.operators.dataproc &mdash; apache-airflow-providers-google Documentation</title>
<link rel="stylesheet" href="../../../../../../_static/_gen/css/main.min.css" type="text/css" />
<link rel="stylesheet" type="text/css" href="../../../../../../_static/pygments.css" />
<link rel="stylesheet" type="text/css" href="../../../../../../_static/_gen/css/main.min.css" />
<link rel="stylesheet" type="text/css" href="../../../../../../_static/copybutton.css" />
<link rel="stylesheet" type="text/css" href="../../../../../../_static/_gen/css/main-custom.min.css" />
<link rel="stylesheet" type="text/css" href="../../../../../../_static/graphviz.css" />
<link rel="shortcut icon" href="../../../../../../_static/pin_32.png"/>
<link rel="index" title="Index" href="../../../../../../genindex.html" />
<link rel="search" title="Search" href="../../../../../../search.html" />
<!-- Matomo -->
<script>
var _paq = window._paq = window._paq || [];
/* tracker methods like "setCustomDimension" should be called before "trackPageView" */
/* We explicitly disable cookie tracking to avoid privacy issues */
_paq.push(['disableCookies']);
_paq.push(['trackPageView']);
_paq.push(['enableLinkTracking']);
(function() {
var u="https://analytics.apache.org/";
_paq.push(['setTrackerUrl', u+'matomo.php']);
_paq.push(['setSiteId', '13']);
var d=document, g=d.createElement('script'), s=d.getElementsByTagName('script')[0];
g.async=true; g.src=u+'matomo.js'; s.parentNode.insertBefore(g,s);
})();
</script>
<!-- End Matomo Code -->
</head><body class="td-section">
<header>
<nav class="js-navbar-scroll navbar">
<div class="navbar__icon-container">
<a href="/">
<svg xmlns="http://www.w3.org/2000/svg" width="155.314" height="60" viewBox="0 0 155.314 60">
<defs>
<clipPath id="clip-path">
<path id="Rectangle_1" d="M0 0h155.314v60H0z" fill="none" data-name="Rectangle 1"></path>
</clipPath>
</defs>
<g id="logo" transform="translate(-1305 -780.355)">
<g id="Group_2" clip-path="url(#clip-path)" data-name="Group 2" transform="translate(1305 780.355)">
<g id="Group_1" data-name="Group 1" transform="translate(.486 .486)">
<path id="Path_1" d="M1307.562 880.867l28.187-28.893a.521.521 0 0 0 .063-.666c-1.714-2.393-4.877-2.808-6.049-4.416-3.472-4.763-4.353-7.459-5.845-7.292a.456.456 0 0 0-.271.143l-10.182 10.438c-5.858 6-6.7 19.225-6.852 30.3a.552.552 0 0 0 .949.386z" fill="#017cee" data-name="Path 1" transform="translate(-1306.613 -822.232)"></path>
<path id="Path_2" d="M1405.512 908.489l-28.893-28.189a.521.521 0 0 0-.667-.063c-2.393 1.715-2.808 4.877-4.416 6.049-4.763 3.472-7.459 4.353-7.292 5.845a.456.456 0 0 0 .143.27l10.438 10.182c6 5.858 19.225 6.7 30.3 6.852a.552.552 0 0 0 .387-.946z" fill="#00ad46" data-name="Path 2" transform="translate(-1346.876 -850.567)"></path>
<path id="Path_3" d="M1373.909 902.252c-3.28-3.2-4.8-9.53 1.486-22.583-10.219 4.567-13.8 10.57-12.039 12.289z" fill="#04d659" data-name="Path 3" transform="translate(-1345.96 -850.233)"></path>
<path id="Path_4" d="M1433.132 782.359l-28.186 28.893a.52.52 0 0 0-.063.666c1.715 2.393 4.876 2.808 6.049 4.416 3.472 4.763 4.354 7.459 5.845 7.292a.454.454 0 0 0 .271-.143l10.182-10.438c5.858-6 6.7-19.225 6.852-30.3a.553.553 0 0 0-.95-.386z" fill="#00c7d4" data-name="Path 4" transform="translate(-1375.21 -782.123)"></path>
<path id="Path_5" d="M1426.9 881.155c-3.2 3.28-9.53 4.8-22.584-1.486 4.567 10.219 10.57 13.8 12.289 12.039z" fill="#11e1ee" data-name="Path 5" transform="translate(-1374.875 -850.233)"></path>
<path id="Path_6" d="M1307 782.919l28.893 28.186a.521.521 0 0 0 .666.063c2.393-1.715 2.808-4.877 4.416-6.049 4.763-3.472 7.459-4.353 7.292-5.845a.459.459 0 0 0-.143-.271l-10.438-10.182c-6-5.858-19.225-6.7-30.3-6.852a.552.552 0 0 0-.386.95z" fill="#e43921" data-name="Path 6" transform="translate(-1306.766 -781.97)"></path>
<path id="Path_7" d="M1405.8 804.711c3.28 3.2 4.8 9.53-1.486 22.584 10.219-4.567 13.8-10.571 12.039-12.289z" fill-rule="evenodd" fill="#ff7557" data-name="Path 7" transform="translate(-1374.875 -797.859)"></path>
<path id="Path_8" d="M1329.355 849.266c3.2-3.28 9.53-4.8 22.584 1.486-4.567-10.219-10.57-13.8-12.289-12.039z" fill="#0cb6ff" data-name="Path 8" transform="translate(-1322.503 -821.316)"></path>
<circle id="Ellipse_1" cx="1.26" cy="1.26" r="1.26" fill="#4a4848" data-name="Ellipse 1" transform="translate(28.18 28.171)"></circle>
<path id="Path_9" d="M1527.558 827.347a.229.229 0 0 1-.223-.223.458.458 0 0 1 .011-.123l2.766-7.214a.346.346 0 0 1 .357-.245h.758a.348.348 0 0 1 .357.245l2.754 7.214.022.123a.228.228 0 0 1-.223.223h-.568a.288.288 0 0 1-.19-.056.352.352 0 0 1-.089-.134l-.613-1.583h-3.657l-.613 1.583a.317.317 0 0 1-.1.134.269.269 0 0 1-.178.056zm4.795-2.732l-1.505-3.958-1.505 3.958zm3.322 4.85a.258.258 0 0 1-.189-.078.241.241 0 0 1-.067-.178v-7.4a.241.241 0 0 1 .067-.178.258.258 0 0 1 .189-.078h.513a.268.268 0 0 1 .256.256v.49a2.118 2.118 0 0 1 1.828-.858 2.092 2.092 0 0 1 1.751.736 3.135 3.135 0 0 1 .636 1.9q.011.122.011.379t-.011.379a3.168 3.168 0 0 1-.636 1.9 2.111 2.111 0 0 1-1.751.736 2.154 2.154 0 0 1-1.806-.836v2.587a.241.241 0 0 1-.067.178.223.223 0 0 1-.179.078zm2.364-2.91a1.324 1.324 0 0 0 1.149-.491 2.266 2.266 0 0 0 .4-1.293q.011-.111.011-.323 0-2.107-1.562-2.107a1.365 1.365 0 0 0-1.159.513 2.111 2.111 0 0 0-.412 1.2l-.012.424.012.435a1.862 1.862 0 0 0 .424 1.149 1.4 1.4 0 0 0 1.148.493zm5.628.9a2.329 2.329 0 0 1-1.015-.223 1.94 1.94 0 0 1-.747-.6 1.487 1.487 0 0 1-.268-.859 1.459 1.459 0 0 1 .6-1.2 3.4 3.4 0 0 1 1.65-.624l1.661-.234v-.323q0-1.137-1.3-1.137a1.4 1.4 0 0 0-.8.212 1.376 1.376 0 0 0-.468.48.305.305 0 0 1-.089.145.18.18 0 0 1-.134.045h-.48a.23.23 0 0 1-.245-.245 1.17 1.17 0 0 1 .245-.6 1.931 1.931 0 0 1 .747-.591 2.7 2.7 0 0 1 1.238-.256 2.351 2.351 0 0 1 1.8.591 2.032 2.032 0 0 1 .547 1.45v3.613a.257.257 0 0 1-.078.19.24.24 0 0 1-.178.067h-.513a.233.233 0 0 1-.257-.256v-.479a1.923 1.923 0 0 1-.714.6 2.557 2.557 0 0 1-1.203.237zm.234-.836a1.579 1.579 0 0 0 1.182-.469 1.881 1.881 0 0 0 .468-1.371v-.312l-1.293.19a2.918 2.918 0 0 0-1.193.379.761.761 0 0 0-.4.658.784.784 0 0 0 .368.691 1.585 1.585 0 0 0 .867.237zm6.643.836a2.556 2.556 0 0 1-1.873-.669 2.738 2.738 0 0 1-.714-1.9l-.011-.446.011-.446a2.7 2.7 0 0 1 .714-1.885 2.531 2.531 0 0 1 1.873-.68 2.917 2.917 0 0 1 1.36.29 2.077 2.077 0 0 1 .825.714 1.7 1.7 0 0 1 .3.848.2.2 0 0 1-.067.178.281.281 0 0 1-.19.067h-.535a.265.265 0 0 1-.168-.045.458.458 0 0 1-.111-.178 1.428 1.428 0 0 0-.535-.758 1.516 1.516 0 0 0-.87-.234 1.45 1.45 0 0 0-1.1.435 1.952 1.952 0 0 0-.435 1.3l-.011.4.011.379a1.969 1.969 0 0 0 .435 1.316 1.446 1.446 0 0 0 1.1.424 1.577 1.577 0 0 0 .87-.223 1.493 1.493 0 0 0 .535-.769.458.458 0 0 1 .111-.178.228.228 0 0 1 .168-.056h.535a.258.258 0 0 1 .19.078.2.2 0 0 1 .067.178 1.75 1.75 0 0 1-.3.847 2.078 2.078 0 0 1-.825.714 2.876 2.876 0 0 1-1.361.302zm4.078-.112a.233.233 0 0 1-.257-.256v-7.4a.241.241 0 0 1 .067-.178.259.259 0 0 1 .19-.078h.557a.267.267 0 0 1 .257.256v2.6a2.167 2.167 0 0 1 .758-.624 2.353 2.353 0 0 1 1.082-.223 2.067 2.067 0 0 1 1.661.691 2.642 2.642 0 0 1 .6 1.818v3.144a.257.257 0 0 1-.078.19.24.24 0 0 1-.178.067h-.557a.233.233 0 0 1-.256-.256V824a1.775 1.775 0 0 0-.39-1.227 1.387 1.387 0 0 0-1.1-.435 1.481 1.481 0 0 0-1.126.446 1.7 1.7 0 0 0-.412 1.215v3.088a.257.257 0 0 1-.078.19.24.24 0 0 1-.178.067zm8.846.112a2.466 2.466 0 0 1-1.84-.7 2.938 2.938 0 0 1-.747-1.94l-.011-.379.011-.368a2.953 2.953 0 0 1 .758-1.918 2.7 2.7 0 0 1 3.735.078 3.114 3.114 0 0 1 .68 2.119v.19a.257.257 0 0 1-.078.189.241.241 0 0 1-.178.067h-3.858v.1a2.11 2.11 0 0 0 .435 1.238 1.332 1.332 0 0 0 1.081.5 1.563 1.563 0 0 0 .836-.2 1.7 1.7 0 0 0 .491-.435.6.6 0 0 1 .145-.156.391.391 0 0 1 .19-.033h.547a.252.252 0 0 1 .167.056.192.192 0 0 1 .067.156.975.975 0 0 1-.312.591 2.51 2.51 0 0 1-.859.6 3.049 3.049 0 0 1-1.26.248zm1.527-3.434v-.033a1.817 1.817 0 0 0-.424-1.249 1.512 1.512 0 0 0-2.23 0 1.883 1.883 0 0 0-.4 1.249v.033z" fill="#51504f" data-name="Path 9" transform="translate(-1460.834 -808.144)"></path>
<path id="Path_10" d="M1527.2 827.081l-.061.061zm-.056-.279l-.08-.031zm2.766-7.214l.08.031zm1.472 0l-.081.029zm2.754 7.214l.084-.015a.064.064 0 0 0 0-.015zm.022.123h.086v-.015zm-.067.156l.06.061zm-.914.011l-.061.061.006.005zm-.089-.134l.081-.027zm-.613-1.583l.08-.031a.086.086 0 0 0-.08-.055zm-3.657 0v-.086a.086.086 0 0 0-.08.055zm-.613 1.583l-.08-.031zm-.1.134l.055.066zm4.047-2.676v.086a.086.086 0 0 0 .08-.116zm-1.505-3.958l.08-.03a.086.086 0 0 0-.16 0zm-1.505 3.958l-.08-.03a.086.086 0 0 0 .08.116zm-1.784 2.646a.128.128 0 0 1-.1-.042l-.122.121a.3.3 0 0 0 .217.092zm-.1-.042a.129.129 0 0 1-.042-.1h-.171a.3.3 0 0 0 .092.217zm-.042-.1a.38.38 0 0 1 .007-.1l-.163-.054a.514.514 0 0 0-.016.15zm.005-.092l2.765-7.214-.16-.061-2.765 7.214zm2.766-7.216a.283.283 0 0 1 .1-.143.3.3 0 0 1 .174-.046v-.172a.47.47 0 0 0-.271.076.453.453 0 0 0-.166.226zm.276-.189h.758v-.172h-.758zm.758 0a.3.3 0 0 1 .175.046.283.283 0 0 1 .1.143l.161-.059a.451.451 0 0 0-.166-.226.47.47 0 0 0-.272-.076zm.277.19l2.754 7.214.16-.061-2.754-7.214zm2.75 7.2l.022.123.169-.031-.022-.123zm.021.107a.13.13 0 0 1-.042.1l.121.121a.3.3 0 0 0 .092-.217zm-.042.1a.13.13 0 0 1-.1.042v.171a.3.3 0 0 0 .217-.092zm-.1.042h-.568v.171h.568zm-.568 0a.206.206 0 0 1-.135-.036l-.11.132a.373.373 0 0 0 .245.076zm-.129-.031a.262.262 0 0 1-.069-.1l-.162.054a.431.431 0 0 0 .11.167zm-.07-.1l-.613-1.584-.16.062.613 1.583zm-.693-1.638h-3.657v.171h3.657zm-3.737.055l-.614 1.584.16.062.613-1.583zm-.615 1.587a.235.235 0 0 1-.075.1l.111.13a.4.4 0 0 0 .126-.172zm-.074.1a.185.185 0 0 1-.124.036v.171a.354.354 0 0 0 .233-.076zm-.124.036h-.569v.171h.569zm4.306-2.677l-1.505-3.958-.16.061 1.505 3.958zm-1.666-3.958l-1.505 3.958.16.061 1.505-3.958zm-1.425 4.075h3.01v-.171h-3.01zm6.143 4.687l-.06.061zm0-7.761l.061.061zm.881 0l-.065.056.01.009zm.078.669h-.086a.086.086 0 0 0 .155.051zm3.579-.123l-.067.053zm.636 1.9h-.086zm0 .758l-.085-.007zm-.636 1.9l.067.054zm-3.557-.1l.068-.052a.086.086 0 0 0-.154.052zm-.067 2.765l-.061-.06zm2.787-3.323l-.069-.051zm.4-1.293l-.085-.008v.005zm-2.709-1.918l-.068-.052zm-.413 1.2h-.086zm-.011.423h-.085zm.011.435h-.086zm.424 1.149l.066-.054zm-1.216 3.315a.173.173 0 0 1-.129-.053l-.121.121a.342.342 0 0 0 .25.1zm-.129-.053a.157.157 0 0 1-.042-.118h-.172a.325.325 0 0 0 .092.239zm-.042-.118v-7.4h-.172v7.4zm0-7.4a.157.157 0 0 1 .042-.118l-.121-.121a.324.324 0 0 0-.092.239zm.042-.118a.173.173 0 0 1 .129-.053v-.172a.342.342 0 0 0-.25.1zm.129-.053h.513v-.172h-.513zm.513 0a.137.137 0 0 1 .113.048l.13-.111a.309.309 0 0 0-.244-.108zm.123.058a.137.137 0 0 1 .048.113h.171a.309.309 0 0 0-.108-.243zm.048.113v.49h.171v-.49zm.155.541a2.033 2.033 0 0 1 1.759-.823v-.171a2.2 2.2 0 0 0-1.9.894zm1.759-.823a2.007 2.007 0 0 1 1.683.7l.135-.106a2.177 2.177 0 0 0-1.818-.768zm1.683.7a3.045 3.045 0 0 1 .617 1.845l.171-.007a3.218 3.218 0 0 0-.654-1.946zm.617 1.85c.007.078.011.2.011.372h.171c0-.171 0-.3-.012-.387zm.011.372c0 .171 0 .294-.011.372l.17.015c.008-.086.012-.216.012-.387zm-.011.376a3.08 3.08 0 0 1-.617 1.846l.134.106a3.25 3.25 0 0 0 .654-1.945zm-.617 1.845a2.025 2.025 0 0 1-1.683.7v.171a2.2 2.2 0 0 0 1.817-.768zm-1.683.7a2.068 2.068 0 0 1-1.739-.8l-.136.1a2.239 2.239 0 0 0 1.874.87zm-1.892-.75v2.587h.172v-2.587zm0 2.587a.156.156 0 0 1-.042.118l.121.121a.325.325 0 0 0 .092-.239zm-.046.123a.138.138 0 0 1-.114.048v.172a.308.308 0 0 0 .244-.108zm-.114.048h-.546v.172h.546zm1.817-2.739a1.408 1.408 0 0 0 1.218-.526l-.138-.1a1.24 1.24 0 0 1-1.079.455zm1.217-.525a2.355 2.355 0 0 0 .419-1.341l-.171-.007a2.182 2.182 0 0 1-.385 1.246zm.418-1.336c.008-.079.012-.19.012-.332h-.172c0 .14 0 .245-.011.315zm.012-.332a2.726 2.726 0 0 0-.407-1.632 1.448 1.448 0 0 0-1.24-.562v.171a1.278 1.278 0 0 1 1.1.492 2.565 2.565 0 0 1 .374 1.53zm-1.647-2.193a1.452 1.452 0 0 0-1.228.547l.136.1a1.282 1.282 0 0 1 1.091-.479zm-1.228.547a2.2 2.2 0 0 0-.43 1.252l.172.008a2.028 2.028 0 0 1 .4-1.157zm-.43 1.254l-.011.424h.171l.011-.424zm-.011.428l.011.435h.172l-.011-.435zm.011.436a1.95 1.95 0 0 0 .443 1.2l.133-.109a1.776 1.776 0 0 1-.4-1.1zm.443 1.2a1.484 1.484 0 0 0 1.214.522v-.171a1.314 1.314 0 0 1-1.082-.459zm5.828 1.117l.037-.077zm-.747-.6l-.07.049zm.335-2.063l.052.068zm1.65-.624l.012.085zm1.661-.234l.012.085a.086.086 0 0 0 .074-.085zm-2.107-1.249l.046.072zm-.468.48l-.075-.042a.083.083 0 0 0-.006.015zm-.089.145l-.054-.067-.007.006zm-.792-.022l-.065.056.009.009zm-.067-.178h-.086zm.245-.6l-.07-.049zm.747-.591l.038.077zm3.033.334l-.063.058zm.468 5.252l.06.061zm-.881 0l-.065.056a.043.043 0 0 0 .009.009zm-.067-.669h.086a.086.086 0 0 0-.156-.048zm-.714.6l-.04-.076zm.223-1.059l-.062-.06zm.468-1.684h.086a.086.086 0 0 0-.1-.085zm-1.293.189l.012.085zm-1.193.379l.046.072zm-.033 1.349l-.047.071zm.635.985a2.241 2.241 0 0 1-.978-.215l-.074.155a2.412 2.412 0 0 0 1.051.231zm-.978-.215a1.859 1.859 0 0 1-.715-.576l-.138.1a2.024 2.024 0 0 0 .779.629zm-.713-.573a1.4 1.4 0 0 1-.253-.81h-.172a1.571 1.571 0 0 0 .283.907zm-.253-.81a1.374 1.374 0 0 1 .569-1.136l-.105-.135a1.544 1.544 0 0 0-.635 1.272zm.569-1.137a3.316 3.316 0 0 1 1.609-.607l-.024-.17a3.481 3.481 0 0 0-1.691.642zm1.609-.607l1.661-.234-.024-.17-1.662.234zm1.735-.319v-.323h-.171v.323zm0-.323a1.156 1.156 0 0 0-.355-.917 1.536 1.536 0 0 0-1.035-.306v.172a1.37 1.37 0 0 1 .922.263.986.986 0 0 1 .3.788zm-1.39-1.223a1.486 1.486 0 0 0-.851.227l.1.142a1.316 1.316 0 0 1 .755-.2zm-.849.226a1.452 1.452 0 0 0-.5.51l.15.084a1.286 1.286 0 0 1 .44-.449zm-.5.524a.226.226 0 0 1-.062.105l.107.134a.391.391 0 0 0 .117-.185zm-.068.112a.1.1 0 0 1-.073.019v.171a.266.266 0 0 0 .194-.07zm-.073.019h-.48v.171h.48zm-.48 0a.18.18 0 0 1-.122-.046l-.112.13a.352.352 0 0 0 .234.087zm-.113-.037a.18.18 0 0 1-.047-.123h-.171a.352.352 0 0 0 .087.234zm-.047-.119a1.1 1.1 0 0 1 .23-.557l-.14-.1a1.253 1.253 0 0 0-.261.648zm.23-.556a1.843 1.843 0 0 1 .715-.564l-.075-.154a2.018 2.018 0 0 0-.78.618zm.716-.564a2.611 2.611 0 0 1 1.2-.247v-.171a2.781 2.781 0 0 0-1.277.266zm1.2-.247a2.268 2.268 0 0 1 1.732.563l.126-.116a2.435 2.435 0 0 0-1.858-.618zm1.733.564a1.945 1.945 0 0 1 .523 1.391h.171a2.117 2.117 0 0 0-.57-1.508zm.523 1.391v3.613h.171v-3.613zm0 3.613a.172.172 0 0 1-.053.129l.121.121a.344.344 0 0 0 .1-.25zm-.053.129a.157.157 0 0 1-.118.042v.171a.326.326 0 0 0 .239-.092zm-.118.042h-.513v.171h.513zm-.513 0a.2.2 0 0 1-.134-.046l-.111.13a.367.367 0 0 0 .245.088zm-.124-.037a.194.194 0 0 1-.047-.134h-.171a.366.366 0 0 0 .087.245zm-.047-.134v-.479h-.171v.479zm-.156-.528a1.846 1.846 0 0 1-.683.575l.079.152a2.012 2.012 0 0 0 .745-.629zm-.683.575a2.476 2.476 0 0 1-1.153.236v.171a2.644 2.644 0 0 0 1.233-.255zm-.919-.429a1.666 1.666 0 0 0 1.244-.494l-.123-.12a1.493 1.493 0 0 1-1.121.442zm1.244-.494a1.969 1.969 0 0 0 .492-1.431h-.171a1.8 1.8 0 0 1-.444 1.312zm.492-1.431v-.312h-.171v.312zm-.1-.4l-1.293.189.025.17 1.293-.189zm-1.293.189a3 3 0 0 0-1.228.393l.095.143a2.837 2.837 0 0 1 1.158-.365zm-1.227.392a.845.845 0 0 0-.441.73h.172a.676.676 0 0 1 .362-.586zm-.441.73a.869.869 0 0 0 .406.762l.095-.142a.7.7 0 0 1-.33-.62zm.408.764a1.673 1.673 0 0 0 .916.247v-.171a1.5 1.5 0 0 1-.823-.221zm5.686.329l-.061.06zm-.714-1.9h-.085zm-.011-.446h-.085zm.011-.446h-.085zm.714-1.885l.061.061zm3.234-.39l-.04.076zm.825.713l-.073.046zm.3.848h-.086zm-.067.178l.056.065zm-.892.022l.054-.067zm-.112-.178l-.081.029zm-.535-.758l-.048.071zm-1.974.2l-.062-.059zm-.435 1.3h-.086zm-.011.4h-.086zm.011.379h-.086zm.435 1.316l-.062.059zm1.974.2l.046.072zm.535-.769l-.079-.033zm.112-.178l.054.067.007-.006zm.892.022l-.061.06zm.067.178l-.085-.009zm-.3.847l-.072-.046zm-.825.714l-.04-.076zm-1.36.2a2.471 2.471 0 0 1-1.814-.644l-.12.122a2.64 2.64 0 0 0 1.933.694zm-1.813-.643a2.653 2.653 0 0 1-.689-1.839l-.171.006a2.822 2.822 0 0 0 .738 1.952zm-.689-1.838l-.011-.446h-.171l.011.446zm-.011-.442l.011-.446h-.171l-.011.446zm.011-.445a2.611 2.611 0 0 1 .689-1.827l-.122-.121a2.78 2.78 0 0 0-.738 1.942zm.689-1.827a2.447 2.447 0 0 1 1.813-.655v-.171a2.617 2.617 0 0 0-1.934.705zm1.813-.655a2.836 2.836 0 0 1 1.32.28l.079-.152a3 3 0 0 0-1.4-.3zm1.32.28a1.99 1.99 0 0 1 .792.683l.145-.091a2.158 2.158 0 0 0-.858-.744zm.793.685a1.617 1.617 0 0 1 .287.8l.171-.009a1.789 1.789 0 0 0-.315-.89zm.287.809a.11.11 0 0 1-.037.1l.112.13a.281.281 0 0 0 .1-.252zm-.037.1a.2.2 0 0 1-.134.046v.171a.369.369 0 0 0 .246-.088zm-.134.046h-.535v.171h.535zm-.535 0a.184.184 0 0 1-.114-.026l-.107.134a.345.345 0 0 0 .221.064zm-.114-.026a.389.389 0 0 1-.086-.144l-.158.066a.533.533 0 0 0 .137.212zm-.084-.14a1.514 1.514 0 0 0-.57-.8l-.093.144a1.343 1.343 0 0 1 .5.715zm-.568-.8a1.6 1.6 0 0 0-.918-.249v.171a1.435 1.435 0 0 1 .822.219zm-.918-.249a1.535 1.535 0 0 0-1.166.462l.124.118a1.364 1.364 0 0 1 1.042-.408zm-1.166.462a2.036 2.036 0 0 0-.458 1.36l.171.006a1.872 1.872 0 0 1 .411-1.249zm-.458 1.361l-.011.4h.171l.011-.4zm-.011.406l.011.379.171-.005-.011-.379zm.011.38a2.052 2.052 0 0 0 .458 1.371l.124-.118a1.889 1.889 0 0 1-.411-1.26zm.458 1.371a1.533 1.533 0 0 0 1.166.451v-.172a1.363 1.363 0 0 1-1.042-.4zm1.166.451a1.661 1.661 0 0 0 .916-.237l-.093-.144a1.491 1.491 0 0 1-.823.209zm.918-.238a1.576 1.576 0 0 0 .568-.812l-.162-.057a1.409 1.409 0 0 1-.5.727zm.566-.807a.39.39 0 0 1 .086-.144l-.107-.134a.533.533 0 0 0-.137.213zm.093-.151a.144.144 0 0 1 .107-.031v-.171a.31.31 0 0 0-.228.081zm.107-.031h.535v-.171h-.535zm.535 0a.173.173 0 0 1 .129.053l.121-.121a.344.344 0 0 0-.25-.1zm.134.057a.11.11 0 0 1 .037.1l.17.017a.281.281 0 0 0-.1-.252zm.037.109a1.664 1.664 0 0 1-.288.806l.144.092a1.839 1.839 0 0 0 .315-.889zm-.288.806a1.989 1.989 0 0 1-.792.683l.079.152a2.162 2.162 0 0 0 .858-.744zm-.793.684a2.8 2.8 0 0 1-1.32.28v.171a2.96 2.96 0 0 0 1.4-.3zm2.568.187l-.065.056.01.009zm0-7.772l.061.06zm.926 0l-.065.056.009.009zm.078 2.776h-.085a.086.086 0 0 0 .153.053zm.758-.624l.038.077zm2.743.468l-.065.056zm.524 5.151l-.061-.06zm-.925 0l-.065.056.009.009zm-.457-4.5l-.065.056zm-2.23.011l-.062-.059zm-.49 4.493l-.061-.06zm-.736-.019a.2.2 0 0 1-.134-.046l-.112.13a.367.367 0 0 0 .245.088zm-.124-.037a.2.2 0 0 1-.046-.134h-.172a.367.367 0 0 0 .087.245zm-.046-.134v-7.4h-.172v7.4zm0-7.4a.156.156 0 0 1 .042-.118l-.121-.121a.326.326 0 0 0-.092.239zm.042-.118a.172.172 0 0 1 .129-.053v-.171a.343.343 0 0 0-.25.1zm.129-.053h.557v-.171h-.557zm.557 0a.137.137 0 0 1 .113.048l.13-.112a.308.308 0 0 0-.244-.108zm.122.057a.137.137 0 0 1 .048.113h.172a.309.309 0 0 0-.108-.243zm.048.113v2.6h.172v-2.6zm.153 2.651a2.076 2.076 0 0 1 .728-.6l-.075-.154a2.248 2.248 0 0 0-.788.649zm.73-.6a2.272 2.272 0 0 1 1.043-.214v-.171a2.438 2.438 0 0 0-1.121.232zm1.043-.214a1.982 1.982 0 0 1 1.6.661l.13-.112a2.152 2.152 0 0 0-1.727-.721zm1.6.663a2.557 2.557 0 0 1 .581 1.761h.171a2.727 2.727 0 0 0-.624-1.874zm.581 1.761v3.144h.171v-3.144zm0 3.144a.173.173 0 0 1-.053.129l.121.121a.345.345 0 0 0 .1-.25zm-.053.129a.156.156 0 0 1-.118.042v.171a.327.327 0 0 0 .239-.092zm-.118.042h-.557v.171h.557zm-.557 0a.2.2 0 0 1-.134-.046l-.112.13a.367.367 0 0 0 .245.088zm-.124-.037a.2.2 0 0 1-.046-.134h-.172a.368.368 0 0 0 .087.245zm-.046-.134V823.8h-.172v3.088zm0-3.088a1.859 1.859 0 0 0-.412-1.284l-.128.114a1.69 1.69 0 0 1 .368 1.169zm-.411-1.283a1.471 1.471 0 0 0-1.169-.464v.171a1.3 1.3 0 0 1 1.039.406zm-1.169-.464a1.566 1.566 0 0 0-1.188.473l.124.118a1.4 1.4 0 0 1 1.064-.419zm-1.188.473a1.779 1.779 0 0 0-.436 1.275h.172a1.609 1.609 0 0 1 .389-1.156zm-.436 1.275v3.088h.172V823.8zm0 3.088a.172.172 0 0 1-.053.129l.121.121a.344.344 0 0 0 .1-.25zm-.053.129a.156.156 0 0 1-.118.042v.171a.327.327 0 0 0 .239-.092zm-.118.042h-.557v.171h.557zm6.449-.505l-.062.059zm-.747-1.94h-.086zm-.012-.379h-.085v.005zm.012-.368l-.086-.006zm.758-1.918l-.061-.06zm3.735.078l-.065.056zm.6 2.5l.061.061zm-4.036.067v-.086a.086.086 0 0 0-.086.086zm0 .1h-.086zm.435 1.238l-.068.053zm1.918.3l.045.073zm.491-.435l-.069-.051zm.145-.156l.039.077h.006zm.9.022l-.055.066zm-.245.747l-.064-.057zm-.858.6l.035.078zm.267-3.189v.086a.086.086 0 0 0 .086-.086zm-.424-1.282l-.066.055zm-2.23 0l-.065-.055zm-.4 1.282h-.086a.086.086 0 0 0 .086.086zm1.528 3.349a2.38 2.38 0 0 1-1.779-.677l-.122.12a2.55 2.55 0 0 0 1.9.728zm-1.778-.676a2.86 2.86 0 0 1-.724-1.886l-.171.009a3.027 3.027 0 0 0 .771 1.995zm-.723-1.884l-.011-.379h-.171l.011.379zm-.011-.374l.011-.368-.172-.005-.011.368zm.011-.365a2.871 2.871 0 0 1 .735-1.864l-.124-.118a3.042 3.042 0 0 0-.782 1.971zm.734-1.864a2.331 2.331 0 0 1 1.756-.687v-.171a2.5 2.5 0 0 0-1.879.74zm1.756-.687a2.307 2.307 0 0 1 1.853.762l.13-.112a2.477 2.477 0 0 0-1.983-.821zm1.854.762a3.03 3.03 0 0 1 .659 2.062h.172a3.2 3.2 0 0 0-.7-2.175zm.659 2.062v.19h.172v-.19zm0 .19a.172.172 0 0 1-.053.129l.121.121a.345.345 0 0 0 .1-.25zm-.053.129a.156.156 0 0 1-.118.042v.171a.327.327 0 0 0 .239-.092zm-.118.042h-3.858v.171h3.858zm-3.944.086v.1h.172v-.1zm0 .1a2.2 2.2 0 0 0 .453 1.287l.135-.106a2.027 2.027 0 0 1-.417-1.189zm.454 1.288a1.418 1.418 0 0 0 1.148.533v-.171a1.247 1.247 0 0 1-1.015-.47zm1.148.533a1.647 1.647 0 0 0 .882-.214l-.09-.146a1.481 1.481 0 0 1-.791.188zm.882-.214a1.777 1.777 0 0 0 .515-.458l-.14-.1a1.613 1.613 0 0 1-.466.412zm.513-.456a1.251 1.251 0 0 1 .081-.1.28.28 0 0 1 .026-.025l.008-.006-.077-.153a.326.326 0 0 0-.083.068 1.55 1.55 0 0 0-.092.113zm.12-.134a.328.328 0 0 1 .146-.021v-.171a.468.468 0 0 0-.234.046zm.146-.021h.547v-.171h-.547zm.547 0a.166.166 0 0 1 .112.036l.11-.132a.337.337 0 0 0-.222-.076zm.112.036a.107.107 0 0 1 .036.09h.171a.277.277 0 0 0-.1-.222zm.036.09a.9.9 0 0 1-.291.534l.128.115a1.055 1.055 0 0 0 .334-.649zm-.291.535a2.42 2.42 0 0 1-.83.581l.072.156a2.6 2.6 0 0 0 .888-.624zm-.829.58a2.964 2.964 0 0 1-1.224.238v.171a3.133 3.133 0 0 0 1.295-.253zm.389-3.111v-.033h-.171v.033zm0-.033a1.9 1.9 0 0 0-.445-1.306l-.129.114a1.731 1.731 0 0 1 .4 1.192zm-.444-1.3a1.466 1.466 0 0 0-1.181-.521v.172a1.3 1.3 0 0 1 1.049.46zm-1.181-.521a1.466 1.466 0 0 0-1.18.521l.131.11a1.3 1.3 0 0 1 1.049-.46zm-1.181.521a1.965 1.965 0 0 0-.422 1.3h.172a1.794 1.794 0 0 1 .382-1.194zm-.422 1.3v.033h.172v-.033zm.086.119h3.055v-.171h-3.055z" fill="#51504f" data-name="Path 10" transform="translate(-1460.636 -807.945)"></path>
<path id="Path_11" d="M1519.066 884.011a.581.581 0 0 1-.567-.567 1.151 1.151 0 0 1 .028-.312l7.026-18.328a.881.881 0 0 1 .906-.623h1.926a.882.882 0 0 1 .907.623l7 18.328.057.312a.583.583 0 0 1-.567.567h-1.445a.735.735 0 0 1-.482-.142.9.9 0 0 1-.226-.34l-1.558-4.023h-9.292l-1.558 4.023a.8.8 0 0 1-.255.34.688.688 0 0 1-.453.142zm12.181-6.94l-3.824-10.056-3.823 10.055zm8.184-10.538a.592.592 0 0 1-.652-.651v-1.53a.714.714 0 0 1 .17-.482.656.656 0 0 1 .482-.2h1.785a.677.677 0 0 1 .68.68v1.53a.655.655 0 0 1-.2.481.713.713 0 0 1-.481.17zm.227 17.479a.593.593 0 0 1-.652-.652v-13.428a.611.611 0 0 1 .17-.453.656.656 0 0 1 .482-.2h1.359a.679.679 0 0 1 .652.651v13.427a.655.655 0 0 1-.2.482.613.613 0 0 1-.453.17zm6.861 0a.592.592 0 0 1-.651-.652v-13.4a.715.715 0 0 1 .17-.481.656.656 0 0 1 .482-.2h1.3a.677.677 0 0 1 .68.68v1.246a4.255 4.255 0 0 1 3.966-1.926h1.1a.679.679 0 0 1 .651.651v1.161a.566.566 0 0 1-.2.453.612.612 0 0 1-.453.17h-1.7a3.2 3.2 0 0 0-2.408.907 3.253 3.253 0 0 0-.879 2.408v8.328a.656.656 0 0 1-.2.482.716.716 0 0 1-.482.17zm12.234 0a.593.593 0 0 1-.651-.652v-11.814h-2.408a.592.592 0 0 1-.651-.651v-.963a.611.611 0 0 1 .17-.453.654.654 0 0 1 .481-.2h2.408v-1.417q0-4.816 4.872-4.815h1.586a.679.679 0 0 1 .652.651v.963a.656.656 0 0 1-.2.481.613.613 0 0 1-.453.17h-1.529a2.1 2.1 0 0 0-1.785.68 3.248 3.248 0 0 0-.51 2.011v1.275h6.062V863.7a.613.613 0 0 1 .17-.453.656.656 0 0 1 .482-.2h1.3a.679.679 0 0 1 .652.651v19.659a.655.655 0 0 1-.2.482.613.613 0 0 1-.454.17h-1.3a.592.592 0 0 1-.652-.652v-11.811h-6.062v11.813a.657.657 0 0 1-.2.482.614.614 0 0 1-.454.17zm20.9.283a6.487 6.487 0 0 1-4.844-1.757 6.837 6.837 0 0 1-1.813-4.674l-.029-1.218.029-1.218a6.732 6.732 0 0 1 1.841-4.646 7.389 7.389 0 0 1 9.631 0 6.736 6.736 0 0 1 1.841 4.646q.028.311.028 1.218t-.028 1.218a6.772 6.772 0 0 1-1.841 4.674 6.391 6.391 0 0 1-4.82 1.756zm0-2.181a3.582 3.582 0 0 0 2.8-1.133 4.931 4.931 0 0 0 1.133-3.258q.028-.283.028-1.076t-.028-1.076a4.931 4.931 0 0 0-1.133-3.258 3.582 3.582 0 0 0-2.8-1.133 3.671 3.671 0 0 0-2.833 1.133 4.83 4.83 0 0 0-1.1 3.258l-.028 1.076.028 1.076a4.83 4.83 0 0 0 1.1 3.258 3.671 3.671 0 0 0 2.828 1.132zm13.755 1.9a.846.846 0 0 1-.566-.17 1.321 1.321 0 0 1-.34-.538l-4.023-13.144-.056-.283a.575.575 0 0 1 .17-.425.641.641 0 0 1 .425-.17h1.246a.612.612 0 0 1 .453.17.646.646 0 0 1 .255.312l3.145 10.679 3.371-10.566a.761.761 0 0 1 .255-.4.726.726 0 0 1 .538-.2h.963a.728.728 0 0 1 .539.2.76.76 0 0 1 .255.4l3.371 10.566 3.144-10.679a.655.655 0 0 1 .2-.312.714.714 0 0 1 .482-.17h1.275a.542.542 0 0 1 .4.17.576.576 0 0 1 .17.425l-.057.283-3.994 13.144a1.323 1.323 0 0 1-.34.538.9.9 0 0 1-.6.17h-1.1a.86.86 0 0 1-.935-.708l-3.286-10.141-3.286 10.141a.928.928 0 0 1-.963.708z" fill="#51504f" data-name="Path 11" transform="translate(-1454.66 -838.62)"></path>
</g>
</g>
</g>
</svg>
</a>
</div>
<div class="desktop-only navbar__menu-container">
<div class="navbar__menu-content" id="main_navbar">
<div class="navbar__links-container">
<a class="navbar__text-link" href="/community/">
Community
</a>
<a class="navbar__text-link" href="/meetups/">
Meetups
</a>
<a class="navbar__text-link" href="/docs/">
Documentation
</a>
<a class="navbar__text-link" href="/use-cases/">
Use-cases
</a>
<a class="navbar__text-link" href="/announcements/">
Announcements
</a>
<a class="navbar__text-link" href="/blog/">
Blog
</a>
<a class="navbar__text-link" href="/ecosystem/">
Ecosystem
</a>
</div>
</div>
</div>
<div class="mobile-only navbar__drawer-container">
<button class="navbar__toggle-button" id="navbar-toggle-button">
<div id="hamburger-icon" class="navbar__toggle-button--icon visible">
<svg xmlns="http://www.w3.org/2000/svg" width="26" height="20" viewBox="0 0 26 20">
<g id="Group_1294" data-name="Group 1294" transform="translate(-38.791 291)">
<g id="Group_1291" data-name="Group 1291" transform="translate(39 -291)">
<rect id="Rectangle_461" width="26" height="2" fill="#51504f" data-name="Rectangle 461" rx="1" transform="translate(-.209)"></rect>
</g>
<g id="Group_1292" data-name="Group 1292" transform="translate(39 -281.822)">
<rect id="Rectangle_462" width="26" height="2" fill="#51504f" data-name="Rectangle 462" rx="1" transform="translate(-.209 -.178)"></rect>
</g>
<g id="Group_1293" data-name="Group 1293" transform="translate(39 -272.644)">
<rect id="Rectangle_463" width="26" height="2" fill="#51504f" data-name="Rectangle 463" rx="1" transform="translate(-.209 -.356)"></rect>
</g>
</g>
</svg>
</div>
<div id="close-icon" class="navbar__toggle-button--icon">
<svg xmlns="http://www.w3.org/2000/svg" width="19.799" height="19.799" viewBox="0 0 19.799 19.799">
<g id="Group_1574" data-name="Group 1574" transform="translate(-41.892 290.899)">
<g id="Group_1291" data-name="Group 1291" transform="rotate(-45 -308.114 -187.077)">
<rect id="Rectangle_461" width="26" height="2" fill="#51504f" data-name="Rectangle 461" rx="1" transform="translate(-.209)"></rect>
</g>
<g id="Group_1292" data-name="Group 1292" transform="rotate(45 372.48 -93.011)">
<rect id="Rectangle_462" width="26" height="2" fill="#51504f" data-name="Rectangle 462" rx="1" transform="translate(-.209 -.178)"></rect>
</g>
</g>
</svg>
</div>
</button>
<div class="navbar__drawer" id="navbar-drawer">
<div class="navbar__menu-content" id="main_navbar">
<div class="navbar__links-container">
<a class="navbar__text-link" href="/community/">
Community
</a>
<a class="navbar__text-link" href="/meetups/">
Meetups
</a>
<a class="navbar__text-link" href="/docs/">
Documentation
</a>
<a class="navbar__text-link" href="/use-cases/">
Use-cases
</a>
<a class="navbar__text-link" href="/announcements/">
Announcements
</a>
<a class="navbar__text-link" href="/blog/">
Blog
</a>
<a class="navbar__text-link" href="/ecosystem/">
Ecosystem
</a>
</div>
</div>
</div>
</div>
</nav>
</header>
<div class="roadmap container-fluid td-default base-layout">
<div class="content-drawer-wrapper">
<button class="content-drawer__toggle-button" id="content-open-button">
<div id="hamburger-icon" class="content-drawer__toggle-button--icon visible">
<svg xmlns="http://www.w3.org/2000/svg" width="20.005" height="13.879" viewBox="0 0 20.005 13.879">
<g id="Group_1619" data-name="Group 1619" transform="translate(271.132 -418.872)">
<g id="Group_1613" data-name="Group 1613" transform="translate(-266.229 431.045)">
<path id="Path_1337" d="M-218.5 513.215h-14.583a.259.259 0 0 1-.259-.26.259.259 0 0 1 .259-.259h14.583a.26.26 0 0 1 .259.259.259.259 0 0 1-.259.26z" fill="#51504f" data-name="Path 1337" transform="translate(233.345 -512.696)"></path>
</g>
<g id="Group_1614" data-name="Group 1614" transform="translate(-266.229 425.684)">
<path id="Path_1338" d="M-218.5 471.9h-14.583a.259.259 0 0 1-.259-.26.26.26 0 0 1 .259-.259h14.583a.26.26 0 0 1 .259.259.26.26 0 0 1-.259.26z" fill="#51504f" data-name="Path 1338" transform="translate(233.345 -471.376)"></path>
</g>
<g id="Group_1615" data-name="Group 1615" transform="translate(-266.229 420.323)">
<path id="Path_1339" d="M-218.5 430.574h-14.583a.259.259 0 0 1-.259-.259.259.259 0 0 1 .259-.259h14.583a.26.26 0 0 1 .259.259.259.259 0 0 1-.259.259z" fill="#51504f" data-name="Path 1339" transform="translate(233.345 -430.055)"></path>
</g>
<g id="Group_1616" data-name="Group 1616" transform="translate(-271.132 429.814)">
<path id="Path_1340" d="M-269.663 506.145a1.47 1.47 0 0 1-1.469-1.469 1.47 1.47 0 0 1 1.469-1.469 1.47 1.47 0 0 1 1.469 1.469 1.47 1.47 0 0 1-1.469 1.469zm0-2.418a.951.951 0 0 0-.95.95.951.951 0 0 0 .95.95.951.951 0 0 0 .95-.95.951.951 0 0 0-.95-.95z" fill="#51504f" data-name="Path 1340" transform="translate(271.132 -503.208)"></path>
</g>
<g id="Group_1617" data-name="Group 1617" transform="translate(-271.132 424.556)">
<path id="Path_1341" d="M-269.663 465.62a1.47 1.47 0 0 1-1.469-1.469 1.47 1.47 0 0 1 1.469-1.469 1.47 1.47 0 0 1 1.469 1.469 1.47 1.47 0 0 1-1.469 1.469zm0-2.418a.951.951 0 0 0-.95.95.951.951 0 0 0 .95.95.951.951 0 0 0 .95-.95.951.951 0 0 0-.95-.952z" fill="#51504f" data-name="Path 1341" transform="translate(271.132 -462.683)"></path>
</g>
<g id="Group_1618" data-name="Group 1618" transform="translate(-271.132 418.872)">
<path id="Path_1342" d="M-269.663 421.809a1.47 1.47 0 0 1-1.469-1.469 1.47 1.47 0 0 1 1.469-1.469 1.47 1.47 0 0 1 1.469 1.469 1.47 1.47 0 0 1-1.469 1.469zm0-2.418a.951.951 0 0 0-.95.95.951.951 0 0 0 .95.95.951.951 0 0 0 .95-.95.951.951 0 0 0-.95-.95z" fill="#51504f" data-name="Path 1342" transform="translate(271.132 -418.872)"></path>
</g>
</g>
</svg>
<span class="bodytext__mobile--brownish-grey">Content</span>
</div>
</button>
<nav id="content-navbar" class="navbar navbar--hidden">
<div class="navbar__icon-container">
<a href="/">
<svg xmlns="http://www.w3.org/2000/svg" width="155.314" height="60" viewBox="0 0 155.314 60">
<defs>
<clipPath id="clip-path">
<path id="Rectangle_1" d="M0 0h155.314v60H0z" fill="none" data-name="Rectangle 1"></path>
</clipPath>
</defs>
<g id="logo" transform="translate(-1305 -780.355)">
<g id="Group_2" clip-path="url(#clip-path)" data-name="Group 2" transform="translate(1305 780.355)">
<g id="Group_1" data-name="Group 1" transform="translate(.486 .486)">
<path id="Path_1" d="M1307.562 880.867l28.187-28.893a.521.521 0 0 0 .063-.666c-1.714-2.393-4.877-2.808-6.049-4.416-3.472-4.763-4.353-7.459-5.845-7.292a.456.456 0 0 0-.271.143l-10.182 10.438c-5.858 6-6.7 19.225-6.852 30.3a.552.552 0 0 0 .949.386z" fill="#017cee" data-name="Path 1" transform="translate(-1306.613 -822.232)"></path>
<path id="Path_2" d="M1405.512 908.489l-28.893-28.189a.521.521 0 0 0-.667-.063c-2.393 1.715-2.808 4.877-4.416 6.049-4.763 3.472-7.459 4.353-7.292 5.845a.456.456 0 0 0 .143.27l10.438 10.182c6 5.858 19.225 6.7 30.3 6.852a.552.552 0 0 0 .387-.946z" fill="#00ad46" data-name="Path 2" transform="translate(-1346.876 -850.567)"></path>
<path id="Path_3" d="M1373.909 902.252c-3.28-3.2-4.8-9.53 1.486-22.583-10.219 4.567-13.8 10.57-12.039 12.289z" fill="#04d659" data-name="Path 3" transform="translate(-1345.96 -850.233)"></path>
<path id="Path_4" d="M1433.132 782.359l-28.186 28.893a.52.52 0 0 0-.063.666c1.715 2.393 4.876 2.808 6.049 4.416 3.472 4.763 4.354 7.459 5.845 7.292a.454.454 0 0 0 .271-.143l10.182-10.438c5.858-6 6.7-19.225 6.852-30.3a.553.553 0 0 0-.95-.386z" fill="#00c7d4" data-name="Path 4" transform="translate(-1375.21 -782.123)"></path>
<path id="Path_5" d="M1426.9 881.155c-3.2 3.28-9.53 4.8-22.584-1.486 4.567 10.219 10.57 13.8 12.289 12.039z" fill="#11e1ee" data-name="Path 5" transform="translate(-1374.875 -850.233)"></path>
<path id="Path_6" d="M1307 782.919l28.893 28.186a.521.521 0 0 0 .666.063c2.393-1.715 2.808-4.877 4.416-6.049 4.763-3.472 7.459-4.353 7.292-5.845a.459.459 0 0 0-.143-.271l-10.438-10.182c-6-5.858-19.225-6.7-30.3-6.852a.552.552 0 0 0-.386.95z" fill="#e43921" data-name="Path 6" transform="translate(-1306.766 -781.97)"></path>
<path id="Path_7" d="M1405.8 804.711c3.28 3.2 4.8 9.53-1.486 22.584 10.219-4.567 13.8-10.571 12.039-12.289z" fill-rule="evenodd" fill="#ff7557" data-name="Path 7" transform="translate(-1374.875 -797.859)"></path>
<path id="Path_8" d="M1329.355 849.266c3.2-3.28 9.53-4.8 22.584 1.486-4.567-10.219-10.57-13.8-12.289-12.039z" fill="#0cb6ff" data-name="Path 8" transform="translate(-1322.503 -821.316)"></path>
<circle id="Ellipse_1" cx="1.26" cy="1.26" r="1.26" fill="#4a4848" data-name="Ellipse 1" transform="translate(28.18 28.171)"></circle>
<path id="Path_9" d="M1527.558 827.347a.229.229 0 0 1-.223-.223.458.458 0 0 1 .011-.123l2.766-7.214a.346.346 0 0 1 .357-.245h.758a.348.348 0 0 1 .357.245l2.754 7.214.022.123a.228.228 0 0 1-.223.223h-.568a.288.288 0 0 1-.19-.056.352.352 0 0 1-.089-.134l-.613-1.583h-3.657l-.613 1.583a.317.317 0 0 1-.1.134.269.269 0 0 1-.178.056zm4.795-2.732l-1.505-3.958-1.505 3.958zm3.322 4.85a.258.258 0 0 1-.189-.078.241.241 0 0 1-.067-.178v-7.4a.241.241 0 0 1 .067-.178.258.258 0 0 1 .189-.078h.513a.268.268 0 0 1 .256.256v.49a2.118 2.118 0 0 1 1.828-.858 2.092 2.092 0 0 1 1.751.736 3.135 3.135 0 0 1 .636 1.9q.011.122.011.379t-.011.379a3.168 3.168 0 0 1-.636 1.9 2.111 2.111 0 0 1-1.751.736 2.154 2.154 0 0 1-1.806-.836v2.587a.241.241 0 0 1-.067.178.223.223 0 0 1-.179.078zm2.364-2.91a1.324 1.324 0 0 0 1.149-.491 2.266 2.266 0 0 0 .4-1.293q.011-.111.011-.323 0-2.107-1.562-2.107a1.365 1.365 0 0 0-1.159.513 2.111 2.111 0 0 0-.412 1.2l-.012.424.012.435a1.862 1.862 0 0 0 .424 1.149 1.4 1.4 0 0 0 1.148.493zm5.628.9a2.329 2.329 0 0 1-1.015-.223 1.94 1.94 0 0 1-.747-.6 1.487 1.487 0 0 1-.268-.859 1.459 1.459 0 0 1 .6-1.2 3.4 3.4 0 0 1 1.65-.624l1.661-.234v-.323q0-1.137-1.3-1.137a1.4 1.4 0 0 0-.8.212 1.376 1.376 0 0 0-.468.48.305.305 0 0 1-.089.145.18.18 0 0 1-.134.045h-.48a.23.23 0 0 1-.245-.245 1.17 1.17 0 0 1 .245-.6 1.931 1.931 0 0 1 .747-.591 2.7 2.7 0 0 1 1.238-.256 2.351 2.351 0 0 1 1.8.591 2.032 2.032 0 0 1 .547 1.45v3.613a.257.257 0 0 1-.078.19.24.24 0 0 1-.178.067h-.513a.233.233 0 0 1-.257-.256v-.479a1.923 1.923 0 0 1-.714.6 2.557 2.557 0 0 1-1.203.237zm.234-.836a1.579 1.579 0 0 0 1.182-.469 1.881 1.881 0 0 0 .468-1.371v-.312l-1.293.19a2.918 2.918 0 0 0-1.193.379.761.761 0 0 0-.4.658.784.784 0 0 0 .368.691 1.585 1.585 0 0 0 .867.237zm6.643.836a2.556 2.556 0 0 1-1.873-.669 2.738 2.738 0 0 1-.714-1.9l-.011-.446.011-.446a2.7 2.7 0 0 1 .714-1.885 2.531 2.531 0 0 1 1.873-.68 2.917 2.917 0 0 1 1.36.29 2.077 2.077 0 0 1 .825.714 1.7 1.7 0 0 1 .3.848.2.2 0 0 1-.067.178.281.281 0 0 1-.19.067h-.535a.265.265 0 0 1-.168-.045.458.458 0 0 1-.111-.178 1.428 1.428 0 0 0-.535-.758 1.516 1.516 0 0 0-.87-.234 1.45 1.45 0 0 0-1.1.435 1.952 1.952 0 0 0-.435 1.3l-.011.4.011.379a1.969 1.969 0 0 0 .435 1.316 1.446 1.446 0 0 0 1.1.424 1.577 1.577 0 0 0 .87-.223 1.493 1.493 0 0 0 .535-.769.458.458 0 0 1 .111-.178.228.228 0 0 1 .168-.056h.535a.258.258 0 0 1 .19.078.2.2 0 0 1 .067.178 1.75 1.75 0 0 1-.3.847 2.078 2.078 0 0 1-.825.714 2.876 2.876 0 0 1-1.361.302zm4.078-.112a.233.233 0 0 1-.257-.256v-7.4a.241.241 0 0 1 .067-.178.259.259 0 0 1 .19-.078h.557a.267.267 0 0 1 .257.256v2.6a2.167 2.167 0 0 1 .758-.624 2.353 2.353 0 0 1 1.082-.223 2.067 2.067 0 0 1 1.661.691 2.642 2.642 0 0 1 .6 1.818v3.144a.257.257 0 0 1-.078.19.24.24 0 0 1-.178.067h-.557a.233.233 0 0 1-.256-.256V824a1.775 1.775 0 0 0-.39-1.227 1.387 1.387 0 0 0-1.1-.435 1.481 1.481 0 0 0-1.126.446 1.7 1.7 0 0 0-.412 1.215v3.088a.257.257 0 0 1-.078.19.24.24 0 0 1-.178.067zm8.846.112a2.466 2.466 0 0 1-1.84-.7 2.938 2.938 0 0 1-.747-1.94l-.011-.379.011-.368a2.953 2.953 0 0 1 .758-1.918 2.7 2.7 0 0 1 3.735.078 3.114 3.114 0 0 1 .68 2.119v.19a.257.257 0 0 1-.078.189.241.241 0 0 1-.178.067h-3.858v.1a2.11 2.11 0 0 0 .435 1.238 1.332 1.332 0 0 0 1.081.5 1.563 1.563 0 0 0 .836-.2 1.7 1.7 0 0 0 .491-.435.6.6 0 0 1 .145-.156.391.391 0 0 1 .19-.033h.547a.252.252 0 0 1 .167.056.192.192 0 0 1 .067.156.975.975 0 0 1-.312.591 2.51 2.51 0 0 1-.859.6 3.049 3.049 0 0 1-1.26.248zm1.527-3.434v-.033a1.817 1.817 0 0 0-.424-1.249 1.512 1.512 0 0 0-2.23 0 1.883 1.883 0 0 0-.4 1.249v.033z" fill="#51504f" data-name="Path 9" transform="translate(-1460.834 -808.144)"></path>
<path id="Path_10" d="M1527.2 827.081l-.061.061zm-.056-.279l-.08-.031zm2.766-7.214l.08.031zm1.472 0l-.081.029zm2.754 7.214l.084-.015a.064.064 0 0 0 0-.015zm.022.123h.086v-.015zm-.067.156l.06.061zm-.914.011l-.061.061.006.005zm-.089-.134l.081-.027zm-.613-1.583l.08-.031a.086.086 0 0 0-.08-.055zm-3.657 0v-.086a.086.086 0 0 0-.08.055zm-.613 1.583l-.08-.031zm-.1.134l.055.066zm4.047-2.676v.086a.086.086 0 0 0 .08-.116zm-1.505-3.958l.08-.03a.086.086 0 0 0-.16 0zm-1.505 3.958l-.08-.03a.086.086 0 0 0 .08.116zm-1.784 2.646a.128.128 0 0 1-.1-.042l-.122.121a.3.3 0 0 0 .217.092zm-.1-.042a.129.129 0 0 1-.042-.1h-.171a.3.3 0 0 0 .092.217zm-.042-.1a.38.38 0 0 1 .007-.1l-.163-.054a.514.514 0 0 0-.016.15zm.005-.092l2.765-7.214-.16-.061-2.765 7.214zm2.766-7.216a.283.283 0 0 1 .1-.143.3.3 0 0 1 .174-.046v-.172a.47.47 0 0 0-.271.076.453.453 0 0 0-.166.226zm.276-.189h.758v-.172h-.758zm.758 0a.3.3 0 0 1 .175.046.283.283 0 0 1 .1.143l.161-.059a.451.451 0 0 0-.166-.226.47.47 0 0 0-.272-.076zm.277.19l2.754 7.214.16-.061-2.754-7.214zm2.75 7.2l.022.123.169-.031-.022-.123zm.021.107a.13.13 0 0 1-.042.1l.121.121a.3.3 0 0 0 .092-.217zm-.042.1a.13.13 0 0 1-.1.042v.171a.3.3 0 0 0 .217-.092zm-.1.042h-.568v.171h.568zm-.568 0a.206.206 0 0 1-.135-.036l-.11.132a.373.373 0 0 0 .245.076zm-.129-.031a.262.262 0 0 1-.069-.1l-.162.054a.431.431 0 0 0 .11.167zm-.07-.1l-.613-1.584-.16.062.613 1.583zm-.693-1.638h-3.657v.171h3.657zm-3.737.055l-.614 1.584.16.062.613-1.583zm-.615 1.587a.235.235 0 0 1-.075.1l.111.13a.4.4 0 0 0 .126-.172zm-.074.1a.185.185 0 0 1-.124.036v.171a.354.354 0 0 0 .233-.076zm-.124.036h-.569v.171h.569zm4.306-2.677l-1.505-3.958-.16.061 1.505 3.958zm-1.666-3.958l-1.505 3.958.16.061 1.505-3.958zm-1.425 4.075h3.01v-.171h-3.01zm6.143 4.687l-.06.061zm0-7.761l.061.061zm.881 0l-.065.056.01.009zm.078.669h-.086a.086.086 0 0 0 .155.051zm3.579-.123l-.067.053zm.636 1.9h-.086zm0 .758l-.085-.007zm-.636 1.9l.067.054zm-3.557-.1l.068-.052a.086.086 0 0 0-.154.052zm-.067 2.765l-.061-.06zm2.787-3.323l-.069-.051zm.4-1.293l-.085-.008v.005zm-2.709-1.918l-.068-.052zm-.413 1.2h-.086zm-.011.423h-.085zm.011.435h-.086zm.424 1.149l.066-.054zm-1.216 3.315a.173.173 0 0 1-.129-.053l-.121.121a.342.342 0 0 0 .25.1zm-.129-.053a.157.157 0 0 1-.042-.118h-.172a.325.325 0 0 0 .092.239zm-.042-.118v-7.4h-.172v7.4zm0-7.4a.157.157 0 0 1 .042-.118l-.121-.121a.324.324 0 0 0-.092.239zm.042-.118a.173.173 0 0 1 .129-.053v-.172a.342.342 0 0 0-.25.1zm.129-.053h.513v-.172h-.513zm.513 0a.137.137 0 0 1 .113.048l.13-.111a.309.309 0 0 0-.244-.108zm.123.058a.137.137 0 0 1 .048.113h.171a.309.309 0 0 0-.108-.243zm.048.113v.49h.171v-.49zm.155.541a2.033 2.033 0 0 1 1.759-.823v-.171a2.2 2.2 0 0 0-1.9.894zm1.759-.823a2.007 2.007 0 0 1 1.683.7l.135-.106a2.177 2.177 0 0 0-1.818-.768zm1.683.7a3.045 3.045 0 0 1 .617 1.845l.171-.007a3.218 3.218 0 0 0-.654-1.946zm.617 1.85c.007.078.011.2.011.372h.171c0-.171 0-.3-.012-.387zm.011.372c0 .171 0 .294-.011.372l.17.015c.008-.086.012-.216.012-.387zm-.011.376a3.08 3.08 0 0 1-.617 1.846l.134.106a3.25 3.25 0 0 0 .654-1.945zm-.617 1.845a2.025 2.025 0 0 1-1.683.7v.171a2.2 2.2 0 0 0 1.817-.768zm-1.683.7a2.068 2.068 0 0 1-1.739-.8l-.136.1a2.239 2.239 0 0 0 1.874.87zm-1.892-.75v2.587h.172v-2.587zm0 2.587a.156.156 0 0 1-.042.118l.121.121a.325.325 0 0 0 .092-.239zm-.046.123a.138.138 0 0 1-.114.048v.172a.308.308 0 0 0 .244-.108zm-.114.048h-.546v.172h.546zm1.817-2.739a1.408 1.408 0 0 0 1.218-.526l-.138-.1a1.24 1.24 0 0 1-1.079.455zm1.217-.525a2.355 2.355 0 0 0 .419-1.341l-.171-.007a2.182 2.182 0 0 1-.385 1.246zm.418-1.336c.008-.079.012-.19.012-.332h-.172c0 .14 0 .245-.011.315zm.012-.332a2.726 2.726 0 0 0-.407-1.632 1.448 1.448 0 0 0-1.24-.562v.171a1.278 1.278 0 0 1 1.1.492 2.565 2.565 0 0 1 .374 1.53zm-1.647-2.193a1.452 1.452 0 0 0-1.228.547l.136.1a1.282 1.282 0 0 1 1.091-.479zm-1.228.547a2.2 2.2 0 0 0-.43 1.252l.172.008a2.028 2.028 0 0 1 .4-1.157zm-.43 1.254l-.011.424h.171l.011-.424zm-.011.428l.011.435h.172l-.011-.435zm.011.436a1.95 1.95 0 0 0 .443 1.2l.133-.109a1.776 1.776 0 0 1-.4-1.1zm.443 1.2a1.484 1.484 0 0 0 1.214.522v-.171a1.314 1.314 0 0 1-1.082-.459zm5.828 1.117l.037-.077zm-.747-.6l-.07.049zm.335-2.063l.052.068zm1.65-.624l.012.085zm1.661-.234l.012.085a.086.086 0 0 0 .074-.085zm-2.107-1.249l.046.072zm-.468.48l-.075-.042a.083.083 0 0 0-.006.015zm-.089.145l-.054-.067-.007.006zm-.792-.022l-.065.056.009.009zm-.067-.178h-.086zm.245-.6l-.07-.049zm.747-.591l.038.077zm3.033.334l-.063.058zm.468 5.252l.06.061zm-.881 0l-.065.056a.043.043 0 0 0 .009.009zm-.067-.669h.086a.086.086 0 0 0-.156-.048zm-.714.6l-.04-.076zm.223-1.059l-.062-.06zm.468-1.684h.086a.086.086 0 0 0-.1-.085zm-1.293.189l.012.085zm-1.193.379l.046.072zm-.033 1.349l-.047.071zm.635.985a2.241 2.241 0 0 1-.978-.215l-.074.155a2.412 2.412 0 0 0 1.051.231zm-.978-.215a1.859 1.859 0 0 1-.715-.576l-.138.1a2.024 2.024 0 0 0 .779.629zm-.713-.573a1.4 1.4 0 0 1-.253-.81h-.172a1.571 1.571 0 0 0 .283.907zm-.253-.81a1.374 1.374 0 0 1 .569-1.136l-.105-.135a1.544 1.544 0 0 0-.635 1.272zm.569-1.137a3.316 3.316 0 0 1 1.609-.607l-.024-.17a3.481 3.481 0 0 0-1.691.642zm1.609-.607l1.661-.234-.024-.17-1.662.234zm1.735-.319v-.323h-.171v.323zm0-.323a1.156 1.156 0 0 0-.355-.917 1.536 1.536 0 0 0-1.035-.306v.172a1.37 1.37 0 0 1 .922.263.986.986 0 0 1 .3.788zm-1.39-1.223a1.486 1.486 0 0 0-.851.227l.1.142a1.316 1.316 0 0 1 .755-.2zm-.849.226a1.452 1.452 0 0 0-.5.51l.15.084a1.286 1.286 0 0 1 .44-.449zm-.5.524a.226.226 0 0 1-.062.105l.107.134a.391.391 0 0 0 .117-.185zm-.068.112a.1.1 0 0 1-.073.019v.171a.266.266 0 0 0 .194-.07zm-.073.019h-.48v.171h.48zm-.48 0a.18.18 0 0 1-.122-.046l-.112.13a.352.352 0 0 0 .234.087zm-.113-.037a.18.18 0 0 1-.047-.123h-.171a.352.352 0 0 0 .087.234zm-.047-.119a1.1 1.1 0 0 1 .23-.557l-.14-.1a1.253 1.253 0 0 0-.261.648zm.23-.556a1.843 1.843 0 0 1 .715-.564l-.075-.154a2.018 2.018 0 0 0-.78.618zm.716-.564a2.611 2.611 0 0 1 1.2-.247v-.171a2.781 2.781 0 0 0-1.277.266zm1.2-.247a2.268 2.268 0 0 1 1.732.563l.126-.116a2.435 2.435 0 0 0-1.858-.618zm1.733.564a1.945 1.945 0 0 1 .523 1.391h.171a2.117 2.117 0 0 0-.57-1.508zm.523 1.391v3.613h.171v-3.613zm0 3.613a.172.172 0 0 1-.053.129l.121.121a.344.344 0 0 0 .1-.25zm-.053.129a.157.157 0 0 1-.118.042v.171a.326.326 0 0 0 .239-.092zm-.118.042h-.513v.171h.513zm-.513 0a.2.2 0 0 1-.134-.046l-.111.13a.367.367 0 0 0 .245.088zm-.124-.037a.194.194 0 0 1-.047-.134h-.171a.366.366 0 0 0 .087.245zm-.047-.134v-.479h-.171v.479zm-.156-.528a1.846 1.846 0 0 1-.683.575l.079.152a2.012 2.012 0 0 0 .745-.629zm-.683.575a2.476 2.476 0 0 1-1.153.236v.171a2.644 2.644 0 0 0 1.233-.255zm-.919-.429a1.666 1.666 0 0 0 1.244-.494l-.123-.12a1.493 1.493 0 0 1-1.121.442zm1.244-.494a1.969 1.969 0 0 0 .492-1.431h-.171a1.8 1.8 0 0 1-.444 1.312zm.492-1.431v-.312h-.171v.312zm-.1-.4l-1.293.189.025.17 1.293-.189zm-1.293.189a3 3 0 0 0-1.228.393l.095.143a2.837 2.837 0 0 1 1.158-.365zm-1.227.392a.845.845 0 0 0-.441.73h.172a.676.676 0 0 1 .362-.586zm-.441.73a.869.869 0 0 0 .406.762l.095-.142a.7.7 0 0 1-.33-.62zm.408.764a1.673 1.673 0 0 0 .916.247v-.171a1.5 1.5 0 0 1-.823-.221zm5.686.329l-.061.06zm-.714-1.9h-.085zm-.011-.446h-.085zm.011-.446h-.085zm.714-1.885l.061.061zm3.234-.39l-.04.076zm.825.713l-.073.046zm.3.848h-.086zm-.067.178l.056.065zm-.892.022l.054-.067zm-.112-.178l-.081.029zm-.535-.758l-.048.071zm-1.974.2l-.062-.059zm-.435 1.3h-.086zm-.011.4h-.086zm.011.379h-.086zm.435 1.316l-.062.059zm1.974.2l.046.072zm.535-.769l-.079-.033zm.112-.178l.054.067.007-.006zm.892.022l-.061.06zm.067.178l-.085-.009zm-.3.847l-.072-.046zm-.825.714l-.04-.076zm-1.36.2a2.471 2.471 0 0 1-1.814-.644l-.12.122a2.64 2.64 0 0 0 1.933.694zm-1.813-.643a2.653 2.653 0 0 1-.689-1.839l-.171.006a2.822 2.822 0 0 0 .738 1.952zm-.689-1.838l-.011-.446h-.171l.011.446zm-.011-.442l.011-.446h-.171l-.011.446zm.011-.445a2.611 2.611 0 0 1 .689-1.827l-.122-.121a2.78 2.78 0 0 0-.738 1.942zm.689-1.827a2.447 2.447 0 0 1 1.813-.655v-.171a2.617 2.617 0 0 0-1.934.705zm1.813-.655a2.836 2.836 0 0 1 1.32.28l.079-.152a3 3 0 0 0-1.4-.3zm1.32.28a1.99 1.99 0 0 1 .792.683l.145-.091a2.158 2.158 0 0 0-.858-.744zm.793.685a1.617 1.617 0 0 1 .287.8l.171-.009a1.789 1.789 0 0 0-.315-.89zm.287.809a.11.11 0 0 1-.037.1l.112.13a.281.281 0 0 0 .1-.252zm-.037.1a.2.2 0 0 1-.134.046v.171a.369.369 0 0 0 .246-.088zm-.134.046h-.535v.171h.535zm-.535 0a.184.184 0 0 1-.114-.026l-.107.134a.345.345 0 0 0 .221.064zm-.114-.026a.389.389 0 0 1-.086-.144l-.158.066a.533.533 0 0 0 .137.212zm-.084-.14a1.514 1.514 0 0 0-.57-.8l-.093.144a1.343 1.343 0 0 1 .5.715zm-.568-.8a1.6 1.6 0 0 0-.918-.249v.171a1.435 1.435 0 0 1 .822.219zm-.918-.249a1.535 1.535 0 0 0-1.166.462l.124.118a1.364 1.364 0 0 1 1.042-.408zm-1.166.462a2.036 2.036 0 0 0-.458 1.36l.171.006a1.872 1.872 0 0 1 .411-1.249zm-.458 1.361l-.011.4h.171l.011-.4zm-.011.406l.011.379.171-.005-.011-.379zm.011.38a2.052 2.052 0 0 0 .458 1.371l.124-.118a1.889 1.889 0 0 1-.411-1.26zm.458 1.371a1.533 1.533 0 0 0 1.166.451v-.172a1.363 1.363 0 0 1-1.042-.4zm1.166.451a1.661 1.661 0 0 0 .916-.237l-.093-.144a1.491 1.491 0 0 1-.823.209zm.918-.238a1.576 1.576 0 0 0 .568-.812l-.162-.057a1.409 1.409 0 0 1-.5.727zm.566-.807a.39.39 0 0 1 .086-.144l-.107-.134a.533.533 0 0 0-.137.213zm.093-.151a.144.144 0 0 1 .107-.031v-.171a.31.31 0 0 0-.228.081zm.107-.031h.535v-.171h-.535zm.535 0a.173.173 0 0 1 .129.053l.121-.121a.344.344 0 0 0-.25-.1zm.134.057a.11.11 0 0 1 .037.1l.17.017a.281.281 0 0 0-.1-.252zm.037.109a1.664 1.664 0 0 1-.288.806l.144.092a1.839 1.839 0 0 0 .315-.889zm-.288.806a1.989 1.989 0 0 1-.792.683l.079.152a2.162 2.162 0 0 0 .858-.744zm-.793.684a2.8 2.8 0 0 1-1.32.28v.171a2.96 2.96 0 0 0 1.4-.3zm2.568.187l-.065.056.01.009zm0-7.772l.061.06zm.926 0l-.065.056.009.009zm.078 2.776h-.085a.086.086 0 0 0 .153.053zm.758-.624l.038.077zm2.743.468l-.065.056zm.524 5.151l-.061-.06zm-.925 0l-.065.056.009.009zm-.457-4.5l-.065.056zm-2.23.011l-.062-.059zm-.49 4.493l-.061-.06zm-.736-.019a.2.2 0 0 1-.134-.046l-.112.13a.367.367 0 0 0 .245.088zm-.124-.037a.2.2 0 0 1-.046-.134h-.172a.367.367 0 0 0 .087.245zm-.046-.134v-7.4h-.172v7.4zm0-7.4a.156.156 0 0 1 .042-.118l-.121-.121a.326.326 0 0 0-.092.239zm.042-.118a.172.172 0 0 1 .129-.053v-.171a.343.343 0 0 0-.25.1zm.129-.053h.557v-.171h-.557zm.557 0a.137.137 0 0 1 .113.048l.13-.112a.308.308 0 0 0-.244-.108zm.122.057a.137.137 0 0 1 .048.113h.172a.309.309 0 0 0-.108-.243zm.048.113v2.6h.172v-2.6zm.153 2.651a2.076 2.076 0 0 1 .728-.6l-.075-.154a2.248 2.248 0 0 0-.788.649zm.73-.6a2.272 2.272 0 0 1 1.043-.214v-.171a2.438 2.438 0 0 0-1.121.232zm1.043-.214a1.982 1.982 0 0 1 1.6.661l.13-.112a2.152 2.152 0 0 0-1.727-.721zm1.6.663a2.557 2.557 0 0 1 .581 1.761h.171a2.727 2.727 0 0 0-.624-1.874zm.581 1.761v3.144h.171v-3.144zm0 3.144a.173.173 0 0 1-.053.129l.121.121a.345.345 0 0 0 .1-.25zm-.053.129a.156.156 0 0 1-.118.042v.171a.327.327 0 0 0 .239-.092zm-.118.042h-.557v.171h.557zm-.557 0a.2.2 0 0 1-.134-.046l-.112.13a.367.367 0 0 0 .245.088zm-.124-.037a.2.2 0 0 1-.046-.134h-.172a.368.368 0 0 0 .087.245zm-.046-.134V823.8h-.172v3.088zm0-3.088a1.859 1.859 0 0 0-.412-1.284l-.128.114a1.69 1.69 0 0 1 .368 1.169zm-.411-1.283a1.471 1.471 0 0 0-1.169-.464v.171a1.3 1.3 0 0 1 1.039.406zm-1.169-.464a1.566 1.566 0 0 0-1.188.473l.124.118a1.4 1.4 0 0 1 1.064-.419zm-1.188.473a1.779 1.779 0 0 0-.436 1.275h.172a1.609 1.609 0 0 1 .389-1.156zm-.436 1.275v3.088h.172V823.8zm0 3.088a.172.172 0 0 1-.053.129l.121.121a.344.344 0 0 0 .1-.25zm-.053.129a.156.156 0 0 1-.118.042v.171a.327.327 0 0 0 .239-.092zm-.118.042h-.557v.171h.557zm6.449-.505l-.062.059zm-.747-1.94h-.086zm-.012-.379h-.085v.005zm.012-.368l-.086-.006zm.758-1.918l-.061-.06zm3.735.078l-.065.056zm.6 2.5l.061.061zm-4.036.067v-.086a.086.086 0 0 0-.086.086zm0 .1h-.086zm.435 1.238l-.068.053zm1.918.3l.045.073zm.491-.435l-.069-.051zm.145-.156l.039.077h.006zm.9.022l-.055.066zm-.245.747l-.064-.057zm-.858.6l.035.078zm.267-3.189v.086a.086.086 0 0 0 .086-.086zm-.424-1.282l-.066.055zm-2.23 0l-.065-.055zm-.4 1.282h-.086a.086.086 0 0 0 .086.086zm1.528 3.349a2.38 2.38 0 0 1-1.779-.677l-.122.12a2.55 2.55 0 0 0 1.9.728zm-1.778-.676a2.86 2.86 0 0 1-.724-1.886l-.171.009a3.027 3.027 0 0 0 .771 1.995zm-.723-1.884l-.011-.379h-.171l.011.379zm-.011-.374l.011-.368-.172-.005-.011.368zm.011-.365a2.871 2.871 0 0 1 .735-1.864l-.124-.118a3.042 3.042 0 0 0-.782 1.971zm.734-1.864a2.331 2.331 0 0 1 1.756-.687v-.171a2.5 2.5 0 0 0-1.879.74zm1.756-.687a2.307 2.307 0 0 1 1.853.762l.13-.112a2.477 2.477 0 0 0-1.983-.821zm1.854.762a3.03 3.03 0 0 1 .659 2.062h.172a3.2 3.2 0 0 0-.7-2.175zm.659 2.062v.19h.172v-.19zm0 .19a.172.172 0 0 1-.053.129l.121.121a.345.345 0 0 0 .1-.25zm-.053.129a.156.156 0 0 1-.118.042v.171a.327.327 0 0 0 .239-.092zm-.118.042h-3.858v.171h3.858zm-3.944.086v.1h.172v-.1zm0 .1a2.2 2.2 0 0 0 .453 1.287l.135-.106a2.027 2.027 0 0 1-.417-1.189zm.454 1.288a1.418 1.418 0 0 0 1.148.533v-.171a1.247 1.247 0 0 1-1.015-.47zm1.148.533a1.647 1.647 0 0 0 .882-.214l-.09-.146a1.481 1.481 0 0 1-.791.188zm.882-.214a1.777 1.777 0 0 0 .515-.458l-.14-.1a1.613 1.613 0 0 1-.466.412zm.513-.456a1.251 1.251 0 0 1 .081-.1.28.28 0 0 1 .026-.025l.008-.006-.077-.153a.326.326 0 0 0-.083.068 1.55 1.55 0 0 0-.092.113zm.12-.134a.328.328 0 0 1 .146-.021v-.171a.468.468 0 0 0-.234.046zm.146-.021h.547v-.171h-.547zm.547 0a.166.166 0 0 1 .112.036l.11-.132a.337.337 0 0 0-.222-.076zm.112.036a.107.107 0 0 1 .036.09h.171a.277.277 0 0 0-.1-.222zm.036.09a.9.9 0 0 1-.291.534l.128.115a1.055 1.055 0 0 0 .334-.649zm-.291.535a2.42 2.42 0 0 1-.83.581l.072.156a2.6 2.6 0 0 0 .888-.624zm-.829.58a2.964 2.964 0 0 1-1.224.238v.171a3.133 3.133 0 0 0 1.295-.253zm.389-3.111v-.033h-.171v.033zm0-.033a1.9 1.9 0 0 0-.445-1.306l-.129.114a1.731 1.731 0 0 1 .4 1.192zm-.444-1.3a1.466 1.466 0 0 0-1.181-.521v.172a1.3 1.3 0 0 1 1.049.46zm-1.181-.521a1.466 1.466 0 0 0-1.18.521l.131.11a1.3 1.3 0 0 1 1.049-.46zm-1.181.521a1.965 1.965 0 0 0-.422 1.3h.172a1.794 1.794 0 0 1 .382-1.194zm-.422 1.3v.033h.172v-.033zm.086.119h3.055v-.171h-3.055z" fill="#51504f" data-name="Path 10" transform="translate(-1460.636 -807.945)"></path>
<path id="Path_11" d="M1519.066 884.011a.581.581 0 0 1-.567-.567 1.151 1.151 0 0 1 .028-.312l7.026-18.328a.881.881 0 0 1 .906-.623h1.926a.882.882 0 0 1 .907.623l7 18.328.057.312a.583.583 0 0 1-.567.567h-1.445a.735.735 0 0 1-.482-.142.9.9 0 0 1-.226-.34l-1.558-4.023h-9.292l-1.558 4.023a.8.8 0 0 1-.255.34.688.688 0 0 1-.453.142zm12.181-6.94l-3.824-10.056-3.823 10.055zm8.184-10.538a.592.592 0 0 1-.652-.651v-1.53a.714.714 0 0 1 .17-.482.656.656 0 0 1 .482-.2h1.785a.677.677 0 0 1 .68.68v1.53a.655.655 0 0 1-.2.481.713.713 0 0 1-.481.17zm.227 17.479a.593.593 0 0 1-.652-.652v-13.428a.611.611 0 0 1 .17-.453.656.656 0 0 1 .482-.2h1.359a.679.679 0 0 1 .652.651v13.427a.655.655 0 0 1-.2.482.613.613 0 0 1-.453.17zm6.861 0a.592.592 0 0 1-.651-.652v-13.4a.715.715 0 0 1 .17-.481.656.656 0 0 1 .482-.2h1.3a.677.677 0 0 1 .68.68v1.246a4.255 4.255 0 0 1 3.966-1.926h1.1a.679.679 0 0 1 .651.651v1.161a.566.566 0 0 1-.2.453.612.612 0 0 1-.453.17h-1.7a3.2 3.2 0 0 0-2.408.907 3.253 3.253 0 0 0-.879 2.408v8.328a.656.656 0 0 1-.2.482.716.716 0 0 1-.482.17zm12.234 0a.593.593 0 0 1-.651-.652v-11.814h-2.408a.592.592 0 0 1-.651-.651v-.963a.611.611 0 0 1 .17-.453.654.654 0 0 1 .481-.2h2.408v-1.417q0-4.816 4.872-4.815h1.586a.679.679 0 0 1 .652.651v.963a.656.656 0 0 1-.2.481.613.613 0 0 1-.453.17h-1.529a2.1 2.1 0 0 0-1.785.68 3.248 3.248 0 0 0-.51 2.011v1.275h6.062V863.7a.613.613 0 0 1 .17-.453.656.656 0 0 1 .482-.2h1.3a.679.679 0 0 1 .652.651v19.659a.655.655 0 0 1-.2.482.613.613 0 0 1-.454.17h-1.3a.592.592 0 0 1-.652-.652v-11.811h-6.062v11.813a.657.657 0 0 1-.2.482.614.614 0 0 1-.454.17zm20.9.283a6.487 6.487 0 0 1-4.844-1.757 6.837 6.837 0 0 1-1.813-4.674l-.029-1.218.029-1.218a6.732 6.732 0 0 1 1.841-4.646 7.389 7.389 0 0 1 9.631 0 6.736 6.736 0 0 1 1.841 4.646q.028.311.028 1.218t-.028 1.218a6.772 6.772 0 0 1-1.841 4.674 6.391 6.391 0 0 1-4.82 1.756zm0-2.181a3.582 3.582 0 0 0 2.8-1.133 4.931 4.931 0 0 0 1.133-3.258q.028-.283.028-1.076t-.028-1.076a4.931 4.931 0 0 0-1.133-3.258 3.582 3.582 0 0 0-2.8-1.133 3.671 3.671 0 0 0-2.833 1.133 4.83 4.83 0 0 0-1.1 3.258l-.028 1.076.028 1.076a4.83 4.83 0 0 0 1.1 3.258 3.671 3.671 0 0 0 2.828 1.132zm13.755 1.9a.846.846 0 0 1-.566-.17 1.321 1.321 0 0 1-.34-.538l-4.023-13.144-.056-.283a.575.575 0 0 1 .17-.425.641.641 0 0 1 .425-.17h1.246a.612.612 0 0 1 .453.17.646.646 0 0 1 .255.312l3.145 10.679 3.371-10.566a.761.761 0 0 1 .255-.4.726.726 0 0 1 .538-.2h.963a.728.728 0 0 1 .539.2.76.76 0 0 1 .255.4l3.371 10.566 3.144-10.679a.655.655 0 0 1 .2-.312.714.714 0 0 1 .482-.17h1.275a.542.542 0 0 1 .4.17.576.576 0 0 1 .17.425l-.057.283-3.994 13.144a1.323 1.323 0 0 1-.34.538.9.9 0 0 1-.6.17h-1.1a.86.86 0 0 1-.935-.708l-3.286-10.141-3.286 10.141a.928.928 0 0 1-.963.708z" fill="#51504f" data-name="Path 11" transform="translate(-1454.66 -838.62)"></path>
</g>
</g>
</g>
</svg>
</a>
</div>
<div id="content-close-button">
<svg xmlns="http://www.w3.org/2000/svg" width="19.799" height="19.799" viewBox="0 0 19.799 19.799">
<g id="Group_1574" data-name="Group 1574" transform="translate(-41.892 290.899)">
<g id="Group_1291" data-name="Group 1291" transform="rotate(-45 -308.114 -187.077)">
<rect id="Rectangle_461" width="26" height="2" fill="#51504f" data-name="Rectangle 461" rx="1" transform="translate(-.209)"></rect>
</g>
<g id="Group_1292" data-name="Group 1292" transform="rotate(45 372.48 -93.011)">
<rect id="Rectangle_462" width="26" height="2" fill="#51504f" data-name="Rectangle 462" rx="1" transform="translate(-.209 -.178)"></rect>
</g>
</g>
</svg>
</div>
</nav>
<div class="content-drawer-container" id="content-drawer">
<div class="content-drawer">
<div class="td-sidebar">
<div id="docs-version-selector" class="docs-version-selector sidebar__version-selector">
<a class="dropdown-toggle" href="#" id="versionDropdown" role="button" data-toggle="dropdown" aria-haspopup="true"
aria-expanded="false">
<span class="bodytext__medium--greyish-brown">Version: </span><span class="version">8.4.0</span>
</a>
<div class="dropdown-menu" aria-labelledby="navbarDropdownMenuLink">
</div>
<script type="application/x-template" id="version-item-template">
<a class="dropdown-item"></a>
</script>
</div>
<div class="searchb-box">
<form class="search-form" action="../../../../../../search.html" method="get">
<input class="search-form__input" type="text" name="q" placeholder="Search docs" size="16">
<input type="hidden" name="check_keywords" value="yes" />
<input type="hidden" name="area" value="default" />
<button class="search-form__button" type="submit">
<svg xmlns="http://www.w3.org/2000/svg" width="20" height="20" viewBox="0 0 20 20">
<g id="Group_1579" data-name="Group 1579" transform="translate(-41.001 -41)">
<path id="Path_169" d="M71.415 64.687a7.215 7.215 0 1 0-6.729 6.728 7.222 7.222 0 0 0 6.729-6.728z" fill="none" data-name="Path 169" transform="translate(-14.277 -14.276)"></path>
<path id="Path_170" d="M60.863 59.8l-6.093-6.09a7.78 7.78 0 1 0-1.06 1.06l6.09 6.093a.468.468 0 0 0 .662 0l.4-.4a.468.468 0 0 0 .001-.663zM42.512 49.183a6.274 6.274 0 1 1 5.851 5.85 6.28 6.28 0 0 1-5.851-5.85z" fill="#51504f" data-name="Path 170"></path>
</g>
</svg>
</button>
</form>
</div>
<style>
.searchb-box {
margin-bottom: 26px;
}
.searchb-box .search-form {
width: 100%;
margin-top: 20px;
}
</style>
<div class="toctree" role="navigation" aria-label="main navigation">
<p class="caption" role="heading"><span class="caption-text">Guides</span></p>
<ul>
<li class="toctree-l1"><a class="reference internal" href="../../../../../../connections/index.html">Connection types</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../../../../logging/index.html">Logging handlers</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../../../../secrets-backends/google-cloud-secret-manager-backend.html">Secrets backends</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../../../../api-auth-backend/google-openid.html">API Authentication backend</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../../../../operators/index.html">Operators</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../../../../sensors/index.html">Sensors</a></li>
</ul>
<p class="caption" role="heading"><span class="caption-text">References</span></p>
<ul>
<li class="toctree-l1"><a class="reference internal" href="../../../../../../_api/airflow/providers/google/index.html">Python API</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../../../../configurations-ref.html">Configuration</a></li>
</ul>
<p class="caption" role="heading"><span class="caption-text">Resources</span></p>
<ul>
<li class="toctree-l1"><a class="reference internal" href="../../../../../../example-dags.html">Example DAGs</a></li>
<li class="toctree-l1"><a class="reference external" href="https://pypi.org/project/apache-airflow-providers-google/">PyPI Repository</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../../../../installing-providers-from-sources.html">Installing from sources</a></li>
</ul>
<p class="caption" role="heading"><span class="caption-text">Commits</span></p>
<ul>
<li class="toctree-l1"><a class="reference internal" href="../../../../../../commits.html">Detailed list of commits</a></li>
</ul>
</div>
<style type="text/css">
.toctree {
}
.toctree li {
list-style: none;
}
.toctree .caption {
font-family: Roboto;
font-size: 18px;
font-weight: bold;
font-stretch: normal;
font-style: normal;
line-height: 1.33;
letter-spacing: normal;
color: #51504f;
padding-bottom: 13px;
text-transform: uppercase;
margin-bottom: 0;
}
.toctree .current > a:not([href="#"]) {
color: #017cee;
}
.toctree > ul {
padding-left: 0;
}
.toctree ul {
padding-left: 15px;
display: none;
}
.toctree > ul,
.toctree li.current > ul {
display: block;
}
.toctree a .toctree-expand {
display: inline-block;
position: relative;
height: 1em;
}
.toctree a .toctree-expand:before {
position: absolute;
top: 6px;
left: -12px;
content: '►';
font-size: 7px;
}
.toctree .current > a > .toctree-expand:before {
content: '▼';
}
.toctree .current {
color: #017cee;
}
.toctree li {
font-family: Roboto;
font-size: 16px;
font-weight: normal;
font-stretch: normal;
font-style: normal;
line-height: 1.63;
letter-spacing: normal;
color: #707070;
}
</style>
</div>
</div>
</div>
</div>
<div class="d-flex">
<div class="td-sidebar desktop-only d-print-none">
<div id="docs-version-selector" class="docs-version-selector sidebar__version-selector">
<a class="dropdown-toggle" href="#" id="versionDropdown" role="button" data-toggle="dropdown" aria-haspopup="true"
aria-expanded="false">
<span class="bodytext__medium--greyish-brown">Version: </span><span class="version">8.4.0</span>
</a>
<div class="dropdown-menu" aria-labelledby="navbarDropdownMenuLink">
</div>
<script type="application/x-template" id="version-item-template">
<a class="dropdown-item"></a>
</script>
</div>
<div class="searchb-box">
<form class="search-form" action="../../../../../../search.html" method="get">
<input class="search-form__input" type="text" name="q" placeholder="Search docs" size="16">
<input type="hidden" name="check_keywords" value="yes" />
<input type="hidden" name="area" value="default" />
<button class="search-form__button" type="submit">
<svg xmlns="http://www.w3.org/2000/svg" width="20" height="20" viewBox="0 0 20 20">
<g id="Group_1579" data-name="Group 1579" transform="translate(-41.001 -41)">
<path id="Path_169" d="M71.415 64.687a7.215 7.215 0 1 0-6.729 6.728 7.222 7.222 0 0 0 6.729-6.728z" fill="none" data-name="Path 169" transform="translate(-14.277 -14.276)"></path>
<path id="Path_170" d="M60.863 59.8l-6.093-6.09a7.78 7.78 0 1 0-1.06 1.06l6.09 6.093a.468.468 0 0 0 .662 0l.4-.4a.468.468 0 0 0 .001-.663zM42.512 49.183a6.274 6.274 0 1 1 5.851 5.85 6.28 6.28 0 0 1-5.851-5.85z" fill="#51504f" data-name="Path 170"></path>
</g>
</svg>
</button>
</form>
</div>
<style>
.searchb-box {
margin-bottom: 26px;
}
.searchb-box .search-form {
width: 100%;
margin-top: 20px;
}
</style>
<div class="toctree" role="navigation" aria-label="main navigation">
<p class="caption" role="heading"><span class="caption-text">Guides</span></p>
<ul>
<li class="toctree-l1"><a class="reference internal" href="../../../../../../connections/index.html">Connection types</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../../../../logging/index.html">Logging handlers</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../../../../secrets-backends/google-cloud-secret-manager-backend.html">Secrets backends</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../../../../api-auth-backend/google-openid.html">API Authentication backend</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../../../../operators/index.html">Operators</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../../../../sensors/index.html">Sensors</a></li>
</ul>
<p class="caption" role="heading"><span class="caption-text">References</span></p>
<ul>
<li class="toctree-l1"><a class="reference internal" href="../../../../../../_api/airflow/providers/google/index.html">Python API</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../../../../configurations-ref.html">Configuration</a></li>
</ul>
<p class="caption" role="heading"><span class="caption-text">Resources</span></p>
<ul>
<li class="toctree-l1"><a class="reference internal" href="../../../../../../example-dags.html">Example DAGs</a></li>
<li class="toctree-l1"><a class="reference external" href="https://pypi.org/project/apache-airflow-providers-google/">PyPI Repository</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../../../../../installing-providers-from-sources.html">Installing from sources</a></li>
</ul>
<p class="caption" role="heading"><span class="caption-text">Commits</span></p>
<ul>
<li class="toctree-l1"><a class="reference internal" href="../../../../../../commits.html">Detailed list of commits</a></li>
</ul>
</div>
<style type="text/css">
.toctree {
}
.toctree li {
list-style: none;
}
.toctree .caption {
font-family: Roboto;
font-size: 18px;
font-weight: bold;
font-stretch: normal;
font-style: normal;
line-height: 1.33;
letter-spacing: normal;
color: #51504f;
padding-bottom: 13px;
text-transform: uppercase;
margin-bottom: 0;
}
.toctree .current > a:not([href="#"]) {
color: #017cee;
}
.toctree > ul {
padding-left: 0;
}
.toctree ul {
padding-left: 15px;
display: none;
}
.toctree > ul,
.toctree li.current > ul {
display: block;
}
.toctree a .toctree-expand {
display: inline-block;
position: relative;
height: 1em;
}
.toctree a .toctree-expand:before {
position: absolute;
top: 6px;
left: -12px;
content: '►';
font-size: 7px;
}
.toctree .current > a > .toctree-expand:before {
content: '▼';
}
.toctree .current {
color: #017cee;
}
.toctree li {
font-family: Roboto;
font-size: 16px;
font-weight: normal;
font-stretch: normal;
font-style: normal;
line-height: 1.63;
letter-spacing: normal;
color: #707070;
}
</style>
</div>
<main class="col-12 col-md-9 col-xl-8" role="main">
<div role="navigation" aria-label="breadcrumbs navigation" class="d-none d-md-block d-print-none">
<ul class="breadcrumb">
<li class="breadcrumb-item"><a href="../../../../../../index.html" class="icon icon-home"> Home</a></li>
<li class="breadcrumb-item"><a href="../../../../../index.html">Module code</a></li>
<li class="breadcrumb-item"><a href="../../../google.html">airflow.providers.google</a></li>
<li class="breadcrumb-item"><a href="dataproc.html"> airflow.providers.google.cloud.operators.dataproc</a></li>
</ul>
</div>
<div class="rst-content">
<div class="document">
<div class="documentwrapper">
<div class="bodywrapper">
<div class="body" role="main">
<h1>Source code for airflow.providers.google.cloud.operators.dataproc</h1><div class="highlight"><pre>
<span></span><span class="c1">#</span>
<span class="c1"># Licensed to the Apache Software Foundation (ASF) under one</span>
<span class="c1"># or more contributor license agreements. See the NOTICE file</span>
<span class="c1"># distributed with this work for additional information</span>
<span class="c1"># regarding copyright ownership. The ASF licenses this file</span>
<span class="c1"># to you under the Apache License, Version 2.0 (the</span>
<span class="c1"># &quot;License&quot;); you may not use this file except in compliance</span>
<span class="c1"># with the License. You may obtain a copy of the License at</span>
<span class="c1">#</span>
<span class="c1"># http://www.apache.org/licenses/LICENSE-2.0</span>
<span class="c1">#</span>
<span class="c1"># Unless required by applicable law or agreed to in writing,</span>
<span class="c1"># software distributed under the License is distributed on an</span>
<span class="c1"># &quot;AS IS&quot; BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY</span>
<span class="c1"># KIND, either express or implied. See the License for the</span>
<span class="c1"># specific language governing permissions and limitations</span>
<span class="c1"># under the License.</span>
<span class="sd">&quot;&quot;&quot;This module contains Google Dataproc operators.&quot;&quot;&quot;</span>
<span class="kn">from</span> <span class="nn">__future__</span> <span class="kn">import</span> <span class="n">annotations</span>
<span class="kn">import</span> <span class="nn">inspect</span>
<span class="kn">import</span> <span class="nn">ntpath</span>
<span class="kn">import</span> <span class="nn">os</span>
<span class="kn">import</span> <span class="nn">re</span>
<span class="kn">import</span> <span class="nn">time</span>
<span class="kn">import</span> <span class="nn">uuid</span>
<span class="kn">import</span> <span class="nn">warnings</span>
<span class="kn">from</span> <span class="nn">datetime</span> <span class="kn">import</span> <span class="n">datetime</span><span class="p">,</span> <span class="n">timedelta</span>
<span class="kn">from</span> <span class="nn">typing</span> <span class="kn">import</span> <span class="n">TYPE_CHECKING</span><span class="p">,</span> <span class="n">Sequence</span>
<span class="kn">from</span> <span class="nn">google.api_core</span> <span class="kn">import</span> <span class="n">operation</span> <span class="c1"># type: ignore</span>
<span class="kn">from</span> <span class="nn">google.api_core.exceptions</span> <span class="kn">import</span> <span class="n">AlreadyExists</span><span class="p">,</span> <span class="n">NotFound</span>
<span class="kn">from</span> <span class="nn">google.api_core.gapic_v1.method</span> <span class="kn">import</span> <span class="n">DEFAULT</span><span class="p">,</span> <span class="n">_MethodDefault</span>
<span class="kn">from</span> <span class="nn">google.api_core.retry</span> <span class="kn">import</span> <span class="n">Retry</span><span class="p">,</span> <span class="n">exponential_sleep_generator</span>
<span class="kn">from</span> <span class="nn">google.cloud.dataproc_v1</span> <span class="kn">import</span> <span class="n">Batch</span><span class="p">,</span> <span class="n">Cluster</span><span class="p">,</span> <span class="n">JobStatus</span>
<span class="kn">from</span> <span class="nn">google.protobuf.duration_pb2</span> <span class="kn">import</span> <span class="n">Duration</span>
<span class="kn">from</span> <span class="nn">google.protobuf.field_mask_pb2</span> <span class="kn">import</span> <span class="n">FieldMask</span>
<span class="kn">from</span> <span class="nn">airflow.exceptions</span> <span class="kn">import</span> <span class="n">AirflowException</span>
<span class="kn">from</span> <span class="nn">airflow.models</span> <span class="kn">import</span> <span class="n">BaseOperator</span>
<span class="kn">from</span> <span class="nn">airflow.providers.google.cloud.hooks.dataproc</span> <span class="kn">import</span> <span class="n">DataprocHook</span><span class="p">,</span> <span class="n">DataProcJobBuilder</span>
<span class="kn">from</span> <span class="nn">airflow.providers.google.cloud.hooks.gcs</span> <span class="kn">import</span> <span class="n">GCSHook</span>
<span class="kn">from</span> <span class="nn">airflow.providers.google.cloud.links.dataproc</span> <span class="kn">import</span> <span class="p">(</span>
<span class="n">DATAPROC_BATCH_LINK</span><span class="p">,</span>
<span class="n">DATAPROC_BATCHES_LINK</span><span class="p">,</span>
<span class="n">DATAPROC_CLUSTER_LINK</span><span class="p">,</span>
<span class="n">DATAPROC_JOB_LOG_LINK</span><span class="p">,</span>
<span class="n">DATAPROC_WORKFLOW_LINK</span><span class="p">,</span>
<span class="n">DATAPROC_WORKFLOW_TEMPLATE_LINK</span><span class="p">,</span>
<span class="n">DataprocLink</span><span class="p">,</span>
<span class="n">DataprocListLink</span><span class="p">,</span>
<span class="p">)</span>
<span class="kn">from</span> <span class="nn">airflow.providers.google.cloud.triggers.dataproc</span> <span class="kn">import</span> <span class="n">DataprocBaseTrigger</span>
<span class="kn">from</span> <span class="nn">airflow.utils</span> <span class="kn">import</span> <span class="n">timezone</span>
<span class="k">if</span> <span class="n">TYPE_CHECKING</span><span class="p">:</span>
<span class="kn">from</span> <span class="nn">airflow.utils.context</span> <span class="kn">import</span> <span class="n">Context</span>
<div class="viewcode-block" id="ClusterGenerator"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.ClusterGenerator">[docs]</a><span class="k">class</span> <span class="nc">ClusterGenerator</span><span class="p">:</span>
<span class="sd">&quot;&quot;&quot;</span>
<span class="sd"> Create a new Dataproc Cluster.</span>
<span class="sd"> :param cluster_name: The name of the DataProc cluster to create. (templated)</span>
<span class="sd"> :param project_id: The ID of the google cloud project in which</span>
<span class="sd"> to create the cluster. (templated)</span>
<span class="sd"> :param num_workers: The # of workers to spin up. If set to zero will</span>
<span class="sd"> spin up cluster in a single node mode</span>
<span class="sd"> :param storage_bucket: The storage bucket to use, setting to None lets dataproc</span>
<span class="sd"> generate a custom one for you</span>
<span class="sd"> :param init_actions_uris: List of GCS uri&#39;s containing</span>
<span class="sd"> dataproc initialization scripts</span>
<span class="sd"> :param init_action_timeout: Amount of time executable scripts in</span>
<span class="sd"> init_actions_uris has to complete</span>
<span class="sd"> :param metadata: dict of key-value google compute engine metadata entries</span>
<span class="sd"> to add to all instances</span>
<span class="sd"> :param image_version: the version of software inside the Dataproc cluster</span>
<span class="sd"> :param custom_image: custom Dataproc image for more info see</span>
<span class="sd"> https://cloud.google.com/dataproc/docs/guides/dataproc-images</span>
<span class="sd"> :param custom_image_project_id: project id for the custom Dataproc image, for more info see</span>
<span class="sd"> https://cloud.google.com/dataproc/docs/guides/dataproc-images</span>
<span class="sd"> :param custom_image_family: family for the custom Dataproc image,</span>
<span class="sd"> family name can be provide using --family flag while creating custom image, for more info see</span>
<span class="sd"> https://cloud.google.com/dataproc/docs/guides/dataproc-images</span>
<span class="sd"> :param autoscaling_policy: The autoscaling policy used by the cluster. Only resource names</span>
<span class="sd"> including projectid and location (region) are valid. Example:</span>
<span class="sd"> ``projects/[projectId]/locations/[dataproc_region]/autoscalingPolicies/[policy_id]``</span>
<span class="sd"> :param properties: dict of properties to set on</span>
<span class="sd"> config files (e.g. spark-defaults.conf), see</span>
<span class="sd"> https://cloud.google.com/dataproc/docs/reference/rest/v1/projects.regions.clusters#SoftwareConfig</span>
<span class="sd"> :param optional_components: List of optional cluster components, for more info see</span>
<span class="sd"> https://cloud.google.com/dataproc/docs/reference/rest/v1/ClusterConfig#Component</span>
<span class="sd"> :param num_masters: The # of master nodes to spin up</span>
<span class="sd"> :param master_machine_type: Compute engine machine type to use for the primary node</span>
<span class="sd"> :param master_disk_type: Type of the boot disk for the primary node</span>
<span class="sd"> (default is ``pd-standard``).</span>
<span class="sd"> Valid values: ``pd-ssd`` (Persistent Disk Solid State Drive) or</span>
<span class="sd"> ``pd-standard`` (Persistent Disk Hard Disk Drive).</span>
<span class="sd"> :param master_disk_size: Disk size for the primary node</span>
<span class="sd"> :param worker_machine_type: Compute engine machine type to use for the worker nodes</span>
<span class="sd"> :param worker_disk_type: Type of the boot disk for the worker node</span>
<span class="sd"> (default is ``pd-standard``).</span>
<span class="sd"> Valid values: ``pd-ssd`` (Persistent Disk Solid State Drive) or</span>
<span class="sd"> ``pd-standard`` (Persistent Disk Hard Disk Drive).</span>
<span class="sd"> :param worker_disk_size: Disk size for the worker nodes</span>
<span class="sd"> :param num_preemptible_workers: The # of preemptible worker nodes to spin up</span>
<span class="sd"> :param labels: dict of labels to add to the cluster</span>
<span class="sd"> :param zone: The zone where the cluster will be located. Set to None to auto-zone. (templated)</span>
<span class="sd"> :param network_uri: The network uri to be used for machine communication, cannot be</span>
<span class="sd"> specified with subnetwork_uri</span>
<span class="sd"> :param subnetwork_uri: The subnetwork uri to be used for machine communication,</span>
<span class="sd"> cannot be specified with network_uri</span>
<span class="sd"> :param internal_ip_only: If true, all instances in the cluster will only</span>
<span class="sd"> have internal IP addresses. This can only be enabled for subnetwork</span>
<span class="sd"> enabled networks</span>
<span class="sd"> :param tags: The GCE tags to add to all instances</span>
<span class="sd"> :param region: The specified region where the dataproc cluster is created.</span>
<span class="sd"> :param gcp_conn_id: The connection ID to use connecting to Google Cloud.</span>
<span class="sd"> :param service_account: The service account of the dataproc instances.</span>
<span class="sd"> :param service_account_scopes: The URIs of service account scopes to be included.</span>
<span class="sd"> :param idle_delete_ttl: The longest duration that cluster would keep alive while</span>
<span class="sd"> staying idle. Passing this threshold will cause cluster to be auto-deleted.</span>
<span class="sd"> A duration in seconds.</span>
<span class="sd"> :param auto_delete_time: The time when cluster will be auto-deleted.</span>
<span class="sd"> :param auto_delete_ttl: The life duration of cluster, the cluster will be</span>
<span class="sd"> auto-deleted at the end of this duration.</span>
<span class="sd"> A duration in seconds. (If auto_delete_time is set this parameter will be ignored)</span>
<span class="sd"> :param customer_managed_key: The customer-managed key used for disk encryption</span>
<span class="sd"> ``projects/[PROJECT_STORING_KEYS]/locations/[LOCATION]/keyRings/[KEY_RING_NAME]/cryptoKeys/[KEY_NAME]`` # noqa</span>
<span class="sd"> :param enable_component_gateway: Provides access to the web interfaces of default and selected optional</span>
<span class="sd"> components on the cluster.</span>
<span class="sd"> &quot;&quot;&quot;</span>
<span class="k">def</span> <span class="fm">__init__</span><span class="p">(</span>
<span class="bp">self</span><span class="p">,</span>
<span class="n">project_id</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span>
<span class="n">num_workers</span><span class="p">:</span> <span class="nb">int</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">zone</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">network_uri</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">subnetwork_uri</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">internal_ip_only</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">tags</span><span class="p">:</span> <span class="nb">list</span><span class="p">[</span><span class="nb">str</span><span class="p">]</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">storage_bucket</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">init_actions_uris</span><span class="p">:</span> <span class="nb">list</span><span class="p">[</span><span class="nb">str</span><span class="p">]</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">init_action_timeout</span><span class="p">:</span> <span class="nb">str</span> <span class="o">=</span> <span class="s2">&quot;10m&quot;</span><span class="p">,</span>
<span class="n">metadata</span><span class="p">:</span> <span class="nb">dict</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">custom_image</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">custom_image_project_id</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">custom_image_family</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">image_version</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">autoscaling_policy</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">properties</span><span class="p">:</span> <span class="nb">dict</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">optional_components</span><span class="p">:</span> <span class="nb">list</span><span class="p">[</span><span class="nb">str</span><span class="p">]</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">num_masters</span><span class="p">:</span> <span class="nb">int</span> <span class="o">=</span> <span class="mi">1</span><span class="p">,</span>
<span class="n">master_machine_type</span><span class="p">:</span> <span class="nb">str</span> <span class="o">=</span> <span class="s1">&#39;n1-standard-4&#39;</span><span class="p">,</span>
<span class="n">master_disk_type</span><span class="p">:</span> <span class="nb">str</span> <span class="o">=</span> <span class="s1">&#39;pd-standard&#39;</span><span class="p">,</span>
<span class="n">master_disk_size</span><span class="p">:</span> <span class="nb">int</span> <span class="o">=</span> <span class="mi">1024</span><span class="p">,</span>
<span class="n">worker_machine_type</span><span class="p">:</span> <span class="nb">str</span> <span class="o">=</span> <span class="s1">&#39;n1-standard-4&#39;</span><span class="p">,</span>
<span class="n">worker_disk_type</span><span class="p">:</span> <span class="nb">str</span> <span class="o">=</span> <span class="s1">&#39;pd-standard&#39;</span><span class="p">,</span>
<span class="n">worker_disk_size</span><span class="p">:</span> <span class="nb">int</span> <span class="o">=</span> <span class="mi">1024</span><span class="p">,</span>
<span class="n">num_preemptible_workers</span><span class="p">:</span> <span class="nb">int</span> <span class="o">=</span> <span class="mi">0</span><span class="p">,</span>
<span class="n">service_account</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">service_account_scopes</span><span class="p">:</span> <span class="nb">list</span><span class="p">[</span><span class="nb">str</span><span class="p">]</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">idle_delete_ttl</span><span class="p">:</span> <span class="nb">int</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">auto_delete_time</span><span class="p">:</span> <span class="n">datetime</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">auto_delete_ttl</span><span class="p">:</span> <span class="nb">int</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">customer_managed_key</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">enable_component_gateway</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">False</span><span class="p">,</span>
<span class="o">**</span><span class="n">kwargs</span><span class="p">,</span>
<span class="p">)</span> <span class="o">-&gt;</span> <span class="kc">None</span><span class="p">:</span>
<span class="bp">self</span><span class="o">.</span><span class="n">project_id</span> <span class="o">=</span> <span class="n">project_id</span>
<span class="bp">self</span><span class="o">.</span><span class="n">num_masters</span> <span class="o">=</span> <span class="n">num_masters</span>
<span class="bp">self</span><span class="o">.</span><span class="n">num_workers</span> <span class="o">=</span> <span class="n">num_workers</span>
<span class="bp">self</span><span class="o">.</span><span class="n">num_preemptible_workers</span> <span class="o">=</span> <span class="n">num_preemptible_workers</span>
<span class="bp">self</span><span class="o">.</span><span class="n">storage_bucket</span> <span class="o">=</span> <span class="n">storage_bucket</span>
<span class="bp">self</span><span class="o">.</span><span class="n">init_actions_uris</span> <span class="o">=</span> <span class="n">init_actions_uris</span>
<span class="bp">self</span><span class="o">.</span><span class="n">init_action_timeout</span> <span class="o">=</span> <span class="n">init_action_timeout</span>
<span class="bp">self</span><span class="o">.</span><span class="n">metadata</span> <span class="o">=</span> <span class="n">metadata</span>
<span class="bp">self</span><span class="o">.</span><span class="n">custom_image</span> <span class="o">=</span> <span class="n">custom_image</span>
<span class="bp">self</span><span class="o">.</span><span class="n">custom_image_project_id</span> <span class="o">=</span> <span class="n">custom_image_project_id</span>
<span class="bp">self</span><span class="o">.</span><span class="n">custom_image_family</span> <span class="o">=</span> <span class="n">custom_image_family</span>
<span class="bp">self</span><span class="o">.</span><span class="n">image_version</span> <span class="o">=</span> <span class="n">image_version</span>
<span class="bp">self</span><span class="o">.</span><span class="n">properties</span> <span class="o">=</span> <span class="n">properties</span> <span class="ow">or</span> <span class="p">{}</span>
<span class="bp">self</span><span class="o">.</span><span class="n">optional_components</span> <span class="o">=</span> <span class="n">optional_components</span>
<span class="bp">self</span><span class="o">.</span><span class="n">master_machine_type</span> <span class="o">=</span> <span class="n">master_machine_type</span>
<span class="bp">self</span><span class="o">.</span><span class="n">master_disk_type</span> <span class="o">=</span> <span class="n">master_disk_type</span>
<span class="bp">self</span><span class="o">.</span><span class="n">master_disk_size</span> <span class="o">=</span> <span class="n">master_disk_size</span>
<span class="bp">self</span><span class="o">.</span><span class="n">autoscaling_policy</span> <span class="o">=</span> <span class="n">autoscaling_policy</span>
<span class="bp">self</span><span class="o">.</span><span class="n">worker_machine_type</span> <span class="o">=</span> <span class="n">worker_machine_type</span>
<span class="bp">self</span><span class="o">.</span><span class="n">worker_disk_type</span> <span class="o">=</span> <span class="n">worker_disk_type</span>
<span class="bp">self</span><span class="o">.</span><span class="n">worker_disk_size</span> <span class="o">=</span> <span class="n">worker_disk_size</span>
<span class="bp">self</span><span class="o">.</span><span class="n">zone</span> <span class="o">=</span> <span class="n">zone</span>
<span class="bp">self</span><span class="o">.</span><span class="n">network_uri</span> <span class="o">=</span> <span class="n">network_uri</span>
<span class="bp">self</span><span class="o">.</span><span class="n">subnetwork_uri</span> <span class="o">=</span> <span class="n">subnetwork_uri</span>
<span class="bp">self</span><span class="o">.</span><span class="n">internal_ip_only</span> <span class="o">=</span> <span class="n">internal_ip_only</span>
<span class="bp">self</span><span class="o">.</span><span class="n">tags</span> <span class="o">=</span> <span class="n">tags</span>
<span class="bp">self</span><span class="o">.</span><span class="n">service_account</span> <span class="o">=</span> <span class="n">service_account</span>
<span class="bp">self</span><span class="o">.</span><span class="n">service_account_scopes</span> <span class="o">=</span> <span class="n">service_account_scopes</span>
<span class="bp">self</span><span class="o">.</span><span class="n">idle_delete_ttl</span> <span class="o">=</span> <span class="n">idle_delete_ttl</span>
<span class="bp">self</span><span class="o">.</span><span class="n">auto_delete_time</span> <span class="o">=</span> <span class="n">auto_delete_time</span>
<span class="bp">self</span><span class="o">.</span><span class="n">auto_delete_ttl</span> <span class="o">=</span> <span class="n">auto_delete_ttl</span>
<span class="bp">self</span><span class="o">.</span><span class="n">customer_managed_key</span> <span class="o">=</span> <span class="n">customer_managed_key</span>
<span class="bp">self</span><span class="o">.</span><span class="n">enable_component_gateway</span> <span class="o">=</span> <span class="n">enable_component_gateway</span>
<span class="bp">self</span><span class="o">.</span><span class="n">single_node</span> <span class="o">=</span> <span class="n">num_workers</span> <span class="o">==</span> <span class="mi">0</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">custom_image</span> <span class="ow">and</span> <span class="bp">self</span><span class="o">.</span><span class="n">image_version</span><span class="p">:</span>
<span class="k">raise</span> <span class="ne">ValueError</span><span class="p">(</span><span class="s2">&quot;The custom_image and image_version can&#39;t be both set&quot;</span><span class="p">)</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">custom_image_family</span> <span class="ow">and</span> <span class="bp">self</span><span class="o">.</span><span class="n">image_version</span><span class="p">:</span>
<span class="k">raise</span> <span class="ne">ValueError</span><span class="p">(</span><span class="s2">&quot;The image_version and custom_image_family can&#39;t be both set&quot;</span><span class="p">)</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">custom_image_family</span> <span class="ow">and</span> <span class="bp">self</span><span class="o">.</span><span class="n">custom_image</span><span class="p">:</span>
<span class="k">raise</span> <span class="ne">ValueError</span><span class="p">(</span><span class="s2">&quot;The custom_image and custom_image_family can&#39;t be both set&quot;</span><span class="p">)</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">single_node</span> <span class="ow">and</span> <span class="bp">self</span><span class="o">.</span><span class="n">num_preemptible_workers</span> <span class="o">&gt;</span> <span class="mi">0</span><span class="p">:</span>
<span class="k">raise</span> <span class="ne">ValueError</span><span class="p">(</span><span class="s2">&quot;Single node cannot have preemptible workers.&quot;</span><span class="p">)</span>
<span class="k">def</span> <span class="nf">_get_init_action_timeout</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="nb">dict</span><span class="p">:</span>
<span class="n">match</span> <span class="o">=</span> <span class="n">re</span><span class="o">.</span><span class="n">match</span><span class="p">(</span><span class="sa">r</span><span class="s2">&quot;^(\d+)([sm])$&quot;</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">init_action_timeout</span><span class="p">)</span>
<span class="k">if</span> <span class="n">match</span><span class="p">:</span>
<span class="n">val</span> <span class="o">=</span> <span class="nb">float</span><span class="p">(</span><span class="n">match</span><span class="o">.</span><span class="n">group</span><span class="p">(</span><span class="mi">1</span><span class="p">))</span>
<span class="k">if</span> <span class="n">match</span><span class="o">.</span><span class="n">group</span><span class="p">(</span><span class="mi">2</span><span class="p">)</span> <span class="o">==</span> <span class="s2">&quot;s&quot;</span><span class="p">:</span>
<span class="k">return</span> <span class="p">{</span><span class="s2">&quot;seconds&quot;</span><span class="p">:</span> <span class="nb">int</span><span class="p">(</span><span class="n">val</span><span class="p">)}</span>
<span class="k">elif</span> <span class="n">match</span><span class="o">.</span><span class="n">group</span><span class="p">(</span><span class="mi">2</span><span class="p">)</span> <span class="o">==</span> <span class="s2">&quot;m&quot;</span><span class="p">:</span>
<span class="k">return</span> <span class="p">{</span><span class="s2">&quot;seconds&quot;</span><span class="p">:</span> <span class="nb">int</span><span class="p">(</span><span class="n">timedelta</span><span class="p">(</span><span class="n">minutes</span><span class="o">=</span><span class="n">val</span><span class="p">)</span><span class="o">.</span><span class="n">total_seconds</span><span class="p">())}</span>
<span class="k">raise</span> <span class="n">AirflowException</span><span class="p">(</span>
<span class="s2">&quot;DataprocClusterCreateOperator init_action_timeout&quot;</span>
<span class="s2">&quot; should be expressed in minutes or seconds. i.e. 10m, 30s&quot;</span>
<span class="p">)</span>
<span class="k">def</span> <span class="nf">_build_gce_cluster_config</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">cluster_data</span><span class="p">):</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">zone</span><span class="p">:</span>
<span class="n">zone_uri</span> <span class="o">=</span> <span class="sa">f</span><span class="s1">&#39;https://www.googleapis.com/compute/v1/projects/</span><span class="si">{</span><span class="bp">self</span><span class="o">.</span><span class="n">project_id</span><span class="si">}</span><span class="s1">/zones/</span><span class="si">{</span><span class="bp">self</span><span class="o">.</span><span class="n">zone</span><span class="si">}</span><span class="s1">&#39;</span>
<span class="n">cluster_data</span><span class="p">[</span><span class="s1">&#39;gce_cluster_config&#39;</span><span class="p">][</span><span class="s1">&#39;zone_uri&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="n">zone_uri</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">metadata</span><span class="p">:</span>
<span class="n">cluster_data</span><span class="p">[</span><span class="s1">&#39;gce_cluster_config&#39;</span><span class="p">][</span><span class="s1">&#39;metadata&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">metadata</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">network_uri</span><span class="p">:</span>
<span class="n">cluster_data</span><span class="p">[</span><span class="s1">&#39;gce_cluster_config&#39;</span><span class="p">][</span><span class="s1">&#39;network_uri&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">network_uri</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">subnetwork_uri</span><span class="p">:</span>
<span class="n">cluster_data</span><span class="p">[</span><span class="s1">&#39;gce_cluster_config&#39;</span><span class="p">][</span><span class="s1">&#39;subnetwork_uri&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">subnetwork_uri</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">internal_ip_only</span><span class="p">:</span>
<span class="k">if</span> <span class="ow">not</span> <span class="bp">self</span><span class="o">.</span><span class="n">subnetwork_uri</span><span class="p">:</span>
<span class="k">raise</span> <span class="n">AirflowException</span><span class="p">(</span><span class="s2">&quot;Set internal_ip_only to true only when you pass a subnetwork_uri.&quot;</span><span class="p">)</span>
<span class="n">cluster_data</span><span class="p">[</span><span class="s1">&#39;gce_cluster_config&#39;</span><span class="p">][</span><span class="s1">&#39;internal_ip_only&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="kc">True</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">tags</span><span class="p">:</span>
<span class="n">cluster_data</span><span class="p">[</span><span class="s1">&#39;gce_cluster_config&#39;</span><span class="p">][</span><span class="s1">&#39;tags&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">tags</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">service_account</span><span class="p">:</span>
<span class="n">cluster_data</span><span class="p">[</span><span class="s1">&#39;gce_cluster_config&#39;</span><span class="p">][</span><span class="s1">&#39;service_account&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">service_account</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">service_account_scopes</span><span class="p">:</span>
<span class="n">cluster_data</span><span class="p">[</span><span class="s1">&#39;gce_cluster_config&#39;</span><span class="p">][</span><span class="s1">&#39;service_account_scopes&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">service_account_scopes</span>
<span class="k">return</span> <span class="n">cluster_data</span>
<span class="k">def</span> <span class="nf">_build_lifecycle_config</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">cluster_data</span><span class="p">):</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">idle_delete_ttl</span><span class="p">:</span>
<span class="n">cluster_data</span><span class="p">[</span><span class="s1">&#39;lifecycle_config&#39;</span><span class="p">][</span><span class="s1">&#39;idle_delete_ttl&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="p">{</span><span class="s2">&quot;seconds&quot;</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">idle_delete_ttl</span><span class="p">}</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">auto_delete_time</span><span class="p">:</span>
<span class="n">utc_auto_delete_time</span> <span class="o">=</span> <span class="n">timezone</span><span class="o">.</span><span class="n">convert_to_utc</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">auto_delete_time</span><span class="p">)</span>
<span class="n">cluster_data</span><span class="p">[</span><span class="s1">&#39;lifecycle_config&#39;</span><span class="p">][</span><span class="s1">&#39;auto_delete_time&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="n">utc_auto_delete_time</span><span class="o">.</span><span class="n">strftime</span><span class="p">(</span>
<span class="s1">&#39;%Y-%m-</span><span class="si">%d</span><span class="s1">T%H:%M:%S.</span><span class="si">%f</span><span class="s1">Z&#39;</span>
<span class="p">)</span>
<span class="k">elif</span> <span class="bp">self</span><span class="o">.</span><span class="n">auto_delete_ttl</span><span class="p">:</span>
<span class="n">cluster_data</span><span class="p">[</span><span class="s1">&#39;lifecycle_config&#39;</span><span class="p">][</span><span class="s1">&#39;auto_delete_ttl&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="p">{</span><span class="s2">&quot;seconds&quot;</span><span class="p">:</span> <span class="nb">int</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">auto_delete_ttl</span><span class="p">)}</span>
<span class="k">return</span> <span class="n">cluster_data</span>
<span class="k">def</span> <span class="nf">_build_cluster_data</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">zone</span><span class="p">:</span>
<span class="n">master_type_uri</span> <span class="o">=</span> <span class="p">(</span>
<span class="sa">f</span><span class="s2">&quot;projects/</span><span class="si">{</span><span class="bp">self</span><span class="o">.</span><span class="n">project_id</span><span class="si">}</span><span class="s2">/zones/</span><span class="si">{</span><span class="bp">self</span><span class="o">.</span><span class="n">zone</span><span class="si">}</span><span class="s2">/machineTypes/</span><span class="si">{</span><span class="bp">self</span><span class="o">.</span><span class="n">master_machine_type</span><span class="si">}</span><span class="s2">&quot;</span>
<span class="p">)</span>
<span class="n">worker_type_uri</span> <span class="o">=</span> <span class="p">(</span>
<span class="sa">f</span><span class="s2">&quot;projects/</span><span class="si">{</span><span class="bp">self</span><span class="o">.</span><span class="n">project_id</span><span class="si">}</span><span class="s2">/zones/</span><span class="si">{</span><span class="bp">self</span><span class="o">.</span><span class="n">zone</span><span class="si">}</span><span class="s2">/machineTypes/</span><span class="si">{</span><span class="bp">self</span><span class="o">.</span><span class="n">worker_machine_type</span><span class="si">}</span><span class="s2">&quot;</span>
<span class="p">)</span>
<span class="k">else</span><span class="p">:</span>
<span class="n">master_type_uri</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">master_machine_type</span>
<span class="n">worker_type_uri</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">worker_machine_type</span>
<span class="n">cluster_data</span> <span class="o">=</span> <span class="p">{</span>
<span class="s1">&#39;gce_cluster_config&#39;</span><span class="p">:</span> <span class="p">{},</span>
<span class="s1">&#39;master_config&#39;</span><span class="p">:</span> <span class="p">{</span>
<span class="s1">&#39;num_instances&#39;</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">num_masters</span><span class="p">,</span>
<span class="s1">&#39;machine_type_uri&#39;</span><span class="p">:</span> <span class="n">master_type_uri</span><span class="p">,</span>
<span class="s1">&#39;disk_config&#39;</span><span class="p">:</span> <span class="p">{</span>
<span class="s1">&#39;boot_disk_type&#39;</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">master_disk_type</span><span class="p">,</span>
<span class="s1">&#39;boot_disk_size_gb&#39;</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">master_disk_size</span><span class="p">,</span>
<span class="p">},</span>
<span class="p">},</span>
<span class="s1">&#39;worker_config&#39;</span><span class="p">:</span> <span class="p">{</span>
<span class="s1">&#39;num_instances&#39;</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">num_workers</span><span class="p">,</span>
<span class="s1">&#39;machine_type_uri&#39;</span><span class="p">:</span> <span class="n">worker_type_uri</span><span class="p">,</span>
<span class="s1">&#39;disk_config&#39;</span><span class="p">:</span> <span class="p">{</span>
<span class="s1">&#39;boot_disk_type&#39;</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">worker_disk_type</span><span class="p">,</span>
<span class="s1">&#39;boot_disk_size_gb&#39;</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">worker_disk_size</span><span class="p">,</span>
<span class="p">},</span>
<span class="p">},</span>
<span class="s1">&#39;secondary_worker_config&#39;</span><span class="p">:</span> <span class="p">{},</span>
<span class="s1">&#39;software_config&#39;</span><span class="p">:</span> <span class="p">{},</span>
<span class="s1">&#39;lifecycle_config&#39;</span><span class="p">:</span> <span class="p">{},</span>
<span class="s1">&#39;encryption_config&#39;</span><span class="p">:</span> <span class="p">{},</span>
<span class="s1">&#39;autoscaling_config&#39;</span><span class="p">:</span> <span class="p">{},</span>
<span class="s1">&#39;endpoint_config&#39;</span><span class="p">:</span> <span class="p">{},</span>
<span class="p">}</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">num_preemptible_workers</span> <span class="o">&gt;</span> <span class="mi">0</span><span class="p">:</span>
<span class="n">cluster_data</span><span class="p">[</span><span class="s1">&#39;secondary_worker_config&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="p">{</span>
<span class="s1">&#39;num_instances&#39;</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">num_preemptible_workers</span><span class="p">,</span>
<span class="s1">&#39;machine_type_uri&#39;</span><span class="p">:</span> <span class="n">worker_type_uri</span><span class="p">,</span>
<span class="s1">&#39;disk_config&#39;</span><span class="p">:</span> <span class="p">{</span>
<span class="s1">&#39;boot_disk_type&#39;</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">worker_disk_type</span><span class="p">,</span>
<span class="s1">&#39;boot_disk_size_gb&#39;</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">worker_disk_size</span><span class="p">,</span>
<span class="p">},</span>
<span class="s1">&#39;is_preemptible&#39;</span><span class="p">:</span> <span class="kc">True</span><span class="p">,</span>
<span class="p">}</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">storage_bucket</span><span class="p">:</span>
<span class="n">cluster_data</span><span class="p">[</span><span class="s1">&#39;config_bucket&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">storage_bucket</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">image_version</span><span class="p">:</span>
<span class="n">cluster_data</span><span class="p">[</span><span class="s1">&#39;software_config&#39;</span><span class="p">][</span><span class="s1">&#39;image_version&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">image_version</span>
<span class="k">elif</span> <span class="bp">self</span><span class="o">.</span><span class="n">custom_image</span><span class="p">:</span>
<span class="n">project_id</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">custom_image_project_id</span> <span class="ow">or</span> <span class="bp">self</span><span class="o">.</span><span class="n">project_id</span>
<span class="n">custom_image_url</span> <span class="o">=</span> <span class="p">(</span>
<span class="sa">f</span><span class="s1">&#39;https://www.googleapis.com/compute/beta/projects/</span><span class="si">{</span><span class="n">project_id</span><span class="si">}</span><span class="s1">&#39;</span>
<span class="sa">f</span><span class="s1">&#39;/global/images/</span><span class="si">{</span><span class="bp">self</span><span class="o">.</span><span class="n">custom_image</span><span class="si">}</span><span class="s1">&#39;</span>
<span class="p">)</span>
<span class="n">cluster_data</span><span class="p">[</span><span class="s1">&#39;master_config&#39;</span><span class="p">][</span><span class="s1">&#39;image_uri&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="n">custom_image_url</span>
<span class="k">if</span> <span class="ow">not</span> <span class="bp">self</span><span class="o">.</span><span class="n">single_node</span><span class="p">:</span>
<span class="n">cluster_data</span><span class="p">[</span><span class="s1">&#39;worker_config&#39;</span><span class="p">][</span><span class="s1">&#39;image_uri&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="n">custom_image_url</span>
<span class="k">elif</span> <span class="bp">self</span><span class="o">.</span><span class="n">custom_image_family</span><span class="p">:</span>
<span class="n">project_id</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">custom_image_project_id</span> <span class="ow">or</span> <span class="bp">self</span><span class="o">.</span><span class="n">project_id</span>
<span class="n">custom_image_url</span> <span class="o">=</span> <span class="p">(</span>
<span class="s1">&#39;https://www.googleapis.com/compute/beta/projects/&#39;</span>
<span class="sa">f</span><span class="s1">&#39;</span><span class="si">{</span><span class="n">project_id</span><span class="si">}</span><span class="s1">/global/images/family/</span><span class="si">{</span><span class="bp">self</span><span class="o">.</span><span class="n">custom_image_family</span><span class="si">}</span><span class="s1">&#39;</span>
<span class="p">)</span>
<span class="n">cluster_data</span><span class="p">[</span><span class="s1">&#39;master_config&#39;</span><span class="p">][</span><span class="s1">&#39;image_uri&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="n">custom_image_url</span>
<span class="k">if</span> <span class="ow">not</span> <span class="bp">self</span><span class="o">.</span><span class="n">single_node</span><span class="p">:</span>
<span class="n">cluster_data</span><span class="p">[</span><span class="s1">&#39;worker_config&#39;</span><span class="p">][</span><span class="s1">&#39;image_uri&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="n">custom_image_url</span>
<span class="n">cluster_data</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_build_gce_cluster_config</span><span class="p">(</span><span class="n">cluster_data</span><span class="p">)</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">single_node</span><span class="p">:</span>
<span class="bp">self</span><span class="o">.</span><span class="n">properties</span><span class="p">[</span><span class="s2">&quot;dataproc:dataproc.allow.zero.workers&quot;</span><span class="p">]</span> <span class="o">=</span> <span class="s2">&quot;true&quot;</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">properties</span><span class="p">:</span>
<span class="n">cluster_data</span><span class="p">[</span><span class="s1">&#39;software_config&#39;</span><span class="p">][</span><span class="s1">&#39;properties&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">properties</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">optional_components</span><span class="p">:</span>
<span class="n">cluster_data</span><span class="p">[</span><span class="s1">&#39;software_config&#39;</span><span class="p">][</span><span class="s1">&#39;optional_components&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">optional_components</span>
<span class="n">cluster_data</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_build_lifecycle_config</span><span class="p">(</span><span class="n">cluster_data</span><span class="p">)</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">init_actions_uris</span><span class="p">:</span>
<span class="n">init_actions_dict</span> <span class="o">=</span> <span class="p">[</span>
<span class="p">{</span><span class="s1">&#39;executable_file&#39;</span><span class="p">:</span> <span class="n">uri</span><span class="p">,</span> <span class="s1">&#39;execution_timeout&#39;</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_get_init_action_timeout</span><span class="p">()}</span>
<span class="k">for</span> <span class="n">uri</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">init_actions_uris</span>
<span class="p">]</span>
<span class="n">cluster_data</span><span class="p">[</span><span class="s1">&#39;initialization_actions&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="n">init_actions_dict</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">customer_managed_key</span><span class="p">:</span>
<span class="n">cluster_data</span><span class="p">[</span><span class="s1">&#39;encryption_config&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="p">{</span><span class="s1">&#39;gce_pd_kms_key_name&#39;</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">customer_managed_key</span><span class="p">}</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">autoscaling_policy</span><span class="p">:</span>
<span class="n">cluster_data</span><span class="p">[</span><span class="s1">&#39;autoscaling_config&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="p">{</span><span class="s1">&#39;policy_uri&#39;</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">autoscaling_policy</span><span class="p">}</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">enable_component_gateway</span><span class="p">:</span>
<span class="n">cluster_data</span><span class="p">[</span><span class="s1">&#39;endpoint_config&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="p">{</span><span class="s1">&#39;enable_http_port_access&#39;</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">enable_component_gateway</span><span class="p">}</span>
<span class="k">return</span> <span class="n">cluster_data</span>
<div class="viewcode-block" id="ClusterGenerator.make"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.ClusterGenerator.make">[docs]</a> <span class="k">def</span> <span class="nf">make</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
<span class="sd">&quot;&quot;&quot;</span>
<span class="sd"> Helper method for easier migration.</span>
<span class="sd"> :return: Dict representing Dataproc cluster.</span>
<span class="sd"> &quot;&quot;&quot;</span>
<span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">_build_cluster_data</span><span class="p">()</span></div></div>
<div class="viewcode-block" id="DataprocCreateClusterOperator"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocCreateClusterOperator">[docs]</a><span class="k">class</span> <span class="nc">DataprocCreateClusterOperator</span><span class="p">(</span><span class="n">BaseOperator</span><span class="p">):</span>
<span class="sd">&quot;&quot;&quot;</span>
<span class="sd"> Create a new cluster on Google Cloud Dataproc. The operator will wait until the</span>
<span class="sd"> creation is successful or an error occurs in the creation process. If the cluster</span>
<span class="sd"> already exists and ``use_if_exists`` is True then the operator will:</span>
<span class="sd"> - if cluster state is ERROR then delete it if specified and raise error</span>
<span class="sd"> - if cluster state is CREATING wait for it and then check for ERROR state</span>
<span class="sd"> - if cluster state is DELETING wait for it and then create new cluster</span>
<span class="sd"> Please refer to</span>
<span class="sd"> https://cloud.google.com/dataproc/docs/reference/rest/v1/projects.regions.clusters</span>
<span class="sd"> for a detailed explanation on the different parameters. Most of the configuration</span>
<span class="sd"> parameters detailed in the link are available as a parameter to this operator.</span>
<span class="sd"> .. seealso::</span>
<span class="sd"> For more information on how to use this operator, take a look at the guide:</span>
<span class="sd"> :ref:`howto/operator:DataprocCreateClusterOperator`</span>
<span class="sd"> :param project_id: The ID of the google cloud project in which</span>
<span class="sd"> to create the cluster. (templated)</span>
<span class="sd"> :param cluster_name: Name of the cluster to create</span>
<span class="sd"> :param labels: Labels that will be assigned to created cluster</span>
<span class="sd"> :param cluster_config: Required. The cluster config to create.</span>
<span class="sd"> If a dict is provided, it must be of the same form as the protobuf message</span>
<span class="sd"> :class:`~google.cloud.dataproc_v1.types.ClusterConfig`</span>
<span class="sd"> :param virtual_cluster_config: Optional. The virtual cluster config, used when creating a Dataproc</span>
<span class="sd"> cluster that does not directly control the underlying compute resources, for example, when creating a</span>
<span class="sd"> `Dataproc-on-GKE cluster</span>
<span class="sd"> &lt;https://cloud.google.com/dataproc/docs/concepts/jobs/dataproc-gke#create-a-dataproc-on-gke-cluster&gt;`</span>
<span class="sd"> :param region: The specified region where the dataproc cluster is created.</span>
<span class="sd"> :param delete_on_error: If true the cluster will be deleted if created with ERROR state. Default</span>
<span class="sd"> value is true.</span>
<span class="sd"> :param use_if_exists: If true use existing cluster</span>
<span class="sd"> :param request_id: Optional. A unique id used to identify the request. If the server receives two</span>
<span class="sd"> ``DeleteClusterRequest`` requests with the same id, then the second request will be ignored and the</span>
<span class="sd"> first ``google.longrunning.Operation`` created and stored in the backend is returned.</span>
<span class="sd"> :param retry: A retry object used to retry requests. If ``None`` is specified, requests will not be</span>
<span class="sd"> retried.</span>
<span class="sd"> :param timeout: The amount of time, in seconds, to wait for the request to complete. Note that if</span>
<span class="sd"> ``retry`` is specified, the timeout applies to each individual attempt.</span>
<span class="sd"> :param metadata: Additional metadata that is provided to the method.</span>
<span class="sd"> :param gcp_conn_id: The connection ID to use connecting to Google Cloud.</span>
<span class="sd"> :param impersonation_chain: Optional service account to impersonate using short-term</span>
<span class="sd"> credentials, or chained list of accounts required to get the access_token</span>
<span class="sd"> of the last account in the list, which will be impersonated in the request.</span>
<span class="sd"> If set as a string, the account must grant the originating account</span>
<span class="sd"> the Service Account Token Creator IAM role.</span>
<span class="sd"> If set as a sequence, the identities from the list must grant</span>
<span class="sd"> Service Account Token Creator IAM role to the directly preceding identity, with first</span>
<span class="sd"> account from the list granting this role to the originating account (templated).</span>
<span class="sd"> &quot;&quot;&quot;</span>
<div class="viewcode-block" id="DataprocCreateClusterOperator.template_fields"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocCreateClusterOperator.template_fields">[docs]</a> <span class="n">template_fields</span><span class="p">:</span> <span class="n">Sequence</span><span class="p">[</span><span class="nb">str</span><span class="p">]</span> <span class="o">=</span> <span class="p">(</span>
<span class="s1">&#39;project_id&#39;</span><span class="p">,</span>
<span class="s1">&#39;region&#39;</span><span class="p">,</span>
<span class="s1">&#39;cluster_config&#39;</span><span class="p">,</span>
<span class="s1">&#39;virtual_cluster_config&#39;</span><span class="p">,</span>
<span class="s1">&#39;cluster_name&#39;</span><span class="p">,</span>
<span class="s1">&#39;labels&#39;</span><span class="p">,</span>
<span class="s1">&#39;impersonation_chain&#39;</span><span class="p">,</span></div>
<span class="p">)</span>
<div class="viewcode-block" id="DataprocCreateClusterOperator.template_fields_renderers"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocCreateClusterOperator.template_fields_renderers">[docs]</a> <span class="n">template_fields_renderers</span> <span class="o">=</span> <span class="p">{</span><span class="s1">&#39;cluster_config&#39;</span><span class="p">:</span> <span class="s1">&#39;json&#39;</span><span class="p">,</span> <span class="s1">&#39;virtual_cluster_config&#39;</span><span class="p">:</span> <span class="s1">&#39;json&#39;</span><span class="p">}</span></div>
<div class="viewcode-block" id="DataprocCreateClusterOperator.operator_extra_links"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocCreateClusterOperator.operator_extra_links">[docs]</a> <span class="n">operator_extra_links</span> <span class="o">=</span> <span class="p">(</span><span class="n">DataprocLink</span><span class="p">(),)</span></div>
<span class="k">def</span> <span class="fm">__init__</span><span class="p">(</span>
<span class="bp">self</span><span class="p">,</span>
<span class="o">*</span><span class="p">,</span>
<span class="n">cluster_name</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span>
<span class="n">region</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span>
<span class="n">project_id</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">cluster_config</span><span class="p">:</span> <span class="nb">dict</span> <span class="o">|</span> <span class="n">Cluster</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">virtual_cluster_config</span><span class="p">:</span> <span class="nb">dict</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">labels</span><span class="p">:</span> <span class="nb">dict</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">request_id</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">delete_on_error</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">True</span><span class="p">,</span>
<span class="n">use_if_exists</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">True</span><span class="p">,</span>
<span class="n">retry</span><span class="p">:</span> <span class="n">Retry</span> <span class="o">|</span> <span class="n">_MethodDefault</span> <span class="o">=</span> <span class="n">DEFAULT</span><span class="p">,</span>
<span class="n">timeout</span><span class="p">:</span> <span class="nb">float</span> <span class="o">=</span> <span class="mi">1</span> <span class="o">*</span> <span class="mi">60</span> <span class="o">*</span> <span class="mi">60</span><span class="p">,</span>
<span class="n">metadata</span><span class="p">:</span> <span class="n">Sequence</span><span class="p">[</span><span class="nb">tuple</span><span class="p">[</span><span class="nb">str</span><span class="p">,</span> <span class="nb">str</span><span class="p">]]</span> <span class="o">=</span> <span class="p">(),</span>
<span class="n">gcp_conn_id</span><span class="p">:</span> <span class="nb">str</span> <span class="o">=</span> <span class="s2">&quot;google_cloud_default&quot;</span><span class="p">,</span>
<span class="n">impersonation_chain</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="n">Sequence</span><span class="p">[</span><span class="nb">str</span><span class="p">]</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="o">**</span><span class="n">kwargs</span><span class="p">,</span>
<span class="p">)</span> <span class="o">-&gt;</span> <span class="kc">None</span><span class="p">:</span>
<span class="c1"># TODO: remove one day</span>
<span class="k">if</span> <span class="n">cluster_config</span> <span class="ow">is</span> <span class="kc">None</span> <span class="ow">and</span> <span class="n">virtual_cluster_config</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
<span class="n">warnings</span><span class="o">.</span><span class="n">warn</span><span class="p">(</span>
<span class="sa">f</span><span class="s2">&quot;Passing cluster parameters by keywords to `</span><span class="si">{</span><span class="nb">type</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span><span class="o">.</span><span class="vm">__name__</span><span class="si">}</span><span class="s2">` will be deprecated. &quot;</span>
<span class="s2">&quot;Please provide cluster_config object using `cluster_config` parameter. &quot;</span>
<span class="s2">&quot;You can use `airflow.dataproc.ClusterGenerator.generate_cluster` &quot;</span>
<span class="s2">&quot;method to obtain cluster object.&quot;</span><span class="p">,</span>
<span class="ne">DeprecationWarning</span><span class="p">,</span>
<span class="n">stacklevel</span><span class="o">=</span><span class="mi">1</span><span class="p">,</span>
<span class="p">)</span>
<span class="c1"># Remove result of apply defaults</span>
<span class="k">if</span> <span class="s1">&#39;params&#39;</span> <span class="ow">in</span> <span class="n">kwargs</span><span class="p">:</span>
<span class="k">del</span> <span class="n">kwargs</span><span class="p">[</span><span class="s1">&#39;params&#39;</span><span class="p">]</span>
<span class="c1"># Create cluster object from kwargs</span>
<span class="k">if</span> <span class="n">project_id</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
<span class="k">raise</span> <span class="n">AirflowException</span><span class="p">(</span>
<span class="s2">&quot;project_id argument is required when building cluster from keywords parameters&quot;</span>
<span class="p">)</span>
<span class="n">kwargs</span><span class="p">[</span><span class="s2">&quot;project_id&quot;</span><span class="p">]</span> <span class="o">=</span> <span class="n">project_id</span>
<span class="n">cluster_config</span> <span class="o">=</span> <span class="n">ClusterGenerator</span><span class="p">(</span><span class="o">**</span><span class="n">kwargs</span><span class="p">)</span><span class="o">.</span><span class="n">make</span><span class="p">()</span>
<span class="c1"># Remove from kwargs cluster params passed for backward compatibility</span>
<span class="n">cluster_params</span> <span class="o">=</span> <span class="n">inspect</span><span class="o">.</span><span class="n">signature</span><span class="p">(</span><span class="n">ClusterGenerator</span><span class="o">.</span><span class="fm">__init__</span><span class="p">)</span><span class="o">.</span><span class="n">parameters</span>
<span class="k">for</span> <span class="n">arg</span> <span class="ow">in</span> <span class="n">cluster_params</span><span class="p">:</span>
<span class="k">if</span> <span class="n">arg</span> <span class="ow">in</span> <span class="n">kwargs</span><span class="p">:</span>
<span class="k">del</span> <span class="n">kwargs</span><span class="p">[</span><span class="n">arg</span><span class="p">]</span>
<span class="nb">super</span><span class="p">()</span><span class="o">.</span><span class="fm">__init__</span><span class="p">(</span><span class="o">**</span><span class="n">kwargs</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">cluster_config</span> <span class="o">=</span> <span class="n">cluster_config</span>
<span class="bp">self</span><span class="o">.</span><span class="n">cluster_name</span> <span class="o">=</span> <span class="n">cluster_name</span>
<span class="bp">self</span><span class="o">.</span><span class="n">labels</span> <span class="o">=</span> <span class="n">labels</span>
<span class="bp">self</span><span class="o">.</span><span class="n">project_id</span> <span class="o">=</span> <span class="n">project_id</span>
<span class="bp">self</span><span class="o">.</span><span class="n">region</span> <span class="o">=</span> <span class="n">region</span>
<span class="bp">self</span><span class="o">.</span><span class="n">request_id</span> <span class="o">=</span> <span class="n">request_id</span>
<span class="bp">self</span><span class="o">.</span><span class="n">retry</span> <span class="o">=</span> <span class="n">retry</span>
<span class="bp">self</span><span class="o">.</span><span class="n">timeout</span> <span class="o">=</span> <span class="n">timeout</span>
<span class="bp">self</span><span class="o">.</span><span class="n">metadata</span> <span class="o">=</span> <span class="n">metadata</span>
<span class="bp">self</span><span class="o">.</span><span class="n">gcp_conn_id</span> <span class="o">=</span> <span class="n">gcp_conn_id</span>
<span class="bp">self</span><span class="o">.</span><span class="n">delete_on_error</span> <span class="o">=</span> <span class="n">delete_on_error</span>
<span class="bp">self</span><span class="o">.</span><span class="n">use_if_exists</span> <span class="o">=</span> <span class="n">use_if_exists</span>
<span class="bp">self</span><span class="o">.</span><span class="n">impersonation_chain</span> <span class="o">=</span> <span class="n">impersonation_chain</span>
<span class="bp">self</span><span class="o">.</span><span class="n">virtual_cluster_config</span> <span class="o">=</span> <span class="n">virtual_cluster_config</span>
<span class="k">def</span> <span class="nf">_create_cluster</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">hook</span><span class="p">:</span> <span class="n">DataprocHook</span><span class="p">):</span>
<span class="n">operation</span> <span class="o">=</span> <span class="n">hook</span><span class="o">.</span><span class="n">create_cluster</span><span class="p">(</span>
<span class="n">project_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">project_id</span><span class="p">,</span>
<span class="n">region</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">region</span><span class="p">,</span>
<span class="n">cluster_name</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">cluster_name</span><span class="p">,</span>
<span class="n">labels</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">labels</span><span class="p">,</span>
<span class="n">cluster_config</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">cluster_config</span><span class="p">,</span>
<span class="n">virtual_cluster_config</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">virtual_cluster_config</span><span class="p">,</span>
<span class="n">request_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">request_id</span><span class="p">,</span>
<span class="n">retry</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">retry</span><span class="p">,</span>
<span class="n">timeout</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">timeout</span><span class="p">,</span>
<span class="n">metadata</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">metadata</span><span class="p">,</span>
<span class="p">)</span>
<span class="n">cluster</span> <span class="o">=</span> <span class="n">operation</span><span class="o">.</span><span class="n">result</span><span class="p">()</span>
<span class="bp">self</span><span class="o">.</span><span class="n">log</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s2">&quot;Cluster created.&quot;</span><span class="p">)</span>
<span class="k">return</span> <span class="n">cluster</span>
<span class="k">def</span> <span class="nf">_delete_cluster</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">hook</span><span class="p">):</span>
<span class="bp">self</span><span class="o">.</span><span class="n">log</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s2">&quot;Deleting the cluster&quot;</span><span class="p">)</span>
<span class="n">hook</span><span class="o">.</span><span class="n">delete_cluster</span><span class="p">(</span><span class="n">region</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">region</span><span class="p">,</span> <span class="n">cluster_name</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">cluster_name</span><span class="p">,</span> <span class="n">project_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">project_id</span><span class="p">)</span>
<span class="k">def</span> <span class="nf">_get_cluster</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">hook</span><span class="p">:</span> <span class="n">DataprocHook</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="n">Cluster</span><span class="p">:</span>
<span class="k">return</span> <span class="n">hook</span><span class="o">.</span><span class="n">get_cluster</span><span class="p">(</span>
<span class="n">project_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">project_id</span><span class="p">,</span>
<span class="n">region</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">region</span><span class="p">,</span>
<span class="n">cluster_name</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">cluster_name</span><span class="p">,</span>
<span class="n">retry</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">retry</span><span class="p">,</span>
<span class="n">timeout</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">timeout</span><span class="p">,</span>
<span class="n">metadata</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">metadata</span><span class="p">,</span>
<span class="p">)</span>
<span class="k">def</span> <span class="nf">_handle_error_state</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">hook</span><span class="p">:</span> <span class="n">DataprocHook</span><span class="p">,</span> <span class="n">cluster</span><span class="p">:</span> <span class="n">Cluster</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="kc">None</span><span class="p">:</span>
<span class="k">if</span> <span class="n">cluster</span><span class="o">.</span><span class="n">status</span><span class="o">.</span><span class="n">state</span> <span class="o">!=</span> <span class="n">cluster</span><span class="o">.</span><span class="n">status</span><span class="o">.</span><span class="n">State</span><span class="o">.</span><span class="n">ERROR</span><span class="p">:</span>
<span class="k">return</span>
<span class="bp">self</span><span class="o">.</span><span class="n">log</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s2">&quot;Cluster is in ERROR state&quot;</span><span class="p">)</span>
<span class="n">gcs_uri</span> <span class="o">=</span> <span class="n">hook</span><span class="o">.</span><span class="n">diagnose_cluster</span><span class="p">(</span>
<span class="n">region</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">region</span><span class="p">,</span> <span class="n">cluster_name</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">cluster_name</span><span class="p">,</span> <span class="n">project_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">project_id</span>
<span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">log</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s1">&#39;Diagnostic information for cluster </span><span class="si">%s</span><span class="s1"> available at: </span><span class="si">%s</span><span class="s1">&#39;</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">cluster_name</span><span class="p">,</span> <span class="n">gcs_uri</span><span class="p">)</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">delete_on_error</span><span class="p">:</span>
<span class="bp">self</span><span class="o">.</span><span class="n">_delete_cluster</span><span class="p">(</span><span class="n">hook</span><span class="p">)</span>
<span class="k">raise</span> <span class="n">AirflowException</span><span class="p">(</span><span class="s2">&quot;Cluster was created but was in ERROR state.&quot;</span><span class="p">)</span>
<span class="k">raise</span> <span class="n">AirflowException</span><span class="p">(</span><span class="s2">&quot;Cluster was created but is in ERROR state&quot;</span><span class="p">)</span>
<span class="k">def</span> <span class="nf">_wait_for_cluster_in_deleting_state</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">hook</span><span class="p">:</span> <span class="n">DataprocHook</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="kc">None</span><span class="p">:</span>
<span class="n">time_left</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">timeout</span>
<span class="k">for</span> <span class="n">time_to_sleep</span> <span class="ow">in</span> <span class="n">exponential_sleep_generator</span><span class="p">(</span><span class="n">initial</span><span class="o">=</span><span class="mi">10</span><span class="p">,</span> <span class="n">maximum</span><span class="o">=</span><span class="mi">120</span><span class="p">):</span>
<span class="k">if</span> <span class="n">time_left</span> <span class="o">&lt;</span> <span class="mi">0</span><span class="p">:</span>
<span class="k">raise</span> <span class="n">AirflowException</span><span class="p">(</span><span class="sa">f</span><span class="s2">&quot;Cluster </span><span class="si">{</span><span class="bp">self</span><span class="o">.</span><span class="n">cluster_name</span><span class="si">}</span><span class="s2"> is still DELETING state, aborting&quot;</span><span class="p">)</span>
<span class="n">time</span><span class="o">.</span><span class="n">sleep</span><span class="p">(</span><span class="n">time_to_sleep</span><span class="p">)</span>
<span class="n">time_left</span> <span class="o">=</span> <span class="n">time_left</span> <span class="o">-</span> <span class="n">time_to_sleep</span>
<span class="k">try</span><span class="p">:</span>
<span class="bp">self</span><span class="o">.</span><span class="n">_get_cluster</span><span class="p">(</span><span class="n">hook</span><span class="p">)</span>
<span class="k">except</span> <span class="n">NotFound</span><span class="p">:</span>
<span class="k">break</span>
<span class="k">def</span> <span class="nf">_wait_for_cluster_in_creating_state</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">hook</span><span class="p">:</span> <span class="n">DataprocHook</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="n">Cluster</span><span class="p">:</span>
<span class="n">time_left</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">timeout</span>
<span class="n">cluster</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_get_cluster</span><span class="p">(</span><span class="n">hook</span><span class="p">)</span>
<span class="k">for</span> <span class="n">time_to_sleep</span> <span class="ow">in</span> <span class="n">exponential_sleep_generator</span><span class="p">(</span><span class="n">initial</span><span class="o">=</span><span class="mi">10</span><span class="p">,</span> <span class="n">maximum</span><span class="o">=</span><span class="mi">120</span><span class="p">):</span>
<span class="k">if</span> <span class="n">cluster</span><span class="o">.</span><span class="n">status</span><span class="o">.</span><span class="n">state</span> <span class="o">!=</span> <span class="n">cluster</span><span class="o">.</span><span class="n">status</span><span class="o">.</span><span class="n">State</span><span class="o">.</span><span class="n">CREATING</span><span class="p">:</span>
<span class="k">break</span>
<span class="k">if</span> <span class="n">time_left</span> <span class="o">&lt;</span> <span class="mi">0</span><span class="p">:</span>
<span class="k">raise</span> <span class="n">AirflowException</span><span class="p">(</span><span class="sa">f</span><span class="s2">&quot;Cluster </span><span class="si">{</span><span class="bp">self</span><span class="o">.</span><span class="n">cluster_name</span><span class="si">}</span><span class="s2"> is still CREATING state, aborting&quot;</span><span class="p">)</span>
<span class="n">time</span><span class="o">.</span><span class="n">sleep</span><span class="p">(</span><span class="n">time_to_sleep</span><span class="p">)</span>
<span class="n">time_left</span> <span class="o">=</span> <span class="n">time_left</span> <span class="o">-</span> <span class="n">time_to_sleep</span>
<span class="n">cluster</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_get_cluster</span><span class="p">(</span><span class="n">hook</span><span class="p">)</span>
<span class="k">return</span> <span class="n">cluster</span>
<div class="viewcode-block" id="DataprocCreateClusterOperator.execute"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocCreateClusterOperator.execute">[docs]</a> <span class="k">def</span> <span class="nf">execute</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">context</span><span class="p">:</span> <span class="n">Context</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="nb">dict</span><span class="p">:</span>
<span class="bp">self</span><span class="o">.</span><span class="n">log</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s1">&#39;Creating cluster: </span><span class="si">%s</span><span class="s1">&#39;</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">cluster_name</span><span class="p">)</span>
<span class="n">hook</span> <span class="o">=</span> <span class="n">DataprocHook</span><span class="p">(</span><span class="n">gcp_conn_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">gcp_conn_id</span><span class="p">,</span> <span class="n">impersonation_chain</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">impersonation_chain</span><span class="p">)</span>
<span class="c1"># Save data required to display extra link no matter what the cluster status will be</span>
<span class="n">DataprocLink</span><span class="o">.</span><span class="n">persist</span><span class="p">(</span>
<span class="n">context</span><span class="o">=</span><span class="n">context</span><span class="p">,</span> <span class="n">task_instance</span><span class="o">=</span><span class="bp">self</span><span class="p">,</span> <span class="n">url</span><span class="o">=</span><span class="n">DATAPROC_CLUSTER_LINK</span><span class="p">,</span> <span class="n">resource</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">cluster_name</span>
<span class="p">)</span>
<span class="k">try</span><span class="p">:</span>
<span class="c1"># First try to create a new cluster</span>
<span class="n">cluster</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_create_cluster</span><span class="p">(</span><span class="n">hook</span><span class="p">)</span>
<span class="k">except</span> <span class="n">AlreadyExists</span><span class="p">:</span>
<span class="k">if</span> <span class="ow">not</span> <span class="bp">self</span><span class="o">.</span><span class="n">use_if_exists</span><span class="p">:</span>
<span class="k">raise</span>
<span class="bp">self</span><span class="o">.</span><span class="n">log</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s2">&quot;Cluster already exists.&quot;</span><span class="p">)</span>
<span class="n">cluster</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_get_cluster</span><span class="p">(</span><span class="n">hook</span><span class="p">)</span>
<span class="c1"># Check if cluster is not in ERROR state</span>
<span class="bp">self</span><span class="o">.</span><span class="n">_handle_error_state</span><span class="p">(</span><span class="n">hook</span><span class="p">,</span> <span class="n">cluster</span><span class="p">)</span>
<span class="k">if</span> <span class="n">cluster</span><span class="o">.</span><span class="n">status</span><span class="o">.</span><span class="n">state</span> <span class="o">==</span> <span class="n">cluster</span><span class="o">.</span><span class="n">status</span><span class="o">.</span><span class="n">State</span><span class="o">.</span><span class="n">CREATING</span><span class="p">:</span>
<span class="c1"># Wait for cluster to be created</span>
<span class="n">cluster</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_wait_for_cluster_in_creating_state</span><span class="p">(</span><span class="n">hook</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">_handle_error_state</span><span class="p">(</span><span class="n">hook</span><span class="p">,</span> <span class="n">cluster</span><span class="p">)</span>
<span class="k">elif</span> <span class="n">cluster</span><span class="o">.</span><span class="n">status</span><span class="o">.</span><span class="n">state</span> <span class="o">==</span> <span class="n">cluster</span><span class="o">.</span><span class="n">status</span><span class="o">.</span><span class="n">State</span><span class="o">.</span><span class="n">DELETING</span><span class="p">:</span>
<span class="c1"># Wait for cluster to be deleted</span>
<span class="bp">self</span><span class="o">.</span><span class="n">_wait_for_cluster_in_deleting_state</span><span class="p">(</span><span class="n">hook</span><span class="p">)</span>
<span class="c1"># Create new cluster</span>
<span class="n">cluster</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_create_cluster</span><span class="p">(</span><span class="n">hook</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">_handle_error_state</span><span class="p">(</span><span class="n">hook</span><span class="p">,</span> <span class="n">cluster</span><span class="p">)</span>
<span class="k">return</span> <span class="n">Cluster</span><span class="o">.</span><span class="n">to_dict</span><span class="p">(</span><span class="n">cluster</span><span class="p">)</span></div></div>
<div class="viewcode-block" id="DataprocScaleClusterOperator"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocScaleClusterOperator">[docs]</a><span class="k">class</span> <span class="nc">DataprocScaleClusterOperator</span><span class="p">(</span><span class="n">BaseOperator</span><span class="p">):</span>
<span class="sd">&quot;&quot;&quot;</span>
<span class="sd"> Scale, up or down, a cluster on Google Cloud Dataproc.</span>
<span class="sd"> The operator will wait until the cluster is re-scaled.</span>
<span class="sd"> **Example**: ::</span>
<span class="sd"> t1 = DataprocClusterScaleOperator(</span>
<span class="sd"> task_id=&#39;dataproc_scale&#39;,</span>
<span class="sd"> project_id=&#39;my-project&#39;,</span>
<span class="sd"> cluster_name=&#39;cluster-1&#39;,</span>
<span class="sd"> num_workers=10,</span>
<span class="sd"> num_preemptible_workers=10,</span>
<span class="sd"> graceful_decommission_timeout=&#39;1h&#39;,</span>
<span class="sd"> dag=dag)</span>
<span class="sd"> .. seealso::</span>
<span class="sd"> For more detail on about scaling clusters have a look at the reference:</span>
<span class="sd"> https://cloud.google.com/dataproc/docs/concepts/configuring-clusters/scaling-clusters</span>
<span class="sd"> :param cluster_name: The name of the cluster to scale. (templated)</span>
<span class="sd"> :param project_id: The ID of the google cloud project in which</span>
<span class="sd"> the cluster runs. (templated)</span>
<span class="sd"> :param region: The region for the dataproc cluster. (templated)</span>
<span class="sd"> :param num_workers: The new number of workers</span>
<span class="sd"> :param num_preemptible_workers: The new number of preemptible workers</span>
<span class="sd"> :param graceful_decommission_timeout: Timeout for graceful YARN decommissioning.</span>
<span class="sd"> Maximum value is 1d</span>
<span class="sd"> :param gcp_conn_id: The connection ID to use connecting to Google Cloud.</span>
<span class="sd"> :param impersonation_chain: Optional service account to impersonate using short-term</span>
<span class="sd"> credentials, or chained list of accounts required to get the access_token</span>
<span class="sd"> of the last account in the list, which will be impersonated in the request.</span>
<span class="sd"> If set as a string, the account must grant the originating account</span>
<span class="sd"> the Service Account Token Creator IAM role.</span>
<span class="sd"> If set as a sequence, the identities from the list must grant</span>
<span class="sd"> Service Account Token Creator IAM role to the directly preceding identity, with first</span>
<span class="sd"> account from the list granting this role to the originating account (templated).</span>
<span class="sd"> &quot;&quot;&quot;</span>
<div class="viewcode-block" id="DataprocScaleClusterOperator.template_fields"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocScaleClusterOperator.template_fields">[docs]</a> <span class="n">template_fields</span><span class="p">:</span> <span class="n">Sequence</span><span class="p">[</span><span class="nb">str</span><span class="p">]</span> <span class="o">=</span> <span class="p">(</span><span class="s1">&#39;cluster_name&#39;</span><span class="p">,</span> <span class="s1">&#39;project_id&#39;</span><span class="p">,</span> <span class="s1">&#39;region&#39;</span><span class="p">,</span> <span class="s1">&#39;impersonation_chain&#39;</span><span class="p">)</span></div>
<div class="viewcode-block" id="DataprocScaleClusterOperator.operator_extra_links"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocScaleClusterOperator.operator_extra_links">[docs]</a> <span class="n">operator_extra_links</span> <span class="o">=</span> <span class="p">(</span><span class="n">DataprocLink</span><span class="p">(),)</span></div>
<span class="k">def</span> <span class="fm">__init__</span><span class="p">(</span>
<span class="bp">self</span><span class="p">,</span>
<span class="o">*</span><span class="p">,</span>
<span class="n">cluster_name</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span>
<span class="n">project_id</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">region</span><span class="p">:</span> <span class="nb">str</span> <span class="o">=</span> <span class="s1">&#39;global&#39;</span><span class="p">,</span>
<span class="n">num_workers</span><span class="p">:</span> <span class="nb">int</span> <span class="o">=</span> <span class="mi">2</span><span class="p">,</span>
<span class="n">num_preemptible_workers</span><span class="p">:</span> <span class="nb">int</span> <span class="o">=</span> <span class="mi">0</span><span class="p">,</span>
<span class="n">graceful_decommission_timeout</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">gcp_conn_id</span><span class="p">:</span> <span class="nb">str</span> <span class="o">=</span> <span class="s2">&quot;google_cloud_default&quot;</span><span class="p">,</span>
<span class="n">impersonation_chain</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="n">Sequence</span><span class="p">[</span><span class="nb">str</span><span class="p">]</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="o">**</span><span class="n">kwargs</span><span class="p">,</span>
<span class="p">)</span> <span class="o">-&gt;</span> <span class="kc">None</span><span class="p">:</span>
<span class="nb">super</span><span class="p">()</span><span class="o">.</span><span class="fm">__init__</span><span class="p">(</span><span class="o">**</span><span class="n">kwargs</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">project_id</span> <span class="o">=</span> <span class="n">project_id</span>
<span class="bp">self</span><span class="o">.</span><span class="n">region</span> <span class="o">=</span> <span class="n">region</span>
<span class="bp">self</span><span class="o">.</span><span class="n">cluster_name</span> <span class="o">=</span> <span class="n">cluster_name</span>
<span class="bp">self</span><span class="o">.</span><span class="n">num_workers</span> <span class="o">=</span> <span class="n">num_workers</span>
<span class="bp">self</span><span class="o">.</span><span class="n">num_preemptible_workers</span> <span class="o">=</span> <span class="n">num_preemptible_workers</span>
<span class="bp">self</span><span class="o">.</span><span class="n">graceful_decommission_timeout</span> <span class="o">=</span> <span class="n">graceful_decommission_timeout</span>
<span class="bp">self</span><span class="o">.</span><span class="n">gcp_conn_id</span> <span class="o">=</span> <span class="n">gcp_conn_id</span>
<span class="bp">self</span><span class="o">.</span><span class="n">impersonation_chain</span> <span class="o">=</span> <span class="n">impersonation_chain</span>
<span class="c1"># TODO: Remove one day</span>
<span class="n">warnings</span><span class="o">.</span><span class="n">warn</span><span class="p">(</span>
<span class="sa">f</span><span class="s2">&quot;The `</span><span class="si">{</span><span class="nb">type</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span><span class="o">.</span><span class="vm">__name__</span><span class="si">}</span><span class="s2">` operator is deprecated, &quot;</span>
<span class="s2">&quot;please use `DataprocUpdateClusterOperator` instead.&quot;</span><span class="p">,</span>
<span class="ne">DeprecationWarning</span><span class="p">,</span>
<span class="n">stacklevel</span><span class="o">=</span><span class="mi">1</span><span class="p">,</span>
<span class="p">)</span>
<span class="k">def</span> <span class="nf">_build_scale_cluster_data</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="nb">dict</span><span class="p">:</span>
<span class="n">scale_data</span> <span class="o">=</span> <span class="p">{</span>
<span class="s1">&#39;config&#39;</span><span class="p">:</span> <span class="p">{</span>
<span class="s1">&#39;worker_config&#39;</span><span class="p">:</span> <span class="p">{</span><span class="s1">&#39;num_instances&#39;</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">num_workers</span><span class="p">},</span>
<span class="s1">&#39;secondary_worker_config&#39;</span><span class="p">:</span> <span class="p">{</span><span class="s1">&#39;num_instances&#39;</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">num_preemptible_workers</span><span class="p">},</span>
<span class="p">}</span>
<span class="p">}</span>
<span class="k">return</span> <span class="n">scale_data</span>
<span class="nd">@property</span>
<span class="k">def</span> <span class="nf">_graceful_decommission_timeout_object</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="nb">dict</span><span class="p">[</span><span class="nb">str</span><span class="p">,</span> <span class="nb">int</span><span class="p">]</span> <span class="o">|</span> <span class="kc">None</span><span class="p">:</span>
<span class="k">if</span> <span class="ow">not</span> <span class="bp">self</span><span class="o">.</span><span class="n">graceful_decommission_timeout</span><span class="p">:</span>
<span class="k">return</span> <span class="kc">None</span>
<span class="n">timeout</span> <span class="o">=</span> <span class="kc">None</span>
<span class="n">match</span> <span class="o">=</span> <span class="n">re</span><span class="o">.</span><span class="n">match</span><span class="p">(</span><span class="sa">r</span><span class="s2">&quot;^(\d+)([smdh])$&quot;</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">graceful_decommission_timeout</span><span class="p">)</span>
<span class="k">if</span> <span class="n">match</span><span class="p">:</span>
<span class="k">if</span> <span class="n">match</span><span class="o">.</span><span class="n">group</span><span class="p">(</span><span class="mi">2</span><span class="p">)</span> <span class="o">==</span> <span class="s2">&quot;s&quot;</span><span class="p">:</span>
<span class="n">timeout</span> <span class="o">=</span> <span class="nb">int</span><span class="p">(</span><span class="n">match</span><span class="o">.</span><span class="n">group</span><span class="p">(</span><span class="mi">1</span><span class="p">))</span>
<span class="k">elif</span> <span class="n">match</span><span class="o">.</span><span class="n">group</span><span class="p">(</span><span class="mi">2</span><span class="p">)</span> <span class="o">==</span> <span class="s2">&quot;m&quot;</span><span class="p">:</span>
<span class="n">val</span> <span class="o">=</span> <span class="nb">float</span><span class="p">(</span><span class="n">match</span><span class="o">.</span><span class="n">group</span><span class="p">(</span><span class="mi">1</span><span class="p">))</span>
<span class="n">timeout</span> <span class="o">=</span> <span class="nb">int</span><span class="p">(</span><span class="n">timedelta</span><span class="p">(</span><span class="n">minutes</span><span class="o">=</span><span class="n">val</span><span class="p">)</span><span class="o">.</span><span class="n">total_seconds</span><span class="p">())</span>
<span class="k">elif</span> <span class="n">match</span><span class="o">.</span><span class="n">group</span><span class="p">(</span><span class="mi">2</span><span class="p">)</span> <span class="o">==</span> <span class="s2">&quot;h&quot;</span><span class="p">:</span>
<span class="n">val</span> <span class="o">=</span> <span class="nb">float</span><span class="p">(</span><span class="n">match</span><span class="o">.</span><span class="n">group</span><span class="p">(</span><span class="mi">1</span><span class="p">))</span>
<span class="n">timeout</span> <span class="o">=</span> <span class="nb">int</span><span class="p">(</span><span class="n">timedelta</span><span class="p">(</span><span class="n">hours</span><span class="o">=</span><span class="n">val</span><span class="p">)</span><span class="o">.</span><span class="n">total_seconds</span><span class="p">())</span>
<span class="k">elif</span> <span class="n">match</span><span class="o">.</span><span class="n">group</span><span class="p">(</span><span class="mi">2</span><span class="p">)</span> <span class="o">==</span> <span class="s2">&quot;d&quot;</span><span class="p">:</span>
<span class="n">val</span> <span class="o">=</span> <span class="nb">float</span><span class="p">(</span><span class="n">match</span><span class="o">.</span><span class="n">group</span><span class="p">(</span><span class="mi">1</span><span class="p">))</span>
<span class="n">timeout</span> <span class="o">=</span> <span class="nb">int</span><span class="p">(</span><span class="n">timedelta</span><span class="p">(</span><span class="n">days</span><span class="o">=</span><span class="n">val</span><span class="p">)</span><span class="o">.</span><span class="n">total_seconds</span><span class="p">())</span>
<span class="k">if</span> <span class="ow">not</span> <span class="n">timeout</span><span class="p">:</span>
<span class="k">raise</span> <span class="n">AirflowException</span><span class="p">(</span>
<span class="s2">&quot;DataprocClusterScaleOperator &quot;</span>
<span class="s2">&quot; should be expressed in day, hours, minutes or seconds. &quot;</span>
<span class="s2">&quot; i.e. 1d, 4h, 10m, 30s&quot;</span>
<span class="p">)</span>
<span class="k">return</span> <span class="p">{</span><span class="s1">&#39;seconds&#39;</span><span class="p">:</span> <span class="n">timeout</span><span class="p">}</span>
<div class="viewcode-block" id="DataprocScaleClusterOperator.execute"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocScaleClusterOperator.execute">[docs]</a> <span class="k">def</span> <span class="nf">execute</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">context</span><span class="p">:</span> <span class="n">Context</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="kc">None</span><span class="p">:</span>
<span class="sd">&quot;&quot;&quot;Scale, up or down, a cluster on Google Cloud Dataproc.&quot;&quot;&quot;</span>
<span class="bp">self</span><span class="o">.</span><span class="n">log</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s2">&quot;Scaling cluster: </span><span class="si">%s</span><span class="s2">&quot;</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">cluster_name</span><span class="p">)</span>
<span class="n">scaling_cluster_data</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_build_scale_cluster_data</span><span class="p">()</span>
<span class="n">update_mask</span> <span class="o">=</span> <span class="p">[</span><span class="s2">&quot;config.worker_config.num_instances&quot;</span><span class="p">,</span> <span class="s2">&quot;config.secondary_worker_config.num_instances&quot;</span><span class="p">]</span>
<span class="n">hook</span> <span class="o">=</span> <span class="n">DataprocHook</span><span class="p">(</span><span class="n">gcp_conn_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">gcp_conn_id</span><span class="p">,</span> <span class="n">impersonation_chain</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">impersonation_chain</span><span class="p">)</span>
<span class="c1"># Save data required to display extra link no matter what the cluster status will be</span>
<span class="n">DataprocLink</span><span class="o">.</span><span class="n">persist</span><span class="p">(</span>
<span class="n">context</span><span class="o">=</span><span class="n">context</span><span class="p">,</span> <span class="n">task_instance</span><span class="o">=</span><span class="bp">self</span><span class="p">,</span> <span class="n">url</span><span class="o">=</span><span class="n">DATAPROC_CLUSTER_LINK</span><span class="p">,</span> <span class="n">resource</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">cluster_name</span>
<span class="p">)</span>
<span class="n">operation</span> <span class="o">=</span> <span class="n">hook</span><span class="o">.</span><span class="n">update_cluster</span><span class="p">(</span>
<span class="n">project_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">project_id</span><span class="p">,</span>
<span class="n">region</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">region</span><span class="p">,</span>
<span class="n">cluster_name</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">cluster_name</span><span class="p">,</span>
<span class="n">cluster</span><span class="o">=</span><span class="n">scaling_cluster_data</span><span class="p">,</span>
<span class="n">graceful_decommission_timeout</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">_graceful_decommission_timeout_object</span><span class="p">,</span>
<span class="n">update_mask</span><span class="o">=</span><span class="p">{</span><span class="s1">&#39;paths&#39;</span><span class="p">:</span> <span class="n">update_mask</span><span class="p">},</span>
<span class="p">)</span>
<span class="n">operation</span><span class="o">.</span><span class="n">result</span><span class="p">()</span>
<span class="bp">self</span><span class="o">.</span><span class="n">log</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s2">&quot;Cluster scaling finished&quot;</span><span class="p">)</span></div></div>
<div class="viewcode-block" id="DataprocDeleteClusterOperator"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocDeleteClusterOperator">[docs]</a><span class="k">class</span> <span class="nc">DataprocDeleteClusterOperator</span><span class="p">(</span><span class="n">BaseOperator</span><span class="p">):</span>
<span class="sd">&quot;&quot;&quot;</span>
<span class="sd"> Deletes a cluster in a project.</span>
<span class="sd"> :param region: Required. The Cloud Dataproc region in which to handle the request (templated).</span>
<span class="sd"> :param cluster_name: Required. The cluster name (templated).</span>
<span class="sd"> :param project_id: Optional. The ID of the Google Cloud project that the cluster belongs to (templated).</span>
<span class="sd"> :param cluster_uuid: Optional. Specifying the ``cluster_uuid`` means the RPC should fail</span>
<span class="sd"> if cluster with specified UUID does not exist.</span>
<span class="sd"> :param request_id: Optional. A unique id used to identify the request. If the server receives two</span>
<span class="sd"> ``DeleteClusterRequest`` requests with the same id, then the second request will be ignored and the</span>
<span class="sd"> first ``google.longrunning.Operation`` created and stored in the backend is returned.</span>
<span class="sd"> :param retry: A retry object used to retry requests. If ``None`` is specified, requests will not be</span>
<span class="sd"> retried.</span>
<span class="sd"> :param timeout: The amount of time, in seconds, to wait for the request to complete. Note that if</span>
<span class="sd"> ``retry`` is specified, the timeout applies to each individual attempt.</span>
<span class="sd"> :param metadata: Additional metadata that is provided to the method.</span>
<span class="sd"> :param gcp_conn_id: The connection ID to use connecting to Google Cloud.</span>
<span class="sd"> :param impersonation_chain: Optional service account to impersonate using short-term</span>
<span class="sd"> credentials, or chained list of accounts required to get the access_token</span>
<span class="sd"> of the last account in the list, which will be impersonated in the request.</span>
<span class="sd"> If set as a string, the account must grant the originating account</span>
<span class="sd"> the Service Account Token Creator IAM role.</span>
<span class="sd"> If set as a sequence, the identities from the list must grant</span>
<span class="sd"> Service Account Token Creator IAM role to the directly preceding identity, with first</span>
<span class="sd"> account from the list granting this role to the originating account (templated).</span>
<span class="sd"> &quot;&quot;&quot;</span>
<div class="viewcode-block" id="DataprocDeleteClusterOperator.template_fields"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocDeleteClusterOperator.template_fields">[docs]</a> <span class="n">template_fields</span><span class="p">:</span> <span class="n">Sequence</span><span class="p">[</span><span class="nb">str</span><span class="p">]</span> <span class="o">=</span> <span class="p">(</span><span class="s1">&#39;project_id&#39;</span><span class="p">,</span> <span class="s1">&#39;region&#39;</span><span class="p">,</span> <span class="s1">&#39;cluster_name&#39;</span><span class="p">,</span> <span class="s1">&#39;impersonation_chain&#39;</span><span class="p">)</span></div>
<span class="k">def</span> <span class="fm">__init__</span><span class="p">(</span>
<span class="bp">self</span><span class="p">,</span>
<span class="o">*</span><span class="p">,</span>
<span class="n">region</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span>
<span class="n">cluster_name</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span>
<span class="n">project_id</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">cluster_uuid</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">request_id</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">retry</span><span class="p">:</span> <span class="n">Retry</span> <span class="o">|</span> <span class="n">_MethodDefault</span> <span class="o">=</span> <span class="n">DEFAULT</span><span class="p">,</span>
<span class="n">timeout</span><span class="p">:</span> <span class="nb">float</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">metadata</span><span class="p">:</span> <span class="n">Sequence</span><span class="p">[</span><span class="nb">tuple</span><span class="p">[</span><span class="nb">str</span><span class="p">,</span> <span class="nb">str</span><span class="p">]]</span> <span class="o">=</span> <span class="p">(),</span>
<span class="n">gcp_conn_id</span><span class="p">:</span> <span class="nb">str</span> <span class="o">=</span> <span class="s2">&quot;google_cloud_default&quot;</span><span class="p">,</span>
<span class="n">impersonation_chain</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="n">Sequence</span><span class="p">[</span><span class="nb">str</span><span class="p">]</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="o">**</span><span class="n">kwargs</span><span class="p">,</span>
<span class="p">):</span>
<span class="nb">super</span><span class="p">()</span><span class="o">.</span><span class="fm">__init__</span><span class="p">(</span><span class="o">**</span><span class="n">kwargs</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">project_id</span> <span class="o">=</span> <span class="n">project_id</span>
<span class="bp">self</span><span class="o">.</span><span class="n">region</span> <span class="o">=</span> <span class="n">region</span>
<span class="bp">self</span><span class="o">.</span><span class="n">cluster_name</span> <span class="o">=</span> <span class="n">cluster_name</span>
<span class="bp">self</span><span class="o">.</span><span class="n">cluster_uuid</span> <span class="o">=</span> <span class="n">cluster_uuid</span>
<span class="bp">self</span><span class="o">.</span><span class="n">request_id</span> <span class="o">=</span> <span class="n">request_id</span>
<span class="bp">self</span><span class="o">.</span><span class="n">retry</span> <span class="o">=</span> <span class="n">retry</span>
<span class="bp">self</span><span class="o">.</span><span class="n">timeout</span> <span class="o">=</span> <span class="n">timeout</span>
<span class="bp">self</span><span class="o">.</span><span class="n">metadata</span> <span class="o">=</span> <span class="n">metadata</span>
<span class="bp">self</span><span class="o">.</span><span class="n">gcp_conn_id</span> <span class="o">=</span> <span class="n">gcp_conn_id</span>
<span class="bp">self</span><span class="o">.</span><span class="n">impersonation_chain</span> <span class="o">=</span> <span class="n">impersonation_chain</span>
<div class="viewcode-block" id="DataprocDeleteClusterOperator.execute"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocDeleteClusterOperator.execute">[docs]</a> <span class="k">def</span> <span class="nf">execute</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">context</span><span class="p">:</span> <span class="n">Context</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="kc">None</span><span class="p">:</span>
<span class="n">hook</span> <span class="o">=</span> <span class="n">DataprocHook</span><span class="p">(</span><span class="n">gcp_conn_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">gcp_conn_id</span><span class="p">,</span> <span class="n">impersonation_chain</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">impersonation_chain</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">log</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s2">&quot;Deleting cluster: </span><span class="si">%s</span><span class="s2">&quot;</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">cluster_name</span><span class="p">)</span>
<span class="n">operation</span> <span class="o">=</span> <span class="n">hook</span><span class="o">.</span><span class="n">delete_cluster</span><span class="p">(</span>
<span class="n">project_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">project_id</span><span class="p">,</span>
<span class="n">region</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">region</span><span class="p">,</span>
<span class="n">cluster_name</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">cluster_name</span><span class="p">,</span>
<span class="n">cluster_uuid</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">cluster_uuid</span><span class="p">,</span>
<span class="n">request_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">request_id</span><span class="p">,</span>
<span class="n">retry</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">retry</span><span class="p">,</span>
<span class="n">timeout</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">timeout</span><span class="p">,</span>
<span class="n">metadata</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">metadata</span><span class="p">,</span>
<span class="p">)</span>
<span class="n">operation</span><span class="o">.</span><span class="n">result</span><span class="p">()</span>
<span class="bp">self</span><span class="o">.</span><span class="n">log</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s2">&quot;Cluster deleted.&quot;</span><span class="p">)</span></div></div>
<div class="viewcode-block" id="DataprocJobBaseOperator"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocJobBaseOperator">[docs]</a><span class="k">class</span> <span class="nc">DataprocJobBaseOperator</span><span class="p">(</span><span class="n">BaseOperator</span><span class="p">):</span>
<span class="sd">&quot;&quot;&quot;</span>
<span class="sd"> The base class for operators that launch job on DataProc.</span>
<span class="sd"> :param region: The specified region where the dataproc cluster is created.</span>
<span class="sd"> :param job_name: The job name used in the DataProc cluster. This name by default</span>
<span class="sd"> is the task_id appended with the execution data, but can be templated. The</span>
<span class="sd"> name will always be appended with a random number to avoid name clashes.</span>
<span class="sd"> :param cluster_name: The name of the DataProc cluster.</span>
<span class="sd"> :param project_id: The ID of the Google Cloud project the cluster belongs to,</span>
<span class="sd"> if not specified the project will be inferred from the provided GCP connection.</span>
<span class="sd"> :param dataproc_properties: Map for the Hive properties. Ideal to put in</span>
<span class="sd"> default arguments (templated)</span>
<span class="sd"> :param dataproc_jars: HCFS URIs of jar files to add to the CLASSPATH of the Hive server and Hadoop</span>
<span class="sd"> MapReduce (MR) tasks. Can contain Hive SerDes and UDFs. (templated)</span>
<span class="sd"> :param gcp_conn_id: The connection ID to use connecting to Google Cloud.</span>
<span class="sd"> :param delegate_to: The account to impersonate using domain-wide delegation of authority,</span>
<span class="sd"> if any. For this to work, the service account making the request must have</span>
<span class="sd"> domain-wide delegation enabled.</span>
<span class="sd"> :param labels: The labels to associate with this job. Label keys must contain 1 to 63 characters,</span>
<span class="sd"> and must conform to RFC 1035. Label values may be empty, but, if present, must contain 1 to 63</span>
<span class="sd"> characters, and must conform to RFC 1035. No more than 32 labels can be associated with a job.</span>
<span class="sd"> :param job_error_states: Job states that should be considered error states.</span>
<span class="sd"> Any states in this set will result in an error being raised and failure of the</span>
<span class="sd"> task. Eg, if the ``CANCELLED`` state should also be considered a task failure,</span>
<span class="sd"> pass in ``{&#39;ERROR&#39;, &#39;CANCELLED&#39;}``. Possible values are currently only</span>
<span class="sd"> ``&#39;ERROR&#39;`` and ``&#39;CANCELLED&#39;``, but could change in the future. Defaults to</span>
<span class="sd"> ``{&#39;ERROR&#39;}``.</span>
<span class="sd"> :param impersonation_chain: Optional service account to impersonate using short-term</span>
<span class="sd"> credentials, or chained list of accounts required to get the access_token</span>
<span class="sd"> of the last account in the list, which will be impersonated in the request.</span>
<span class="sd"> If set as a string, the account must grant the originating account</span>
<span class="sd"> the Service Account Token Creator IAM role.</span>
<span class="sd"> If set as a sequence, the identities from the list must grant</span>
<span class="sd"> Service Account Token Creator IAM role to the directly preceding identity, with first</span>
<span class="sd"> account from the list granting this role to the originating account (templated).</span>
<span class="sd"> :param asynchronous: Flag to return after submitting the job to the Dataproc API.</span>
<span class="sd"> This is useful for submitting long running jobs and</span>
<span class="sd"> waiting on them asynchronously using the DataprocJobSensor</span>
<span class="sd"> :param deferrable: Run operator in the deferrable mode</span>
<span class="sd"> :param polling_interval_seconds: time in seconds between polling for job completion.</span>
<span class="sd"> The value is considered only when running in deferrable mode. Must be greater than 0.</span>
<span class="sd"> :var dataproc_job_id: The actual &quot;jobId&quot; as submitted to the Dataproc API.</span>
<span class="sd"> This is useful for identifying or linking to the job in the Google Cloud Console</span>
<span class="sd"> Dataproc UI, as the actual &quot;jobId&quot; submitted to the Dataproc API is appended with</span>
<span class="sd"> an 8 character random string.</span>
<span class="sd"> :vartype dataproc_job_id: str</span>
<span class="sd"> &quot;&quot;&quot;</span>
<div class="viewcode-block" id="DataprocJobBaseOperator.job_type"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocJobBaseOperator.job_type">[docs]</a> <span class="n">job_type</span> <span class="o">=</span> <span class="s2">&quot;&quot;</span></div>
<div class="viewcode-block" id="DataprocJobBaseOperator.operator_extra_links"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocJobBaseOperator.operator_extra_links">[docs]</a> <span class="n">operator_extra_links</span> <span class="o">=</span> <span class="p">(</span><span class="n">DataprocLink</span><span class="p">(),)</span></div>
<span class="k">def</span> <span class="fm">__init__</span><span class="p">(</span>
<span class="bp">self</span><span class="p">,</span>
<span class="o">*</span><span class="p">,</span>
<span class="n">region</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span>
<span class="n">job_name</span><span class="p">:</span> <span class="nb">str</span> <span class="o">=</span> <span class="s1">&#39;{{task.task_id}}_{{ds_nodash}}&#39;</span><span class="p">,</span>
<span class="n">cluster_name</span><span class="p">:</span> <span class="nb">str</span> <span class="o">=</span> <span class="s2">&quot;cluster-1&quot;</span><span class="p">,</span>
<span class="n">project_id</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">dataproc_properties</span><span class="p">:</span> <span class="nb">dict</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">dataproc_jars</span><span class="p">:</span> <span class="nb">list</span><span class="p">[</span><span class="nb">str</span><span class="p">]</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">gcp_conn_id</span><span class="p">:</span> <span class="nb">str</span> <span class="o">=</span> <span class="s1">&#39;google_cloud_default&#39;</span><span class="p">,</span>
<span class="n">delegate_to</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">labels</span><span class="p">:</span> <span class="nb">dict</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">job_error_states</span><span class="p">:</span> <span class="nb">set</span><span class="p">[</span><span class="nb">str</span><span class="p">]</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">impersonation_chain</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="n">Sequence</span><span class="p">[</span><span class="nb">str</span><span class="p">]</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">asynchronous</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">False</span><span class="p">,</span>
<span class="n">deferrable</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">False</span><span class="p">,</span>
<span class="n">polling_interval_seconds</span><span class="p">:</span> <span class="nb">int</span> <span class="o">=</span> <span class="mi">10</span><span class="p">,</span>
<span class="o">**</span><span class="n">kwargs</span><span class="p">,</span>
<span class="p">)</span> <span class="o">-&gt;</span> <span class="kc">None</span><span class="p">:</span>
<span class="nb">super</span><span class="p">()</span><span class="o">.</span><span class="fm">__init__</span><span class="p">(</span><span class="o">**</span><span class="n">kwargs</span><span class="p">)</span>
<span class="k">if</span> <span class="n">deferrable</span> <span class="ow">and</span> <span class="n">polling_interval_seconds</span> <span class="o">&lt;=</span> <span class="mi">0</span><span class="p">:</span>
<span class="k">raise</span> <span class="ne">ValueError</span><span class="p">(</span><span class="s2">&quot;Invalid value for polling_interval_seconds. Expected value greater than 0&quot;</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">gcp_conn_id</span> <span class="o">=</span> <span class="n">gcp_conn_id</span>
<span class="bp">self</span><span class="o">.</span><span class="n">delegate_to</span> <span class="o">=</span> <span class="n">delegate_to</span>
<span class="bp">self</span><span class="o">.</span><span class="n">labels</span> <span class="o">=</span> <span class="n">labels</span>
<span class="bp">self</span><span class="o">.</span><span class="n">job_name</span> <span class="o">=</span> <span class="n">job_name</span>
<span class="bp">self</span><span class="o">.</span><span class="n">cluster_name</span> <span class="o">=</span> <span class="n">cluster_name</span>
<span class="bp">self</span><span class="o">.</span><span class="n">dataproc_properties</span> <span class="o">=</span> <span class="n">dataproc_properties</span>
<span class="bp">self</span><span class="o">.</span><span class="n">dataproc_jars</span> <span class="o">=</span> <span class="n">dataproc_jars</span>
<span class="bp">self</span><span class="o">.</span><span class="n">region</span> <span class="o">=</span> <span class="n">region</span>
<span class="bp">self</span><span class="o">.</span><span class="n">job_error_states</span> <span class="o">=</span> <span class="n">job_error_states</span> <span class="k">if</span> <span class="n">job_error_states</span> <span class="ow">is</span> <span class="ow">not</span> <span class="kc">None</span> <span class="k">else</span> <span class="p">{</span><span class="s1">&#39;ERROR&#39;</span><span class="p">}</span>
<span class="bp">self</span><span class="o">.</span><span class="n">impersonation_chain</span> <span class="o">=</span> <span class="n">impersonation_chain</span>
<span class="bp">self</span><span class="o">.</span><span class="n">hook</span> <span class="o">=</span> <span class="n">DataprocHook</span><span class="p">(</span><span class="n">gcp_conn_id</span><span class="o">=</span><span class="n">gcp_conn_id</span><span class="p">,</span> <span class="n">impersonation_chain</span><span class="o">=</span><span class="n">impersonation_chain</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">project_id</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">hook</span><span class="o">.</span><span class="n">project_id</span> <span class="k">if</span> <span class="n">project_id</span> <span class="ow">is</span> <span class="kc">None</span> <span class="k">else</span> <span class="n">project_id</span>
<span class="bp">self</span><span class="o">.</span><span class="n">job_template</span><span class="p">:</span> <span class="n">DataProcJobBuilder</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span>
<span class="bp">self</span><span class="o">.</span><span class="n">job</span><span class="p">:</span> <span class="nb">dict</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span>
<span class="bp">self</span><span class="o">.</span><span class="n">dataproc_job_id</span> <span class="o">=</span> <span class="kc">None</span>
<span class="bp">self</span><span class="o">.</span><span class="n">asynchronous</span> <span class="o">=</span> <span class="n">asynchronous</span>
<span class="bp">self</span><span class="o">.</span><span class="n">deferrable</span> <span class="o">=</span> <span class="n">deferrable</span>
<span class="bp">self</span><span class="o">.</span><span class="n">polling_interval_seconds</span> <span class="o">=</span> <span class="n">polling_interval_seconds</span>
<div class="viewcode-block" id="DataprocJobBaseOperator.create_job_template"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocJobBaseOperator.create_job_template">[docs]</a> <span class="k">def</span> <span class="nf">create_job_template</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="n">DataProcJobBuilder</span><span class="p">:</span>
<span class="sd">&quot;&quot;&quot;Initialize `self.job_template` with default values&quot;&quot;&quot;</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">project_id</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
<span class="k">raise</span> <span class="n">AirflowException</span><span class="p">(</span>
<span class="s2">&quot;project id should either be set via project_id &quot;</span>
<span class="s2">&quot;parameter or retrieved from the connection,&quot;</span>
<span class="p">)</span>
<span class="n">job_template</span> <span class="o">=</span> <span class="n">DataProcJobBuilder</span><span class="p">(</span>
<span class="n">project_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">project_id</span><span class="p">,</span>
<span class="n">task_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">task_id</span><span class="p">,</span>
<span class="n">cluster_name</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">cluster_name</span><span class="p">,</span>
<span class="n">job_type</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">job_type</span><span class="p">,</span>
<span class="n">properties</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">dataproc_properties</span><span class="p">,</span>
<span class="p">)</span>
<span class="n">job_template</span><span class="o">.</span><span class="n">set_job_name</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">job_name</span><span class="p">)</span>
<span class="n">job_template</span><span class="o">.</span><span class="n">add_jar_file_uris</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">dataproc_jars</span><span class="p">)</span>
<span class="n">job_template</span><span class="o">.</span><span class="n">add_labels</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">labels</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">job_template</span> <span class="o">=</span> <span class="n">job_template</span>
<span class="k">return</span> <span class="n">job_template</span></div>
<span class="k">def</span> <span class="nf">_generate_job_template</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="nb">str</span><span class="p">:</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">job_template</span><span class="p">:</span>
<span class="n">job</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">job_template</span><span class="o">.</span><span class="n">build</span><span class="p">()</span>
<span class="k">return</span> <span class="n">job</span><span class="p">[</span><span class="s1">&#39;job&#39;</span><span class="p">]</span>
<span class="k">raise</span> <span class="ne">Exception</span><span class="p">(</span><span class="s2">&quot;Create a job template before&quot;</span><span class="p">)</span>
<div class="viewcode-block" id="DataprocJobBaseOperator.execute"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocJobBaseOperator.execute">[docs]</a> <span class="k">def</span> <span class="nf">execute</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">context</span><span class="p">:</span> <span class="n">Context</span><span class="p">):</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">job_template</span><span class="p">:</span>
<span class="bp">self</span><span class="o">.</span><span class="n">job</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">job_template</span><span class="o">.</span><span class="n">build</span><span class="p">()</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">job</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
<span class="k">raise</span> <span class="ne">Exception</span><span class="p">(</span><span class="s2">&quot;The job should be set here.&quot;</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">dataproc_job_id</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">job</span><span class="p">[</span><span class="s2">&quot;job&quot;</span><span class="p">][</span><span class="s2">&quot;reference&quot;</span><span class="p">][</span><span class="s2">&quot;job_id&quot;</span><span class="p">]</span>
<span class="bp">self</span><span class="o">.</span><span class="n">log</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s1">&#39;Submitting </span><span class="si">%s</span><span class="s1"> job </span><span class="si">%s</span><span class="s1">&#39;</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">job_type</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">dataproc_job_id</span><span class="p">)</span>
<span class="n">job_object</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">hook</span><span class="o">.</span><span class="n">submit_job</span><span class="p">(</span>
<span class="n">project_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">project_id</span><span class="p">,</span> <span class="n">job</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">job</span><span class="p">[</span><span class="s2">&quot;job&quot;</span><span class="p">],</span> <span class="n">region</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">region</span>
<span class="p">)</span>
<span class="n">job_id</span> <span class="o">=</span> <span class="n">job_object</span><span class="o">.</span><span class="n">reference</span><span class="o">.</span><span class="n">job_id</span>
<span class="bp">self</span><span class="o">.</span><span class="n">log</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s1">&#39;Job </span><span class="si">%s</span><span class="s1"> submitted successfully.&#39;</span><span class="p">,</span> <span class="n">job_id</span><span class="p">)</span>
<span class="c1"># Save data required for extra links no matter what the job status will be</span>
<span class="n">DataprocLink</span><span class="o">.</span><span class="n">persist</span><span class="p">(</span>
<span class="n">context</span><span class="o">=</span><span class="n">context</span><span class="p">,</span> <span class="n">task_instance</span><span class="o">=</span><span class="bp">self</span><span class="p">,</span> <span class="n">url</span><span class="o">=</span><span class="n">DATAPROC_JOB_LOG_LINK</span><span class="p">,</span> <span class="n">resource</span><span class="o">=</span><span class="n">job_id</span>
<span class="p">)</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">deferrable</span><span class="p">:</span>
<span class="bp">self</span><span class="o">.</span><span class="n">defer</span><span class="p">(</span>
<span class="n">trigger</span><span class="o">=</span><span class="n">DataprocBaseTrigger</span><span class="p">(</span>
<span class="n">job_id</span><span class="o">=</span><span class="n">job_id</span><span class="p">,</span>
<span class="n">project_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">project_id</span><span class="p">,</span>
<span class="n">region</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">region</span><span class="p">,</span>
<span class="n">delegate_to</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">delegate_to</span><span class="p">,</span>
<span class="n">gcp_conn_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">gcp_conn_id</span><span class="p">,</span>
<span class="n">impersonation_chain</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">impersonation_chain</span><span class="p">,</span>
<span class="n">polling_interval_seconds</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">polling_interval_seconds</span><span class="p">,</span>
<span class="p">),</span>
<span class="n">method_name</span><span class="o">=</span><span class="s2">&quot;execute_complete&quot;</span><span class="p">,</span>
<span class="p">)</span>
<span class="k">if</span> <span class="ow">not</span> <span class="bp">self</span><span class="o">.</span><span class="n">asynchronous</span><span class="p">:</span>
<span class="bp">self</span><span class="o">.</span><span class="n">log</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s1">&#39;Waiting for job </span><span class="si">%s</span><span class="s1"> to complete&#39;</span><span class="p">,</span> <span class="n">job_id</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">hook</span><span class="o">.</span><span class="n">wait_for_job</span><span class="p">(</span><span class="n">job_id</span><span class="o">=</span><span class="n">job_id</span><span class="p">,</span> <span class="n">region</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">region</span><span class="p">,</span> <span class="n">project_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">project_id</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">log</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s1">&#39;Job </span><span class="si">%s</span><span class="s1"> completed successfully.&#39;</span><span class="p">,</span> <span class="n">job_id</span><span class="p">)</span>
<span class="k">return</span> <span class="n">job_id</span>
<span class="k">else</span><span class="p">:</span>
<span class="k">raise</span> <span class="n">AirflowException</span><span class="p">(</span><span class="s2">&quot;Create a job template before&quot;</span><span class="p">)</span></div>
<div class="viewcode-block" id="DataprocJobBaseOperator.execute_complete"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocJobBaseOperator.execute_complete">[docs]</a> <span class="k">def</span> <span class="nf">execute_complete</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">context</span><span class="p">,</span> <span class="n">event</span><span class="o">=</span><span class="kc">None</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="kc">None</span><span class="p">:</span>
<span class="sd">&quot;&quot;&quot;</span>
<span class="sd"> Callback for when the trigger fires - returns immediately.</span>
<span class="sd"> Relies on trigger to throw an exception, otherwise it assumes execution was</span>
<span class="sd"> successful.</span>
<span class="sd"> &quot;&quot;&quot;</span>
<span class="n">job_state</span> <span class="o">=</span> <span class="n">event</span><span class="p">[</span><span class="s2">&quot;job_state&quot;</span><span class="p">]</span>
<span class="n">job_id</span> <span class="o">=</span> <span class="n">event</span><span class="p">[</span><span class="s2">&quot;job_id&quot;</span><span class="p">]</span>
<span class="k">if</span> <span class="n">job_state</span> <span class="o">==</span> <span class="n">JobStatus</span><span class="o">.</span><span class="n">State</span><span class="o">.</span><span class="n">ERROR</span><span class="p">:</span>
<span class="k">raise</span> <span class="n">AirflowException</span><span class="p">(</span><span class="sa">f</span><span class="s1">&#39;Job failed:</span><span class="se">\n</span><span class="si">{</span><span class="n">job_id</span><span class="si">}</span><span class="s1">&#39;</span><span class="p">)</span>
<span class="k">if</span> <span class="n">job_state</span> <span class="o">==</span> <span class="n">JobStatus</span><span class="o">.</span><span class="n">State</span><span class="o">.</span><span class="n">CANCELLED</span><span class="p">:</span>
<span class="k">raise</span> <span class="n">AirflowException</span><span class="p">(</span><span class="sa">f</span><span class="s1">&#39;Job was cancelled:</span><span class="se">\n</span><span class="si">{</span><span class="n">job_id</span><span class="si">}</span><span class="s1">&#39;</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">log</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s2">&quot;</span><span class="si">%s</span><span class="s2"> completed successfully.&quot;</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">task_id</span><span class="p">)</span></div>
<div class="viewcode-block" id="DataprocJobBaseOperator.on_kill"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocJobBaseOperator.on_kill">[docs]</a> <span class="k">def</span> <span class="nf">on_kill</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="kc">None</span><span class="p">:</span>
<span class="sd">&quot;&quot;&quot;</span>
<span class="sd"> Callback called when the operator is killed.</span>
<span class="sd"> Cancel any running job.</span>
<span class="sd"> &quot;&quot;&quot;</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">dataproc_job_id</span><span class="p">:</span>
<span class="bp">self</span><span class="o">.</span><span class="n">hook</span><span class="o">.</span><span class="n">cancel_job</span><span class="p">(</span><span class="n">project_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">project_id</span><span class="p">,</span> <span class="n">job_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">dataproc_job_id</span><span class="p">,</span> <span class="n">region</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">region</span><span class="p">)</span></div></div>
<div class="viewcode-block" id="DataprocSubmitPigJobOperator"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocSubmitPigJobOperator">[docs]</a><span class="k">class</span> <span class="nc">DataprocSubmitPigJobOperator</span><span class="p">(</span><span class="n">DataprocJobBaseOperator</span><span class="p">):</span>
<span class="sd">&quot;&quot;&quot;</span>
<span class="sd"> Start a Pig query Job on a Cloud DataProc cluster. The parameters of the operation</span>
<span class="sd"> will be passed to the cluster.</span>
<span class="sd"> It&#39;s a good practice to define dataproc_* parameters in the default_args of the dag</span>
<span class="sd"> like the cluster name and UDFs.</span>
<span class="sd"> .. code-block:: python</span>
<span class="sd"> default_args = {</span>
<span class="sd"> &quot;cluster_name&quot;: &quot;cluster-1&quot;,</span>
<span class="sd"> &quot;dataproc_pig_jars&quot;: [</span>
<span class="sd"> &quot;gs://example/udf/jar/datafu/1.2.0/datafu.jar&quot;,</span>
<span class="sd"> &quot;gs://example/udf/jar/gpig/1.2/gpig.jar&quot;,</span>
<span class="sd"> ],</span>
<span class="sd"> }</span>
<span class="sd"> You can pass a pig script as string or file reference. Use variables to pass on</span>
<span class="sd"> variables for the pig script to be resolved on the cluster or use the parameters to</span>
<span class="sd"> be resolved in the script as template parameters.</span>
<span class="sd"> **Example**: ::</span>
<span class="sd"> t1 = DataProcPigOperator(</span>
<span class="sd"> task_id=&#39;dataproc_pig&#39;,</span>
<span class="sd"> query=&#39;a_pig_script.pig&#39;,</span>
<span class="sd"> variables={&#39;out&#39;: &#39;gs://example/output/{{ds}}&#39;},</span>
<span class="sd"> dag=dag)</span>
<span class="sd"> .. seealso::</span>
<span class="sd"> For more detail on about job submission have a look at the reference:</span>
<span class="sd"> https://cloud.google.com/dataproc/reference/rest/v1/projects.regions.jobs</span>
<span class="sd"> :param query: The query or reference to the query</span>
<span class="sd"> file (pg or pig extension). (templated)</span>
<span class="sd"> :param query_uri: The HCFS URI of the script that contains the Pig queries.</span>
<span class="sd"> :param variables: Map of named parameters for the query. (templated)</span>
<span class="sd"> &quot;&quot;&quot;</span>
<div class="viewcode-block" id="DataprocSubmitPigJobOperator.template_fields"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocSubmitPigJobOperator.template_fields">[docs]</a> <span class="n">template_fields</span><span class="p">:</span> <span class="n">Sequence</span><span class="p">[</span><span class="nb">str</span><span class="p">]</span> <span class="o">=</span> <span class="p">(</span>
<span class="s1">&#39;query&#39;</span><span class="p">,</span>
<span class="s1">&#39;variables&#39;</span><span class="p">,</span>
<span class="s1">&#39;job_name&#39;</span><span class="p">,</span>
<span class="s1">&#39;cluster_name&#39;</span><span class="p">,</span>
<span class="s1">&#39;region&#39;</span><span class="p">,</span>
<span class="s1">&#39;dataproc_jars&#39;</span><span class="p">,</span>
<span class="s1">&#39;dataproc_properties&#39;</span><span class="p">,</span>
<span class="s1">&#39;impersonation_chain&#39;</span><span class="p">,</span></div>
<span class="p">)</span>
<div class="viewcode-block" id="DataprocSubmitPigJobOperator.template_ext"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocSubmitPigJobOperator.template_ext">[docs]</a> <span class="n">template_ext</span> <span class="o">=</span> <span class="p">(</span><span class="s1">&#39;.pg&#39;</span><span class="p">,</span> <span class="s1">&#39;.pig&#39;</span><span class="p">)</span></div>
<div class="viewcode-block" id="DataprocSubmitPigJobOperator.ui_color"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocSubmitPigJobOperator.ui_color">[docs]</a> <span class="n">ui_color</span> <span class="o">=</span> <span class="s1">&#39;#0273d4&#39;</span></div>
<div class="viewcode-block" id="DataprocSubmitPigJobOperator.job_type"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocSubmitPigJobOperator.job_type">[docs]</a> <span class="n">job_type</span> <span class="o">=</span> <span class="s1">&#39;pig_job&#39;</span></div>
<div class="viewcode-block" id="DataprocSubmitPigJobOperator.operator_extra_links"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocSubmitPigJobOperator.operator_extra_links">[docs]</a> <span class="n">operator_extra_links</span> <span class="o">=</span> <span class="p">(</span><span class="n">DataprocLink</span><span class="p">(),)</span></div>
<span class="k">def</span> <span class="fm">__init__</span><span class="p">(</span>
<span class="bp">self</span><span class="p">,</span>
<span class="o">*</span><span class="p">,</span>
<span class="n">query</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">query_uri</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">variables</span><span class="p">:</span> <span class="nb">dict</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="o">**</span><span class="n">kwargs</span><span class="p">,</span>
<span class="p">)</span> <span class="o">-&gt;</span> <span class="kc">None</span><span class="p">:</span>
<span class="c1"># TODO: Remove one day</span>
<span class="n">warnings</span><span class="o">.</span><span class="n">warn</span><span class="p">(</span>
<span class="s2">&quot;The `</span><span class="si">{cls}</span><span class="s2">` operator is deprecated, please use `DataprocSubmitJobOperator` instead. You can use&quot;</span>
<span class="s2">&quot; `generate_job` method of `</span><span class="si">{cls}</span><span class="s2">` to generate dictionary representing your job&quot;</span>
<span class="s2">&quot; and use it with the new operator.&quot;</span><span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="bp">cls</span><span class="o">=</span><span class="nb">type</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span><span class="o">.</span><span class="vm">__name__</span><span class="p">),</span>
<span class="ne">DeprecationWarning</span><span class="p">,</span>
<span class="n">stacklevel</span><span class="o">=</span><span class="mi">1</span><span class="p">,</span>
<span class="p">)</span>
<span class="nb">super</span><span class="p">()</span><span class="o">.</span><span class="fm">__init__</span><span class="p">(</span><span class="o">**</span><span class="n">kwargs</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">query</span> <span class="o">=</span> <span class="n">query</span>
<span class="bp">self</span><span class="o">.</span><span class="n">query_uri</span> <span class="o">=</span> <span class="n">query_uri</span>
<span class="bp">self</span><span class="o">.</span><span class="n">variables</span> <span class="o">=</span> <span class="n">variables</span>
<div class="viewcode-block" id="DataprocSubmitPigJobOperator.generate_job"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocSubmitPigJobOperator.generate_job">[docs]</a> <span class="k">def</span> <span class="nf">generate_job</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
<span class="sd">&quot;&quot;&quot;</span>
<span class="sd"> Helper method for easier migration to `DataprocSubmitJobOperator`.</span>
<span class="sd"> :return: Dict representing Dataproc job</span>
<span class="sd"> &quot;&quot;&quot;</span>
<span class="n">job_template</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">create_job_template</span><span class="p">()</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">query</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">query_uri</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
<span class="k">raise</span> <span class="n">AirflowException</span><span class="p">(</span><span class="s1">&#39;One of query or query_uri should be set here&#39;</span><span class="p">)</span>
<span class="n">job_template</span><span class="o">.</span><span class="n">add_query_uri</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">query_uri</span><span class="p">)</span>
<span class="k">else</span><span class="p">:</span>
<span class="n">job_template</span><span class="o">.</span><span class="n">add_query</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">query</span><span class="p">)</span>
<span class="n">job_template</span><span class="o">.</span><span class="n">add_variables</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">variables</span><span class="p">)</span>
<span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">_generate_job_template</span><span class="p">()</span></div>
<div class="viewcode-block" id="DataprocSubmitPigJobOperator.execute"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocSubmitPigJobOperator.execute">[docs]</a> <span class="k">def</span> <span class="nf">execute</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">context</span><span class="p">:</span> <span class="n">Context</span><span class="p">):</span>
<span class="n">job_template</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">create_job_template</span><span class="p">()</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">query</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">query_uri</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
<span class="k">raise</span> <span class="n">AirflowException</span><span class="p">(</span><span class="s1">&#39;One of query or query_uri should be set here&#39;</span><span class="p">)</span>
<span class="n">job_template</span><span class="o">.</span><span class="n">add_query_uri</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">query_uri</span><span class="p">)</span>
<span class="k">else</span><span class="p">:</span>
<span class="n">job_template</span><span class="o">.</span><span class="n">add_query</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">query</span><span class="p">)</span>
<span class="n">job_template</span><span class="o">.</span><span class="n">add_variables</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">variables</span><span class="p">)</span>
<span class="nb">super</span><span class="p">()</span><span class="o">.</span><span class="n">execute</span><span class="p">(</span><span class="n">context</span><span class="p">)</span></div></div>
<div class="viewcode-block" id="DataprocSubmitHiveJobOperator"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocSubmitHiveJobOperator">[docs]</a><span class="k">class</span> <span class="nc">DataprocSubmitHiveJobOperator</span><span class="p">(</span><span class="n">DataprocJobBaseOperator</span><span class="p">):</span>
<span class="sd">&quot;&quot;&quot;</span>
<span class="sd"> Start a Hive query Job on a Cloud DataProc cluster.</span>
<span class="sd"> :param query: The query or reference to the query file (q extension).</span>
<span class="sd"> :param query_uri: The HCFS URI of the script that contains the Hive queries.</span>
<span class="sd"> :param variables: Map of named parameters for the query.</span>
<span class="sd"> &quot;&quot;&quot;</span>
<div class="viewcode-block" id="DataprocSubmitHiveJobOperator.template_fields"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocSubmitHiveJobOperator.template_fields">[docs]</a> <span class="n">template_fields</span><span class="p">:</span> <span class="n">Sequence</span><span class="p">[</span><span class="nb">str</span><span class="p">]</span> <span class="o">=</span> <span class="p">(</span>
<span class="s1">&#39;query&#39;</span><span class="p">,</span>
<span class="s1">&#39;variables&#39;</span><span class="p">,</span>
<span class="s1">&#39;job_name&#39;</span><span class="p">,</span>
<span class="s1">&#39;cluster_name&#39;</span><span class="p">,</span>
<span class="s1">&#39;region&#39;</span><span class="p">,</span>
<span class="s1">&#39;dataproc_jars&#39;</span><span class="p">,</span>
<span class="s1">&#39;dataproc_properties&#39;</span><span class="p">,</span>
<span class="s1">&#39;impersonation_chain&#39;</span><span class="p">,</span></div>
<span class="p">)</span>
<div class="viewcode-block" id="DataprocSubmitHiveJobOperator.template_ext"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocSubmitHiveJobOperator.template_ext">[docs]</a> <span class="n">template_ext</span> <span class="o">=</span> <span class="p">(</span><span class="s1">&#39;.q&#39;</span><span class="p">,</span> <span class="s1">&#39;.hql&#39;</span><span class="p">)</span></div>
<div class="viewcode-block" id="DataprocSubmitHiveJobOperator.ui_color"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocSubmitHiveJobOperator.ui_color">[docs]</a> <span class="n">ui_color</span> <span class="o">=</span> <span class="s1">&#39;#0273d4&#39;</span></div>
<div class="viewcode-block" id="DataprocSubmitHiveJobOperator.job_type"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocSubmitHiveJobOperator.job_type">[docs]</a> <span class="n">job_type</span> <span class="o">=</span> <span class="s1">&#39;hive_job&#39;</span></div>
<span class="k">def</span> <span class="fm">__init__</span><span class="p">(</span>
<span class="bp">self</span><span class="p">,</span>
<span class="o">*</span><span class="p">,</span>
<span class="n">query</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">query_uri</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">variables</span><span class="p">:</span> <span class="nb">dict</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="o">**</span><span class="n">kwargs</span><span class="p">,</span>
<span class="p">)</span> <span class="o">-&gt;</span> <span class="kc">None</span><span class="p">:</span>
<span class="c1"># TODO: Remove one day</span>
<span class="n">warnings</span><span class="o">.</span><span class="n">warn</span><span class="p">(</span>
<span class="s2">&quot;The `</span><span class="si">{cls}</span><span class="s2">` operator is deprecated, please use `DataprocSubmitJobOperator` instead. You can use&quot;</span>
<span class="s2">&quot; `generate_job` method of `</span><span class="si">{cls}</span><span class="s2">` to generate dictionary representing your job&quot;</span>
<span class="s2">&quot; and use it with the new operator.&quot;</span><span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="bp">cls</span><span class="o">=</span><span class="nb">type</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span><span class="o">.</span><span class="vm">__name__</span><span class="p">),</span>
<span class="ne">DeprecationWarning</span><span class="p">,</span>
<span class="n">stacklevel</span><span class="o">=</span><span class="mi">1</span><span class="p">,</span>
<span class="p">)</span>
<span class="nb">super</span><span class="p">()</span><span class="o">.</span><span class="fm">__init__</span><span class="p">(</span><span class="o">**</span><span class="n">kwargs</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">query</span> <span class="o">=</span> <span class="n">query</span>
<span class="bp">self</span><span class="o">.</span><span class="n">query_uri</span> <span class="o">=</span> <span class="n">query_uri</span>
<span class="bp">self</span><span class="o">.</span><span class="n">variables</span> <span class="o">=</span> <span class="n">variables</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">query</span> <span class="ow">is</span> <span class="ow">not</span> <span class="kc">None</span> <span class="ow">and</span> <span class="bp">self</span><span class="o">.</span><span class="n">query_uri</span> <span class="ow">is</span> <span class="ow">not</span> <span class="kc">None</span><span class="p">:</span>
<span class="k">raise</span> <span class="n">AirflowException</span><span class="p">(</span><span class="s1">&#39;Only one of `query` and `query_uri` can be passed.&#39;</span><span class="p">)</span>
<div class="viewcode-block" id="DataprocSubmitHiveJobOperator.generate_job"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocSubmitHiveJobOperator.generate_job">[docs]</a> <span class="k">def</span> <span class="nf">generate_job</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
<span class="sd">&quot;&quot;&quot;</span>
<span class="sd"> Helper method for easier migration to `DataprocSubmitJobOperator`.</span>
<span class="sd"> :return: Dict representing Dataproc job</span>
<span class="sd"> &quot;&quot;&quot;</span>
<span class="n">job_template</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">create_job_template</span><span class="p">()</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">query</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">query_uri</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
<span class="k">raise</span> <span class="n">AirflowException</span><span class="p">(</span><span class="s1">&#39;One of query or query_uri should be set here&#39;</span><span class="p">)</span>
<span class="n">job_template</span><span class="o">.</span><span class="n">add_query_uri</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">query_uri</span><span class="p">)</span>
<span class="k">else</span><span class="p">:</span>
<span class="n">job_template</span><span class="o">.</span><span class="n">add_query</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">query</span><span class="p">)</span>
<span class="n">job_template</span><span class="o">.</span><span class="n">add_variables</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">variables</span><span class="p">)</span>
<span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">_generate_job_template</span><span class="p">()</span></div>
<div class="viewcode-block" id="DataprocSubmitHiveJobOperator.execute"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocSubmitHiveJobOperator.execute">[docs]</a> <span class="k">def</span> <span class="nf">execute</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">context</span><span class="p">:</span> <span class="n">Context</span><span class="p">):</span>
<span class="n">job_template</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">create_job_template</span><span class="p">()</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">query</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">query_uri</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
<span class="k">raise</span> <span class="n">AirflowException</span><span class="p">(</span><span class="s1">&#39;One of query or query_uri should be set here&#39;</span><span class="p">)</span>
<span class="n">job_template</span><span class="o">.</span><span class="n">add_query_uri</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">query_uri</span><span class="p">)</span>
<span class="k">else</span><span class="p">:</span>
<span class="n">job_template</span><span class="o">.</span><span class="n">add_query</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">query</span><span class="p">)</span>
<span class="n">job_template</span><span class="o">.</span><span class="n">add_variables</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">variables</span><span class="p">)</span>
<span class="nb">super</span><span class="p">()</span><span class="o">.</span><span class="n">execute</span><span class="p">(</span><span class="n">context</span><span class="p">)</span></div></div>
<div class="viewcode-block" id="DataprocSubmitSparkSqlJobOperator"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocSubmitSparkSqlJobOperator">[docs]</a><span class="k">class</span> <span class="nc">DataprocSubmitSparkSqlJobOperator</span><span class="p">(</span><span class="n">DataprocJobBaseOperator</span><span class="p">):</span>
<span class="sd">&quot;&quot;&quot;</span>
<span class="sd"> Start a Spark SQL query Job on a Cloud DataProc cluster.</span>
<span class="sd"> :param query: The query or reference to the query file (q extension). (templated)</span>
<span class="sd"> :param query_uri: The HCFS URI of the script that contains the SQL queries.</span>
<span class="sd"> :param variables: Map of named parameters for the query. (templated)</span>
<span class="sd"> &quot;&quot;&quot;</span>
<div class="viewcode-block" id="DataprocSubmitSparkSqlJobOperator.template_fields"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocSubmitSparkSqlJobOperator.template_fields">[docs]</a> <span class="n">template_fields</span><span class="p">:</span> <span class="n">Sequence</span><span class="p">[</span><span class="nb">str</span><span class="p">]</span> <span class="o">=</span> <span class="p">(</span>
<span class="s1">&#39;query&#39;</span><span class="p">,</span>
<span class="s1">&#39;variables&#39;</span><span class="p">,</span>
<span class="s1">&#39;job_name&#39;</span><span class="p">,</span>
<span class="s1">&#39;cluster_name&#39;</span><span class="p">,</span>
<span class="s1">&#39;region&#39;</span><span class="p">,</span>
<span class="s1">&#39;dataproc_jars&#39;</span><span class="p">,</span>
<span class="s1">&#39;dataproc_properties&#39;</span><span class="p">,</span>
<span class="s1">&#39;impersonation_chain&#39;</span><span class="p">,</span></div>
<span class="p">)</span>
<div class="viewcode-block" id="DataprocSubmitSparkSqlJobOperator.template_ext"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocSubmitSparkSqlJobOperator.template_ext">[docs]</a> <span class="n">template_ext</span> <span class="o">=</span> <span class="p">(</span><span class="s1">&#39;.q&#39;</span><span class="p">,)</span></div>
<div class="viewcode-block" id="DataprocSubmitSparkSqlJobOperator.template_fields_renderers"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocSubmitSparkSqlJobOperator.template_fields_renderers">[docs]</a> <span class="n">template_fields_renderers</span> <span class="o">=</span> <span class="p">{</span><span class="s1">&#39;sql&#39;</span><span class="p">:</span> <span class="s1">&#39;sql&#39;</span><span class="p">}</span></div>
<div class="viewcode-block" id="DataprocSubmitSparkSqlJobOperator.ui_color"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocSubmitSparkSqlJobOperator.ui_color">[docs]</a> <span class="n">ui_color</span> <span class="o">=</span> <span class="s1">&#39;#0273d4&#39;</span></div>
<div class="viewcode-block" id="DataprocSubmitSparkSqlJobOperator.job_type"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocSubmitSparkSqlJobOperator.job_type">[docs]</a> <span class="n">job_type</span> <span class="o">=</span> <span class="s1">&#39;spark_sql_job&#39;</span></div>
<span class="k">def</span> <span class="fm">__init__</span><span class="p">(</span>
<span class="bp">self</span><span class="p">,</span>
<span class="o">*</span><span class="p">,</span>
<span class="n">query</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">query_uri</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">variables</span><span class="p">:</span> <span class="nb">dict</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="o">**</span><span class="n">kwargs</span><span class="p">,</span>
<span class="p">)</span> <span class="o">-&gt;</span> <span class="kc">None</span><span class="p">:</span>
<span class="c1"># TODO: Remove one day</span>
<span class="n">warnings</span><span class="o">.</span><span class="n">warn</span><span class="p">(</span>
<span class="s2">&quot;The `</span><span class="si">{cls}</span><span class="s2">` operator is deprecated, please use `DataprocSubmitJobOperator` instead. You can use&quot;</span>
<span class="s2">&quot; `generate_job` method of `</span><span class="si">{cls}</span><span class="s2">` to generate dictionary representing your job&quot;</span>
<span class="s2">&quot; and use it with the new operator.&quot;</span><span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="bp">cls</span><span class="o">=</span><span class="nb">type</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span><span class="o">.</span><span class="vm">__name__</span><span class="p">),</span>
<span class="ne">DeprecationWarning</span><span class="p">,</span>
<span class="n">stacklevel</span><span class="o">=</span><span class="mi">1</span><span class="p">,</span>
<span class="p">)</span>
<span class="nb">super</span><span class="p">()</span><span class="o">.</span><span class="fm">__init__</span><span class="p">(</span><span class="o">**</span><span class="n">kwargs</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">query</span> <span class="o">=</span> <span class="n">query</span>
<span class="bp">self</span><span class="o">.</span><span class="n">query_uri</span> <span class="o">=</span> <span class="n">query_uri</span>
<span class="bp">self</span><span class="o">.</span><span class="n">variables</span> <span class="o">=</span> <span class="n">variables</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">query</span> <span class="ow">is</span> <span class="ow">not</span> <span class="kc">None</span> <span class="ow">and</span> <span class="bp">self</span><span class="o">.</span><span class="n">query_uri</span> <span class="ow">is</span> <span class="ow">not</span> <span class="kc">None</span><span class="p">:</span>
<span class="k">raise</span> <span class="n">AirflowException</span><span class="p">(</span><span class="s1">&#39;Only one of `query` and `query_uri` can be passed.&#39;</span><span class="p">)</span>
<div class="viewcode-block" id="DataprocSubmitSparkSqlJobOperator.generate_job"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocSubmitSparkSqlJobOperator.generate_job">[docs]</a> <span class="k">def</span> <span class="nf">generate_job</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
<span class="sd">&quot;&quot;&quot;</span>
<span class="sd"> Helper method for easier migration to `DataprocSubmitJobOperator`.</span>
<span class="sd"> :return: Dict representing Dataproc job</span>
<span class="sd"> &quot;&quot;&quot;</span>
<span class="n">job_template</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">create_job_template</span><span class="p">()</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">query</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
<span class="n">job_template</span><span class="o">.</span><span class="n">add_query_uri</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">query_uri</span><span class="p">)</span>
<span class="k">else</span><span class="p">:</span>
<span class="n">job_template</span><span class="o">.</span><span class="n">add_query</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">query</span><span class="p">)</span>
<span class="n">job_template</span><span class="o">.</span><span class="n">add_variables</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">variables</span><span class="p">)</span>
<span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">_generate_job_template</span><span class="p">()</span></div>
<div class="viewcode-block" id="DataprocSubmitSparkSqlJobOperator.execute"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocSubmitSparkSqlJobOperator.execute">[docs]</a> <span class="k">def</span> <span class="nf">execute</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">context</span><span class="p">:</span> <span class="n">Context</span><span class="p">):</span>
<span class="n">job_template</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">create_job_template</span><span class="p">()</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">query</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">query_uri</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
<span class="k">raise</span> <span class="n">AirflowException</span><span class="p">(</span><span class="s1">&#39;One of query or query_uri should be set here&#39;</span><span class="p">)</span>
<span class="n">job_template</span><span class="o">.</span><span class="n">add_query_uri</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">query_uri</span><span class="p">)</span>
<span class="k">else</span><span class="p">:</span>
<span class="n">job_template</span><span class="o">.</span><span class="n">add_query</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">query</span><span class="p">)</span>
<span class="n">job_template</span><span class="o">.</span><span class="n">add_variables</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">variables</span><span class="p">)</span>
<span class="nb">super</span><span class="p">()</span><span class="o">.</span><span class="n">execute</span><span class="p">(</span><span class="n">context</span><span class="p">)</span></div></div>
<div class="viewcode-block" id="DataprocSubmitSparkJobOperator"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocSubmitSparkJobOperator">[docs]</a><span class="k">class</span> <span class="nc">DataprocSubmitSparkJobOperator</span><span class="p">(</span><span class="n">DataprocJobBaseOperator</span><span class="p">):</span>
<span class="sd">&quot;&quot;&quot;</span>
<span class="sd"> Start a Spark Job on a Cloud DataProc cluster.</span>
<span class="sd"> :param main_jar: The HCFS URI of the jar file that contains the main class</span>
<span class="sd"> (use this or the main_class, not both together).</span>
<span class="sd"> :param main_class: Name of the job class. (use this or the main_jar, not both</span>
<span class="sd"> together).</span>
<span class="sd"> :param arguments: Arguments for the job. (templated)</span>
<span class="sd"> :param archives: List of archived files that will be unpacked in the work</span>
<span class="sd"> directory. Should be stored in Cloud Storage.</span>
<span class="sd"> :param files: List of files to be copied to the working directory</span>
<span class="sd"> &quot;&quot;&quot;</span>
<div class="viewcode-block" id="DataprocSubmitSparkJobOperator.template_fields"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocSubmitSparkJobOperator.template_fields">[docs]</a> <span class="n">template_fields</span><span class="p">:</span> <span class="n">Sequence</span><span class="p">[</span><span class="nb">str</span><span class="p">]</span> <span class="o">=</span> <span class="p">(</span>
<span class="s1">&#39;arguments&#39;</span><span class="p">,</span>
<span class="s1">&#39;job_name&#39;</span><span class="p">,</span>
<span class="s1">&#39;cluster_name&#39;</span><span class="p">,</span>
<span class="s1">&#39;region&#39;</span><span class="p">,</span>
<span class="s1">&#39;dataproc_jars&#39;</span><span class="p">,</span>
<span class="s1">&#39;dataproc_properties&#39;</span><span class="p">,</span>
<span class="s1">&#39;impersonation_chain&#39;</span><span class="p">,</span></div>
<span class="p">)</span>
<div class="viewcode-block" id="DataprocSubmitSparkJobOperator.ui_color"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocSubmitSparkJobOperator.ui_color">[docs]</a> <span class="n">ui_color</span> <span class="o">=</span> <span class="s1">&#39;#0273d4&#39;</span></div>
<div class="viewcode-block" id="DataprocSubmitSparkJobOperator.job_type"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocSubmitSparkJobOperator.job_type">[docs]</a> <span class="n">job_type</span> <span class="o">=</span> <span class="s1">&#39;spark_job&#39;</span></div>
<span class="k">def</span> <span class="fm">__init__</span><span class="p">(</span>
<span class="bp">self</span><span class="p">,</span>
<span class="o">*</span><span class="p">,</span>
<span class="n">main_jar</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">main_class</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">arguments</span><span class="p">:</span> <span class="nb">list</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">archives</span><span class="p">:</span> <span class="nb">list</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">files</span><span class="p">:</span> <span class="nb">list</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="o">**</span><span class="n">kwargs</span><span class="p">,</span>
<span class="p">)</span> <span class="o">-&gt;</span> <span class="kc">None</span><span class="p">:</span>
<span class="c1"># TODO: Remove one day</span>
<span class="n">warnings</span><span class="o">.</span><span class="n">warn</span><span class="p">(</span>
<span class="s2">&quot;The `</span><span class="si">{cls}</span><span class="s2">` operator is deprecated, please use `DataprocSubmitJobOperator` instead. You can use&quot;</span>
<span class="s2">&quot; `generate_job` method of `</span><span class="si">{cls}</span><span class="s2">` to generate dictionary representing your job&quot;</span>
<span class="s2">&quot; and use it with the new operator.&quot;</span><span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="bp">cls</span><span class="o">=</span><span class="nb">type</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span><span class="o">.</span><span class="vm">__name__</span><span class="p">),</span>
<span class="ne">DeprecationWarning</span><span class="p">,</span>
<span class="n">stacklevel</span><span class="o">=</span><span class="mi">1</span><span class="p">,</span>
<span class="p">)</span>
<span class="nb">super</span><span class="p">()</span><span class="o">.</span><span class="fm">__init__</span><span class="p">(</span><span class="o">**</span><span class="n">kwargs</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">main_jar</span> <span class="o">=</span> <span class="n">main_jar</span>
<span class="bp">self</span><span class="o">.</span><span class="n">main_class</span> <span class="o">=</span> <span class="n">main_class</span>
<span class="bp">self</span><span class="o">.</span><span class="n">arguments</span> <span class="o">=</span> <span class="n">arguments</span>
<span class="bp">self</span><span class="o">.</span><span class="n">archives</span> <span class="o">=</span> <span class="n">archives</span>
<span class="bp">self</span><span class="o">.</span><span class="n">files</span> <span class="o">=</span> <span class="n">files</span>
<div class="viewcode-block" id="DataprocSubmitSparkJobOperator.generate_job"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocSubmitSparkJobOperator.generate_job">[docs]</a> <span class="k">def</span> <span class="nf">generate_job</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
<span class="sd">&quot;&quot;&quot;</span>
<span class="sd"> Helper method for easier migration to `DataprocSubmitJobOperator`.</span>
<span class="sd"> :return: Dict representing Dataproc job</span>
<span class="sd"> &quot;&quot;&quot;</span>
<span class="n">job_template</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">create_job_template</span><span class="p">()</span>
<span class="n">job_template</span><span class="o">.</span><span class="n">set_main</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">main_jar</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">main_class</span><span class="p">)</span>
<span class="n">job_template</span><span class="o">.</span><span class="n">add_args</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">arguments</span><span class="p">)</span>
<span class="n">job_template</span><span class="o">.</span><span class="n">add_archive_uris</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">archives</span><span class="p">)</span>
<span class="n">job_template</span><span class="o">.</span><span class="n">add_file_uris</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">files</span><span class="p">)</span>
<span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">_generate_job_template</span><span class="p">()</span></div>
<div class="viewcode-block" id="DataprocSubmitSparkJobOperator.execute"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocSubmitSparkJobOperator.execute">[docs]</a> <span class="k">def</span> <span class="nf">execute</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">context</span><span class="p">:</span> <span class="n">Context</span><span class="p">):</span>
<span class="n">job_template</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">create_job_template</span><span class="p">()</span>
<span class="n">job_template</span><span class="o">.</span><span class="n">set_main</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">main_jar</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">main_class</span><span class="p">)</span>
<span class="n">job_template</span><span class="o">.</span><span class="n">add_args</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">arguments</span><span class="p">)</span>
<span class="n">job_template</span><span class="o">.</span><span class="n">add_archive_uris</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">archives</span><span class="p">)</span>
<span class="n">job_template</span><span class="o">.</span><span class="n">add_file_uris</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">files</span><span class="p">)</span>
<span class="nb">super</span><span class="p">()</span><span class="o">.</span><span class="n">execute</span><span class="p">(</span><span class="n">context</span><span class="p">)</span></div></div>
<div class="viewcode-block" id="DataprocSubmitHadoopJobOperator"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocSubmitHadoopJobOperator">[docs]</a><span class="k">class</span> <span class="nc">DataprocSubmitHadoopJobOperator</span><span class="p">(</span><span class="n">DataprocJobBaseOperator</span><span class="p">):</span>
<span class="sd">&quot;&quot;&quot;</span>
<span class="sd"> Start a Hadoop Job on a Cloud DataProc cluster.</span>
<span class="sd"> :param main_jar: The HCFS URI of the jar file containing the main class</span>
<span class="sd"> (use this or the main_class, not both together).</span>
<span class="sd"> :param main_class: Name of the job class. (use this or the main_jar, not both</span>
<span class="sd"> together).</span>
<span class="sd"> :param arguments: Arguments for the job. (templated)</span>
<span class="sd"> :param archives: List of archived files that will be unpacked in the work</span>
<span class="sd"> directory. Should be stored in Cloud Storage.</span>
<span class="sd"> :param files: List of files to be copied to the working directory</span>
<span class="sd"> &quot;&quot;&quot;</span>
<div class="viewcode-block" id="DataprocSubmitHadoopJobOperator.template_fields"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocSubmitHadoopJobOperator.template_fields">[docs]</a> <span class="n">template_fields</span><span class="p">:</span> <span class="n">Sequence</span><span class="p">[</span><span class="nb">str</span><span class="p">]</span> <span class="o">=</span> <span class="p">(</span>
<span class="s1">&#39;arguments&#39;</span><span class="p">,</span>
<span class="s1">&#39;job_name&#39;</span><span class="p">,</span>
<span class="s1">&#39;cluster_name&#39;</span><span class="p">,</span>
<span class="s1">&#39;region&#39;</span><span class="p">,</span>
<span class="s1">&#39;dataproc_jars&#39;</span><span class="p">,</span>
<span class="s1">&#39;dataproc_properties&#39;</span><span class="p">,</span>
<span class="s1">&#39;impersonation_chain&#39;</span><span class="p">,</span></div>
<span class="p">)</span>
<div class="viewcode-block" id="DataprocSubmitHadoopJobOperator.ui_color"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocSubmitHadoopJobOperator.ui_color">[docs]</a> <span class="n">ui_color</span> <span class="o">=</span> <span class="s1">&#39;#0273d4&#39;</span></div>
<div class="viewcode-block" id="DataprocSubmitHadoopJobOperator.job_type"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocSubmitHadoopJobOperator.job_type">[docs]</a> <span class="n">job_type</span> <span class="o">=</span> <span class="s1">&#39;hadoop_job&#39;</span></div>
<span class="k">def</span> <span class="fm">__init__</span><span class="p">(</span>
<span class="bp">self</span><span class="p">,</span>
<span class="o">*</span><span class="p">,</span>
<span class="n">main_jar</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">main_class</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">arguments</span><span class="p">:</span> <span class="nb">list</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">archives</span><span class="p">:</span> <span class="nb">list</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">files</span><span class="p">:</span> <span class="nb">list</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="o">**</span><span class="n">kwargs</span><span class="p">,</span>
<span class="p">)</span> <span class="o">-&gt;</span> <span class="kc">None</span><span class="p">:</span>
<span class="c1"># TODO: Remove one day</span>
<span class="n">warnings</span><span class="o">.</span><span class="n">warn</span><span class="p">(</span>
<span class="s2">&quot;The `</span><span class="si">{cls}</span><span class="s2">` operator is deprecated, please use `DataprocSubmitJobOperator` instead. You can use&quot;</span>
<span class="s2">&quot; `generate_job` method of `</span><span class="si">{cls}</span><span class="s2">` to generate dictionary representing your job&quot;</span>
<span class="s2">&quot; and use it with the new operator.&quot;</span><span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="bp">cls</span><span class="o">=</span><span class="nb">type</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span><span class="o">.</span><span class="vm">__name__</span><span class="p">),</span>
<span class="ne">DeprecationWarning</span><span class="p">,</span>
<span class="n">stacklevel</span><span class="o">=</span><span class="mi">1</span><span class="p">,</span>
<span class="p">)</span>
<span class="nb">super</span><span class="p">()</span><span class="o">.</span><span class="fm">__init__</span><span class="p">(</span><span class="o">**</span><span class="n">kwargs</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">main_jar</span> <span class="o">=</span> <span class="n">main_jar</span>
<span class="bp">self</span><span class="o">.</span><span class="n">main_class</span> <span class="o">=</span> <span class="n">main_class</span>
<span class="bp">self</span><span class="o">.</span><span class="n">arguments</span> <span class="o">=</span> <span class="n">arguments</span>
<span class="bp">self</span><span class="o">.</span><span class="n">archives</span> <span class="o">=</span> <span class="n">archives</span>
<span class="bp">self</span><span class="o">.</span><span class="n">files</span> <span class="o">=</span> <span class="n">files</span>
<div class="viewcode-block" id="DataprocSubmitHadoopJobOperator.generate_job"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocSubmitHadoopJobOperator.generate_job">[docs]</a> <span class="k">def</span> <span class="nf">generate_job</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
<span class="sd">&quot;&quot;&quot;</span>
<span class="sd"> Helper method for easier migration to `DataprocSubmitJobOperator`.</span>
<span class="sd"> :return: Dict representing Dataproc job</span>
<span class="sd"> &quot;&quot;&quot;</span>
<span class="n">job_template</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">create_job_template</span><span class="p">()</span>
<span class="n">job_template</span><span class="o">.</span><span class="n">set_main</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">main_jar</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">main_class</span><span class="p">)</span>
<span class="n">job_template</span><span class="o">.</span><span class="n">add_args</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">arguments</span><span class="p">)</span>
<span class="n">job_template</span><span class="o">.</span><span class="n">add_archive_uris</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">archives</span><span class="p">)</span>
<span class="n">job_template</span><span class="o">.</span><span class="n">add_file_uris</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">files</span><span class="p">)</span>
<span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">_generate_job_template</span><span class="p">()</span></div>
<div class="viewcode-block" id="DataprocSubmitHadoopJobOperator.execute"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocSubmitHadoopJobOperator.execute">[docs]</a> <span class="k">def</span> <span class="nf">execute</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">context</span><span class="p">:</span> <span class="n">Context</span><span class="p">):</span>
<span class="n">job_template</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">create_job_template</span><span class="p">()</span>
<span class="n">job_template</span><span class="o">.</span><span class="n">set_main</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">main_jar</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">main_class</span><span class="p">)</span>
<span class="n">job_template</span><span class="o">.</span><span class="n">add_args</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">arguments</span><span class="p">)</span>
<span class="n">job_template</span><span class="o">.</span><span class="n">add_archive_uris</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">archives</span><span class="p">)</span>
<span class="n">job_template</span><span class="o">.</span><span class="n">add_file_uris</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">files</span><span class="p">)</span>
<span class="nb">super</span><span class="p">()</span><span class="o">.</span><span class="n">execute</span><span class="p">(</span><span class="n">context</span><span class="p">)</span></div></div>
<div class="viewcode-block" id="DataprocSubmitPySparkJobOperator"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocSubmitPySparkJobOperator">[docs]</a><span class="k">class</span> <span class="nc">DataprocSubmitPySparkJobOperator</span><span class="p">(</span><span class="n">DataprocJobBaseOperator</span><span class="p">):</span>
<span class="sd">&quot;&quot;&quot;</span>
<span class="sd"> Start a PySpark Job on a Cloud DataProc cluster.</span>
<span class="sd"> :param main: [Required] The Hadoop Compatible Filesystem (HCFS) URI of the main</span>
<span class="sd"> Python file to use as the driver. Must be a .py file. (templated)</span>
<span class="sd"> :param arguments: Arguments for the job. (templated)</span>
<span class="sd"> :param archives: List of archived files that will be unpacked in the work</span>
<span class="sd"> directory. Should be stored in Cloud Storage.</span>
<span class="sd"> :param files: List of files to be copied to the working directory</span>
<span class="sd"> :param pyfiles: List of Python files to pass to the PySpark framework.</span>
<span class="sd"> Supported file types: .py, .egg, and .zip</span>
<span class="sd"> &quot;&quot;&quot;</span>
<div class="viewcode-block" id="DataprocSubmitPySparkJobOperator.template_fields"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocSubmitPySparkJobOperator.template_fields">[docs]</a> <span class="n">template_fields</span><span class="p">:</span> <span class="n">Sequence</span><span class="p">[</span><span class="nb">str</span><span class="p">]</span> <span class="o">=</span> <span class="p">(</span>
<span class="s1">&#39;main&#39;</span><span class="p">,</span>
<span class="s1">&#39;arguments&#39;</span><span class="p">,</span>
<span class="s1">&#39;job_name&#39;</span><span class="p">,</span>
<span class="s1">&#39;cluster_name&#39;</span><span class="p">,</span>
<span class="s1">&#39;region&#39;</span><span class="p">,</span>
<span class="s1">&#39;dataproc_jars&#39;</span><span class="p">,</span>
<span class="s1">&#39;dataproc_properties&#39;</span><span class="p">,</span>
<span class="s1">&#39;impersonation_chain&#39;</span><span class="p">,</span></div>
<span class="p">)</span>
<div class="viewcode-block" id="DataprocSubmitPySparkJobOperator.ui_color"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocSubmitPySparkJobOperator.ui_color">[docs]</a> <span class="n">ui_color</span> <span class="o">=</span> <span class="s1">&#39;#0273d4&#39;</span></div>
<div class="viewcode-block" id="DataprocSubmitPySparkJobOperator.job_type"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocSubmitPySparkJobOperator.job_type">[docs]</a> <span class="n">job_type</span> <span class="o">=</span> <span class="s1">&#39;pyspark_job&#39;</span></div>
<span class="nd">@staticmethod</span>
<span class="k">def</span> <span class="nf">_generate_temp_filename</span><span class="p">(</span><span class="n">filename</span><span class="p">):</span>
<span class="n">date</span> <span class="o">=</span> <span class="n">time</span><span class="o">.</span><span class="n">strftime</span><span class="p">(</span><span class="s1">&#39;%Y%m</span><span class="si">%d</span><span class="s1">%H%M%S&#39;</span><span class="p">)</span>
<span class="k">return</span> <span class="sa">f</span><span class="s2">&quot;</span><span class="si">{</span><span class="n">date</span><span class="si">}</span><span class="s2">_</span><span class="si">{</span><span class="nb">str</span><span class="p">(</span><span class="n">uuid</span><span class="o">.</span><span class="n">uuid4</span><span class="p">())[:</span><span class="mi">8</span><span class="p">]</span><span class="si">}</span><span class="s2">_</span><span class="si">{</span><span class="n">ntpath</span><span class="o">.</span><span class="n">basename</span><span class="p">(</span><span class="n">filename</span><span class="p">)</span><span class="si">}</span><span class="s2">&quot;</span>
<span class="k">def</span> <span class="nf">_upload_file_temp</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">bucket</span><span class="p">,</span> <span class="n">local_file</span><span class="p">):</span>
<span class="sd">&quot;&quot;&quot;Upload a local file to a Google Cloud Storage bucket.&quot;&quot;&quot;</span>
<span class="n">temp_filename</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_generate_temp_filename</span><span class="p">(</span><span class="n">local_file</span><span class="p">)</span>
<span class="k">if</span> <span class="ow">not</span> <span class="n">bucket</span><span class="p">:</span>
<span class="k">raise</span> <span class="n">AirflowException</span><span class="p">(</span>
<span class="s2">&quot;If you want Airflow to upload the local file to a temporary bucket, set &quot;</span>
<span class="s2">&quot;the &#39;temp_bucket&#39; key in the connection string&quot;</span>
<span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">log</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s2">&quot;Uploading </span><span class="si">%s</span><span class="s2"> to </span><span class="si">%s</span><span class="s2">&quot;</span><span class="p">,</span> <span class="n">local_file</span><span class="p">,</span> <span class="n">temp_filename</span><span class="p">)</span>
<span class="n">GCSHook</span><span class="p">(</span><span class="n">gcp_conn_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">gcp_conn_id</span><span class="p">,</span> <span class="n">impersonation_chain</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">impersonation_chain</span><span class="p">)</span><span class="o">.</span><span class="n">upload</span><span class="p">(</span>
<span class="n">bucket_name</span><span class="o">=</span><span class="n">bucket</span><span class="p">,</span>
<span class="n">object_name</span><span class="o">=</span><span class="n">temp_filename</span><span class="p">,</span>
<span class="n">mime_type</span><span class="o">=</span><span class="s1">&#39;application/x-python&#39;</span><span class="p">,</span>
<span class="n">filename</span><span class="o">=</span><span class="n">local_file</span><span class="p">,</span>
<span class="p">)</span>
<span class="k">return</span> <span class="sa">f</span><span class="s2">&quot;gs://</span><span class="si">{</span><span class="n">bucket</span><span class="si">}</span><span class="s2">/</span><span class="si">{</span><span class="n">temp_filename</span><span class="si">}</span><span class="s2">&quot;</span>
<span class="k">def</span> <span class="fm">__init__</span><span class="p">(</span>
<span class="bp">self</span><span class="p">,</span>
<span class="o">*</span><span class="p">,</span>
<span class="n">main</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span>
<span class="n">arguments</span><span class="p">:</span> <span class="nb">list</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">archives</span><span class="p">:</span> <span class="nb">list</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">pyfiles</span><span class="p">:</span> <span class="nb">list</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">files</span><span class="p">:</span> <span class="nb">list</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="o">**</span><span class="n">kwargs</span><span class="p">,</span>
<span class="p">)</span> <span class="o">-&gt;</span> <span class="kc">None</span><span class="p">:</span>
<span class="c1"># TODO: Remove one day</span>
<span class="n">warnings</span><span class="o">.</span><span class="n">warn</span><span class="p">(</span>
<span class="s2">&quot;The `</span><span class="si">{cls}</span><span class="s2">` operator is deprecated, please use `DataprocSubmitJobOperator` instead. You can use&quot;</span>
<span class="s2">&quot; `generate_job` method of `</span><span class="si">{cls}</span><span class="s2">` to generate dictionary representing your job&quot;</span>
<span class="s2">&quot; and use it with the new operator.&quot;</span><span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="bp">cls</span><span class="o">=</span><span class="nb">type</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span><span class="o">.</span><span class="vm">__name__</span><span class="p">),</span>
<span class="ne">DeprecationWarning</span><span class="p">,</span>
<span class="n">stacklevel</span><span class="o">=</span><span class="mi">1</span><span class="p">,</span>
<span class="p">)</span>
<span class="nb">super</span><span class="p">()</span><span class="o">.</span><span class="fm">__init__</span><span class="p">(</span><span class="o">**</span><span class="n">kwargs</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">main</span> <span class="o">=</span> <span class="n">main</span>
<span class="bp">self</span><span class="o">.</span><span class="n">arguments</span> <span class="o">=</span> <span class="n">arguments</span>
<span class="bp">self</span><span class="o">.</span><span class="n">archives</span> <span class="o">=</span> <span class="n">archives</span>
<span class="bp">self</span><span class="o">.</span><span class="n">files</span> <span class="o">=</span> <span class="n">files</span>
<span class="bp">self</span><span class="o">.</span><span class="n">pyfiles</span> <span class="o">=</span> <span class="n">pyfiles</span>
<div class="viewcode-block" id="DataprocSubmitPySparkJobOperator.generate_job"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocSubmitPySparkJobOperator.generate_job">[docs]</a> <span class="k">def</span> <span class="nf">generate_job</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
<span class="sd">&quot;&quot;&quot;</span>
<span class="sd"> Helper method for easier migration to `DataprocSubmitJobOperator`.</span>
<span class="sd"> :return: Dict representing Dataproc job</span>
<span class="sd"> &quot;&quot;&quot;</span>
<span class="n">job_template</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">create_job_template</span><span class="p">()</span>
<span class="c1"># Check if the file is local, if that is the case, upload it to a bucket</span>
<span class="k">if</span> <span class="n">os</span><span class="o">.</span><span class="n">path</span><span class="o">.</span><span class="n">isfile</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">main</span><span class="p">):</span>
<span class="n">cluster_info</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">hook</span><span class="o">.</span><span class="n">get_cluster</span><span class="p">(</span>
<span class="n">project_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">project_id</span><span class="p">,</span> <span class="n">region</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">region</span><span class="p">,</span> <span class="n">cluster_name</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">cluster_name</span>
<span class="p">)</span>
<span class="n">bucket</span> <span class="o">=</span> <span class="n">cluster_info</span><span class="p">[</span><span class="s1">&#39;config&#39;</span><span class="p">][</span><span class="s1">&#39;config_bucket&#39;</span><span class="p">]</span>
<span class="bp">self</span><span class="o">.</span><span class="n">main</span> <span class="o">=</span> <span class="sa">f</span><span class="s2">&quot;gs://</span><span class="si">{</span><span class="n">bucket</span><span class="si">}</span><span class="s2">/</span><span class="si">{</span><span class="bp">self</span><span class="o">.</span><span class="n">main</span><span class="si">}</span><span class="s2">&quot;</span>
<span class="n">job_template</span><span class="o">.</span><span class="n">set_python_main</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">main</span><span class="p">)</span>
<span class="n">job_template</span><span class="o">.</span><span class="n">add_args</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">arguments</span><span class="p">)</span>
<span class="n">job_template</span><span class="o">.</span><span class="n">add_archive_uris</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">archives</span><span class="p">)</span>
<span class="n">job_template</span><span class="o">.</span><span class="n">add_file_uris</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">files</span><span class="p">)</span>
<span class="n">job_template</span><span class="o">.</span><span class="n">add_python_file_uris</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">pyfiles</span><span class="p">)</span>
<span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">_generate_job_template</span><span class="p">()</span></div>
<div class="viewcode-block" id="DataprocSubmitPySparkJobOperator.execute"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocSubmitPySparkJobOperator.execute">[docs]</a> <span class="k">def</span> <span class="nf">execute</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">context</span><span class="p">:</span> <span class="n">Context</span><span class="p">):</span>
<span class="n">job_template</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">create_job_template</span><span class="p">()</span>
<span class="c1"># Check if the file is local, if that is the case, upload it to a bucket</span>
<span class="k">if</span> <span class="n">os</span><span class="o">.</span><span class="n">path</span><span class="o">.</span><span class="n">isfile</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">main</span><span class="p">):</span>
<span class="n">cluster_info</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">hook</span><span class="o">.</span><span class="n">get_cluster</span><span class="p">(</span>
<span class="n">project_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">project_id</span><span class="p">,</span> <span class="n">region</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">region</span><span class="p">,</span> <span class="n">cluster_name</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">cluster_name</span>
<span class="p">)</span>
<span class="n">bucket</span> <span class="o">=</span> <span class="n">cluster_info</span><span class="p">[</span><span class="s1">&#39;config&#39;</span><span class="p">][</span><span class="s1">&#39;config_bucket&#39;</span><span class="p">]</span>
<span class="bp">self</span><span class="o">.</span><span class="n">main</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_upload_file_temp</span><span class="p">(</span><span class="n">bucket</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">main</span><span class="p">)</span>
<span class="n">job_template</span><span class="o">.</span><span class="n">set_python_main</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">main</span><span class="p">)</span>
<span class="n">job_template</span><span class="o">.</span><span class="n">add_args</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">arguments</span><span class="p">)</span>
<span class="n">job_template</span><span class="o">.</span><span class="n">add_archive_uris</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">archives</span><span class="p">)</span>
<span class="n">job_template</span><span class="o">.</span><span class="n">add_file_uris</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">files</span><span class="p">)</span>
<span class="n">job_template</span><span class="o">.</span><span class="n">add_python_file_uris</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">pyfiles</span><span class="p">)</span>
<span class="nb">super</span><span class="p">()</span><span class="o">.</span><span class="n">execute</span><span class="p">(</span><span class="n">context</span><span class="p">)</span></div></div>
<div class="viewcode-block" id="DataprocCreateWorkflowTemplateOperator"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocCreateWorkflowTemplateOperator">[docs]</a><span class="k">class</span> <span class="nc">DataprocCreateWorkflowTemplateOperator</span><span class="p">(</span><span class="n">BaseOperator</span><span class="p">):</span>
<span class="sd">&quot;&quot;&quot;</span>
<span class="sd"> Creates new workflow template.</span>
<span class="sd"> :param project_id: Optional. The ID of the Google Cloud project the cluster belongs to.</span>
<span class="sd"> :param region: Required. The Cloud Dataproc region in which to handle the request.</span>
<span class="sd"> :param template: The Dataproc workflow template to create. If a dict is provided,</span>
<span class="sd"> it must be of the same form as the protobuf message WorkflowTemplate.</span>
<span class="sd"> :param retry: A retry object used to retry requests. If ``None`` is specified, requests will not be</span>
<span class="sd"> retried.</span>
<span class="sd"> :param timeout: The amount of time, in seconds, to wait for the request to complete. Note that if</span>
<span class="sd"> ``retry`` is specified, the timeout applies to each individual attempt.</span>
<span class="sd"> :param metadata: Additional metadata that is provided to the method.</span>
<span class="sd"> &quot;&quot;&quot;</span>
<div class="viewcode-block" id="DataprocCreateWorkflowTemplateOperator.template_fields"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocCreateWorkflowTemplateOperator.template_fields">[docs]</a> <span class="n">template_fields</span><span class="p">:</span> <span class="n">Sequence</span><span class="p">[</span><span class="nb">str</span><span class="p">]</span> <span class="o">=</span> <span class="p">(</span><span class="s2">&quot;region&quot;</span><span class="p">,</span> <span class="s2">&quot;template&quot;</span><span class="p">)</span></div>
<div class="viewcode-block" id="DataprocCreateWorkflowTemplateOperator.template_fields_renderers"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocCreateWorkflowTemplateOperator.template_fields_renderers">[docs]</a> <span class="n">template_fields_renderers</span> <span class="o">=</span> <span class="p">{</span><span class="s2">&quot;template&quot;</span><span class="p">:</span> <span class="s2">&quot;json&quot;</span><span class="p">}</span></div>
<div class="viewcode-block" id="DataprocCreateWorkflowTemplateOperator.operator_extra_links"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocCreateWorkflowTemplateOperator.operator_extra_links">[docs]</a> <span class="n">operator_extra_links</span> <span class="o">=</span> <span class="p">(</span><span class="n">DataprocLink</span><span class="p">(),)</span></div>
<span class="k">def</span> <span class="fm">__init__</span><span class="p">(</span>
<span class="bp">self</span><span class="p">,</span>
<span class="o">*</span><span class="p">,</span>
<span class="n">template</span><span class="p">:</span> <span class="nb">dict</span><span class="p">,</span>
<span class="n">region</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span>
<span class="n">project_id</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">retry</span><span class="p">:</span> <span class="n">Retry</span> <span class="o">|</span> <span class="n">_MethodDefault</span> <span class="o">=</span> <span class="n">DEFAULT</span><span class="p">,</span>
<span class="n">timeout</span><span class="p">:</span> <span class="nb">float</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">metadata</span><span class="p">:</span> <span class="n">Sequence</span><span class="p">[</span><span class="nb">tuple</span><span class="p">[</span><span class="nb">str</span><span class="p">,</span> <span class="nb">str</span><span class="p">]]</span> <span class="o">=</span> <span class="p">(),</span>
<span class="n">gcp_conn_id</span><span class="p">:</span> <span class="nb">str</span> <span class="o">=</span> <span class="s2">&quot;google_cloud_default&quot;</span><span class="p">,</span>
<span class="n">impersonation_chain</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="n">Sequence</span><span class="p">[</span><span class="nb">str</span><span class="p">]</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="o">**</span><span class="n">kwargs</span><span class="p">,</span>
<span class="p">):</span>
<span class="nb">super</span><span class="p">()</span><span class="o">.</span><span class="fm">__init__</span><span class="p">(</span><span class="o">**</span><span class="n">kwargs</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">region</span> <span class="o">=</span> <span class="n">region</span>
<span class="bp">self</span><span class="o">.</span><span class="n">template</span> <span class="o">=</span> <span class="n">template</span>
<span class="bp">self</span><span class="o">.</span><span class="n">project_id</span> <span class="o">=</span> <span class="n">project_id</span>
<span class="bp">self</span><span class="o">.</span><span class="n">retry</span> <span class="o">=</span> <span class="n">retry</span>
<span class="bp">self</span><span class="o">.</span><span class="n">timeout</span> <span class="o">=</span> <span class="n">timeout</span>
<span class="bp">self</span><span class="o">.</span><span class="n">metadata</span> <span class="o">=</span> <span class="n">metadata</span>
<span class="bp">self</span><span class="o">.</span><span class="n">gcp_conn_id</span> <span class="o">=</span> <span class="n">gcp_conn_id</span>
<span class="bp">self</span><span class="o">.</span><span class="n">impersonation_chain</span> <span class="o">=</span> <span class="n">impersonation_chain</span>
<div class="viewcode-block" id="DataprocCreateWorkflowTemplateOperator.execute"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocCreateWorkflowTemplateOperator.execute">[docs]</a> <span class="k">def</span> <span class="nf">execute</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">context</span><span class="p">:</span> <span class="n">Context</span><span class="p">):</span>
<span class="n">hook</span> <span class="o">=</span> <span class="n">DataprocHook</span><span class="p">(</span><span class="n">gcp_conn_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">gcp_conn_id</span><span class="p">,</span> <span class="n">impersonation_chain</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">impersonation_chain</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">log</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s2">&quot;Creating template&quot;</span><span class="p">)</span>
<span class="k">try</span><span class="p">:</span>
<span class="n">workflow</span> <span class="o">=</span> <span class="n">hook</span><span class="o">.</span><span class="n">create_workflow_template</span><span class="p">(</span>
<span class="n">region</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">region</span><span class="p">,</span>
<span class="n">template</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">template</span><span class="p">,</span>
<span class="n">project_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">project_id</span><span class="p">,</span>
<span class="n">retry</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">retry</span><span class="p">,</span>
<span class="n">timeout</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">timeout</span><span class="p">,</span>
<span class="n">metadata</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">metadata</span><span class="p">,</span>
<span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">log</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s2">&quot;Workflow </span><span class="si">%s</span><span class="s2"> created&quot;</span><span class="p">,</span> <span class="n">workflow</span><span class="o">.</span><span class="n">name</span><span class="p">)</span>
<span class="k">except</span> <span class="n">AlreadyExists</span><span class="p">:</span>
<span class="bp">self</span><span class="o">.</span><span class="n">log</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s2">&quot;Workflow with given id already exists&quot;</span><span class="p">)</span>
<span class="n">DataprocLink</span><span class="o">.</span><span class="n">persist</span><span class="p">(</span>
<span class="n">context</span><span class="o">=</span><span class="n">context</span><span class="p">,</span>
<span class="n">task_instance</span><span class="o">=</span><span class="bp">self</span><span class="p">,</span>
<span class="n">url</span><span class="o">=</span><span class="n">DATAPROC_WORKFLOW_TEMPLATE_LINK</span><span class="p">,</span>
<span class="n">resource</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">template</span><span class="p">[</span><span class="s2">&quot;id&quot;</span><span class="p">],</span></div></div>
<span class="p">)</span>
<div class="viewcode-block" id="DataprocInstantiateWorkflowTemplateOperator"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocInstantiateWorkflowTemplateOperator">[docs]</a><span class="k">class</span> <span class="nc">DataprocInstantiateWorkflowTemplateOperator</span><span class="p">(</span><span class="n">BaseOperator</span><span class="p">):</span>
<span class="sd">&quot;&quot;&quot;</span>
<span class="sd"> Instantiate a WorkflowTemplate on Google Cloud Dataproc. The operator will wait</span>
<span class="sd"> until the WorkflowTemplate is finished executing.</span>
<span class="sd"> .. seealso::</span>
<span class="sd"> Please refer to:</span>
<span class="sd"> https://cloud.google.com/dataproc/docs/reference/rest/v1beta2/projects.regions.workflowTemplates/instantiate</span>
<span class="sd"> :param template_id: The id of the template. (templated)</span>
<span class="sd"> :param project_id: The ID of the google cloud project in which</span>
<span class="sd"> the template runs</span>
<span class="sd"> :param region: The specified region where the dataproc cluster is created.</span>
<span class="sd"> :param parameters: a map of parameters for Dataproc Template in key-value format:</span>
<span class="sd"> map (key: string, value: string)</span>
<span class="sd"> Example: { &quot;date_from&quot;: &quot;2019-08-01&quot;, &quot;date_to&quot;: &quot;2019-08-02&quot;}.</span>
<span class="sd"> Values may not exceed 100 characters. Please refer to:</span>
<span class="sd"> https://cloud.google.com/dataproc/docs/concepts/workflows/workflow-parameters</span>
<span class="sd"> :param request_id: Optional. A unique id used to identify the request. If the server receives two</span>
<span class="sd"> ``SubmitJobRequest`` requests with the same id, then the second request will be ignored and the first</span>
<span class="sd"> ``Job`` created and stored in the backend is returned.</span>
<span class="sd"> It is recommended to always set this value to a UUID.</span>
<span class="sd"> :param retry: A retry object used to retry requests. If ``None`` is specified, requests will not be</span>
<span class="sd"> retried.</span>
<span class="sd"> :param timeout: The amount of time, in seconds, to wait for the request to complete. Note that if</span>
<span class="sd"> ``retry`` is specified, the timeout applies to each individual attempt.</span>
<span class="sd"> :param metadata: Additional metadata that is provided to the method.</span>
<span class="sd"> :param gcp_conn_id: The connection ID to use connecting to Google Cloud.</span>
<span class="sd"> :param impersonation_chain: Optional service account to impersonate using short-term</span>
<span class="sd"> credentials, or chained list of accounts required to get the access_token</span>
<span class="sd"> of the last account in the list, which will be impersonated in the request.</span>
<span class="sd"> If set as a string, the account must grant the originating account</span>
<span class="sd"> the Service Account Token Creator IAM role.</span>
<span class="sd"> If set as a sequence, the identities from the list must grant</span>
<span class="sd"> Service Account Token Creator IAM role to the directly preceding identity, with first</span>
<span class="sd"> account from the list granting this role to the originating account (templated).</span>
<span class="sd"> &quot;&quot;&quot;</span>
<div class="viewcode-block" id="DataprocInstantiateWorkflowTemplateOperator.template_fields"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocInstantiateWorkflowTemplateOperator.template_fields">[docs]</a> <span class="n">template_fields</span><span class="p">:</span> <span class="n">Sequence</span><span class="p">[</span><span class="nb">str</span><span class="p">]</span> <span class="o">=</span> <span class="p">(</span><span class="s1">&#39;template_id&#39;</span><span class="p">,</span> <span class="s1">&#39;impersonation_chain&#39;</span><span class="p">,</span> <span class="s1">&#39;request_id&#39;</span><span class="p">,</span> <span class="s1">&#39;parameters&#39;</span><span class="p">)</span></div>
<div class="viewcode-block" id="DataprocInstantiateWorkflowTemplateOperator.template_fields_renderers"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocInstantiateWorkflowTemplateOperator.template_fields_renderers">[docs]</a> <span class="n">template_fields_renderers</span> <span class="o">=</span> <span class="p">{</span><span class="s2">&quot;parameters&quot;</span><span class="p">:</span> <span class="s2">&quot;json&quot;</span><span class="p">}</span></div>
<div class="viewcode-block" id="DataprocInstantiateWorkflowTemplateOperator.operator_extra_links"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocInstantiateWorkflowTemplateOperator.operator_extra_links">[docs]</a> <span class="n">operator_extra_links</span> <span class="o">=</span> <span class="p">(</span><span class="n">DataprocLink</span><span class="p">(),)</span></div>
<span class="k">def</span> <span class="fm">__init__</span><span class="p">(</span>
<span class="bp">self</span><span class="p">,</span>
<span class="o">*</span><span class="p">,</span>
<span class="n">template_id</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span>
<span class="n">region</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span>
<span class="n">project_id</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">version</span><span class="p">:</span> <span class="nb">int</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">request_id</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">parameters</span><span class="p">:</span> <span class="nb">dict</span><span class="p">[</span><span class="nb">str</span><span class="p">,</span> <span class="nb">str</span><span class="p">]</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">retry</span><span class="p">:</span> <span class="n">Retry</span> <span class="o">|</span> <span class="n">_MethodDefault</span> <span class="o">=</span> <span class="n">DEFAULT</span><span class="p">,</span>
<span class="n">timeout</span><span class="p">:</span> <span class="nb">float</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">metadata</span><span class="p">:</span> <span class="n">Sequence</span><span class="p">[</span><span class="nb">tuple</span><span class="p">[</span><span class="nb">str</span><span class="p">,</span> <span class="nb">str</span><span class="p">]]</span> <span class="o">=</span> <span class="p">(),</span>
<span class="n">gcp_conn_id</span><span class="p">:</span> <span class="nb">str</span> <span class="o">=</span> <span class="s2">&quot;google_cloud_default&quot;</span><span class="p">,</span>
<span class="n">impersonation_chain</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="n">Sequence</span><span class="p">[</span><span class="nb">str</span><span class="p">]</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="o">**</span><span class="n">kwargs</span><span class="p">,</span>
<span class="p">)</span> <span class="o">-&gt;</span> <span class="kc">None</span><span class="p">:</span>
<span class="nb">super</span><span class="p">()</span><span class="o">.</span><span class="fm">__init__</span><span class="p">(</span><span class="o">**</span><span class="n">kwargs</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">template_id</span> <span class="o">=</span> <span class="n">template_id</span>
<span class="bp">self</span><span class="o">.</span><span class="n">parameters</span> <span class="o">=</span> <span class="n">parameters</span>
<span class="bp">self</span><span class="o">.</span><span class="n">version</span> <span class="o">=</span> <span class="n">version</span>
<span class="bp">self</span><span class="o">.</span><span class="n">project_id</span> <span class="o">=</span> <span class="n">project_id</span>
<span class="bp">self</span><span class="o">.</span><span class="n">region</span> <span class="o">=</span> <span class="n">region</span>
<span class="bp">self</span><span class="o">.</span><span class="n">retry</span> <span class="o">=</span> <span class="n">retry</span>
<span class="bp">self</span><span class="o">.</span><span class="n">timeout</span> <span class="o">=</span> <span class="n">timeout</span>
<span class="bp">self</span><span class="o">.</span><span class="n">metadata</span> <span class="o">=</span> <span class="n">metadata</span>
<span class="bp">self</span><span class="o">.</span><span class="n">request_id</span> <span class="o">=</span> <span class="n">request_id</span>
<span class="bp">self</span><span class="o">.</span><span class="n">gcp_conn_id</span> <span class="o">=</span> <span class="n">gcp_conn_id</span>
<span class="bp">self</span><span class="o">.</span><span class="n">impersonation_chain</span> <span class="o">=</span> <span class="n">impersonation_chain</span>
<div class="viewcode-block" id="DataprocInstantiateWorkflowTemplateOperator.execute"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocInstantiateWorkflowTemplateOperator.execute">[docs]</a> <span class="k">def</span> <span class="nf">execute</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">context</span><span class="p">:</span> <span class="n">Context</span><span class="p">):</span>
<span class="n">hook</span> <span class="o">=</span> <span class="n">DataprocHook</span><span class="p">(</span><span class="n">gcp_conn_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">gcp_conn_id</span><span class="p">,</span> <span class="n">impersonation_chain</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">impersonation_chain</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">log</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s1">&#39;Instantiating template </span><span class="si">%s</span><span class="s1">&#39;</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">template_id</span><span class="p">)</span>
<span class="n">operation</span> <span class="o">=</span> <span class="n">hook</span><span class="o">.</span><span class="n">instantiate_workflow_template</span><span class="p">(</span>
<span class="n">project_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">project_id</span><span class="p">,</span>
<span class="n">region</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">region</span><span class="p">,</span>
<span class="n">template_name</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">template_id</span><span class="p">,</span>
<span class="n">version</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">version</span><span class="p">,</span>
<span class="n">request_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">request_id</span><span class="p">,</span>
<span class="n">parameters</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">parameters</span><span class="p">,</span>
<span class="n">retry</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">retry</span><span class="p">,</span>
<span class="n">timeout</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">timeout</span><span class="p">,</span>
<span class="n">metadata</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">metadata</span><span class="p">,</span>
<span class="p">)</span>
<span class="n">operation</span><span class="o">.</span><span class="n">result</span><span class="p">()</span>
<span class="n">workflow_id</span> <span class="o">=</span> <span class="n">operation</span><span class="o">.</span><span class="n">operation</span><span class="o">.</span><span class="n">name</span><span class="o">.</span><span class="n">split</span><span class="p">(</span><span class="s1">&#39;/&#39;</span><span class="p">)[</span><span class="o">-</span><span class="mi">1</span><span class="p">]</span>
<span class="n">DataprocLink</span><span class="o">.</span><span class="n">persist</span><span class="p">(</span>
<span class="n">context</span><span class="o">=</span><span class="n">context</span><span class="p">,</span> <span class="n">task_instance</span><span class="o">=</span><span class="bp">self</span><span class="p">,</span> <span class="n">url</span><span class="o">=</span><span class="n">DATAPROC_WORKFLOW_LINK</span><span class="p">,</span> <span class="n">resource</span><span class="o">=</span><span class="n">workflow_id</span>
<span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">log</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s1">&#39;Template instantiated.&#39;</span><span class="p">)</span></div></div>
<div class="viewcode-block" id="DataprocInstantiateInlineWorkflowTemplateOperator"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocInstantiateInlineWorkflowTemplateOperator">[docs]</a><span class="k">class</span> <span class="nc">DataprocInstantiateInlineWorkflowTemplateOperator</span><span class="p">(</span><span class="n">BaseOperator</span><span class="p">):</span>
<span class="sd">&quot;&quot;&quot;</span>
<span class="sd"> Instantiate a WorkflowTemplate Inline on Google Cloud Dataproc. The operator will</span>
<span class="sd"> wait until the WorkflowTemplate is finished executing.</span>
<span class="sd"> .. seealso::</span>
<span class="sd"> For more information on how to use this operator, take a look at the guide:</span>
<span class="sd"> :ref:`howto/operator:DataprocInstantiateInlineWorkflowTemplateOperator`</span>
<span class="sd"> For more detail on about instantiate inline have a look at the reference:</span>
<span class="sd"> https://cloud.google.com/dataproc/docs/reference/rest/v1/projects.regions.workflowTemplates/instantiateInline</span>
<span class="sd"> :param template: The template contents. (templated)</span>
<span class="sd"> :param project_id: The ID of the google cloud project in which</span>
<span class="sd"> the template runs</span>
<span class="sd"> :param region: The specified region where the dataproc cluster is created.</span>
<span class="sd"> :param parameters: a map of parameters for Dataproc Template in key-value format:</span>
<span class="sd"> map (key: string, value: string)</span>
<span class="sd"> Example: { &quot;date_from&quot;: &quot;2019-08-01&quot;, &quot;date_to&quot;: &quot;2019-08-02&quot;}.</span>
<span class="sd"> Values may not exceed 100 characters. Please refer to:</span>
<span class="sd"> https://cloud.google.com/dataproc/docs/concepts/workflows/workflow-parameters</span>
<span class="sd"> :param request_id: Optional. A unique id used to identify the request. If the server receives two</span>
<span class="sd"> ``SubmitJobRequest`` requests with the same id, then the second request will be ignored and the first</span>
<span class="sd"> ``Job`` created and stored in the backend is returned.</span>
<span class="sd"> It is recommended to always set this value to a UUID.</span>
<span class="sd"> :param retry: A retry object used to retry requests. If ``None`` is specified, requests will not be</span>
<span class="sd"> retried.</span>
<span class="sd"> :param timeout: The amount of time, in seconds, to wait for the request to complete. Note that if</span>
<span class="sd"> ``retry`` is specified, the timeout applies to each individual attempt.</span>
<span class="sd"> :param metadata: Additional metadata that is provided to the method.</span>
<span class="sd"> :param gcp_conn_id: The connection ID to use connecting to Google Cloud.</span>
<span class="sd"> :param impersonation_chain: Optional service account to impersonate using short-term</span>
<span class="sd"> credentials, or chained list of accounts required to get the access_token</span>
<span class="sd"> of the last account in the list, which will be impersonated in the request.</span>
<span class="sd"> If set as a string, the account must grant the originating account</span>
<span class="sd"> the Service Account Token Creator IAM role.</span>
<span class="sd"> If set as a sequence, the identities from the list must grant</span>
<span class="sd"> Service Account Token Creator IAM role to the directly preceding identity, with first</span>
<span class="sd"> account from the list granting this role to the originating account (templated).</span>
<span class="sd"> &quot;&quot;&quot;</span>
<div class="viewcode-block" id="DataprocInstantiateInlineWorkflowTemplateOperator.template_fields"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocInstantiateInlineWorkflowTemplateOperator.template_fields">[docs]</a> <span class="n">template_fields</span><span class="p">:</span> <span class="n">Sequence</span><span class="p">[</span><span class="nb">str</span><span class="p">]</span> <span class="o">=</span> <span class="p">(</span><span class="s1">&#39;template&#39;</span><span class="p">,</span> <span class="s1">&#39;impersonation_chain&#39;</span><span class="p">)</span></div>
<div class="viewcode-block" id="DataprocInstantiateInlineWorkflowTemplateOperator.template_fields_renderers"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocInstantiateInlineWorkflowTemplateOperator.template_fields_renderers">[docs]</a> <span class="n">template_fields_renderers</span> <span class="o">=</span> <span class="p">{</span><span class="s2">&quot;template&quot;</span><span class="p">:</span> <span class="s2">&quot;json&quot;</span><span class="p">}</span></div>
<div class="viewcode-block" id="DataprocInstantiateInlineWorkflowTemplateOperator.operator_extra_links"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocInstantiateInlineWorkflowTemplateOperator.operator_extra_links">[docs]</a> <span class="n">operator_extra_links</span> <span class="o">=</span> <span class="p">(</span><span class="n">DataprocLink</span><span class="p">(),)</span></div>
<span class="k">def</span> <span class="fm">__init__</span><span class="p">(</span>
<span class="bp">self</span><span class="p">,</span>
<span class="o">*</span><span class="p">,</span>
<span class="n">template</span><span class="p">:</span> <span class="nb">dict</span><span class="p">,</span>
<span class="n">region</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span>
<span class="n">project_id</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">request_id</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">retry</span><span class="p">:</span> <span class="n">Retry</span> <span class="o">|</span> <span class="n">_MethodDefault</span> <span class="o">=</span> <span class="n">DEFAULT</span><span class="p">,</span>
<span class="n">timeout</span><span class="p">:</span> <span class="nb">float</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">metadata</span><span class="p">:</span> <span class="n">Sequence</span><span class="p">[</span><span class="nb">tuple</span><span class="p">[</span><span class="nb">str</span><span class="p">,</span> <span class="nb">str</span><span class="p">]]</span> <span class="o">=</span> <span class="p">(),</span>
<span class="n">gcp_conn_id</span><span class="p">:</span> <span class="nb">str</span> <span class="o">=</span> <span class="s2">&quot;google_cloud_default&quot;</span><span class="p">,</span>
<span class="n">impersonation_chain</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="n">Sequence</span><span class="p">[</span><span class="nb">str</span><span class="p">]</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="o">**</span><span class="n">kwargs</span><span class="p">,</span>
<span class="p">)</span> <span class="o">-&gt;</span> <span class="kc">None</span><span class="p">:</span>
<span class="nb">super</span><span class="p">()</span><span class="o">.</span><span class="fm">__init__</span><span class="p">(</span><span class="o">**</span><span class="n">kwargs</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">template</span> <span class="o">=</span> <span class="n">template</span>
<span class="bp">self</span><span class="o">.</span><span class="n">project_id</span> <span class="o">=</span> <span class="n">project_id</span>
<span class="bp">self</span><span class="o">.</span><span class="n">region</span> <span class="o">=</span> <span class="n">region</span>
<span class="bp">self</span><span class="o">.</span><span class="n">template</span> <span class="o">=</span> <span class="n">template</span>
<span class="bp">self</span><span class="o">.</span><span class="n">request_id</span> <span class="o">=</span> <span class="n">request_id</span>
<span class="bp">self</span><span class="o">.</span><span class="n">retry</span> <span class="o">=</span> <span class="n">retry</span>
<span class="bp">self</span><span class="o">.</span><span class="n">timeout</span> <span class="o">=</span> <span class="n">timeout</span>
<span class="bp">self</span><span class="o">.</span><span class="n">metadata</span> <span class="o">=</span> <span class="n">metadata</span>
<span class="bp">self</span><span class="o">.</span><span class="n">gcp_conn_id</span> <span class="o">=</span> <span class="n">gcp_conn_id</span>
<span class="bp">self</span><span class="o">.</span><span class="n">impersonation_chain</span> <span class="o">=</span> <span class="n">impersonation_chain</span>
<div class="viewcode-block" id="DataprocInstantiateInlineWorkflowTemplateOperator.execute"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocInstantiateInlineWorkflowTemplateOperator.execute">[docs]</a> <span class="k">def</span> <span class="nf">execute</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">context</span><span class="p">:</span> <span class="n">Context</span><span class="p">):</span>
<span class="bp">self</span><span class="o">.</span><span class="n">log</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s1">&#39;Instantiating Inline Template&#39;</span><span class="p">)</span>
<span class="n">hook</span> <span class="o">=</span> <span class="n">DataprocHook</span><span class="p">(</span><span class="n">gcp_conn_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">gcp_conn_id</span><span class="p">,</span> <span class="n">impersonation_chain</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">impersonation_chain</span><span class="p">)</span>
<span class="n">operation</span> <span class="o">=</span> <span class="n">hook</span><span class="o">.</span><span class="n">instantiate_inline_workflow_template</span><span class="p">(</span>
<span class="n">template</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">template</span><span class="p">,</span>
<span class="n">project_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">project_id</span><span class="p">,</span>
<span class="n">region</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">region</span><span class="p">,</span>
<span class="n">request_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">request_id</span><span class="p">,</span>
<span class="n">retry</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">retry</span><span class="p">,</span>
<span class="n">timeout</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">timeout</span><span class="p">,</span>
<span class="n">metadata</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">metadata</span><span class="p">,</span>
<span class="p">)</span>
<span class="n">operation</span><span class="o">.</span><span class="n">result</span><span class="p">()</span>
<span class="n">workflow_id</span> <span class="o">=</span> <span class="n">operation</span><span class="o">.</span><span class="n">operation</span><span class="o">.</span><span class="n">name</span><span class="o">.</span><span class="n">split</span><span class="p">(</span><span class="s1">&#39;/&#39;</span><span class="p">)[</span><span class="o">-</span><span class="mi">1</span><span class="p">]</span>
<span class="n">DataprocLink</span><span class="o">.</span><span class="n">persist</span><span class="p">(</span>
<span class="n">context</span><span class="o">=</span><span class="n">context</span><span class="p">,</span> <span class="n">task_instance</span><span class="o">=</span><span class="bp">self</span><span class="p">,</span> <span class="n">url</span><span class="o">=</span><span class="n">DATAPROC_WORKFLOW_LINK</span><span class="p">,</span> <span class="n">resource</span><span class="o">=</span><span class="n">workflow_id</span>
<span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">log</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s1">&#39;Template instantiated.&#39;</span><span class="p">)</span></div></div>
<div class="viewcode-block" id="DataprocSubmitJobOperator"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocSubmitJobOperator">[docs]</a><span class="k">class</span> <span class="nc">DataprocSubmitJobOperator</span><span class="p">(</span><span class="n">BaseOperator</span><span class="p">):</span>
<span class="sd">&quot;&quot;&quot;</span>
<span class="sd"> Submits a job to a cluster.</span>
<span class="sd"> :param project_id: Optional. The ID of the Google Cloud project that the job belongs to.</span>
<span class="sd"> :param region: Required. The Cloud Dataproc region in which to handle the request.</span>
<span class="sd"> :param job: Required. The job resource.</span>
<span class="sd"> If a dict is provided, it must be of the same form as the protobuf message</span>
<span class="sd"> :class:`~google.cloud.dataproc_v1.types.Job`</span>
<span class="sd"> :param request_id: Optional. A unique id used to identify the request. If the server receives two</span>
<span class="sd"> ``SubmitJobRequest`` requests with the same id, then the second request will be ignored and the first</span>
<span class="sd"> ``Job`` created and stored in the backend is returned.</span>
<span class="sd"> It is recommended to always set this value to a UUID.</span>
<span class="sd"> :param retry: A retry object used to retry requests. If ``None`` is specified, requests will not be</span>
<span class="sd"> retried.</span>
<span class="sd"> :param timeout: The amount of time, in seconds, to wait for the request to complete. Note that if</span>
<span class="sd"> ``retry`` is specified, the timeout applies to each individual attempt.</span>
<span class="sd"> :param metadata: Additional metadata that is provided to the method.</span>
<span class="sd"> :param gcp_conn_id:</span>
<span class="sd"> :param impersonation_chain: Optional service account to impersonate using short-term</span>
<span class="sd"> credentials, or chained list of accounts required to get the access_token</span>
<span class="sd"> of the last account in the list, which will be impersonated in the request.</span>
<span class="sd"> If set as a string, the account must grant the originating account</span>
<span class="sd"> the Service Account Token Creator IAM role.</span>
<span class="sd"> If set as a sequence, the identities from the list must grant</span>
<span class="sd"> Service Account Token Creator IAM role to the directly preceding identity, with first</span>
<span class="sd"> account from the list granting this role to the originating account (templated).</span>
<span class="sd"> :param asynchronous: Flag to return after submitting the job to the Dataproc API.</span>
<span class="sd"> This is useful for submitting long running jobs and</span>
<span class="sd"> waiting on them asynchronously using the DataprocJobSensor</span>
<span class="sd"> :param deferrable: Run operator in the deferrable mode</span>
<span class="sd"> :param polling_interval_seconds: time in seconds between polling for job completion.</span>
<span class="sd"> The value is considered only when running in deferrable mode. Must be greater than 0.</span>
<span class="sd"> :param cancel_on_kill: Flag which indicates whether cancel the hook&#39;s job or not, when on_kill is called</span>
<span class="sd"> :param wait_timeout: How many seconds wait for job to be ready. Used only if ``asynchronous`` is False</span>
<span class="sd"> &quot;&quot;&quot;</span>
<div class="viewcode-block" id="DataprocSubmitJobOperator.template_fields"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocSubmitJobOperator.template_fields">[docs]</a> <span class="n">template_fields</span><span class="p">:</span> <span class="n">Sequence</span><span class="p">[</span><span class="nb">str</span><span class="p">]</span> <span class="o">=</span> <span class="p">(</span><span class="s1">&#39;project_id&#39;</span><span class="p">,</span> <span class="s1">&#39;region&#39;</span><span class="p">,</span> <span class="s1">&#39;job&#39;</span><span class="p">,</span> <span class="s1">&#39;impersonation_chain&#39;</span><span class="p">,</span> <span class="s1">&#39;request_id&#39;</span><span class="p">)</span></div>
<div class="viewcode-block" id="DataprocSubmitJobOperator.template_fields_renderers"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocSubmitJobOperator.template_fields_renderers">[docs]</a> <span class="n">template_fields_renderers</span> <span class="o">=</span> <span class="p">{</span><span class="s2">&quot;job&quot;</span><span class="p">:</span> <span class="s2">&quot;json&quot;</span><span class="p">}</span></div>
<div class="viewcode-block" id="DataprocSubmitJobOperator.operator_extra_links"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocSubmitJobOperator.operator_extra_links">[docs]</a> <span class="n">operator_extra_links</span> <span class="o">=</span> <span class="p">(</span><span class="n">DataprocLink</span><span class="p">(),)</span></div>
<span class="k">def</span> <span class="fm">__init__</span><span class="p">(</span>
<span class="bp">self</span><span class="p">,</span>
<span class="o">*</span><span class="p">,</span>
<span class="n">job</span><span class="p">:</span> <span class="nb">dict</span><span class="p">,</span>
<span class="n">region</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span>
<span class="n">project_id</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">request_id</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">retry</span><span class="p">:</span> <span class="n">Retry</span> <span class="o">|</span> <span class="n">_MethodDefault</span> <span class="o">=</span> <span class="n">DEFAULT</span><span class="p">,</span>
<span class="n">timeout</span><span class="p">:</span> <span class="nb">float</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">metadata</span><span class="p">:</span> <span class="n">Sequence</span><span class="p">[</span><span class="nb">tuple</span><span class="p">[</span><span class="nb">str</span><span class="p">,</span> <span class="nb">str</span><span class="p">]]</span> <span class="o">=</span> <span class="p">(),</span>
<span class="n">gcp_conn_id</span><span class="p">:</span> <span class="nb">str</span> <span class="o">=</span> <span class="s2">&quot;google_cloud_default&quot;</span><span class="p">,</span>
<span class="n">impersonation_chain</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="n">Sequence</span><span class="p">[</span><span class="nb">str</span><span class="p">]</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">asynchronous</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">False</span><span class="p">,</span>
<span class="n">deferrable</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">False</span><span class="p">,</span>
<span class="n">polling_interval_seconds</span><span class="p">:</span> <span class="nb">int</span> <span class="o">=</span> <span class="mi">10</span><span class="p">,</span>
<span class="n">cancel_on_kill</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">True</span><span class="p">,</span>
<span class="n">wait_timeout</span><span class="p">:</span> <span class="nb">int</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="o">**</span><span class="n">kwargs</span><span class="p">,</span>
<span class="p">)</span> <span class="o">-&gt;</span> <span class="kc">None</span><span class="p">:</span>
<span class="nb">super</span><span class="p">()</span><span class="o">.</span><span class="fm">__init__</span><span class="p">(</span><span class="o">**</span><span class="n">kwargs</span><span class="p">)</span>
<span class="k">if</span> <span class="n">deferrable</span> <span class="ow">and</span> <span class="n">polling_interval_seconds</span> <span class="o">&lt;=</span> <span class="mi">0</span><span class="p">:</span>
<span class="k">raise</span> <span class="ne">ValueError</span><span class="p">(</span><span class="s2">&quot;Invalid value for polling_interval_seconds. Expected value greater than 0&quot;</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">project_id</span> <span class="o">=</span> <span class="n">project_id</span>
<span class="bp">self</span><span class="o">.</span><span class="n">region</span> <span class="o">=</span> <span class="n">region</span>
<span class="bp">self</span><span class="o">.</span><span class="n">job</span> <span class="o">=</span> <span class="n">job</span>
<span class="bp">self</span><span class="o">.</span><span class="n">request_id</span> <span class="o">=</span> <span class="n">request_id</span>
<span class="bp">self</span><span class="o">.</span><span class="n">retry</span> <span class="o">=</span> <span class="n">retry</span>
<span class="bp">self</span><span class="o">.</span><span class="n">timeout</span> <span class="o">=</span> <span class="n">timeout</span>
<span class="bp">self</span><span class="o">.</span><span class="n">metadata</span> <span class="o">=</span> <span class="n">metadata</span>
<span class="bp">self</span><span class="o">.</span><span class="n">gcp_conn_id</span> <span class="o">=</span> <span class="n">gcp_conn_id</span>
<span class="bp">self</span><span class="o">.</span><span class="n">impersonation_chain</span> <span class="o">=</span> <span class="n">impersonation_chain</span>
<span class="bp">self</span><span class="o">.</span><span class="n">asynchronous</span> <span class="o">=</span> <span class="n">asynchronous</span>
<span class="bp">self</span><span class="o">.</span><span class="n">deferrable</span> <span class="o">=</span> <span class="n">deferrable</span>
<span class="bp">self</span><span class="o">.</span><span class="n">polling_interval_seconds</span> <span class="o">=</span> <span class="n">polling_interval_seconds</span>
<span class="bp">self</span><span class="o">.</span><span class="n">cancel_on_kill</span> <span class="o">=</span> <span class="n">cancel_on_kill</span>
<span class="bp">self</span><span class="o">.</span><span class="n">hook</span><span class="p">:</span> <span class="n">DataprocHook</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span>
<span class="bp">self</span><span class="o">.</span><span class="n">job_id</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span>
<span class="bp">self</span><span class="o">.</span><span class="n">wait_timeout</span> <span class="o">=</span> <span class="n">wait_timeout</span>
<div class="viewcode-block" id="DataprocSubmitJobOperator.execute"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocSubmitJobOperator.execute">[docs]</a> <span class="k">def</span> <span class="nf">execute</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">context</span><span class="p">:</span> <span class="n">Context</span><span class="p">):</span>
<span class="bp">self</span><span class="o">.</span><span class="n">log</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s2">&quot;Submitting job&quot;</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">hook</span> <span class="o">=</span> <span class="n">DataprocHook</span><span class="p">(</span><span class="n">gcp_conn_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">gcp_conn_id</span><span class="p">,</span> <span class="n">impersonation_chain</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">impersonation_chain</span><span class="p">)</span>
<span class="n">job_object</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">hook</span><span class="o">.</span><span class="n">submit_job</span><span class="p">(</span>
<span class="n">project_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">project_id</span><span class="p">,</span>
<span class="n">region</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">region</span><span class="p">,</span>
<span class="n">job</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">job</span><span class="p">,</span>
<span class="n">request_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">request_id</span><span class="p">,</span>
<span class="n">retry</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">retry</span><span class="p">,</span>
<span class="n">timeout</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">timeout</span><span class="p">,</span>
<span class="n">metadata</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">metadata</span><span class="p">,</span>
<span class="p">)</span>
<span class="n">new_job_id</span><span class="p">:</span> <span class="nb">str</span> <span class="o">=</span> <span class="n">job_object</span><span class="o">.</span><span class="n">reference</span><span class="o">.</span><span class="n">job_id</span>
<span class="bp">self</span><span class="o">.</span><span class="n">log</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s1">&#39;Job </span><span class="si">%s</span><span class="s1"> submitted successfully.&#39;</span><span class="p">,</span> <span class="n">new_job_id</span><span class="p">)</span>
<span class="c1"># Save data required by extra links no matter what the job status will be</span>
<span class="n">DataprocLink</span><span class="o">.</span><span class="n">persist</span><span class="p">(</span>
<span class="n">context</span><span class="o">=</span><span class="n">context</span><span class="p">,</span> <span class="n">task_instance</span><span class="o">=</span><span class="bp">self</span><span class="p">,</span> <span class="n">url</span><span class="o">=</span><span class="n">DATAPROC_JOB_LOG_LINK</span><span class="p">,</span> <span class="n">resource</span><span class="o">=</span><span class="n">new_job_id</span>
<span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">job_id</span> <span class="o">=</span> <span class="n">new_job_id</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">deferrable</span><span class="p">:</span>
<span class="bp">self</span><span class="o">.</span><span class="n">defer</span><span class="p">(</span>
<span class="n">trigger</span><span class="o">=</span><span class="n">DataprocBaseTrigger</span><span class="p">(</span>
<span class="n">job_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">job_id</span><span class="p">,</span>
<span class="n">project_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">project_id</span><span class="p">,</span>
<span class="n">region</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">region</span><span class="p">,</span>
<span class="n">gcp_conn_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">gcp_conn_id</span><span class="p">,</span>
<span class="n">impersonation_chain</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">impersonation_chain</span><span class="p">,</span>
<span class="n">polling_interval_seconds</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">polling_interval_seconds</span><span class="p">,</span>
<span class="p">),</span>
<span class="n">method_name</span><span class="o">=</span><span class="s2">&quot;execute_complete&quot;</span><span class="p">,</span>
<span class="p">)</span>
<span class="k">elif</span> <span class="ow">not</span> <span class="bp">self</span><span class="o">.</span><span class="n">asynchronous</span><span class="p">:</span>
<span class="bp">self</span><span class="o">.</span><span class="n">log</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s1">&#39;Waiting for job </span><span class="si">%s</span><span class="s1"> to complete&#39;</span><span class="p">,</span> <span class="n">new_job_id</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">hook</span><span class="o">.</span><span class="n">wait_for_job</span><span class="p">(</span>
<span class="n">job_id</span><span class="o">=</span><span class="n">new_job_id</span><span class="p">,</span> <span class="n">region</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">region</span><span class="p">,</span> <span class="n">project_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">project_id</span><span class="p">,</span> <span class="n">timeout</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">wait_timeout</span>
<span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">log</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s1">&#39;Job </span><span class="si">%s</span><span class="s1"> completed successfully.&#39;</span><span class="p">,</span> <span class="n">new_job_id</span><span class="p">)</span>
<span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">job_id</span></div>
<div class="viewcode-block" id="DataprocSubmitJobOperator.execute_complete"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocSubmitJobOperator.execute_complete">[docs]</a> <span class="k">def</span> <span class="nf">execute_complete</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">context</span><span class="p">,</span> <span class="n">event</span><span class="o">=</span><span class="kc">None</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="kc">None</span><span class="p">:</span>
<span class="sd">&quot;&quot;&quot;</span>
<span class="sd"> Callback for when the trigger fires - returns immediately.</span>
<span class="sd"> Relies on trigger to throw an exception, otherwise it assumes execution was</span>
<span class="sd"> successful.</span>
<span class="sd"> &quot;&quot;&quot;</span>
<span class="n">job_state</span> <span class="o">=</span> <span class="n">event</span><span class="p">[</span><span class="s2">&quot;job_state&quot;</span><span class="p">]</span>
<span class="n">job_id</span> <span class="o">=</span> <span class="n">event</span><span class="p">[</span><span class="s2">&quot;job_id&quot;</span><span class="p">]</span>
<span class="k">if</span> <span class="n">job_state</span> <span class="o">==</span> <span class="n">JobStatus</span><span class="o">.</span><span class="n">State</span><span class="o">.</span><span class="n">ERROR</span><span class="p">:</span>
<span class="k">raise</span> <span class="n">AirflowException</span><span class="p">(</span><span class="sa">f</span><span class="s1">&#39;Job failed:</span><span class="se">\n</span><span class="si">{</span><span class="n">job_id</span><span class="si">}</span><span class="s1">&#39;</span><span class="p">)</span>
<span class="k">if</span> <span class="n">job_state</span> <span class="o">==</span> <span class="n">JobStatus</span><span class="o">.</span><span class="n">State</span><span class="o">.</span><span class="n">CANCELLED</span><span class="p">:</span>
<span class="k">raise</span> <span class="n">AirflowException</span><span class="p">(</span><span class="sa">f</span><span class="s1">&#39;Job was cancelled:</span><span class="se">\n</span><span class="si">{</span><span class="n">job_id</span><span class="si">}</span><span class="s1">&#39;</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">log</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s2">&quot;</span><span class="si">%s</span><span class="s2"> completed successfully.&quot;</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">task_id</span><span class="p">)</span></div>
<div class="viewcode-block" id="DataprocSubmitJobOperator.on_kill"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocSubmitJobOperator.on_kill">[docs]</a> <span class="k">def</span> <span class="nf">on_kill</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">job_id</span> <span class="ow">and</span> <span class="bp">self</span><span class="o">.</span><span class="n">cancel_on_kill</span><span class="p">:</span>
<span class="bp">self</span><span class="o">.</span><span class="n">hook</span><span class="o">.</span><span class="n">cancel_job</span><span class="p">(</span><span class="n">job_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">job_id</span><span class="p">,</span> <span class="n">project_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">project_id</span><span class="p">,</span> <span class="n">region</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">region</span><span class="p">)</span></div></div>
<div class="viewcode-block" id="DataprocUpdateClusterOperator"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocUpdateClusterOperator">[docs]</a><span class="k">class</span> <span class="nc">DataprocUpdateClusterOperator</span><span class="p">(</span><span class="n">BaseOperator</span><span class="p">):</span>
<span class="sd">&quot;&quot;&quot;</span>
<span class="sd"> Updates a cluster in a project.</span>
<span class="sd"> :param region: Required. The Cloud Dataproc region in which to handle the request.</span>
<span class="sd"> :param project_id: Optional. The ID of the Google Cloud project the cluster belongs to.</span>
<span class="sd"> :param cluster_name: Required. The cluster name.</span>
<span class="sd"> :param cluster: Required. The changes to the cluster.</span>
<span class="sd"> If a dict is provided, it must be of the same form as the protobuf message</span>
<span class="sd"> :class:`~google.cloud.dataproc_v1.types.Cluster`</span>
<span class="sd"> :param update_mask: Required. Specifies the path, relative to ``Cluster``, of the field to update. For</span>
<span class="sd"> example, to change the number of workers in a cluster to 5, the ``update_mask`` parameter would be</span>
<span class="sd"> specified as ``config.worker_config.num_instances``, and the ``PATCH`` request body would specify the</span>
<span class="sd"> new value. If a dict is provided, it must be of the same form as the protobuf message</span>
<span class="sd"> :class:`~google.protobuf.field_mask_pb2.FieldMask`</span>
<span class="sd"> :param graceful_decommission_timeout: Optional. Timeout for graceful YARN decommissioning. Graceful</span>
<span class="sd"> decommissioning allows removing nodes from the cluster without interrupting jobs in progress. Timeout</span>
<span class="sd"> specifies how long to wait for jobs in progress to finish before forcefully removing nodes (and</span>
<span class="sd"> potentially interrupting jobs). Default timeout is 0 (for forceful decommission), and the maximum</span>
<span class="sd"> allowed timeout is 1 day.</span>
<span class="sd"> :param request_id: Optional. A unique id used to identify the request. If the server receives two</span>
<span class="sd"> ``UpdateClusterRequest`` requests with the same id, then the second request will be ignored and the</span>
<span class="sd"> first ``google.longrunning.Operation`` created and stored in the backend is returned.</span>
<span class="sd"> :param retry: A retry object used to retry requests. If ``None`` is specified, requests will not be</span>
<span class="sd"> retried.</span>
<span class="sd"> :param timeout: The amount of time, in seconds, to wait for the request to complete. Note that if</span>
<span class="sd"> ``retry`` is specified, the timeout applies to each individual attempt.</span>
<span class="sd"> :param metadata: Additional metadata that is provided to the method.</span>
<span class="sd"> :param gcp_conn_id: The connection ID to use connecting to Google Cloud.</span>
<span class="sd"> :param impersonation_chain: Optional service account to impersonate using short-term</span>
<span class="sd"> credentials, or chained list of accounts required to get the access_token</span>
<span class="sd"> of the last account in the list, which will be impersonated in the request.</span>
<span class="sd"> If set as a string, the account must grant the originating account</span>
<span class="sd"> the Service Account Token Creator IAM role.</span>
<span class="sd"> If set as a sequence, the identities from the list must grant</span>
<span class="sd"> Service Account Token Creator IAM role to the directly preceding identity, with first</span>
<span class="sd"> account from the list granting this role to the originating account (templated).</span>
<span class="sd"> &quot;&quot;&quot;</span>
<div class="viewcode-block" id="DataprocUpdateClusterOperator.template_fields"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocUpdateClusterOperator.template_fields">[docs]</a> <span class="n">template_fields</span><span class="p">:</span> <span class="n">Sequence</span><span class="p">[</span><span class="nb">str</span><span class="p">]</span> <span class="o">=</span> <span class="p">(</span>
<span class="s1">&#39;cluster_name&#39;</span><span class="p">,</span>
<span class="s1">&#39;cluster&#39;</span><span class="p">,</span>
<span class="s1">&#39;region&#39;</span><span class="p">,</span>
<span class="s1">&#39;request_id&#39;</span><span class="p">,</span>
<span class="s1">&#39;project_id&#39;</span><span class="p">,</span>
<span class="s1">&#39;impersonation_chain&#39;</span><span class="p">,</span></div>
<span class="p">)</span>
<div class="viewcode-block" id="DataprocUpdateClusterOperator.operator_extra_links"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocUpdateClusterOperator.operator_extra_links">[docs]</a> <span class="n">operator_extra_links</span> <span class="o">=</span> <span class="p">(</span><span class="n">DataprocLink</span><span class="p">(),)</span></div>
<span class="k">def</span> <span class="fm">__init__</span><span class="p">(</span>
<span class="bp">self</span><span class="p">,</span>
<span class="o">*</span><span class="p">,</span>
<span class="n">cluster_name</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span>
<span class="n">cluster</span><span class="p">:</span> <span class="nb">dict</span> <span class="o">|</span> <span class="n">Cluster</span><span class="p">,</span>
<span class="n">update_mask</span><span class="p">:</span> <span class="nb">dict</span> <span class="o">|</span> <span class="n">FieldMask</span><span class="p">,</span>
<span class="n">graceful_decommission_timeout</span><span class="p">:</span> <span class="nb">dict</span> <span class="o">|</span> <span class="n">Duration</span><span class="p">,</span>
<span class="n">region</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span>
<span class="n">request_id</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">project_id</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">retry</span><span class="p">:</span> <span class="n">Retry</span> <span class="o">|</span> <span class="n">_MethodDefault</span> <span class="o">=</span> <span class="n">DEFAULT</span><span class="p">,</span>
<span class="n">timeout</span><span class="p">:</span> <span class="nb">float</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">metadata</span><span class="p">:</span> <span class="n">Sequence</span><span class="p">[</span><span class="nb">tuple</span><span class="p">[</span><span class="nb">str</span><span class="p">,</span> <span class="nb">str</span><span class="p">]]</span> <span class="o">=</span> <span class="p">(),</span>
<span class="n">gcp_conn_id</span><span class="p">:</span> <span class="nb">str</span> <span class="o">=</span> <span class="s2">&quot;google_cloud_default&quot;</span><span class="p">,</span>
<span class="n">impersonation_chain</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="n">Sequence</span><span class="p">[</span><span class="nb">str</span><span class="p">]</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="o">**</span><span class="n">kwargs</span><span class="p">,</span>
<span class="p">):</span>
<span class="nb">super</span><span class="p">()</span><span class="o">.</span><span class="fm">__init__</span><span class="p">(</span><span class="o">**</span><span class="n">kwargs</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">project_id</span> <span class="o">=</span> <span class="n">project_id</span>
<span class="bp">self</span><span class="o">.</span><span class="n">region</span> <span class="o">=</span> <span class="n">region</span>
<span class="bp">self</span><span class="o">.</span><span class="n">cluster_name</span> <span class="o">=</span> <span class="n">cluster_name</span>
<span class="bp">self</span><span class="o">.</span><span class="n">cluster</span> <span class="o">=</span> <span class="n">cluster</span>
<span class="bp">self</span><span class="o">.</span><span class="n">update_mask</span> <span class="o">=</span> <span class="n">update_mask</span>
<span class="bp">self</span><span class="o">.</span><span class="n">graceful_decommission_timeout</span> <span class="o">=</span> <span class="n">graceful_decommission_timeout</span>
<span class="bp">self</span><span class="o">.</span><span class="n">request_id</span> <span class="o">=</span> <span class="n">request_id</span>
<span class="bp">self</span><span class="o">.</span><span class="n">retry</span> <span class="o">=</span> <span class="n">retry</span>
<span class="bp">self</span><span class="o">.</span><span class="n">timeout</span> <span class="o">=</span> <span class="n">timeout</span>
<span class="bp">self</span><span class="o">.</span><span class="n">metadata</span> <span class="o">=</span> <span class="n">metadata</span>
<span class="bp">self</span><span class="o">.</span><span class="n">gcp_conn_id</span> <span class="o">=</span> <span class="n">gcp_conn_id</span>
<span class="bp">self</span><span class="o">.</span><span class="n">impersonation_chain</span> <span class="o">=</span> <span class="n">impersonation_chain</span>
<div class="viewcode-block" id="DataprocUpdateClusterOperator.execute"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocUpdateClusterOperator.execute">[docs]</a> <span class="k">def</span> <span class="nf">execute</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">context</span><span class="p">:</span> <span class="n">Context</span><span class="p">):</span>
<span class="n">hook</span> <span class="o">=</span> <span class="n">DataprocHook</span><span class="p">(</span><span class="n">gcp_conn_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">gcp_conn_id</span><span class="p">,</span> <span class="n">impersonation_chain</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">impersonation_chain</span><span class="p">)</span>
<span class="c1"># Save data required by extra links no matter what the cluster status will be</span>
<span class="n">DataprocLink</span><span class="o">.</span><span class="n">persist</span><span class="p">(</span>
<span class="n">context</span><span class="o">=</span><span class="n">context</span><span class="p">,</span> <span class="n">task_instance</span><span class="o">=</span><span class="bp">self</span><span class="p">,</span> <span class="n">url</span><span class="o">=</span><span class="n">DATAPROC_CLUSTER_LINK</span><span class="p">,</span> <span class="n">resource</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">cluster_name</span>
<span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">log</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s2">&quot;Updating </span><span class="si">%s</span><span class="s2"> cluster.&quot;</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">cluster_name</span><span class="p">)</span>
<span class="n">operation</span> <span class="o">=</span> <span class="n">hook</span><span class="o">.</span><span class="n">update_cluster</span><span class="p">(</span>
<span class="n">project_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">project_id</span><span class="p">,</span>
<span class="n">region</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">region</span><span class="p">,</span>
<span class="n">cluster_name</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">cluster_name</span><span class="p">,</span>
<span class="n">cluster</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">cluster</span><span class="p">,</span>
<span class="n">update_mask</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">update_mask</span><span class="p">,</span>
<span class="n">graceful_decommission_timeout</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">graceful_decommission_timeout</span><span class="p">,</span>
<span class="n">request_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">request_id</span><span class="p">,</span>
<span class="n">retry</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">retry</span><span class="p">,</span>
<span class="n">timeout</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">timeout</span><span class="p">,</span>
<span class="n">metadata</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">metadata</span><span class="p">,</span>
<span class="p">)</span>
<span class="n">operation</span><span class="o">.</span><span class="n">result</span><span class="p">()</span>
<span class="bp">self</span><span class="o">.</span><span class="n">log</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s2">&quot;Updated </span><span class="si">%s</span><span class="s2"> cluster.&quot;</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">cluster_name</span><span class="p">)</span></div></div>
<div class="viewcode-block" id="DataprocCreateBatchOperator"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocCreateBatchOperator">[docs]</a><span class="k">class</span> <span class="nc">DataprocCreateBatchOperator</span><span class="p">(</span><span class="n">BaseOperator</span><span class="p">):</span>
<span class="sd">&quot;&quot;&quot;</span>
<span class="sd"> Creates a batch workload.</span>
<span class="sd"> :param project_id: Optional. The ID of the Google Cloud project that the cluster belongs to. (templated)</span>
<span class="sd"> :param region: Required. The Cloud Dataproc region in which to handle the request. (templated)</span>
<span class="sd"> :param batch: Required. The batch to create. (templated)</span>
<span class="sd"> :param batch_id: Optional. The ID to use for the batch, which will become the final component</span>
<span class="sd"> of the batch&#39;s resource name.</span>
<span class="sd"> This value must be 4-63 characters. Valid characters are /[a-z][0-9]-/. (templated)</span>
<span class="sd"> :param request_id: Optional. A unique id used to identify the request. If the server receives two</span>
<span class="sd"> ``CreateBatchRequest`` requests with the same id, then the second request will be ignored and</span>
<span class="sd"> the first ``google.longrunning.Operation`` created and stored in the backend is returned.</span>
<span class="sd"> :param retry: A retry object used to retry requests. If ``None`` is specified, requests will not be</span>
<span class="sd"> retried.</span>
<span class="sd"> :param timeout: The amount of time, in seconds, to wait for the request to complete. Note that if</span>
<span class="sd"> ``retry`` is specified, the timeout applies to each individual attempt.</span>
<span class="sd"> :param metadata: Additional metadata that is provided to the method.</span>
<span class="sd"> :param gcp_conn_id: The connection ID to use connecting to Google Cloud.</span>
<span class="sd"> :param impersonation_chain: Optional service account to impersonate using short-term</span>
<span class="sd"> credentials, or chained list of accounts required to get the access_token</span>
<span class="sd"> of the last account in the list, which will be impersonated in the request.</span>
<span class="sd"> If set as a string, the account must grant the originating account</span>
<span class="sd"> the Service Account Token Creator IAM role.</span>
<span class="sd"> If set as a sequence, the identities from the list must grant</span>
<span class="sd"> Service Account Token Creator IAM role to the directly preceding identity, with first</span>
<span class="sd"> account from the list granting this role to the originating account (templated).</span>
<span class="sd"> &quot;&quot;&quot;</span>
<div class="viewcode-block" id="DataprocCreateBatchOperator.template_fields"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocCreateBatchOperator.template_fields">[docs]</a> <span class="n">template_fields</span><span class="p">:</span> <span class="n">Sequence</span><span class="p">[</span><span class="nb">str</span><span class="p">]</span> <span class="o">=</span> <span class="p">(</span>
<span class="s1">&#39;project_id&#39;</span><span class="p">,</span>
<span class="s1">&#39;batch&#39;</span><span class="p">,</span>
<span class="s1">&#39;batch_id&#39;</span><span class="p">,</span>
<span class="s1">&#39;region&#39;</span><span class="p">,</span>
<span class="s1">&#39;impersonation_chain&#39;</span><span class="p">,</span></div>
<span class="p">)</span>
<div class="viewcode-block" id="DataprocCreateBatchOperator.operator_extra_links"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocCreateBatchOperator.operator_extra_links">[docs]</a> <span class="n">operator_extra_links</span> <span class="o">=</span> <span class="p">(</span><span class="n">DataprocLink</span><span class="p">(),)</span></div>
<span class="k">def</span> <span class="fm">__init__</span><span class="p">(</span>
<span class="bp">self</span><span class="p">,</span>
<span class="o">*</span><span class="p">,</span>
<span class="n">region</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">project_id</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">batch</span><span class="p">:</span> <span class="nb">dict</span> <span class="o">|</span> <span class="n">Batch</span><span class="p">,</span>
<span class="n">batch_id</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">request_id</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">retry</span><span class="p">:</span> <span class="n">Retry</span> <span class="o">|</span> <span class="n">_MethodDefault</span> <span class="o">=</span> <span class="n">DEFAULT</span><span class="p">,</span>
<span class="n">timeout</span><span class="p">:</span> <span class="nb">float</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">metadata</span><span class="p">:</span> <span class="n">Sequence</span><span class="p">[</span><span class="nb">tuple</span><span class="p">[</span><span class="nb">str</span><span class="p">,</span> <span class="nb">str</span><span class="p">]]</span> <span class="o">=</span> <span class="p">(),</span>
<span class="n">gcp_conn_id</span><span class="p">:</span> <span class="nb">str</span> <span class="o">=</span> <span class="s2">&quot;google_cloud_default&quot;</span><span class="p">,</span>
<span class="n">impersonation_chain</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="n">Sequence</span><span class="p">[</span><span class="nb">str</span><span class="p">]</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="o">**</span><span class="n">kwargs</span><span class="p">,</span>
<span class="p">):</span>
<span class="nb">super</span><span class="p">()</span><span class="o">.</span><span class="fm">__init__</span><span class="p">(</span><span class="o">**</span><span class="n">kwargs</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">region</span> <span class="o">=</span> <span class="n">region</span>
<span class="bp">self</span><span class="o">.</span><span class="n">project_id</span> <span class="o">=</span> <span class="n">project_id</span>
<span class="bp">self</span><span class="o">.</span><span class="n">batch</span> <span class="o">=</span> <span class="n">batch</span>
<span class="bp">self</span><span class="o">.</span><span class="n">batch_id</span> <span class="o">=</span> <span class="n">batch_id</span>
<span class="bp">self</span><span class="o">.</span><span class="n">request_id</span> <span class="o">=</span> <span class="n">request_id</span>
<span class="bp">self</span><span class="o">.</span><span class="n">retry</span> <span class="o">=</span> <span class="n">retry</span>
<span class="bp">self</span><span class="o">.</span><span class="n">timeout</span> <span class="o">=</span> <span class="n">timeout</span>
<span class="bp">self</span><span class="o">.</span><span class="n">metadata</span> <span class="o">=</span> <span class="n">metadata</span>
<span class="bp">self</span><span class="o">.</span><span class="n">gcp_conn_id</span> <span class="o">=</span> <span class="n">gcp_conn_id</span>
<span class="bp">self</span><span class="o">.</span><span class="n">impersonation_chain</span> <span class="o">=</span> <span class="n">impersonation_chain</span>
<span class="bp">self</span><span class="o">.</span><span class="n">operation</span><span class="p">:</span> <span class="n">operation</span><span class="o">.</span><span class="n">Operation</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span>
<div class="viewcode-block" id="DataprocCreateBatchOperator.execute"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocCreateBatchOperator.execute">[docs]</a> <span class="k">def</span> <span class="nf">execute</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">context</span><span class="p">:</span> <span class="n">Context</span><span class="p">):</span>
<span class="n">hook</span> <span class="o">=</span> <span class="n">DataprocHook</span><span class="p">(</span><span class="n">gcp_conn_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">gcp_conn_id</span><span class="p">,</span> <span class="n">impersonation_chain</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">impersonation_chain</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">log</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s2">&quot;Creating batch&quot;</span><span class="p">)</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">region</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
<span class="k">raise</span> <span class="n">AirflowException</span><span class="p">(</span><span class="s1">&#39;Region should be set here&#39;</span><span class="p">)</span>
<span class="k">try</span><span class="p">:</span>
<span class="bp">self</span><span class="o">.</span><span class="n">operation</span> <span class="o">=</span> <span class="n">hook</span><span class="o">.</span><span class="n">create_batch</span><span class="p">(</span>
<span class="n">region</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">region</span><span class="p">,</span>
<span class="n">project_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">project_id</span><span class="p">,</span>
<span class="n">batch</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">batch</span><span class="p">,</span>
<span class="n">batch_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">batch_id</span><span class="p">,</span>
<span class="n">request_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">request_id</span><span class="p">,</span>
<span class="n">retry</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">retry</span><span class="p">,</span>
<span class="n">timeout</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">timeout</span><span class="p">,</span>
<span class="n">metadata</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">metadata</span><span class="p">,</span>
<span class="p">)</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">operation</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
<span class="k">raise</span> <span class="ne">RuntimeError</span><span class="p">(</span><span class="s2">&quot;The operation should be set here!&quot;</span><span class="p">)</span>
<span class="n">result</span> <span class="o">=</span> <span class="n">hook</span><span class="o">.</span><span class="n">wait_for_operation</span><span class="p">(</span><span class="n">timeout</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">timeout</span><span class="p">,</span> <span class="n">operation</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">operation</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">log</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s2">&quot;Batch </span><span class="si">%s</span><span class="s2"> created&quot;</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">batch_id</span><span class="p">)</span>
<span class="k">except</span> <span class="n">AlreadyExists</span><span class="p">:</span>
<span class="bp">self</span><span class="o">.</span><span class="n">log</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s2">&quot;Batch with given id already exists&quot;</span><span class="p">)</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">batch_id</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
<span class="k">raise</span> <span class="n">AirflowException</span><span class="p">(</span><span class="s1">&#39;Batch Id should be set here&#39;</span><span class="p">)</span>
<span class="n">result</span> <span class="o">=</span> <span class="n">hook</span><span class="o">.</span><span class="n">get_batch</span><span class="p">(</span>
<span class="n">batch_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">batch_id</span><span class="p">,</span>
<span class="n">region</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">region</span><span class="p">,</span>
<span class="n">project_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">project_id</span><span class="p">,</span>
<span class="n">retry</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">retry</span><span class="p">,</span>
<span class="n">timeout</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">timeout</span><span class="p">,</span>
<span class="n">metadata</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">metadata</span><span class="p">,</span>
<span class="p">)</span>
<span class="c1"># The existing batch may be a in a number of states other than &#39;SUCCEEDED&#39;</span>
<span class="k">if</span> <span class="n">result</span><span class="o">.</span><span class="n">state</span> <span class="o">!=</span> <span class="n">Batch</span><span class="o">.</span><span class="n">State</span><span class="o">.</span><span class="n">SUCCEEDED</span><span class="p">:</span>
<span class="k">if</span> <span class="n">result</span><span class="o">.</span><span class="n">state</span> <span class="o">==</span> <span class="n">Batch</span><span class="o">.</span><span class="n">State</span><span class="o">.</span><span class="n">FAILED</span> <span class="ow">or</span> <span class="n">result</span><span class="o">.</span><span class="n">state</span> <span class="o">==</span> <span class="n">Batch</span><span class="o">.</span><span class="n">State</span><span class="o">.</span><span class="n">CANCELLED</span><span class="p">:</span>
<span class="k">raise</span> <span class="n">AirflowException</span><span class="p">(</span>
<span class="sa">f</span><span class="s2">&quot;Existing Batch </span><span class="si">{</span><span class="bp">self</span><span class="o">.</span><span class="n">batch_id</span><span class="si">}</span><span class="s2"> failed or cancelled. &quot;</span>
<span class="sa">f</span><span class="s2">&quot;Error: </span><span class="si">{</span><span class="n">result</span><span class="o">.</span><span class="n">state_message</span><span class="si">}</span><span class="s2">&quot;</span>
<span class="p">)</span>
<span class="k">else</span><span class="p">:</span>
<span class="c1"># Batch state is either: RUNNING, PENDING, CANCELLING, or UNSPECIFIED</span>
<span class="bp">self</span><span class="o">.</span><span class="n">log</span><span class="o">.</span><span class="n">info</span><span class="p">(</span>
<span class="sa">f</span><span class="s2">&quot;Batch </span><span class="si">{</span><span class="bp">self</span><span class="o">.</span><span class="n">batch_id</span><span class="si">}</span><span class="s2"> is in state </span><span class="si">{</span><span class="n">result</span><span class="o">.</span><span class="n">state</span><span class="o">.</span><span class="n">name</span><span class="si">}</span><span class="s2">.&quot;</span>
<span class="s2">&quot;Waiting for state change...&quot;</span>
<span class="p">)</span>
<span class="n">result</span> <span class="o">=</span> <span class="n">hook</span><span class="o">.</span><span class="n">wait_for_operation</span><span class="p">(</span><span class="n">timeout</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">timeout</span><span class="p">,</span> <span class="n">operation</span><span class="o">=</span><span class="n">result</span><span class="p">)</span>
<span class="n">batch_id</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">batch_id</span> <span class="ow">or</span> <span class="n">result</span><span class="o">.</span><span class="n">name</span><span class="o">.</span><span class="n">split</span><span class="p">(</span><span class="s1">&#39;/&#39;</span><span class="p">)[</span><span class="o">-</span><span class="mi">1</span><span class="p">]</span>
<span class="n">DataprocLink</span><span class="o">.</span><span class="n">persist</span><span class="p">(</span><span class="n">context</span><span class="o">=</span><span class="n">context</span><span class="p">,</span> <span class="n">task_instance</span><span class="o">=</span><span class="bp">self</span><span class="p">,</span> <span class="n">url</span><span class="o">=</span><span class="n">DATAPROC_BATCH_LINK</span><span class="p">,</span> <span class="n">resource</span><span class="o">=</span><span class="n">batch_id</span><span class="p">)</span>
<span class="k">return</span> <span class="n">Batch</span><span class="o">.</span><span class="n">to_dict</span><span class="p">(</span><span class="n">result</span><span class="p">)</span></div>
<div class="viewcode-block" id="DataprocCreateBatchOperator.on_kill"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocCreateBatchOperator.on_kill">[docs]</a> <span class="k">def</span> <span class="nf">on_kill</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">operation</span><span class="p">:</span>
<span class="bp">self</span><span class="o">.</span><span class="n">operation</span><span class="o">.</span><span class="n">cancel</span><span class="p">()</span></div></div>
<div class="viewcode-block" id="DataprocDeleteBatchOperator"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocDeleteBatchOperator">[docs]</a><span class="k">class</span> <span class="nc">DataprocDeleteBatchOperator</span><span class="p">(</span><span class="n">BaseOperator</span><span class="p">):</span>
<span class="sd">&quot;&quot;&quot;</span>
<span class="sd"> Deletes the batch workload resource.</span>
<span class="sd"> :param batch_id: Required. The ID to use for the batch, which will become the final component</span>
<span class="sd"> of the batch&#39;s resource name.</span>
<span class="sd"> This value must be 4-63 characters. Valid characters are /[a-z][0-9]-/.</span>
<span class="sd"> :param region: Required. The Cloud Dataproc region in which to handle the request.</span>
<span class="sd"> :param project_id: Optional. The ID of the Google Cloud project that the cluster belongs to.</span>
<span class="sd"> :param retry: A retry object used to retry requests. If ``None`` is specified, requests will not be</span>
<span class="sd"> retried.</span>
<span class="sd"> :param timeout: The amount of time, in seconds, to wait for the request to complete. Note that if</span>
<span class="sd"> ``retry`` is specified, the timeout applies to each individual attempt.</span>
<span class="sd"> :param metadata: Additional metadata that is provided to the method.</span>
<span class="sd"> :param gcp_conn_id: The connection ID to use connecting to Google Cloud.</span>
<span class="sd"> :param impersonation_chain: Optional service account to impersonate using short-term</span>
<span class="sd"> credentials, or chained list of accounts required to get the access_token</span>
<span class="sd"> of the last account in the list, which will be impersonated in the request.</span>
<span class="sd"> If set as a string, the account must grant the originating account</span>
<span class="sd"> the Service Account Token Creator IAM role.</span>
<span class="sd"> If set as a sequence, the identities from the list must grant</span>
<span class="sd"> Service Account Token Creator IAM role to the directly preceding identity, with first</span>
<span class="sd"> account from the list granting this role to the originating account (templated).</span>
<span class="sd"> &quot;&quot;&quot;</span>
<div class="viewcode-block" id="DataprocDeleteBatchOperator.template_fields"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocDeleteBatchOperator.template_fields">[docs]</a> <span class="n">template_fields</span><span class="p">:</span> <span class="n">Sequence</span><span class="p">[</span><span class="nb">str</span><span class="p">]</span> <span class="o">=</span> <span class="p">(</span><span class="s2">&quot;batch_id&quot;</span><span class="p">,</span> <span class="s2">&quot;region&quot;</span><span class="p">,</span> <span class="s2">&quot;project_id&quot;</span><span class="p">,</span> <span class="s2">&quot;impersonation_chain&quot;</span><span class="p">)</span></div>
<span class="k">def</span> <span class="fm">__init__</span><span class="p">(</span>
<span class="bp">self</span><span class="p">,</span>
<span class="o">*</span><span class="p">,</span>
<span class="n">batch_id</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span>
<span class="n">region</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span>
<span class="n">project_id</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">retry</span><span class="p">:</span> <span class="n">Retry</span> <span class="o">|</span> <span class="n">_MethodDefault</span> <span class="o">=</span> <span class="n">DEFAULT</span><span class="p">,</span>
<span class="n">timeout</span><span class="p">:</span> <span class="nb">float</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">metadata</span><span class="p">:</span> <span class="n">Sequence</span><span class="p">[</span><span class="nb">tuple</span><span class="p">[</span><span class="nb">str</span><span class="p">,</span> <span class="nb">str</span><span class="p">]]</span> <span class="o">=</span> <span class="p">(),</span>
<span class="n">gcp_conn_id</span><span class="p">:</span> <span class="nb">str</span> <span class="o">=</span> <span class="s2">&quot;google_cloud_default&quot;</span><span class="p">,</span>
<span class="n">impersonation_chain</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="n">Sequence</span><span class="p">[</span><span class="nb">str</span><span class="p">]</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="o">**</span><span class="n">kwargs</span><span class="p">,</span>
<span class="p">):</span>
<span class="nb">super</span><span class="p">()</span><span class="o">.</span><span class="fm">__init__</span><span class="p">(</span><span class="o">**</span><span class="n">kwargs</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">batch_id</span> <span class="o">=</span> <span class="n">batch_id</span>
<span class="bp">self</span><span class="o">.</span><span class="n">region</span> <span class="o">=</span> <span class="n">region</span>
<span class="bp">self</span><span class="o">.</span><span class="n">project_id</span> <span class="o">=</span> <span class="n">project_id</span>
<span class="bp">self</span><span class="o">.</span><span class="n">retry</span> <span class="o">=</span> <span class="n">retry</span>
<span class="bp">self</span><span class="o">.</span><span class="n">timeout</span> <span class="o">=</span> <span class="n">timeout</span>
<span class="bp">self</span><span class="o">.</span><span class="n">metadata</span> <span class="o">=</span> <span class="n">metadata</span>
<span class="bp">self</span><span class="o">.</span><span class="n">gcp_conn_id</span> <span class="o">=</span> <span class="n">gcp_conn_id</span>
<span class="bp">self</span><span class="o">.</span><span class="n">impersonation_chain</span> <span class="o">=</span> <span class="n">impersonation_chain</span>
<div class="viewcode-block" id="DataprocDeleteBatchOperator.execute"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocDeleteBatchOperator.execute">[docs]</a> <span class="k">def</span> <span class="nf">execute</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">context</span><span class="p">:</span> <span class="n">Context</span><span class="p">):</span>
<span class="n">hook</span> <span class="o">=</span> <span class="n">DataprocHook</span><span class="p">(</span><span class="n">gcp_conn_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">gcp_conn_id</span><span class="p">,</span> <span class="n">impersonation_chain</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">impersonation_chain</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">log</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s2">&quot;Deleting batch: </span><span class="si">%s</span><span class="s2">&quot;</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">batch_id</span><span class="p">)</span>
<span class="n">hook</span><span class="o">.</span><span class="n">delete_batch</span><span class="p">(</span>
<span class="n">batch_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">batch_id</span><span class="p">,</span>
<span class="n">region</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">region</span><span class="p">,</span>
<span class="n">project_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">project_id</span><span class="p">,</span>
<span class="n">retry</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">retry</span><span class="p">,</span>
<span class="n">timeout</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">timeout</span><span class="p">,</span>
<span class="n">metadata</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">metadata</span><span class="p">,</span>
<span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">log</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s2">&quot;Batch deleted.&quot;</span><span class="p">)</span></div></div>
<div class="viewcode-block" id="DataprocGetBatchOperator"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocGetBatchOperator">[docs]</a><span class="k">class</span> <span class="nc">DataprocGetBatchOperator</span><span class="p">(</span><span class="n">BaseOperator</span><span class="p">):</span>
<span class="sd">&quot;&quot;&quot;</span>
<span class="sd"> Gets the batch workload resource representation.</span>
<span class="sd"> :param batch_id: Required. The ID to use for the batch, which will become the final component</span>
<span class="sd"> of the batch&#39;s resource name.</span>
<span class="sd"> This value must be 4-63 characters. Valid characters are /[a-z][0-9]-/.</span>
<span class="sd"> :param region: Required. The Cloud Dataproc region in which to handle the request.</span>
<span class="sd"> :param project_id: Optional. The ID of the Google Cloud project that the cluster belongs to.</span>
<span class="sd"> :param retry: A retry object used to retry requests. If ``None`` is specified, requests will not be</span>
<span class="sd"> retried.</span>
<span class="sd"> :param timeout: The amount of time, in seconds, to wait for the request to complete. Note that if</span>
<span class="sd"> ``retry`` is specified, the timeout applies to each individual attempt.</span>
<span class="sd"> :param metadata: Additional metadata that is provided to the method.</span>
<span class="sd"> :param gcp_conn_id: The connection ID to use connecting to Google Cloud.</span>
<span class="sd"> :param impersonation_chain: Optional service account to impersonate using short-term</span>
<span class="sd"> credentials, or chained list of accounts required to get the access_token</span>
<span class="sd"> of the last account in the list, which will be impersonated in the request.</span>
<span class="sd"> If set as a string, the account must grant the originating account</span>
<span class="sd"> the Service Account Token Creator IAM role.</span>
<span class="sd"> If set as a sequence, the identities from the list must grant</span>
<span class="sd"> Service Account Token Creator IAM role to the directly preceding identity, with first</span>
<span class="sd"> account from the list granting this role to the originating account (templated).</span>
<span class="sd"> &quot;&quot;&quot;</span>
<div class="viewcode-block" id="DataprocGetBatchOperator.template_fields"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocGetBatchOperator.template_fields">[docs]</a> <span class="n">template_fields</span><span class="p">:</span> <span class="n">Sequence</span><span class="p">[</span><span class="nb">str</span><span class="p">]</span> <span class="o">=</span> <span class="p">(</span><span class="s2">&quot;batch_id&quot;</span><span class="p">,</span> <span class="s2">&quot;region&quot;</span><span class="p">,</span> <span class="s2">&quot;project_id&quot;</span><span class="p">,</span> <span class="s2">&quot;impersonation_chain&quot;</span><span class="p">)</span></div>
<div class="viewcode-block" id="DataprocGetBatchOperator.operator_extra_links"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocGetBatchOperator.operator_extra_links">[docs]</a> <span class="n">operator_extra_links</span> <span class="o">=</span> <span class="p">(</span><span class="n">DataprocLink</span><span class="p">(),)</span></div>
<span class="k">def</span> <span class="fm">__init__</span><span class="p">(</span>
<span class="bp">self</span><span class="p">,</span>
<span class="o">*</span><span class="p">,</span>
<span class="n">batch_id</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span>
<span class="n">region</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span>
<span class="n">project_id</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">retry</span><span class="p">:</span> <span class="n">Retry</span> <span class="o">|</span> <span class="n">_MethodDefault</span> <span class="o">=</span> <span class="n">DEFAULT</span><span class="p">,</span>
<span class="n">timeout</span><span class="p">:</span> <span class="nb">float</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">metadata</span><span class="p">:</span> <span class="n">Sequence</span><span class="p">[</span><span class="nb">tuple</span><span class="p">[</span><span class="nb">str</span><span class="p">,</span> <span class="nb">str</span><span class="p">]]</span> <span class="o">=</span> <span class="p">(),</span>
<span class="n">gcp_conn_id</span><span class="p">:</span> <span class="nb">str</span> <span class="o">=</span> <span class="s2">&quot;google_cloud_default&quot;</span><span class="p">,</span>
<span class="n">impersonation_chain</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="n">Sequence</span><span class="p">[</span><span class="nb">str</span><span class="p">]</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="o">**</span><span class="n">kwargs</span><span class="p">,</span>
<span class="p">):</span>
<span class="nb">super</span><span class="p">()</span><span class="o">.</span><span class="fm">__init__</span><span class="p">(</span><span class="o">**</span><span class="n">kwargs</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">batch_id</span> <span class="o">=</span> <span class="n">batch_id</span>
<span class="bp">self</span><span class="o">.</span><span class="n">region</span> <span class="o">=</span> <span class="n">region</span>
<span class="bp">self</span><span class="o">.</span><span class="n">project_id</span> <span class="o">=</span> <span class="n">project_id</span>
<span class="bp">self</span><span class="o">.</span><span class="n">retry</span> <span class="o">=</span> <span class="n">retry</span>
<span class="bp">self</span><span class="o">.</span><span class="n">timeout</span> <span class="o">=</span> <span class="n">timeout</span>
<span class="bp">self</span><span class="o">.</span><span class="n">metadata</span> <span class="o">=</span> <span class="n">metadata</span>
<span class="bp">self</span><span class="o">.</span><span class="n">gcp_conn_id</span> <span class="o">=</span> <span class="n">gcp_conn_id</span>
<span class="bp">self</span><span class="o">.</span><span class="n">impersonation_chain</span> <span class="o">=</span> <span class="n">impersonation_chain</span>
<div class="viewcode-block" id="DataprocGetBatchOperator.execute"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocGetBatchOperator.execute">[docs]</a> <span class="k">def</span> <span class="nf">execute</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">context</span><span class="p">:</span> <span class="n">Context</span><span class="p">):</span>
<span class="n">hook</span> <span class="o">=</span> <span class="n">DataprocHook</span><span class="p">(</span><span class="n">gcp_conn_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">gcp_conn_id</span><span class="p">,</span> <span class="n">impersonation_chain</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">impersonation_chain</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">log</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s2">&quot;Getting batch: </span><span class="si">%s</span><span class="s2">&quot;</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">batch_id</span><span class="p">)</span>
<span class="n">batch</span> <span class="o">=</span> <span class="n">hook</span><span class="o">.</span><span class="n">get_batch</span><span class="p">(</span>
<span class="n">batch_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">batch_id</span><span class="p">,</span>
<span class="n">region</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">region</span><span class="p">,</span>
<span class="n">project_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">project_id</span><span class="p">,</span>
<span class="n">retry</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">retry</span><span class="p">,</span>
<span class="n">timeout</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">timeout</span><span class="p">,</span>
<span class="n">metadata</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">metadata</span><span class="p">,</span>
<span class="p">)</span>
<span class="n">DataprocLink</span><span class="o">.</span><span class="n">persist</span><span class="p">(</span>
<span class="n">context</span><span class="o">=</span><span class="n">context</span><span class="p">,</span> <span class="n">task_instance</span><span class="o">=</span><span class="bp">self</span><span class="p">,</span> <span class="n">url</span><span class="o">=</span><span class="n">DATAPROC_BATCH_LINK</span><span class="p">,</span> <span class="n">resource</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">batch_id</span>
<span class="p">)</span>
<span class="k">return</span> <span class="n">Batch</span><span class="o">.</span><span class="n">to_dict</span><span class="p">(</span><span class="n">batch</span><span class="p">)</span></div></div>
<div class="viewcode-block" id="DataprocListBatchesOperator"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocListBatchesOperator">[docs]</a><span class="k">class</span> <span class="nc">DataprocListBatchesOperator</span><span class="p">(</span><span class="n">BaseOperator</span><span class="p">):</span>
<span class="sd">&quot;&quot;&quot;</span>
<span class="sd"> Lists batch workloads.</span>
<span class="sd"> :param region: Required. The Cloud Dataproc region in which to handle the request.</span>
<span class="sd"> :param project_id: Optional. The ID of the Google Cloud project that the cluster belongs to.</span>
<span class="sd"> :param page_size: Optional. The maximum number of batches to return in each response. The service may</span>
<span class="sd"> return fewer than this value. The default page size is 20; the maximum page size is 1000.</span>
<span class="sd"> :param page_token: Optional. A page token received from a previous ``ListBatches`` call.</span>
<span class="sd"> Provide this token to retrieve the subsequent page.</span>
<span class="sd"> :param retry: Optional, a retry object used to retry requests. If `None` is specified, requests</span>
<span class="sd"> will not be retried.</span>
<span class="sd"> :param timeout: Optional, the amount of time, in seconds, to wait for the request to complete.</span>
<span class="sd"> Note that if `retry` is specified, the timeout applies to each individual attempt.</span>
<span class="sd"> :param metadata: Optional, additional metadata that is provided to the method.</span>
<span class="sd"> :param gcp_conn_id: Optional, the connection ID used to connect to Google Cloud Platform.</span>
<span class="sd"> :param impersonation_chain: Optional service account to impersonate using short-term</span>
<span class="sd"> credentials, or chained list of accounts required to get the access_token</span>
<span class="sd"> of the last account in the list, which will be impersonated in the request.</span>
<span class="sd"> If set as a string, the account must grant the originating account</span>
<span class="sd"> the Service Account Token Creator IAM role.</span>
<span class="sd"> If set as a sequence, the identities from the list must grant</span>
<span class="sd"> Service Account Token Creator IAM role to the directly preceding identity, with first</span>
<span class="sd"> account from the list granting this role to the originating account (templated).</span>
<span class="sd"> :rtype: List[dict]</span>
<span class="sd"> &quot;&quot;&quot;</span>
<div class="viewcode-block" id="DataprocListBatchesOperator.template_fields"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocListBatchesOperator.template_fields">[docs]</a> <span class="n">template_fields</span><span class="p">:</span> <span class="n">Sequence</span><span class="p">[</span><span class="nb">str</span><span class="p">]</span> <span class="o">=</span> <span class="p">(</span><span class="s2">&quot;region&quot;</span><span class="p">,</span> <span class="s2">&quot;project_id&quot;</span><span class="p">,</span> <span class="s2">&quot;impersonation_chain&quot;</span><span class="p">)</span></div>
<div class="viewcode-block" id="DataprocListBatchesOperator.operator_extra_links"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocListBatchesOperator.operator_extra_links">[docs]</a> <span class="n">operator_extra_links</span> <span class="o">=</span> <span class="p">(</span><span class="n">DataprocListLink</span><span class="p">(),)</span></div>
<span class="k">def</span> <span class="fm">__init__</span><span class="p">(</span>
<span class="bp">self</span><span class="p">,</span>
<span class="o">*</span><span class="p">,</span>
<span class="n">region</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span>
<span class="n">project_id</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">page_size</span><span class="p">:</span> <span class="nb">int</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">page_token</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">retry</span><span class="p">:</span> <span class="n">Retry</span> <span class="o">|</span> <span class="n">_MethodDefault</span> <span class="o">=</span> <span class="n">DEFAULT</span><span class="p">,</span>
<span class="n">timeout</span><span class="p">:</span> <span class="nb">float</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="n">metadata</span><span class="p">:</span> <span class="n">Sequence</span><span class="p">[</span><span class="nb">tuple</span><span class="p">[</span><span class="nb">str</span><span class="p">,</span> <span class="nb">str</span><span class="p">]]</span> <span class="o">=</span> <span class="p">(),</span>
<span class="n">gcp_conn_id</span><span class="p">:</span> <span class="nb">str</span> <span class="o">=</span> <span class="s2">&quot;google_cloud_default&quot;</span><span class="p">,</span>
<span class="n">impersonation_chain</span><span class="p">:</span> <span class="nb">str</span> <span class="o">|</span> <span class="n">Sequence</span><span class="p">[</span><span class="nb">str</span><span class="p">]</span> <span class="o">|</span> <span class="kc">None</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
<span class="o">**</span><span class="n">kwargs</span><span class="p">,</span>
<span class="p">)</span> <span class="o">-&gt;</span> <span class="kc">None</span><span class="p">:</span>
<span class="nb">super</span><span class="p">()</span><span class="o">.</span><span class="fm">__init__</span><span class="p">(</span><span class="o">**</span><span class="n">kwargs</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">region</span> <span class="o">=</span> <span class="n">region</span>
<span class="bp">self</span><span class="o">.</span><span class="n">project_id</span> <span class="o">=</span> <span class="n">project_id</span>
<span class="bp">self</span><span class="o">.</span><span class="n">page_size</span> <span class="o">=</span> <span class="n">page_size</span>
<span class="bp">self</span><span class="o">.</span><span class="n">page_token</span> <span class="o">=</span> <span class="n">page_token</span>
<span class="bp">self</span><span class="o">.</span><span class="n">retry</span> <span class="o">=</span> <span class="n">retry</span>
<span class="bp">self</span><span class="o">.</span><span class="n">timeout</span> <span class="o">=</span> <span class="n">timeout</span>
<span class="bp">self</span><span class="o">.</span><span class="n">metadata</span> <span class="o">=</span> <span class="n">metadata</span>
<span class="bp">self</span><span class="o">.</span><span class="n">gcp_conn_id</span> <span class="o">=</span> <span class="n">gcp_conn_id</span>
<span class="bp">self</span><span class="o">.</span><span class="n">impersonation_chain</span> <span class="o">=</span> <span class="n">impersonation_chain</span>
<div class="viewcode-block" id="DataprocListBatchesOperator.execute"><a class="viewcode-back" href="../../../../../../_api/airflow/providers/google/cloud/operators/dataproc/index.html#airflow.providers.google.cloud.operators.dataproc.DataprocListBatchesOperator.execute">[docs]</a> <span class="k">def</span> <span class="nf">execute</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">context</span><span class="p">:</span> <span class="n">Context</span><span class="p">):</span>
<span class="n">hook</span> <span class="o">=</span> <span class="n">DataprocHook</span><span class="p">(</span><span class="n">gcp_conn_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">gcp_conn_id</span><span class="p">,</span> <span class="n">impersonation_chain</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">impersonation_chain</span><span class="p">)</span>
<span class="n">results</span> <span class="o">=</span> <span class="n">hook</span><span class="o">.</span><span class="n">list_batches</span><span class="p">(</span>
<span class="n">region</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">region</span><span class="p">,</span>
<span class="n">project_id</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">project_id</span><span class="p">,</span>
<span class="n">page_size</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">page_size</span><span class="p">,</span>
<span class="n">page_token</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">page_token</span><span class="p">,</span>
<span class="n">retry</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">retry</span><span class="p">,</span>
<span class="n">timeout</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">timeout</span><span class="p">,</span>
<span class="n">metadata</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">metadata</span><span class="p">,</span>
<span class="p">)</span>
<span class="n">DataprocListLink</span><span class="o">.</span><span class="n">persist</span><span class="p">(</span><span class="n">context</span><span class="o">=</span><span class="n">context</span><span class="p">,</span> <span class="n">task_instance</span><span class="o">=</span><span class="bp">self</span><span class="p">,</span> <span class="n">url</span><span class="o">=</span><span class="n">DATAPROC_BATCHES_LINK</span><span class="p">)</span>
<span class="k">return</span> <span class="p">[</span><span class="n">Batch</span><span class="o">.</span><span class="n">to_dict</span><span class="p">(</span><span class="n">result</span><span class="p">)</span> <span class="k">for</span> <span class="n">result</span> <span class="ow">in</span> <span class="n">results</span><span class="p">]</span></div></div>
</pre></div>
<div class="pager" role="navigation" aria-label="related navigation"><button class="btn-hollow btn-blue bodytext__medium--cerulean-blue" disabled>Previous</button><button class="btn-hollow btn-blue bodytext__medium--cerulean-blue" disabled>Next</button>
</div>
</div>
</div>
</div>
</div>
</div>
<div class="rating-container">
<p class="bodytext__medium--greyish-brown font-weight-500">Was this entry helpful?</p>
<div class="rating">
<div id="rate-star-5" class="rate-star">
<svg xmlns="http://www.w3.org/2000/svg" width="21.05" height="20.02" viewBox="0 0 21.05 20.02">
<g id="Group_806" data-name="Group 806" transform="translate(-774.404 -13.178)">
<path id="Path_715" d="M791.434 33.2l-6.5-3.42-6.5 3.42 1.242-7.243-5.262-5.13 7.273-1.057 3.252-6.59 3.252 6.59 7.273 1.057-5.262 5.13z" fill="#017cee" data-name="Path 715"></path>
</g>
</svg>
</div>
<div id="rate-star-4" class="rate-star">
<svg xmlns="http://www.w3.org/2000/svg" width="21.05" height="20.02" viewBox="0 0 21.05 20.02">
<g id="Group_806" data-name="Group 806" transform="translate(-774.404 -13.178)">
<path id="Path_715" d="M791.434 33.2l-6.5-3.42-6.5 3.42 1.242-7.243-5.262-5.13 7.273-1.057 3.252-6.59 3.252 6.59 7.273 1.057-5.262 5.13z" fill="#017cee" data-name="Path 715"></path>
</g>
</svg>
</div>
<div id="rate-star-3" class="rate-star">
<svg xmlns="http://www.w3.org/2000/svg" width="21.05" height="20.02" viewBox="0 0 21.05 20.02">
<g id="Group_806" data-name="Group 806" transform="translate(-774.404 -13.178)">
<path id="Path_715" d="M791.434 33.2l-6.5-3.42-6.5 3.42 1.242-7.243-5.262-5.13 7.273-1.057 3.252-6.59 3.252 6.59 7.273 1.057-5.262 5.13z" fill="#017cee" data-name="Path 715"></path>
</g>
</svg>
</div>
<div id="rate-star-2" class="rate-star">
<svg xmlns="http://www.w3.org/2000/svg" width="21.05" height="20.02" viewBox="0 0 21.05 20.02">
<g id="Group_806" data-name="Group 806" transform="translate(-774.404 -13.178)">
<path id="Path_715" d="M791.434 33.2l-6.5-3.42-6.5 3.42 1.242-7.243-5.262-5.13 7.273-1.057 3.252-6.59 3.252 6.59 7.273 1.057-5.262 5.13z" fill="#017cee" data-name="Path 715"></path>
</g>
</svg>
</div>
<div id="rate-star-1" class="rate-star">
<svg xmlns="http://www.w3.org/2000/svg" width="21.05" height="20.02" viewBox="0 0 21.05 20.02">
<g id="Group_806" data-name="Group 806" transform="translate(-774.404 -13.178)">
<path id="Path_715" d="M791.434 33.2l-6.5-3.42-6.5 3.42 1.242-7.243-5.262-5.13 7.273-1.057 3.252-6.59 3.252 6.59 7.273 1.057-5.262 5.13z" fill="#017cee" data-name="Path 715"></path>
</g>
</svg>
</div>
</div>
</div>
</main>
<nav class="wy-nav-side-toc">
<div class="wy-menu-vertical">
</div>
</nav>
</div>
</div>
<footer>
<div class="footer-section footer-section__media-section">
<div class="d-flex align-items-center">
<a class="footer-section__media-section--link" target="_blank" href="https://github.com/apache/airflow">
<svg xmlns="http://www.w3.org/2000/svg" width="46.14" height="45" viewBox="0 0 46.14 45">
<path id="Path_207" d="M228.962 1078.578a23.072 23.072 0 0 0-7.29 44.96c1.154.212 1.574-.5 1.574-1.112 0-.548-.02-2-.031-3.924-6.417 1.394-7.771-3.093-7.771-3.093a6.109 6.109 0 0 0-2.562-3.375c-2.095-1.431.159-1.4.159-1.4a4.846 4.846 0 0 1 3.533 2.377c2.058 3.525 5.4 2.507 6.714 1.917a4.926 4.926 0 0 1 1.464-3.084c-5.123-.582-10.508-2.562-10.508-11.4a8.919 8.919 0 0 1 2.374-6.191 8.3 8.3 0 0 1 .226-6.105s1.937-.62 6.344 2.365a21.857 21.857 0 0 1 11.551 0c4.4-2.985 6.338-2.365 6.338-2.365a8.284 8.284 0 0 1 .23 6.105 8.9 8.9 0 0 1 2.371 6.191c0 8.862-5.393 10.812-10.533 11.384a5.506 5.506 0 0 1 1.566 4.272c0 3.084-.028 5.572-.028 6.329 0 .617.415 1.334 1.586 1.109a23.073 23.073 0 0 0-7.308-44.958z" fill="#fff" fill-rule="evenodd" data-name="Path 207" transform="translate(-205.894 -1078.578)"></path>
</svg>
</a>
<a class="footer-section__media-section--link" target="_blank" href="https://github.com/apache/airflow/issues">
<svg xmlns="http://www.w3.org/2000/svg" width="45" height="45" viewBox="0 0 45 45">
<g id="Group_210" data-name="Group 210" transform="translate(-339.789 -1315.282)">
<path id="Path_218" d="M394.82 1315.282h-21.671a9.784 9.784 0 0 0 9.784 9.778h3.986v3.857a9.784 9.784 0 0 0 9.784 9.771v-21.523a1.884 1.884 0 0 0-1.883-1.883z" fill="#fff" data-name="Path 218" transform="translate(-11.914)"></path>
<path id="Path_219" d="M378.14 1332.072h-21.671a9.778 9.778 0 0 0 9.778 9.778h4.018v3.857a9.784 9.784 0 0 0 9.752 9.778v-21.536a1.877 1.877 0 0 0-1.877-1.877z" fill="#fff" data-name="Path 219" transform="translate(-5.957 -5.996)"></path>
<path id="Path_220" d="M361.46 1348.862h-21.671a9.778 9.778 0 0 0 9.778 9.778h3.992v3.857a9.778 9.778 0 0 0 9.778 9.778v-21.529a1.883 1.883 0 0 0-1.877-1.884z" fill="#fff" data-name="Path 220" transform="translate(0 -11.993)"></path>
</g>
</svg>
</a>
<a class="footer-section__media-section--link" target="_blank" href="https://s.apache.org/airflow-slack">
<svg xmlns="http://www.w3.org/2000/svg" width="45.073" height="45.073" viewBox="0 0 45.073 45.073">
<g id="Group_208" data-name="Group 208" transform="translate(-661.145 -806.287)">
<g id="Group_204" data-name="Group 204" transform="translate(661.145 830.01)">
<path id="Path_208" d="M670.634 856.859a4.744 4.744 0 1 1-4.744-4.744h4.744z" fill="#fff" data-name="Path 208" transform="translate(-661.145 -852.115)"></path>
<path id="Path_209" d="M684.059 856.859a4.744 4.744 0 0 1 9.489 0v11.861a4.744 4.744 0 1 1-9.489 0z" fill="#fff" data-name="Path 209" transform="translate(-672.198 -852.115)"></path>
</g>
<g id="Group_205" data-name="Group 205" transform="translate(661.145 806.287)">
<path id="Path_210" d="M688.8 815.776a4.744 4.744 0 1 1 4.744-4.745v4.745z" fill="#fff" data-name="Path 210" transform="translate(-672.198 -806.287)"></path>
<path id="Path_211" d="M677.751 829.2a4.744 4.744 0 0 1 0 9.489H665.89a4.744 4.744 0 1 1 0-9.489z" fill="#fff" data-name="Path 211" transform="translate(-661.145 -817.34)"></path>
</g>
<g id="Group_206" data-name="Group 206" transform="translate(684.868 806.287)">
<path id="Path_212" d="M729.887 833.945a4.744 4.744 0 1 1 4.745 4.745h-4.745z" fill="#fff" data-name="Path 212" transform="translate(-718.026 -817.34)"></path>
<path id="Path_213" d="M716.462 822.893a4.744 4.744 0 1 1-9.489 0v-11.862a4.744 4.744 0 0 1 9.489 0z" fill="#fff" data-name="Path 213" transform="translate(-706.973 -806.287)"></path>
</g>
<g id="Group_207" data-name="Group 207" transform="translate(684.868 830.01)">
<path id="Path_214" d="M711.718 875.029a4.744 4.744 0 1 1-4.745 4.744v-4.744z" fill="#fff" data-name="Path 214" transform="translate(-706.973 -863.168)"></path>
<path id="Path_215" d="M711.718 861.6a4.744 4.744 0 1 1 0-9.489h11.861a4.744 4.744 0 0 1 0 9.489z" fill="#fff" data-name="Path 215" transform="translate(-706.973 -852.115)"></path>
</g>
</g>
</svg>
</a>
<a class="footer-section__media-section--link" target="_blank" href="https://stackoverflow.com/questions/tagged/airflow">
<svg xmlns="http://www.w3.org/2000/svg" width="37.647" height="44.6" viewBox="0 0 37.647 44.6">
<g id="Group_209" data-name="Group 209" transform="translate(-645.2 -975.455)">
<path id="Path_216" d="M677.028 1043.1v-11.948h3.966v15.914H645.2v-15.914h3.966v11.948z" fill="#fff" data-name="Path 216" transform="translate(0 -27.014)"></path>
<path id="Path_217" d="M661.012 1003.008l19.467 4.069.824-3.914-19.467-4.069zm2.575-9.27l18.025 8.395 1.648-3.605-18.025-8.446zm5-8.858l15.3 12.721 2.524-3.039-15.3-12.721zm9.888-9.425l-3.193 2.369 11.845 15.965 3.193-2.369zm-17.875 36.617h19.879v-3.966H660.6z" fill="#fff" data-name="Path 217" transform="translate(-7.469)"></path>
</g>
</svg>
</a>
<a class="footer-section__media-section--link" target="_blank" href="https://twitter.com/ApacheAirflow">
<svg xmlns="http://www.w3.org/2000/svg" width="44.355" height="44.355" viewBox="0 0 44.355 44.355">
<g id="Group_211" data-name="Group 211" transform="translate(-503 -1382.2)">
<path id="Path_221" d="M541.811 1382.2h-33.267a5.546 5.546 0 0 0-5.544 5.544v33.266a5.547 5.547 0 0 0 5.544 5.545h33.267a5.546 5.546 0 0 0 5.544-5.545v-33.266a5.546 5.546 0 0 0-5.544-5.544zm-5.611 16.533c.011.244.011.488.011.732 0 7.507-5.722 16.178-16.179 16.178a16.173 16.173 0 0 1-8.7-2.539 10.92 10.92 0 0 0 1.353.078 11.4 11.4 0 0 0 7.064-2.44 5.685 5.685 0 0 1-5.311-3.947 5.725 5.725 0 0 0 2.561-.1 5.7 5.7 0 0 1-4.557-5.578v-.078a5.752 5.752 0 0 0 2.572.71 5.709 5.709 0 0 1-1.763-7.6 16.146 16.146 0 0 0 11.721 5.944 6.282 6.282 0 0 1-.144-1.3 5.688 5.688 0 0 1 9.836-3.892 11.3 11.3 0 0 0 3.615-1.375 5.721 5.721 0 0 1-2.506 3.149 11.538 11.538 0 0 0 3.271-.9 11.585 11.585 0 0 1-2.844 2.958z" fill="#fff" data-name="Path 221"></path>
</g>
</svg>
</a>
<a class="footer-section__media-section--link" target="_blank" href="https://www.youtube.com/channel/UCSXwxpWZQ7XZ1WL3wqevChA">
<svg xmlns="http://www.w3.org/2000/svg" width="49.594" height="34.941" viewBox="0 0 49.594 34.941">
<path id="Path_223" d="M1124.557 1230a6.232 6.232 0 0 0-4.385-4.413c-3.867-1.043-19.376-1.043-19.376-1.043s-15.508 0-19.376 1.043a6.232 6.232 0 0 0-4.385 4.413c-1.036 3.893-1.036 12.014-1.036 12.014s0 8.122 1.036 12.015a6.232 6.232 0 0 0 4.385 4.413c3.867 1.043 19.376 1.043 19.376 1.043s15.509 0 19.376-1.043a6.232 6.232 0 0 0 4.385-4.413c1.036-3.893 1.036-12.015 1.036-12.015s.001-8.123-1.036-12.014zm-28.833 19.388v-14.748l12.962 7.374z" fill="#fff" data-name="Path 223" transform="translate(-1076 -1224.542)"></path>
</svg>
</a>
</div>
<div class="footer-section__media-section--button-with-text">
<span class="footer-section__media-section--text">Want to be a part of Apache Airflow?</span>
<a href="/community">
<button id="" class="btn-filled bodytext__medium--white ">Join community</button>
</a>
</div>
</div>
<div class="footer-section footer-section__policies-section">
<div class="footer-section">
<span>© The Apache Software Foundation <script>document.write(new Date().getFullYear())</script></span>
<div class="footer-section__policies-section--policies">
<a href="https://www.apache.org/licenses/" class="footer-section__policies-section--policy-item">
<span>License</span>
</a>
<a href="https://www.apache.org/foundation/sponsorship.html" class="footer-section__policies-section--policy-item">
<span>Donate</span>
</a>
<a href="https://www.apache.org/foundation/thanks.html" class="footer-section__policies-section--policy-item">
<span>Thanks</span>
</a>
<a href="https://www.apache.org/security/" class="footer-section__policies-section--policy-item">
<span>Security</span>
</a>
</div>
</div>
<span class="footer-section__policies-section--disclaimer">
Apache Airflow, Apache, Airflow, the Airflow logo, and the Apache feather logo are either registered trademarks or trademarks of The Apache Software Foundation.
All other products or name brands are trademarks of their respective holders, including The Apache Software Foundation.
</span>
</div>
</footer>
<script type="text/javascript" src="../../../../../../_static/_gen/js/docs.js"></script>
<script type="text/javascript" id="documentation_options" data-url_root="../../../../../../" src="../../../../../../_static/documentation_options.js"></script>
<script src="https://code.jquery.com/jquery-3.4.1.min.js" integrity="sha256-CSXorXvZcTkaix6Yvo6HppcZGetbYMGWSFlBw8HfCJo=" crossorigin="anonymous"></script>
<script src="https://cdnjs.cloudflare.com/ajax/libs/popper.js/1.14.3/umd/popper.min.js" integrity="sha384-ZMP7rVo3mIykV+2+9J3UJ46jBk0WLaUAdn689aCwoqbBJiSnjAK/l8WvCWPIPm49" crossorigin="anonymous"></script>
<script src="https://stackpath.bootstrapcdn.com/bootstrap/4.1.3/js/bootstrap.min.js" integrity="sha384-ChfqqxuZUCnJSK3+MXmPNIyE6ZbWh2IMqE241rYiqJxyMiZ6OW/JmZQ5stwEULTy" crossorigin="anonymous"></script>
<script data-url_root="../../../../../../" id="documentation_options" src="../../../../../../_static/documentation_options.js"></script>
<script src="../../../../../../_static/jquery.js"></script>
<script src="../../../../../../_static/underscore.js"></script>
<script src="../../../../../../_static/_sphinx_javascript_frameworks_compat.js"></script>
<script src="../../../../../../_static/doctools.js"></script>
<script src="../../../../../../_static/sphinx_highlight.js"></script>
<script src="../../../../../../_static/clipboard.min.js"></script>
<script src="../../../../../../_static/copybutton.js"></script>
<script src="../../../../../../_static/js/globaltoc.js"></script>
</body>
</html>