docs-archive/apache-airflow/1.10.2/plugins.html (333 lines of code) (raw):

<!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>Plugins &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" /> <link rel="next" title="Security" href="security.html" /> <link rel="prev" title="Scheduling &amp; Triggers" href="scheduler.html" /> <script src="_static/js/modernizr.min.js"></script> <!-- Matomo --> <script> var _paq = window._paq = window._paq || []; _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 --> <link rel="canonical" href="https://airflow.apache.org/docs/apache-airflow/stable/plugins.html" /> </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 class="version"> 1.10.2 </div> <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 class="current"> <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 current"><a class="current reference internal" href="#">Plugins</a><ul> <li class="toctree-l2"><a class="reference internal" href="#what-for">What for?</a></li> <li class="toctree-l2"><a class="reference internal" href="#why-build-on-top-of-airflow">Why build on top of Airflow?</a></li> <li class="toctree-l2"><a class="reference internal" href="#interface">Interface</a></li> <li class="toctree-l2"><a class="reference internal" href="#example">Example</a></li> <li class="toctree-l2"><a class="reference internal" href="#note-on-role-based-views">Note on role based views</a></li> <li class="toctree-l2"><a class="reference internal" href="#plugins-as-python-packages">Plugins as Python packages</a></li> </ul> </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="metrics.html">Metrics</a></li> <li class="toctree-l1"><a class="reference internal" href="kubernetes.html">Kubernetes</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="changelog.html">Changelog</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>Plugins</li> <li class="wy-breadcrumbs-aside"> <a href="_sources/plugins.rst.txt" rel="nofollow"> View page source</a> </li> </ul> <hr/> </div> <div role="main" class="document" itemscope="itemscope" itemtype="http://schema.org/Article"> <div itemprop="articleBody"> <div class="section" id="plugins"> <h1>Plugins<a class="headerlink" href="#plugins" title="Permalink to this headline">¶</a></h1> <p>Airflow has a simple plugin manager built-in that can integrate external features to its core by simply dropping files in your <code class="docutils literal notranslate"><span class="pre">$AIRFLOW_HOME/plugins</span></code> folder.</p> <p>The python modules in the <code class="docutils literal notranslate"><span class="pre">plugins</span></code> folder get imported, and <strong>hooks</strong>, <strong>operators</strong>, <strong>sensors</strong>, <strong>macros</strong>, <strong>executors</strong> and web <strong>views</strong> get integrated to Airflow’s main collections and become available for use.</p> <div class="section" id="what-for"> <h2>What for?<a class="headerlink" href="#what-for" title="Permalink to this headline">¶</a></h2> <p>Airflow offers a generic toolbox for working with data. Different organizations have different stacks and different needs. Using Airflow plugins can be a way for companies to customize their Airflow installation to reflect their ecosystem.</p> <p>Plugins can be used as an easy way to write, share and activate new sets of features.</p> <p>There’s also a need for a set of more complex applications to interact with different flavors of data and metadata.</p> <p>Examples:</p> <ul class="simple"> <li>A set of tools to parse Hive logs and expose Hive metadata (CPU /IO / phases/ skew /…)</li> <li>An anomaly detection framework, allowing people to collect metrics, set thresholds and alerts</li> <li>An auditing tool, helping understand who accesses what</li> <li>A config-driven SLA monitoring tool, allowing you to set monitored tables and at what time they should land, alert people, and expose visualizations of outages</li> <li>…</li> </ul> </div> <div class="section" id="why-build-on-top-of-airflow"> <h2>Why build on top of Airflow?<a class="headerlink" href="#why-build-on-top-of-airflow" title="Permalink to this headline">¶</a></h2> <p>Airflow has many components that can be reused when building an application:</p> <ul class="simple"> <li>A web server you can use to render your views</li> <li>A metadata database to store your models</li> <li>Access to your databases, and knowledge of how to connect to them</li> <li>An array of workers that your application can push workload to</li> <li>Airflow is deployed, you can just piggy back on its deployment logistics</li> <li>Basic charting capabilities, underlying libraries and abstractions</li> </ul> </div> <div class="section" id="interface"> <h2>Interface<a class="headerlink" href="#interface" title="Permalink to this headline">¶</a></h2> <p>To create a plugin you will need to derive the <code class="docutils literal notranslate"><span class="pre">airflow.plugins_manager.AirflowPlugin</span></code> class and reference the objects you want to plug into Airflow. Here’s what the class you need to derive looks like:</p> <div class="code python highlight-default notranslate"><div class="highlight"><pre><span></span><span class="k">class</span> <span class="nc">AirflowPlugin</span><span class="p">(</span><span class="nb">object</span><span class="p">):</span> <span class="c1"># The name of your plugin (str)</span> <span class="n">name</span> <span class="o">=</span> <span class="kc">None</span> <span class="c1"># A list of class(es) derived from BaseOperator</span> <span class="n">operators</span> <span class="o">=</span> <span class="p">[]</span> <span class="c1"># A list of class(es) derived from BaseSensorOperator</span> <span class="n">sensors</span> <span class="o">=</span> <span class="p">[]</span> <span class="c1"># A list of class(es) derived from BaseHook</span> <span class="n">hooks</span> <span class="o">=</span> <span class="p">[]</span> <span class="c1"># A list of class(es) derived from BaseExecutor</span> <span class="n">executors</span> <span class="o">=</span> <span class="p">[]</span> <span class="c1"># A list of references to inject into the macros namespace</span> <span class="n">macros</span> <span class="o">=</span> <span class="p">[]</span> <span class="c1"># A list of objects created from a class derived</span> <span class="c1"># from flask_admin.BaseView</span> <span class="n">admin_views</span> <span class="o">=</span> <span class="p">[]</span> <span class="c1"># A list of Blueprint object created from flask.Blueprint. For use with the flask_admin based GUI</span> <span class="n">flask_blueprints</span> <span class="o">=</span> <span class="p">[]</span> <span class="c1"># A list of menu links (flask_admin.base.MenuLink). For use with the flask_admin based GUI</span> <span class="n">menu_links</span> <span class="o">=</span> <span class="p">[]</span> <span class="c1"># A list of dictionaries containing FlaskAppBuilder BaseView object and some metadata. See example below</span> <span class="n">appbuilder_views</span> <span class="o">=</span> <span class="p">[]</span> <span class="c1"># A list of dictionaries containing FlaskAppBuilder BaseView object and some metadata. See example below</span> <span class="n">appbuilder_menu_items</span> <span class="o">=</span> <span class="p">[]</span> </pre></div> </div> </div> <div class="section" id="example"> <h2>Example<a class="headerlink" href="#example" title="Permalink to this headline">¶</a></h2> <p>The code below defines a plugin that injects a set of dummy object definitions in Airflow.</p> <div class="code python highlight-default notranslate"><div class="highlight"><pre><span></span><span class="c1"># This is the class you derive to create a plugin</span> <span class="kn">from</span> <span class="nn">airflow.plugins_manager</span> <span class="k">import</span> <span class="n">AirflowPlugin</span> <span class="kn">from</span> <span class="nn">flask</span> <span class="k">import</span> <span class="n">Blueprint</span> <span class="kn">from</span> <span class="nn">flask_admin</span> <span class="k">import</span> <span class="n">BaseView</span><span class="p">,</span> <span class="n">expose</span> <span class="kn">from</span> <span class="nn">flask_admin.base</span> <span class="k">import</span> <span class="n">MenuLink</span> <span class="c1"># Importing base classes that we need to derive</span> <span class="kn">from</span> <span class="nn">airflow.hooks.base_hook</span> <span class="k">import</span> <span class="n">BaseHook</span> <span class="kn">from</span> <span class="nn">airflow.models</span> <span class="k">import</span> <span class="n">BaseOperator</span> <span class="kn">from</span> <span class="nn">airflow.sensors.base_sensor_operator</span> <span class="k">import</span> <span class="n">BaseSensorOperator</span> <span class="kn">from</span> <span class="nn">airflow.executors.base_executor</span> <span class="k">import</span> <span class="n">BaseExecutor</span> <span class="c1"># Will show up under airflow.hooks.test_plugin.PluginHook</span> <span class="k">class</span> <span class="nc">PluginHook</span><span class="p">(</span><span class="n">BaseHook</span><span class="p">):</span> <span class="k">pass</span> <span class="c1"># Will show up under airflow.operators.test_plugin.PluginOperator</span> <span class="k">class</span> <span class="nc">PluginOperator</span><span class="p">(</span><span class="n">BaseOperator</span><span class="p">):</span> <span class="k">pass</span> <span class="c1"># Will show up under airflow.sensors.test_plugin.PluginSensorOperator</span> <span class="k">class</span> <span class="nc">PluginSensorOperator</span><span class="p">(</span><span class="n">BaseSensorOperator</span><span class="p">):</span> <span class="k">pass</span> <span class="c1"># Will show up under airflow.executors.test_plugin.PluginExecutor</span> <span class="k">class</span> <span class="nc">PluginExecutor</span><span class="p">(</span><span class="n">BaseExecutor</span><span class="p">):</span> <span class="k">pass</span> <span class="c1"># Will show up under airflow.macros.test_plugin.plugin_macro</span> <span class="k">def</span> <span class="nf">plugin_macro</span><span class="p">():</span> <span class="k">pass</span> <span class="c1"># Creating a flask admin BaseView</span> <span class="k">class</span> <span class="nc">TestView</span><span class="p">(</span><span class="n">BaseView</span><span class="p">):</span> <span class="nd">@expose</span><span class="p">(</span><span class="s1">&#39;/&#39;</span><span class="p">)</span> <span class="k">def</span> <span class="nf">test</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span> <span class="c1"># in this example, put your test_plugin/test.html template at airflow/plugins/templates/test_plugin/test.html</span> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">render</span><span class="p">(</span><span class="s2">&quot;test_plugin/test.html&quot;</span><span class="p">,</span> <span class="n">content</span><span class="o">=</span><span class="s2">&quot;Hello galaxy!&quot;</span><span class="p">)</span> <span class="n">v</span> <span class="o">=</span> <span class="n">TestView</span><span class="p">(</span><span class="n">category</span><span class="o">=</span><span class="s2">&quot;Test Plugin&quot;</span><span class="p">,</span> <span class="n">name</span><span class="o">=</span><span class="s2">&quot;Test View&quot;</span><span class="p">)</span> <span class="c1"># Creating a flask blueprint to integrate the templates and static folder</span> <span class="n">bp</span> <span class="o">=</span> <span class="n">Blueprint</span><span class="p">(</span> <span class="s2">&quot;test_plugin&quot;</span><span class="p">,</span> <span class="vm">__name__</span><span class="p">,</span> <span class="n">template_folder</span><span class="o">=</span><span class="s1">&#39;templates&#39;</span><span class="p">,</span> <span class="c1"># registers airflow/plugins/templates as a Jinja template folder</span> <span class="n">static_folder</span><span class="o">=</span><span class="s1">&#39;static&#39;</span><span class="p">,</span> <span class="n">static_url_path</span><span class="o">=</span><span class="s1">&#39;/static/test_plugin&#39;</span><span class="p">)</span> <span class="n">ml</span> <span class="o">=</span> <span class="n">MenuLink</span><span class="p">(</span> <span class="n">category</span><span class="o">=</span><span class="s1">&#39;Test Plugin&#39;</span><span class="p">,</span> <span class="n">name</span><span class="o">=</span><span class="s1">&#39;Test Menu Link&#39;</span><span class="p">,</span> <span class="n">url</span><span class="o">=</span><span class="s1">&#39;https://airflow.apache.org/&#39;</span><span class="p">)</span> <span class="c1"># Creating a flask appbuilder BaseView</span> <span class="k">class</span> <span class="nc">TestAppBuilderBaseView</span><span class="p">(</span><span class="n">AppBuilderBaseView</span><span class="p">):</span> <span class="nd">@expose</span><span class="p">(</span><span class="s2">&quot;/&quot;</span><span class="p">)</span> <span class="k">def</span> <span class="nf">test</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">render</span><span class="p">(</span><span class="s2">&quot;test_plugin/test.html&quot;</span><span class="p">,</span> <span class="n">content</span><span class="o">=</span><span class="s2">&quot;Hello galaxy!&quot;</span><span class="p">)</span> <span class="n">v_appbuilder_view</span> <span class="o">=</span> <span class="n">TestAppBuilderBaseView</span><span class="p">()</span> <span class="n">v_appbuilder_package</span> <span class="o">=</span> <span class="p">{</span><span class="s2">&quot;name&quot;</span><span class="p">:</span> <span class="s2">&quot;Test View&quot;</span><span class="p">,</span> <span class="s2">&quot;category&quot;</span><span class="p">:</span> <span class="s2">&quot;Test Plugin&quot;</span><span class="p">,</span> <span class="s2">&quot;view&quot;</span><span class="p">:</span> <span class="n">v_appbuilder_view</span><span class="p">}</span> <span class="c1"># Creating a flask appbuilder Menu Item</span> <span class="n">appbuilder_mitem</span> <span class="o">=</span> <span class="p">{</span><span class="s2">&quot;name&quot;</span><span class="p">:</span> <span class="s2">&quot;Google&quot;</span><span class="p">,</span> <span class="s2">&quot;category&quot;</span><span class="p">:</span> <span class="s2">&quot;Search&quot;</span><span class="p">,</span> <span class="s2">&quot;category_icon&quot;</span><span class="p">:</span> <span class="s2">&quot;fa-th&quot;</span><span class="p">,</span> <span class="s2">&quot;href&quot;</span><span class="p">:</span> <span class="s2">&quot;https://www.google.com&quot;</span><span class="p">}</span> <span class="c1"># Defining the plugin class</span> <span class="k">class</span> <span class="nc">AirflowTestPlugin</span><span class="p">(</span><span class="n">AirflowPlugin</span><span class="p">):</span> <span class="n">name</span> <span class="o">=</span> <span class="s2">&quot;test_plugin&quot;</span> <span class="n">operators</span> <span class="o">=</span> <span class="p">[</span><span class="n">PluginOperator</span><span class="p">]</span> <span class="n">sensors</span> <span class="o">=</span> <span class="p">[</span><span class="n">PluginSensorOperator</span><span class="p">]</span> <span class="n">hooks</span> <span class="o">=</span> <span class="p">[</span><span class="n">PluginHook</span><span class="p">]</span> <span class="n">executors</span> <span class="o">=</span> <span class="p">[</span><span class="n">PluginExecutor</span><span class="p">]</span> <span class="n">macros</span> <span class="o">=</span> <span class="p">[</span><span class="n">plugin_macro</span><span class="p">]</span> <span class="n">admin_views</span> <span class="o">=</span> <span class="p">[</span><span class="n">v</span><span class="p">]</span> <span class="n">flask_blueprints</span> <span class="o">=</span> <span class="p">[</span><span class="n">bp</span><span class="p">]</span> <span class="n">menu_links</span> <span class="o">=</span> <span class="p">[</span><span class="n">ml</span><span class="p">]</span> <span class="n">appbuilder_views</span> <span class="o">=</span> <span class="p">[</span><span class="n">v_appbuilder_package</span><span class="p">]</span> <span class="n">appbuilder_menu_items</span> <span class="o">=</span> <span class="p">[</span><span class="n">appbuilder_mitem</span><span class="p">]</span> </pre></div> </div> </div> <div class="section" id="note-on-role-based-views"> <h2>Note on role based views<a class="headerlink" href="#note-on-role-based-views" title="Permalink to this headline">¶</a></h2> <p>Airflow 1.10 introduced role based views using FlaskAppBuilder. You can configure which UI is used by setting rbac = True. To support plugin views and links for both versions of the UI and maintain backwards compatibility, the fields appbuilder_views and appbuilder_menu_items were added to the AirflowTestPlugin class.</p> </div> <div class="section" id="plugins-as-python-packages"> <h2>Plugins as Python packages<a class="headerlink" href="#plugins-as-python-packages" title="Permalink to this headline">¶</a></h2> <p>It is possible to load plugins via <a href="#id1"><span class="problematic" id="id2">`setuptools' entrypoint&lt;https://packaging.python.org/guides/creating-and-discovering-plugins/#using-package-metadata&gt;`_</span></a> mechanism. To do this link your plugin using an entrypoint in your package. If the package is installed, airflow will automatically load the registered plugins from the entrypoint list.</p> <p>_Note_: Neither the entrypoint name (eg, <cite>my_plugin</cite>) nor the name of the plugin class will contribute towards the module and class name of the plugin itself. The structure is determined by <cite>airflow.plugins_manager.AirflowPlugin.name</cite> and the class name of the plugin component with the pattern <cite>airflow.{component}.{name}.{component_class_name}</cite>.</p> <div class="highlight-python notranslate"><div class="highlight"><pre><span></span><span class="c1"># my_package/my_plugin.py</span> <span class="kn">from</span> <span class="nn">airflow.plugins_manager</span> <span class="kn">import</span> <span class="n">AirflowPlugin</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.hooks.base_hook</span> <span class="kn">import</span> <span class="n">BaseHook</span> <span class="k">class</span> <span class="nc">MyOperator</span><span class="p">(</span><span class="n">BaseOperator</span><span class="p">):</span> <span class="k">pass</span> <span class="k">class</span> <span class="nc">MyHook</span><span class="p">(</span><span class="n">BaseHook</span><span class="p">):</span> <span class="k">pass</span> <span class="k">class</span> <span class="nc">MyAirflowPlugin</span><span class="p">(</span><span class="n">AirflowPlugin</span><span class="p">):</span> <span class="n">name</span> <span class="o">=</span> <span class="s1">&#39;my_namespace&#39;</span> <span class="n">operators</span> <span class="o">=</span> <span class="p">[</span><span class="n">MyOperator</span><span class="p">]</span> <span class="n">hooks</span> <span class="o">=</span> <span class="p">[</span><span class="n">MyHook</span><span class="p">]</span> </pre></div> </div> <div class="highlight-python notranslate"><div class="highlight"><pre><span></span><span class="kn">from</span> <span class="nn">setuptools</span> <span class="kn">import</span> <span class="n">setup</span> <span class="n">setup</span><span class="p">(</span> <span class="n">name</span><span class="o">=</span><span class="s2">&quot;my-package&quot;</span><span class="p">,</span> <span class="o">...</span> <span class="n">entry_points</span> <span class="o">=</span> <span class="p">{</span> <span class="s1">&#39;airflow.plugins&#39;</span><span class="p">:</span> <span class="p">[</span> <span class="s1">&#39;my_plugin = my_package.my_plugin:MyAirflowPlugin&#39;</span> <span class="p">]</span> <span class="p">}</span> <span class="p">)</span> </pre></div> </div> <dl class="docutils"> <dt>This will create a hook, and an operator accessible at:</dt> <dd><ul class="first last simple"> <li><cite>airflow.hooks.my_namespace.MyHook</cite></li> <li><cite>airflow.operators.my_namespace.MyOperator</cite></li> </ul> </dd> </dl> </div> </div> </div> </div> <footer> <div class="rst-footer-buttons" role="navigation" aria-label="footer navigation"> <a href="security.html" class="btn btn-neutral float-right" title="Security" accesskey="n" rel="next">Next <span class="fa fa-arrow-circle-right"></span></a> <a href="scheduler.html" class="btn btn-neutral" title="Scheduling &amp; Triggers" accesskey="p" rel="prev"><span class="fa fa-arrow-circle-left"></span> Previous</a> </div> <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" id="documentation_options" data-url_root="./" src="_static/documentation_options.js"></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/language_data.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>