In my case, the problem was how I was importing the Timetable - airflow apparently expects you to import it relative to the $PLUGINS_FOLDER, not from any other directory, even if that other directory is also on the PYTHONPATH.įor a concrete example: export PYTHONPATH=/path/to/my/code:$PYTHONPATH answer pointed me in the right direction. airflow plugins reports that the plugin is registered, and running the DAG script on the command line works fine, but the web UI reports that the plugin is not registered. Workday_timetable_plugin | $PLUGINS_FOLDER/AfterWorkdayTimetable.py Return DagRunInfo.interval(start=next_start, end=(next_start + timedelta(days=1)))Ĭlass WorkdayTimetablePlugin(AirflowPlugin):ĭetails of dags/test_afterwork_timetable.py: import datetimeįrom AfterWorkdayTimetable import AfterWorkdayTimetableįrom import DummyOperator Return None # Over the DAG's scheduled end don't schedule. Last_start_weekday = last_start.weekday() Last_start = last_automated_data_interval.start

If last_automated_data_interval is not None: # There was a previous run on the regular schedule. Return DataInterval(start=start, end=(start + timedelta(days=1))) Start = bine((run_after - delta).date(), Time.min).replace(tzinfo=UTC)

If weekday in (0, 6): # Monday and Sunday - interval is last Friday.ĭays_since_friday = (run_after.weekday() - 4) % 7ĭelta = timedelta(days=days_since_friday)Įlse: # Otherwise the interval is yesterday. In the web UI, the last line of detailed error message: : Failed to serialize DAG 'example_timetable_dag2': Timetable class 'AfterWorkdayTimetable' is not registeredīut if I run airflow plugins, I can see the timetable is in the name and source list.ĭetail of plugins/AfterWorkdayTimetable.py: from datetime import timedeltaįrom pendulum import Date, DateTime, Time, timezoneįrom ugins_manager import AirflowPluginįrom import DagRunInfo, DataInterval, TimeRestriction, Timetableĭef infer_data_interval(self, run_after: DateTime) -> DataInterval: create the example dag file, and put it in $Home/airflow/dagsĪfter restart scheduler and webserver, I get DAG import error.create the example timetable py file, and put it in the $Home/airflow/plugins.

Airflow scheduler timezone