http://git-wip-us.apache.org/repos/asf/incubator-airflow-site/blob/7d4d7628/_modules/airflow/contrib/operators/dataproc_operator.html
----------------------------------------------------------------------
diff --git a/_modules/airflow/contrib/operators/dataproc_operator.html 
b/_modules/airflow/contrib/operators/dataproc_operator.html
index f26e459..6bc5d70 100644
--- a/_modules/airflow/contrib/operators/dataproc_operator.html
+++ b/_modules/airflow/contrib/operators/dataproc_operator.html
@@ -24,26 +24,17 @@
 
   
 
-  
-  
-    <link rel="stylesheet" href="../../../../_static/css/theme.css" 
type="text/css" />
-  
-
-  
-
-  
-        <link rel="index" title="Index"
-              href="../../../../genindex.html"/>
-        <link rel="search" title="Search" href="../../../../search.html"/>
-    <link rel="top" title="Airflow Documentation" 
href="../../../../index.html"/>
-        <link rel="up" title="Module code" href="../../../index.html"/> 
+  <link rel="stylesheet" href="../../../../_static/css/theme.css" 
type="text/css" />
+  <link rel="stylesheet" href="../../../../_static/pygments.css" 
type="text/css" />
+    <link rel="index" title="Index" href="../../../../genindex.html" />
+    <link rel="search" title="Search" href="../../../../search.html" /> 
 
   
   <script src="../../../../_static/js/modernizr.min.js"></script>
 
 </head>
 
-<body class="wy-body-for-nav" role="document">
+<body class="wy-body-for-nav">
 
    
   <div class="wy-grid-for-nav">
@@ -116,7 +107,7 @@
     <section data-toggle="wy-nav-shift" class="wy-nav-content-wrap">
 
       
-      <nav class="wy-nav-top" role="navigation" aria-label="top navigation">
+      <nav class="wy-nav-top" aria-label="top navigation">
         
           <i data-toggle="wy-nav-top" class="fa fa-bars"></i>
           <a href="../../../../index.html">Airflow</a>
@@ -124,9 +115,10 @@
       </nav>
 
 
-      
       <div class="wy-nav-content">
+        
         <div class="rst-content">
+        
           
 
 
@@ -156,8 +148,6 @@
     
       <li class="wy-breadcrumbs-aside">
         
-            
-        
       </li>
     
   </ul>
@@ -218,10 +208,10 @@
 <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">    :param cluster_name: The name of the DataProc cluster to 
create.</span>
+<span class="sd">    :param cluster_name: The name of the DataProc cluster to 
create. (templated)</span>
 <span class="sd">    :type cluster_name: string</span>
 <span class="sd">    :param project_id: The ID of the google cloud project in 
which</span>
-<span class="sd">        to create the cluster</span>
+<span class="sd">        to create the cluster. (templated)</span>
 <span class="sd">    :type project_id: string</span>
 <span class="sd">    :param num_workers: The # of workers to spin up</span>
 <span class="sd">    :type num_workers: int</span>
@@ -256,7 +246,7 @@
 <span class="sd">    :type num_preemptible_workers: int</span>
 <span class="sd">    :param labels: dict of labels to add to the cluster</span>
 <span class="sd">    :type labels: dict</span>
-<span class="sd">    :param zone: The zone where the cluster will be 
located</span>
+<span class="sd">    :param zone: The zone where the cluster will be located. 
(templated)</span>
 <span class="sd">    :type zone: string</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>
@@ -264,9 +254,13 @@
 <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">    :type subnetwork_uri: string</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">    :type internal_ip_only: bool</span>
 <span class="sd">    :param tags: The GCE tags to add to all instances</span>
 <span class="sd">    :type tags: list[string]</span>
-<span class="sd">    :param region: leave as &#39;global&#39;, might become 
relevant in the future</span>
+<span class="sd">    :param region: leave as &#39;global&#39;, might become 
relevant in the future. (templated)</span>
 <span class="sd">    :param gcp_conn_id: The connection ID to use connecting 
to Google Cloud Platform.</span>
 <span class="sd">    :type gcp_conn_id: string</span>
 <span class="sd">    :param delegate_to: The account to impersonate, if 
any.</span>
@@ -282,7 +276,7 @@
 <span class="sd">        A duration in seconds.</span>
 <span class="sd">    :type idle_delete_ttl: int</span>
 <span class="sd">    :param auto_delete_time:  The time when cluster will be 
auto-deleted.</span>
-<span class="sd">    :type auto_delete_time: datetime</span>
+<span class="sd">    :type auto_delete_time: datetime.datetime</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>
@@ -299,6 +293,7 @@
                  <span class="n">zone</span><span class="p">,</span>
                  <span class="n">network_uri</span><span 
class="o">=</span><span class="kc">None</span><span class="p">,</span>
                  <span class="n">subnetwork_uri</span><span 
class="o">=</span><span class="kc">None</span><span class="p">,</span>
+                 <span class="n">internal_ip_only</span><span 
class="o">=</span><span class="kc">None</span><span class="p">,</span>
                  <span class="n">tags</span><span class="o">=</span><span 
class="kc">None</span><span class="p">,</span>
                  <span class="n">storage_bucket</span><span 
class="o">=</span><span class="kc">None</span><span class="p">,</span>
                  <span class="n">init_actions_uris</span><span 
class="o">=</span><span class="kc">None</span><span class="p">,</span>
@@ -344,6 +339,7 @@
         <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">region</span> <span class="o">=</span> <span class="n">region</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>
@@ -420,13 +416,12 @@
                 <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">zone</span>
             <span class="p">)</span>
         <span class="n">master_type_uri</span> <span class="o">=</span> \
-            <span 
class="s2">&quot;https://www.googleapis.com/compute/v1/projects/</span><span 
class="si">{}</span><span class="s2">/zones/</span><span 
class="si">{}</span><span class="s2">/machineTypes/</span><span 
class="si">{}</span><span class="s2">&quot;</span><span class="o">.</span><span 
class="n">format</span><span class="p">(</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">zone</span><span 
class="p">,</span> <span class="bp">self</span><span class="o">.</span><span 
class="n">master_machine_type</span>
-            <span class="p">)</span>
+            <span 
class="s2">&quot;https://www.googleapis.com/compute/v1/projects/</span><span 
class="si">{}</span><span class="s2">/zones/</span><span 
class="si">{}</span><span class="s2">/machineTypes/</span><span 
class="si">{}</span><span class="s2">&quot;</span>\
+            <span class="o">.</span><span class="n">format</span><span 
class="p">(</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">zone</span><span 
class="p">,</span> <span class="bp">self</span><span class="o">.</span><span 
class="n">master_machine_type</span><span class="p">)</span>
         <span class="n">worker_type_uri</span> <span class="o">=</span> \
-            <span 
class="s2">&quot;https://www.googleapis.com/compute/v1/projects/</span><span 
class="si">{}</span><span class="s2">/zones/</span><span 
class="si">{}</span><span class="s2">/machineTypes/</span><span 
class="si">{}</span><span class="s2">&quot;</span><span class="o">.</span><span 
class="n">format</span><span class="p">(</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">zone</span><span 
class="p">,</span> <span class="bp">self</span><span class="o">.</span><span 
class="n">worker_machine_type</span>
-            <span class="p">)</span>
+            <span 
class="s2">&quot;https://www.googleapis.com/compute/v1/projects/</span><span 
class="si">{}</span><span class="s2">/zones/</span><span 
class="si">{}</span><span class="s2">/machineTypes/</span><span 
class="si">{}</span><span class="s2">&quot;</span>\
+            <span class="o">.</span><span class="n">format</span><span 
class="p">(</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">zone</span><span 
class="p">,</span> <span class="bp">self</span><span class="o">.</span><span 
class="n">worker_machine_type</span><span class="p">)</span>
+
         <span class="n">cluster_data</span> <span class="o">=</span> <span 
