Fix import error for backup dags without xcom or task instance

[holly-evans]

What type of PR is this? (check all applicable)

• Refactor
• Feature
• Bug Fix
• Optimization
• Documentation Update
• Dependency Update
• Release

Description

Custom DAG factories cannot create backup dags unless they include the xcom and task_instance tables.

This PR fixes create_backup_dag to catch the KeyError if either xcom or task_instance is not one of the export_tasks.

Related Tickets & Documents

• Related Issue [BUG] create_backup_dag produces IndexError if xcom and task_instance are not included in tables #40
• Closes [BUG] create_backup_dag produces IndexError if xcom and task_instance are not included in tables #40

Linting

Have you done linting by issuing ./build.sh lint command?

• Yes
• No, I need help with linting

Testing

Have you run testing by issuing ./build.sh unit command?

• Yes
• No, I need help with writing tests

Documentation

Have you updated the README appropriate for this PR?

• Yes
• No, README does not need any changes for this PR

Changelog

Have you updated the CHANGELOG with the changes you are making in this PR?
Please use Added, Removed, and/or Changed subsections under the Unreleased section.

• Yes
• No, I need help

Terms of Contribution

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

[crupakheti]

@holly-evans Thank you for your contribution! We will get started with the review soon.

[github-actions]

Coverage report

Click to see where and how coverage changed

File	Statements	Missing	Coverage	Coverage (new stmts)	Lines missing
assets/dags/mwaa_dr/framework/factory
base_dr_factory.py					348-349
Project Total

This report was generated by python-coverage-comment-action