[AIRFLOW-2525] Fix PostgresHook.copy_expert to work with "COPY FROM" #3421
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Make sure you have checked all steps below.
JIRA
Description
For now PostgresHook.copy_expert supports
"COPY TO" but not "COPY FROM", because it
opens a file with write mode and doesn't
commit operations. This PR fixes it by
opening a file with read and write mode
and committing operations at last.
Tests
Fixed tests.hooks.test_postgres_hook:TestPostgresHook.test_copy_expert to confirm that the open() function is called with read and write mode and operations are committed at last.
In addition, I confirmed that the fixed version worked with both COPY FROM and COPY TO with PostgreSQL 9.5.12.
Commits
Documentation
Code Quality
git diff upstream/master -u -- "*.py" | flake8 --diff