class="p">{</span>
             <span class="s1">&#39;projectId&#39;</span><span 
class="p">:</span> <span class="bp">self</span><span class="o">.</span><span 
class="n">project_id</span><span class="p">,</span>
             <span class="s1">&#39;clusterName&#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>
@@ -468,7 +463,7 @@
         <span class="c1"># [a-z]([-a-z0-9]*[a-z0-9])? (current airflow version 
string follows</span>
         <span class="c1"># semantic versioning spec: x.y.z).</span>
         <span class="n">cluster_data</span><span class="p">[</span><span 
class="s1">&#39;labels&#39;</span><span class="p">]</span><span 
class="o">.</span><span class="n">update</span><span class="p">({</span><span 
class="s1">&#39;airflow-version&#39;</span><span class="p">:</span>
-                                       <span class="s1">&#39;v&#39;</span> 
<span class="o">+</span> <span class="n">version</span><span 
class="o">.</span><span class="n">replace</span><span class="p">(</span><span 
class="s1">&#39;.&#39;</span><span class="p">,</span> <span 
class="s1">&#39;-&#39;</span><span class="p">)</span><span 
class="o">.</span><span class="n">replace</span><span class="p">(</span><span 
class="s1">&#39;+&#39;</span><span class="p">,</span><span 
class="s1">&#39;-&#39;</span><span class="p">)})</span>
+                                       <span class="s1">&#39;v&#39;</span> 
<span class="o">+</span> <span class="n">version</span><span 
class="o">.</span><span class="n">replace</span><span class="p">(</span><span 
class="s1">&#39;.&#39;</span><span class="p">,</span> <span 
class="s1">&#39;-&#39;</span><span class="p">)</span><span 
class="o">.</span><span class="n">replace</span><span class="p">(</span><span 
class="s1">&#39;+&#39;</span><span class="p">,</span> <span 
class="s1">&#39;-&#39;</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&#39;</span><span class="p">][</span><span 
class="s1">&#39;configBucket&#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">metadata</span><span class="p">:</span>
@@ -476,7 +471,13 @@
         <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;config&#39;</span><span class="p">][</span><span 
class="s1">&#39;gceClusterConfig&#39;</span><span class="p">][</span><span 
class="s1">&#39;networkUri&#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;config&#39;</span><span class="p">][</span><span 
class="s1">&#39;gceClusterConfig&#39;</span><span class="p">][</span><span 
class="s1">&#39;subnetworkUri&#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="n">cluster_data</span><span class="p">[</span><span 
class="s1">&#39;config&#39;</span><span class="p">][</span><span 
class="s1">&#39;gceClusterConfig&#39;</span><span class="p">][</span><span 
class="s1">&#39;subnetworkUri&#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&quot;</span>
+                                       <span class="s2">&quot; 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;config&#39;</span><span class="p">][</span><span 
class="s1">&#39;gceClusterConfig&#39;</span><span class="p">][</span><span 
class="s1">&#39;internalIpOnly&#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;config&#39;</span><span class="p">][</span><span 
class="s1">&#39;gceClusterConfig&#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">image_version</span><span class="p">:</span>
@@ -503,10 +504,10 @@
             <span class="n">cluster_data</span><span class="p">[</span><span 
class="s1">&#39;config&#39;</span><span class="p">][</span><span 
class="s1">&#39;initializationActions&#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">service_account</span><span class="p">:</span>
             <span class="n">cluster_data</span><span class="p">[</span><span 
class="s1">&#39;config&#39;</span><span class="p">][</span><span 
class="s1">&#39;gceClusterConfig&#39;</span><span class="p">][</span><span 
class="s1">&#39;serviceAccount&#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="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;config&#39;</span><span class="p">][</span><span 
class="s1">&#39;gceClusterConfig&#39;</span><span class="p">][</span><span 
class="s1">&#39;serviceAccountScopes&#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="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">execute</span><span 
class="p">(</span><span class="bp">self</span><span class="p">,</span> <span 
class="n">context</span><span class="p">):</span>
@@ -539,7 +540,7 @@
                 <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;Cluster </span><span 
class="si">{}</span><span class="s1"> already exists... Checking 
status...&#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="p">)</span>
                 <span class="bp">self</span><span class="o">.</span><span 
class="n">_wait_for_done</span><span class="p">(</span><span 
class="n">service</span><span class="p">)</span>
                 <span class="k">return</span> <span class="kc">True</span>
             <span class="k">else</span><span class="p">:</span>
@@ -548,17 +549,165 @@
         <span class="bp">self</span><span class="o">.</span><span 
class="n">_wait_for_done</span><span class="p">(</span><span 
class="n">service</span><span class="p">)</span></div>
 
 
+<div class="viewcode-block" id="DataprocClusterScaleOperator"><a 
class="viewcode-back" 
href="../../../../integration.html#airflow.contrib.operators.dataproc_operator.DataprocClusterScaleOperator">[docs]</a><span
 class="k">class</span> <span 
class="nc">DataprocClusterScaleOperator</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">    :type cluster_name: string</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">    :type project_id: string</span>
+<span class="sd">    :param region: The region for the dataproc cluster. 
(templated)</span>
+<span class="sd">    :type region: string</span>
+<span class="sd">    :param gcp_conn_id: The connection ID to use connecting 
to Google Cloud Platform.</span>
+<span class="sd">    :type gcp_conn_id: string</span>
+<span class="sd">    :param num_workers: The new number of workers</span>
+<span class="sd">    :type num_workers: int</span>
+<span class="sd">    :param num_preemptible_workers: The new number of 
preemptible workers</span>
+<span class="sd">    :type num_preemptible_workers: int</span>
+<span class="sd">    :param graceful_decommission_timeout: Timeout for 
graceful YARN decomissioning.</span>
+<span class="sd">        Maximum value is 1d</span>
+<span class="sd">    :type graceful_decommission_timeout: string</span>
+<span class="sd">    :param delegate_to: The account to impersonate, if 
any.</span>
+<span class="sd">        For this to work, the service account making the 
request must have domain-wide</span>
+<span class="sd">        delegation enabled.</span>
+<span class="sd">    :type delegate_to: string</span>
+<span class="sd">    &quot;&quot;&quot;</span>
+
+    <span class="n">template_fields</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="nd">@apply_defaults</span>
+    <span class="k">def</span> <span class="nf">__init__</span><span 
class="p">(</span><span class="bp">self</span><span class="p">,</span>
+                 <span class="n">cluster_name</span><span class="p">,</span>
+                 <span class="n">project_id</span><span class="p">,</span>
+                 <span class="n">region</span><span class="o">=</span><span 
class="s1">&#39;global&#39;</span><span class="p">,</span>
+                 <span class="n">gcp_conn_id</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="o">=</span><span class="kc">None</span><span class="p">,</span>
+                 <span class="n">num_workers</span><span 
class="o">=</span><span class="mi">2</span><span class="p">,</span>
+                 <span class="n">num_preemptible_workers</span><span 
class="o">=</span><span class="mi">0</span><span class="p">,</span>
+                 <span class="n">graceful_decommission_timeout</span><span 
class="o">=</span><span class="kc">None</span><span class="p">,</span>
+                 <span class="o">*</span><span class="n">args</span><span 
class="p">,</span>
+                 <span class="o">**</span><span class="n">kwargs</span><span 
class="p">):</span>
+        <span class="nb">super</span><span class="p">(</span><span 
class="n">DataprocClusterScaleOperator</span><span class="p">,</span> <span 
class="bp">self</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">args</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">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">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">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">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="c1"># Optional</span>
+        <span class="bp">self</span><span class="o">.</span><span 
class="n">optional_arguments</span> <span class="o">=</span> <span 
class="p">{}</span>
+        <span class="k">if</span> <span 
class="n">graceful_decommission_timeout</span><span class="p">:</span>
+            <span class="bp">self</span><span class="o">.</span><span 
class="n">optional_arguments</span><span class="p">[</span><span 
class="s1">&#39;gracefulDecommissionTimeout&#39;</span><span class="p">]</span> 
<span class="o">=</span> \
+                <span class="bp">self</span><span class="o">.</span><span 
class="n">_get_graceful_decommission_timeout</span><span class="p">(</span>
+                    <span class="n">graceful_decommission_timeout</span><span 
class="p">)</span>
+
+    <span class="k">def</span> <span class="nf">_wait_for_done</span><span 
class="p">(</span><span class="bp">self</span><span class="p">,</span> <span 
class="n">service</span><span class="p">,</span> <span 
class="n">operation_name</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="mi">15</span><span 
class="p">)</span>
+        <span class="k">while</span> <span class="kc">True</span><span 
class="p">:</span>
+            <span class="k">try</span><span class="p">:</span>
+                <span class="n">response</span> <span class="o">=</span> <span 
class="n">service</span><span class="o">.</span><span 
class="n">projects</span><span class="p">()</span><span class="o">.</span><span 
class="n">regions</span><span class="p">()</span><span class="o">.</span><span 
class="n">operations</span><span class="p">()</span><span 
class="o">.</span><span class="n">get</span><span class="p">(</span>
+                    <span class="n">name</span><span class="o">=</span><span 
class="n">operation_name</span>
+                <span class="p">)</span><span class="o">.</span><span 
class="n">execute</span><span class="p">()</span>
+
+                <span class="k">if</span> <span 
class="s1">&#39;done&#39;</span> <span class="ow">in</span> <span 
class="n">response</span> <span class="ow">and</span> <span 
class="n">response</span><span class="p">[</span><span 
class="s1">&#39;done&#39;</span><span class="p">]:</span>
+                    <span class="k">if</span> <span 
class="s1">&#39;error&#39;</span> <span class="ow">in</span> <span 
class="n">response</span><span class="p">:</span>
+                        <span class="k">raise</span> <span 
class="ne">Exception</span><span class="p">(</span><span 
class="nb">str</span><span class="p">(</span><span 
class="n">response</span><span class="p">[</span><span 
class="s1">&#39;error&#39;</span><span class="p">]))</span>
+                    <span class="k">else</span><span class="p">:</span>
+                        <span class="k">return</span>
+                <span class="n">time</span><span class="o">.</span><span 
class="n">sleep</span><span class="p">(</span><span class="mi">15</span><span 
class="p">)</span>
+            <span class="k">except</span> <span class="n">HttpError</span> 
<span class="k">as</span> <span class="n">e</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">error</span><span 
class="p">(</span><span class="s2">&quot;Operation not found.&quot;</span><span 
class="p">)</span>
+                <span class="k">raise</span> <span class="n">e</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="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;workerConfig&#39;</span><span 
class="p">:</span> <span class="p">{</span>
+                    <span class="s1">&#39;numInstances&#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;secondaryWorkerConfig&#39;</span><span 
class="p">:</span> <span class="p">{</span>
+                    <span class="s1">&#39;numInstances&#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="k">def</span> <span 
class="nf">_get_graceful_decommission_timeout</span><span 
class="p">(</span><span class="bp">self</span><span class="p">,</span> <span 
class="n">timeout</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+)(s|m|h|d)$&quot;</span><span class="p">,</span> <span 
class="n">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="k">return</span> <span class="n">timeout</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="k">return</span> <span 
class="s2">&quot;</span><span class="si">{}</span><span 
class="s2">s&quot;</span><span class="o">.</span><span 
class="n">format</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">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="k">return</span> <span 
class="s2">&quot;</span><span class="si">{}</span><span 
class="s2">s&quot;</span><span class="o">.</span><span 
class="n">format</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">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="k">return</span> <span 
class="s2">&quot;</span><span class="si">{}</span><span 
class="s2">s&quot;</span><span class="o">.</span><span 
class="n">format</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">seconds</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">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="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">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">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">service</span> <span class="o">=</span> <span 
class="n">hook</span><span class="o">.</span><span 
class="n">get_conn</span><span class="p">()</span>
+
+        <span class="n">update_mask</span> <span class="o">=</span> <span 
class="s2">&quot;config.worker_config.num_instances,&quot;</span> \
+                      <span class="o">+</span> <span 
class="s2">&quot;config.secondary_worker_config.num_instances&quot;</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">response</span> <span class="o">=</span> <span 
class="n">service</span><span class="o">.</span><span 
class="n">projects</span><span class="p">()</span><span class="o">.</span><span 
class="n">regions</span><span class="p">()</span><span class="o">.</span><span 
class="n">clusters</span><span class="p">()</span><span class="o">.</span><span 
class="n">patch</span><span class="p">(</span>
+            <span class="n">projectId</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">clusterName</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">updateMask</span><span class="o">=</span><span 
class="n">update_mask</span><span class="p">,</span>
+            <span class="n">body</span><span class="o">=</span><span 
class="n">scaling_cluster_data</span><span class="p">,</span>
+            <span class="o">**</span><span class="bp">self</span><span 
class="o">.</span><span class="n">optional_arguments</span>
+        <span class="p">)</span><span class="o">.</span><span 
class="n">execute</span><span class="p">()</span>
+        <span class="n">operation_name</span> <span class="o">=</span> <span 
class="n">response</span><span class="p">[</span><span 
class="s1">&#39;name&#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;Cluster scale operation name: 
</span><span class="si">%s</span><span class="s2">&quot;</span><span 
class="p">,</span> <span class="n">operation_name</span><span class="p">)</span>
+        <span class="bp">self</span><span class="o">.</span><span 
class="n">_wait_for_done</span><span class="p">(</span><span 
class="n">service</span><span class="p">,</span> <span 
class="n">operation_name</span><span class="p">)</span></div>
+
+
 <div class="viewcode-block" id="DataprocClusterDeleteOperator"><a 
class="viewcode-back" 
href="../../../../integration.html#airflow.contrib.operators.dataproc_operator.DataprocClusterDeleteOperator">[docs]</a><span
 class="k">class</span> <span 
class="nc">DataprocClusterDeleteOperator</span><span class="p">(</span><span 
class="n">BaseOperator</span><span class="p">):</span>
     <span class="sd">&quot;&quot;&quot;</span>
 <span class="sd">    Delete a cluster on Google Cloud Dataproc. The operator 
will wait until the</span>
 <span class="sd">    cluster is destroyed.</span>
 
-<span class="sd">    :param cluster_name: The name of the cluster to 
create.</span>
+<span class="sd">    :param cluster_name: The name of the cluster to create. 
(templated)</span>
 <span class="sd">    :type cluster_name: string</span>
 <span class="sd">    :param project_id: The ID of the google cloud project in 
which</span>
-<span class="sd">        the cluster runs</span>
+<span class="sd">        the cluster runs. (templated)</span>
 <span class="sd">    :type project_id: string</span>
-<span class="sd">    :param region: leave as &#39;global&#39;, might become 
relevant in the future</span>
+<span class="sd">    :param region: leave as &#39;global&#39;, might become 
relevant in the future. (templated)</span>
 <span class="sd">    :type region: string</span>
 <span class="sd">    :param gcp_conn_id: The connection ID to use connecting 
to Google Cloud Platform.</span>
 <span class="sd">    :type gcp_conn_id: string</span>
@@ -653,17 +802,19 @@
 <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 file 
(pg or pig extension).</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">    :type query: string</span>
 <span class="sd">    :param query_uri: The uri of a pig script on Cloud 
Storage.</span>
 <span class="sd">    :type query_uri: string</span>
-<span class="sd">    :param variables: Map of named parameters for the 
query.</span>
+<span class="sd">    :param variables: Map of named parameters for the query. 
(templated)</span>
 <span class="sd">    :type variables: dict</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 job_name: The job name used in the DataProc 
cluster. This</span>
+<span class="sd">        name by default is the task_id appended with the 
execution data, but can</span>
+<span class="sd">        be templated. The name will always be appended with a 
random number to</span>
+<span class="sd">        avoid name clashes. (templated)</span>
 <span class="sd">    :type job_name: string</span>
-<span class="sd">    :param cluster_name: The name of the DataProc 
cluster.</span>
+<span class="sd">    :param cluster_name: The name of the DataProc cluster. 
(templated)</span>
 <span class="sd">    :type cluster_name: string</span>
 <span class="sd">    :param dataproc_pig_properties: Map for the Pig 
properties. Ideal to put in</span>
 <span class="sd">        default arguments</span>
@@ -814,17 +965,18 @@
     <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).</span>
+<span class="sd">    :param query: The query or reference to the query file (q 
extension). (templated)</span>
 <span class="sd">    :type query: string</span>
 <span class="sd">    :param query_uri: The uri of a spark sql script on Cloud 
Storage.</span>
 <span class="sd">    :type query_uri: string</span>
-<span class="sd">    :param variables: Map of named parameters for the 
query.</span>
+<span class="sd">    :param variables: Map of named parameters for the query. 
(templated)</span>
 <span class="sd">    :type variables: dict</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 job_name: The job name used in the DataProc 
cluster. This</span>
+<span class="sd">        name by default is the task_id appended with the 
execution data, but can</span>
+<span class="sd">        be templated. The name will always be appended with a 
random number to</span>
+<span class="sd">        avoid name clashes. (templated)</span>
 <span class="sd">    :type job_name: string</span>
-<span class="sd">    :param cluster_name: The name of the DataProc 
cluster.</span>
+<span class="sd">    :param cluster_name: The name of the DataProc cluster. 
(templated)</span>
 <span class="sd">    :type cluster_name: string</span>
 <span class="sd">    :param dataproc_spark_properties: Map for the Pig 
properties. Ideal to put in</span>
 <span class="sd">        default arguments</span>
@@ -901,18 +1053,19 @@
 <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">    :type main_class: string</span>
-<span class="sd">    :param arguments: Arguments for the job.</span>
+<span class="sd">    :param arguments: Arguments for the job. 
(templated)</span>
 <span class="sd">    :type arguments: list</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">    :type archives: list</span>
 <span class="sd">    :param files: List of files to be copied to the working 
directory</span>
 <span class="sd">    :type files: list</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 job_name: The job name used in the DataProc 
cluster. This</span>
+<span class="sd">        name by default is the task_id appended with the 
execution data, but can</span>
+<span class="sd">        be templated. The name will always be appended with a 
random number to</span>
+<span class="sd">        avoid name clashes. (templated)</span>
 <span class="sd">    :type job_name: string</span>
-<span class="sd">    :param cluster_name: The name of the DataProc 
cluster.</span>
+<span class="sd">    :param cluster_name: The name of the DataProc cluster. 
(templated)</span>
 <span class="sd">    :type cluster_name: string</span>
 <span class="sd">    :param dataproc_spark_properties: Map for the Pig 
properties. Ideal to put in</span>
 <span class="sd">        default arguments</span>
@@ -991,18 +1144,19 @@
 <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">    :type main_class: string</span>
-<span class="sd">    :param arguments: Arguments for the job.</span>
+<span class="sd">    :param arguments: Arguments for the job. 
(templated)</span>
 <span class="sd">    :type arguments: list</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">    :type archives: list</span>
 <span class="sd">    :param files: List of files to be copied to the working 
directory</span>
 <span class="sd">    :type files: list</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 job_name: The job name used in the DataProc 
cluster. This</span>
+<span class="sd">        name by default is the task_id appended with the 
execution data, but can</span>
+<span class="sd">        be templated. The name will always be appended with a 
random number to</span>
+<span class="sd">        avoid name clashes. (templated)</span>
 <span class="sd">    :type job_name: string</span>
-<span class="sd">    :param cluster_name: The name of the DataProc 
cluster.</span>
+<span class="sd">    :param cluster_name: The name of the DataProc cluster. 
(templated)</span>
 <span class="sd">    :type cluster_name: string</span>
 <span class="sd">    :param dataproc_hadoop_properties: Map for the Pig 
properties. Ideal to put in</span>
 <span class="sd">        default arguments</span>
@@ -1078,7 +1232,7 @@
 <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.</span>
 <span class="sd">    :type main: string</span>
-<span class="sd">    :param arguments: Arguments for the job.</span>
+<span class="sd">    :param arguments: Arguments for the job. 
(templated)</span>
 <span class="sd">    :type arguments: list</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>
@@ -1088,9 +1242,10 @@
 <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">    :type pyfiles: list</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 job_name: The job name used in the DataProc 
cluster. This</span>
+<span class="sd">        name by default is the task_id appended with the 
execution data, but can</span>
+<span class="sd">        be templated. The name will always be appended with a 
random number to</span>
+<span class="sd">        avoid name clashes. (templated)</span>
 <span class="sd">    :type job_name: string</span>
 <span class="sd">    :param cluster_name: The name of the DataProc 
cluster.</span>
 <span class="sd">    :type cluster_name: string</span>
@@ -1222,7 +1377,7 @@
         <span class="p">)</span>
 
     <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="bp">self</span><span class="o">.</span><span 
class="n">hook</span><span class="o">.</span><span class="k">await</span><span 
class="p">(</span><span class="bp">self</span><span class="o">.</span><span 
class="n">start</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</span><span 
class="p">(</span><span class="bp">self</span><span class="o">.</span><span 
class="n">start</span><span class="p">())</span>
 
     <span class="k">def</span> <span class="nf">start</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="k">raise</span> <span 
class="n">AirflowException</span><span class="p">(</span><span 
class="s1">&#39;plese start a workflow operation&#39;</span><span 
class="p">)</span></div>
@@ -1237,7 +1392,7 @@
 <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.</span>
+<span class="sd">    :param template_id: The id of the template. 
(templated)</span>
 <span class="sd">    :type template_id: string</span>
 <span class="sd">    :param project_id: The ID of the google cloud project in 
which</span>
 <span class="sd">        the template runs</span>
@@ -1281,7 +1436,7 @@
 <span class="sd">        Please refer to:</span>
 <span class="sd">        
https://cloud.google.com/dataproc/docs/reference/rest/v1beta2/projects.regions.workflowTemplates/instantiateInline</span>
 
-<span class="sd">    :param template: The template contents.</span>
+<span class="sd">    :param template: The template contents. (templated)</span>
 <span class="sd">    :type template: map</span>
 <span class="sd">    :param project_id: The ID of the google cloud project in 
which</span>
 <span class="sd">        the template runs</span>
@@ -1316,9 +1471,7 @@
 </pre></div>
 
            </div>
-           <div class="articleComments">
-            
-           </div>
+           
           </div>
           <footer>
   
@@ -1330,7 +1483,7 @@
 
     </p>
   </div>
-  Built with <a href="http://sphinx-doc.org/";>Sphinx</a> using a <a 
href="https://github.com/snide/sphinx_rtd_theme";>theme</a> provided by <a 
href="https://readthedocs.org";>Read the Docs</a>. 
+  Built with <a href="http://sphinx-doc.org/";>Sphinx</a> using a <a 
href="https://github.com/rtfd/sphinx_rtd_theme";>theme</a> provided by <a 
href="https://readthedocs.org";>Read the Docs</a>. 
 
 </footer>
 
@@ -1349,6 +1502,7 @@
         var DOCUMENTATION_OPTIONS = {
             URL_ROOT:'../../../../',
             VERSION:'',
+            LANGUAGE:'None',
             COLLAPSE_INDEX:false,
             FILE_SUFFIX:'.html',
             HAS_SOURCE:  true,
@@ -1361,19 +1515,13 @@
 
   
 
-  
-  
-    <script type="text/javascript" 
src="../../../../_static/js/theme.js"></script>
-  
+  <script type="text/javascript" 
src="../../../../_static/js/theme.js"></script>
 
-  
-  
   <script type="text/javascript">
       jQuery(function () {
-          SphinxRtdTheme.StickyNav.enable();
+          SphinxRtdTheme.Navigation.enable(true);
       });
-  </script>
-   
+  </script> 
 
 </body>
 </html>
\ No newline at end of file

http://git-wip-us.apache.org/repos/asf/incubator-airflow-site/blob/7d4d7628/_modules/airflow/contrib/operators/datastore_export_operator.html
----------------------------------------------------------------------
diff --git a/_modules/airflow/contrib/operators/datastore_export_operator.html 
b/_modules/airflow/contrib/operators/datastore_export_operator.html
index 94e2d02..915c753 100644
--- a/_modules/airflow/contrib/operators/datastore_export_operator.html
+++ b/_modules/airflow/contrib/operators/datastore_export_operator.html
@@ -24,26 +24,17 @@
 
   
 
-  
-  
-    <link rel="stylesheet" href="../../../../_static/css/theme.css" 
type="text/css" />
-  
-
-  
-
-  
-        <link rel="index" title="Index"
-              href="../../../../genindex.html"/>
-        <link rel="search" title="Search" href="../../../../search.html"/>
-    <link rel="top" title="Airflow Documentation" 
href="../../../../index.html"/>
-        <link rel="up" title="Module code" href="../../../index.html"/> 
+  <link rel="stylesheet" href="../../../../_static/css/theme.css" 
type="text/css" />
+  <link rel="stylesheet" href="../../../../_static/pygments.css" 
type="text/css" />
+    <link rel="index" title="Index" href="../../../../genindex.html" />
+    <link rel="search" title="Search" href="../../../../search.html" /> 
 
   
   <script src="../../../../_static/js/modernizr.min.js"></script>
 
 </head>
 
-<body class="wy-body-for-nav" role="document">
+<body class="wy-body-for-nav">
 
    
   <div class="wy-grid-for-nav">
@@ -116,7 +107,7 @@
     <section data-toggle="wy-nav-shift" class="wy-nav-content-wrap">
 
       
-      <nav class="wy-nav-top" role="navigation" aria-label="top navigation">
+      <nav class="wy-nav-top" aria-label="top navigation">
         
           <i data-toggle="wy-nav-top" class="fa fa-bars"></i>
           <a href="../../../../index.html">Airflow</a>
@@ -124,9 +115,10 @@
       </nav>
 
 
-      
       <div class="wy-nav-content">
+        
         <div class="rst-content">
+        
           
 
 
@@ -156,8 +148,6 @@
     
       <li class="wy-breadcrumbs-aside">
         
-            
-        
       </li>
     
   </ul>
@@ -178,9 +168,9 @@
 <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">#</span>
 <span class="c1">#   http://www.apache.org/licenses/LICENSE-2.0</span>
-<span class="c1"># </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>
@@ -206,15 +196,16 @@
 <span class="sd">    :type namespace: str</span>
 <span class="sd">    :param datastore_conn_id: the name of the Datastore 
connection id to use</span>
 <span class="sd">    :type datastore_conn_id: string</span>
-<span class="sd">    :param cloud_storage_conn_id: the name of the cloud 
storage connection id to force-write</span>
-<span class="sd">        backup</span>
+<span class="sd">    :param cloud_storage_conn_id: the name of the cloud 
storage connection id to</span>
+<span class="sd">        force-write backup</span>
 <span class="sd">    :type cloud_storage_conn_id: string</span>
 <span class="sd">    :param delegate_to: The account to impersonate, if 
any.</span>
 <span class="sd">        For this to work, the service account making the 
request must have domain-wide</span>
 <span class="sd">        delegation enabled.</span>
 <span class="sd">    :type delegate_to: string</span>
-<span class="sd">    :param entity_filter: description of what data from the 
project is included in the export,</span>
-<span class="sd">        refer to 
https://cloud.google.com/datastore/docs/reference/rest/Shared.Types/EntityFilter</span>
+<span class="sd">    :param entity_filter: description of what data from the 
project is included in the</span>
+<span class="sd">        export, refer to</span>
+<span class="sd">        
https://cloud.google.com/datastore/docs/reference/rest/Shared.Types/EntityFilter</span>
 <span class="sd">    :type entity_filter: dict</span>
 <span class="sd">    :param labels: client-assigned labels for cloud 
storage</span>
 <span class="sd">    :type labels: dict</span>
@@ -263,7 +254,7 @@
             <span class="k">for</span> <span class="n">o</span> <span 
class="ow">in</span> <span class="n">objects</span><span class="p">:</span>
                 <span class="n">gcs_hook</span><span class="o">.</span><span 
class="n">delete</span><span class="p">(</span><span 
class="bp">self</span><span class="o">.</span><span 
class="n">bucket</span><span class="p">,</span> <span class="n">o</span><span 
class="p">)</span>
 
-        <span class="n">ds_hook</span> <span class="o">=</span> <span 
class="n">DatastoreHook</span><span class="p">(</span><span 
class="bp">self</span><span class="o">.</span><span 
class="n">datastore_conn_id</span><span class="p">,</span><span 
class="bp">self</span><span class="o">.</span><span 
class="n">delegate_to</span><span class="p">)</span>
+        <span class="n">ds_hook</span> <span class="o">=</span> <span 
class="n">DatastoreHook</span><span class="p">(</span><span 
class="bp">self</span><span class="o">.</span><span 
class="n">datastore_conn_id</span><span class="p">,</span> <span 
class="bp">self</span><span class="o">.</span><span 
class="n">delegate_to</span><span class="p">)</span>
         <span class="n">result</span> <span class="o">=</span> <span 
class="n">ds_hook</span><span class="o">.</span><span 
class="n">export_to_storage_bucket</span><span class="p">(</span><span 
class="n">bucket</span><span class="o">=</span><span 
class="bp">self</span><span class="o">.</span><span 
class="n">bucket</span><span class="p">,</span>
                                                   <span 
class="n">namespace</span><span class="o">=</span><span 
class="bp">self</span><span class="o">.</span><span 
class="n">namespace</span><span class="p">,</span>
                                                   <span 
class="n">entity_filter</span><span class="o">=</span><span 
class="bp">self</span><span class="o">.</span><span 
class="n">entity_filter</span><span class="p">,</span>
@@ -281,9 +272,7 @@
 </pre></div>
 
            </div>
-           <div class="articleComments">
-            
-           </div>
+           
           </div>
           <footer>
   
@@ -295,7 +284,7 @@
 
     </p>
   </div>
-  Built with <a href="http://sphinx-doc.org/";>Sphinx</a> using a <a 
href="https://github.com/snide/sphinx_rtd_theme";>theme</a> provided by <a 
href="https://readthedocs.org";>Read the Docs</a>. 
+  Built with <a href="http://sphinx-doc.org/";>Sphinx</a> using a <a 
href="https://github.com/rtfd/sphinx_rtd_theme";>theme</a> provided by <a 
href="https://readthedocs.org";>Read the Docs</a>. 
 
 </footer>
 
@@ -314,6 +303,7 @@
         var DOCUMENTATION_OPTIONS = {
             URL_ROOT:'../../../../',
             VERSION:'',
+            LANGUAGE:'None',
             COLLAPSE_INDEX:false,
             FILE_SUFFIX:'.html',
             HAS_SOURCE:  true,
@@ -326,19 +316,13 @@
 
   
 
-  
-  
-    <script type="text/javascript" 
src="../../../../_static/js/theme.js"></script>
-  
+  <script type="text/javascript" 
src="../../../../_static/js/theme.js"></script>
 
-  
-  
   <script type="text/javascript">
       jQuery(function () {
-          SphinxRtdTheme.StickyNav.enable();
+          SphinxRtdTheme.Navigation.enable(true);
       });
-  </script>
-   
+  </script> 
 
 </body>
 </html>
\ No newline at end of file

http://git-wip-us.apache.org/repos/asf/incubator-airflow-site/blob/7d4d7628/_modules/airflow/contrib/operators/datastore_import_operator.html
----------------------------------------------------------------------
diff --git a/_modules/airflow/contrib/operators/datastore_import_operator.html 
b/_modules/airflow/contrib/operators/datastore_import_operator.html
index 3b6d62c..c7d84b4 100644
--- a/_modules/airflow/contrib/operators/datastore_import_operator.html
+++ b/_modules/airflow/contrib/operators/datastore_import_operator.html
@@ -24,26 +24,17 @@
 
   
 
-  
-  
-    <link rel="stylesheet" href="../../../../_static/css/theme.css" 
type="text/css" />
-  
-
-  
-
-  
-        <link rel="index" title="Index"
-              href="../../../../genindex.html"/>
-        <link rel="search" title="Search" href="../../../../search.html"/>
-    <link rel="top" title="Airflow Documentation" 
href="../../../../index.html"/>
-        <link rel="up" title="Module code" href="../../../index.html"/> 
+  <link rel="stylesheet" href="../../../../_static/css/theme.css" 
type="text/css" />
+  <link rel="stylesheet" href="../../../../_static/pygments.css" 
type="text/css" />
+    <link rel="index" title="Index" href="../../../../genindex.html" />
+    <link rel="search" title="Search" href="../../../../search.html" /> 
 
   
   <script src="../../../../_static/js/modernizr.min.js"></script>
 
 </head>
 
-<body class="wy-body-for-nav" role="document">
+<body class="wy-body-for-nav">
 
    
   <div class="wy-grid-for-nav">
@@ -116,7 +107,7 @@
     <section data-toggle="wy-nav-shift" class="wy-nav-content-wrap">
 
       
-      <nav class="wy-nav-top" role="navigation" aria-label="top navigation">
+      <nav class="wy-nav-top" aria-label="top navigation">
         
           <i data-toggle="wy-nav-top" class="fa fa-bars"></i>
           <a href="../../../../index.html">Airflow</a>
@@ -124,9 +115,10 @@
       </nav>
 
 
-      
       <div class="wy-nav-content">
+        
         <div class="rst-content">
+        
           
 
 
@@ -156,8 +148,6 @@
     
       <li class="wy-breadcrumbs-aside">
         
-            
-        
       </li>
     
   </ul>
@@ -178,9 +168,9 @@
 <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">#</span>
 <span class="c1">#   http://www.apache.org/licenses/LICENSE-2.0</span>
-<span class="c1"># </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>
@@ -206,8 +196,9 @@
 <span class="sd">    :param namespace: optional namespace of the backup 
metadata file in</span>
 <span class="sd">        the specified Cloud Storage bucket.</span>
 <span class="sd">    :type namespace: str</span>
-<span class="sd">    :param entity_filter: description of what data from the 
project is included in the export,</span>
-<span class="sd">        refer to 
https://cloud.google.com/datastore/docs/reference/rest/Shared.Types/EntityFilter</span>
+<span class="sd">    :param entity_filter: description of what data from the 
project is included in</span>
+<span class="sd">        the export, refer to</span>
+<span class="sd">        
https://cloud.google.com/datastore/docs/reference/rest/Shared.Types/EntityFilter</span>
 <span class="sd">    :type entity_filter: dict</span>
 <span class="sd">    :param labels: client-assigned labels for cloud 
storage</span>
 <span class="sd">    :type labels: dict</span>
@@ -269,9 +260,7 @@
 </pre></div>
 
            </div>
-           <div class="articleComments">
-            
-           </div>
+           
           </div>
           <footer>
   
@@ -283,7 +272,7 @@
 
     </p>
   </div>
-  Built with <a href="http://sphinx-doc.org/";>Sphinx</a> using a <a 
href="https://github.com/snide/sphinx_rtd_theme";>theme</a> provided by <a 
href="https://readthedocs.org";>Read the Docs</a>. 
+  Built with <a href="http://sphinx-doc.org/";>Sphinx</a> using a <a 
href="https://github.com/rtfd/sphinx_rtd_theme";>theme</a> provided by <a 
href="https://readthedocs.org";>Read the Docs</a>. 
 
 </footer>
 
@@ -302,6 +291,7 @@
         var DOCUMENTATION_OPTIONS = {
             URL_ROOT:'../../../../',
             VERSION:'',
+            LANGUAGE:'None',
             COLLAPSE_INDEX:false,
             FILE_SUFFIX:'.html',
             HAS_SOURCE:  true,
@@ -314,19 +304,13 @@
 
   
 
-  
-  
-    <script type="text/javascript" 
src="../../../../_static/js/theme.js"></script>
-  
+  <script type="text/javascript" 
src="../../../../_static/js/theme.js"></script>
 
-  
-  
   <script type="text/javascript">
       jQuery(function () {
-          SphinxRtdTheme.StickyNav.enable();
+          SphinxRtdTheme.Navigation.enable(true);
       });
-  </script>
-   
+  </script> 
 
 </body>
 </html>
\ No newline at end of file

http://git-wip-us.apache.org/repos/asf/incubator-airflow-site/blob/7d4d7628/_modules/airflow/contrib/operators/discord_webhook_operator.html
----------------------------------------------------------------------
diff --git a/_modules/airflow/contrib/operators/discord_webhook_operator.html 
b/_modules/airflow/contrib/operators/discord_webhook_operator.html
index 8aa057e..3122f2a 100644
--- a/_modules/airflow/contrib/operators/discord_webhook_operator.html
+++ b/_modules/airflow/contrib/operators/discord_webhook_operator.html
@@ -24,26 +24,17 @@
 
   
 
-  
-  
-    <link rel="stylesheet" href="../../../../_static/css/theme.css" 
type="text/css" />
-  
-
-  
-
-  
-        <link rel="index" title="Index"
-              href="../../../../genindex.html"/>
-        <link rel="search" title="Search" href="../../../../search.html"/>
-    <link rel="top" title="Airflow Documentation" 
href="../../../../index.html"/>
-        <link rel="up" title="Module code" href="../../../index.html"/> 
+  <link rel="stylesheet" href="../../../../_static/css/theme.css" 
type="text/css" />
+  <link rel="stylesheet" href="../../../../_static/pygments.css" 
type="text/css" />
+    <link rel="index" title="Index" href="../../../../genindex.html" />
+    <link rel="search" title="Search" href="../../../../search.html" /> 
 
   
   <script src="../../../../_static/js/modernizr.min.js"></script>
 
 </head>
 
-<body class="wy-body-for-nav" role="document">
+<body class="wy-body-for-nav">
 
    
   <div class="wy-grid-for-nav">
@@ -116,7 +107,7 @@
     <section data-toggle="wy-nav-shift" class="wy-nav-content-wrap">
 
       
-      <nav class="wy-nav-top" role="navigation" aria-label="top navigation">
+      <nav class="wy-nav-top" aria-label="top navigation">
         
           <i data-toggle="wy-nav-top" class="fa fa-bars"></i>
           <a href="../../../../index.html">Airflow</a>
@@ -124,9 +115,10 @@
       </nav>
 
 
-      
       <div class="wy-nav-content">
+        
         <div class="rst-content">
+        
           
 
 
@@ -156,8 +148,6 @@
     
       <li class="wy-breadcrumbs-aside">
         
-            
-        
       </li>
     
   </ul>
@@ -178,9 +168,9 @@
 <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">#</span>
 <span class="c1">#   http://www.apache.org/licenses/LICENSE-2.0</span>
-<span class="c1"># </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>
@@ -212,9 +202,9 @@
 <span class="sd">                             
&quot;webhooks/{webhook.id}/{webhook.token}&quot;</span>
 <span class="sd">    :type webhook_endpoint: str</span>
 <span class="sd">    :param message: The message you want to send to your 
Discord channel</span>
-<span class="sd">                    (max 2000 characters)</span>
+<span class="sd">                    (max 2000 characters). (templated)</span>
 <span class="sd">    :type message: str</span>
-<span class="sd">    :param username: Override the default username of the 
webhook</span>
+<span class="sd">    :param username: Override the default username of the 
webhook. (templated)</span>
 <span class="sd">    :type username: str</span>
 <span class="sd">    :param avatar_url: Override the default avatar of the 
webhook</span>
 <span class="sd">    :type avatar_url: str</span>
@@ -270,9 +260,7 @@
 </pre></div>
 
            </div>
-           <div class="articleComments">
-            
-           </div>
+           
           </div>
           <footer>
   
@@ -284,7 +272,7 @@
 
     </p>
   </div>
-  Built with <a href="http://sphinx-doc.org/";>Sphinx</a> using a <a 
href="https://github.com/snide/sphinx_rtd_theme";>theme</a> provided by <a 
href="https://readthedocs.org";>Read the Docs</a>. 
+  Built with <a href="http://sphinx-doc.org/";>Sphinx</a> using a <a 
href="https://github.com/rtfd/sphinx_rtd_theme";>theme</a> provided by <a 
href="https://readthedocs.org";>Read the Docs</a>. 
 
 </footer>
 
@@ -303,6 +291,7 @@
         var DOCUMENTATION_OPTIONS = {
             URL_ROOT:'../../../../',
             VERSION:'',
+            LANGUAGE:'None',
             COLLAPSE_INDEX:false,
             FILE_SUFFIX:'.html',
             HAS_SOURCE:  true,
@@ -315,19 +304,13 @@
 
   
 
-  
-  
-    <script type="text/javascript" 
src="../../../../_static/js/theme.js"></script>
-  
+  <script type="text/javascript" 
src="../../../../_static/js/theme.js"></script>
 
-  
-  
   <script type="text/javascript">
       jQuery(function () {
-          SphinxRtdTheme.StickyNav.enable();
+          SphinxRtdTheme.Navigation.enable(true);
       });
-  </script>
-   
+  </script> 
 
 </body>
 </html>
\ No newline at end of file

http://git-wip-us.apache.org/repos/asf/incubator-airflow-site/blob/7d4d7628/_modules/airflow/contrib/operators/druid_operator.html
----------------------------------------------------------------------
diff --git a/_modules/airflow/contrib/operators/druid_operator.html 
b/_modules/airflow/contrib/operators/druid_operator.html
new file mode 100644
index 0000000..dd9de81
--- /dev/null
+++ b/_modules/airflow/contrib/operators/druid_operator.html
@@ -0,0 +1,280 @@
+
+
+<!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]-->
+<head>
+  <meta charset="utf-8">
+  
+  <meta name="viewport" content="width=device-width, initial-scale=1.0">
+  
+  <title>airflow.contrib.operators.druid_operator &mdash; Airflow 
Documentation</title>
+  
+
+  
+  
+  
+  
+
+  
+
+  
+  
+    
+
+  
+
+  <link rel="stylesheet" href="../../../../_static/css/theme.css" 
type="text/css" />
+  <link rel="stylesheet" href="../../../../_static/pygments.css" 
type="text/css" />
+    <link rel="index" title="Index" href="../../../../genindex.html" />
+    <link rel="search" title="Search" href="../../../../search.html" /> 
+
+  
+  <script src="../../../../_static/js/modernizr.min.js"></script>
+
+</head>
+
+<body class="wy-body-for-nav">
+
+   
+  <div class="wy-grid-for-nav">
+
+    
+    <nav data-toggle="wy-nav-shift" class="wy-nav-side">
+      <div class="wy-side-scroll">
+        <div class="wy-side-nav-search">
+          
+
+          
+            <a href="../../../../index.html" class="icon icon-home"> Airflow
+          
+
+          
+          </a>
+
+          
+            
+            
+          
+
+          
+<div role="search">
+  <form id="rtd-search-form" class="wy-form" action="../../../../search.html" 
method="get">
+    <input type="text" name="q" placeholder="Search docs" />
+    <input type="hidden" name="check_keywords" value="yes" />
+    <input type="hidden" name="area" value="default" />
+  </form>
+</div>
+
+          
+        </div>
+
+        <div class="wy-menu wy-menu-vertical" data-spy="affix" 
role="navigation" aria-label="main navigation">
+          
+            
+            
+              
+            
+            
+              <ul>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../project.html">Project</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../license.html">License</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../start.html">Quick Start</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../installation.html">Installation</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../tutorial.html">Tutorial</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../howto/index.html">How-to Guides</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../ui.html">UI / Screenshots</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../concepts.html">Concepts</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../profiling.html">Data Profiling</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../cli.html">Command Line Interface</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../scheduler.html">Scheduling &amp; Triggers</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../plugins.html">Plugins</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../security.html">Security</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../timezone.html">Time zones</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../api.html">Experimental Rest API</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../integration.html">Integration</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../lineage.html">Lineage</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../faq.html">FAQ</a></li>
+<li class="toctree-l1"><a class="reference internal" 
href="../../../../code.html">API Reference</a></li>
+</ul>
+
+            
+          
+        </div>
+      </div>
+    </nav>
+
+    <section data-toggle="wy-nav-shift" class="wy-nav-content-wrap">
+
+      
+      <nav class="wy-nav-top" aria-label="top navigation">
+        
+          <i data-toggle="wy-nav-top" class="fa fa-bars"></i>
+          <a href="../../../../index.html">Airflow</a>
+        
+      </nav>
+
+
+      <div class="wy-nav-content">
+        
+        <div class="rst-content">
+        
+          
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+<div role="navigation" aria-label="breadcrumbs navigation">
+
+  <ul class="wy-breadcrumbs">
+    
+      <li><a href="../../../../index.html">Docs</a> &raquo;</li>
+        
+          <li><a href="../../../index.html">Module code</a> &raquo;</li>
+        
+      <li>airflow.contrib.operators.druid_operator</li>
+    
+    
+      <li class="wy-breadcrumbs-aside">
+        
+      </li>
+    
+  </ul>
+
+  
+  <hr/>
+</div>
+          <div role="main" class="document" itemscope="itemscope" 
itemtype="http://schema.org/Article";>
+           <div itemprop="articleBody">
+            
+  <h1>Source code for airflow.contrib.operators.druid_operator</h1><div 
class="highlight"><pre>
+<span></span><span class="c1"># -*- coding: utf-8 -*-</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="kn">import</span> <span class="nn">json</span>
+
+<span class="kn">from</span> <span class="nn">airflow.hooks.druid_hook</span> 
<span class="k">import</span> <span class="n">DruidHook</span>
+<span class="kn">from</span> <span class="nn">airflow.models</span> <span 
class="k">import</span> <span class="n">BaseOperator</span>
+
+
+<div class="viewcode-block" id="DruidOperator"><a class="viewcode-back" 
href="../../../../code.html#airflow.contrib.operators.druid_operator.DruidOperator">[docs]</a><span
 class="k">class</span> <span class="nc">DruidOperator</span><span 
class="p">(</span><span class="n">BaseOperator</span><span class="p">):</span>
+    <span class="sd">&quot;&quot;&quot;</span>
+<span class="sd">    Allows to submit a task directly to druid</span>
+
+<span class="sd">    :param json_index_file: The filepath to the druid index 
specification</span>
+<span class="sd">    :type json_index_file: str</span>
+<span class="sd">    :param druid_ingest_conn_id: The connection id of the 
Druid overlord which</span>
+<span class="sd">        accepts index jobs</span>
+<span class="sd">    :type druid_ingest_conn_id: str</span>
+<span class="sd">    &quot;&quot;&quot;</span>
+    <span class="n">template_fields</span> <span class="o">=</span> <span 
class="p">(</span><span class="s1">&#39;index_spec_str&#39;</span><span 
class="p">,)</span>
+    <span class="n">template_ext</span> <span class="o">=</span> <span 
class="p">(</span><span class="s1">&#39;.json&#39;</span><span 
class="p">,)</span>
+
+    <span class="k">def</span> <span class="nf">__init__</span><span 
class="p">(</span><span class="bp">self</span><span class="p">,</span> <span 
class="n">json_index_file</span><span class="p">,</span>
+                 <span class="n">druid_ingest_conn_id</span><span 
class="o">=</span><span class="s1">&#39;druid_ingest_default&#39;</span><span 
class="p">,</span>
+                 <span class="n">max_ingestion_time</span><span 
class="o">=</span><span class="kc">None</span><span class="p">,</span>
+                 <span class="o">*</span><span class="n">args</span><span 
class="p">,</span> <span class="o">**</span><span class="n">kwargs</span><span 
class="p">):</span>
+        <span class="nb">super</span><span class="p">(</span><span 
class="n">DruidOperator</span><span class="p">,</span> <span 
class="bp">self</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">args</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">conn_id</span> <span class="o">=</span> <span 
class="n">druid_ingest_conn_id</span>
+        <span class="bp">self</span><span class="o">.</span><span 
class="n">max_ingestion_time</span> <span class="o">=</span> <span 
class="n">max_ingestion_time</span>
+
+        <span class="k">with</span> <span class="nb">open</span><span 
class="p">(</span><span class="n">json_index_file</span><span 
class="p">)</span> <span class="k">as</span> <span 
class="n">data_file</span><span class="p">:</span>
+            <span class="n">index_spec</span> <span class="o">=</span> <span 
class="n">json</span><span class="o">.</span><span class="n">load</span><span 
class="p">(</span><span class="n">data_file</span><span class="p">)</span>
+        <span class="bp">self</span><span class="o">.</span><span 
class="n">index_spec_str</span> <span class="o">=</span> <span 
class="n">json</span><span class="o">.</span><span class="n">dumps</span><span 
class="p">(</span>
+            <span class="n">index_spec</span><span class="p">,</span>
+            <span class="n">sort_keys</span><span class="o">=</span><span 
class="kc">True</span><span class="p">,</span>
+            <span class="n">indent</span><span class="o">=</span><span 
class="mi">4</span><span class="p">,</span>
+            <span class="n">separators</span><span class="o">=</span><span 
class="p">(</span><span class="s1">&#39;,&#39;</span><span class="p">,</span> 
<span class="s1">&#39;: &#39;</span><span class="p">)</span>
+        <span class="p">)</span>
+
+    <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">hook</span> <span class="o">=</span> <span 
class="n">DruidHook</span><span class="p">(</span>
+            <span class="n">druid_ingest_conn_id</span><span 
class="o">=</span><span class="bp">self</span><span class="o">.</span><span 
class="n">conn_id</span><span class="p">,</span>
+            <span class="n">max_ingestion_time</span><span 
class="o">=</span><span class="bp">self</span><span class="o">.</span><span 
class="n">max_ingestion_time</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;Sumitting </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">index_spec_str</span><span class="p">)</span>
+        <span class="n">hook</span><span class="o">.</span><span 
class="n">submit_indexing_job</span><span class="p">(</span><span 
class="bp">self</span><span class="o">.</span><span 
class="n">index_spec_str</span><span class="p">)</span></div>
+</pre></div>
+
+           </div>
+           
+          </div>
+          <footer>
+  
+
+  <hr/>
+
+  <div role="contentinfo">
+    <p>
+
+    </p>
+  </div>
+  Built with <a href="http://sphinx-doc.org/";>Sphinx</a> using a <a 
href="https://github.com/rtfd/sphinx_rtd_theme";>theme</a> provided by <a 
href="https://readthedocs.org";>Read the Docs</a>. 
+
+</footer>
+
+        </div>
+      </div>
+
+    </section>
+
+  </div>
+  
+
+
+  
+
+    <script type="text/javascript">
+        var DOCUMENTATION_OPTIONS = {
+            URL_ROOT:'../../../../',
+            VERSION:'',
+            LANGUAGE:'None',
+            COLLAPSE_INDEX:false,
+            FILE_SUFFIX:'.html',
+            HAS_SOURCE:  true,
+            SOURCELINK_SUFFIX: '.txt'
+        };
+    </script>
+      <script type="text/javascript" 
src="../../../../_static/jquery.js"></script>
+      <script type="text/javascript" 
src="../../../../_static/underscore.js"></script>
+      <script type="text/javascript" 
src="../../../../_static/doctools.js"></script>
+
+  
+
+  <script type="text/javascript" 
src="../../../../_static/js/theme.js"></script>
+
+  <script type="text/javascript">
+      jQuery(function () {
+          SphinxRtdTheme.Navigation.enable(true);
+      });
+  </script> 
+
+</body>
+</html>
\ No newline at end of file

Reply via email to