opencultureconsulting
/
openrefine-client
mirror of https://github.com/opencultureconsulting/openrefine-client.git
2
0
Fork 0
Code Issues Releases Activity

fix #14 export --format --output

This commit is contained in:
Felix Lohmeier 2021-01-02 17:25:21 +01:00
parent 82da3f7b4e
commit bc89a98776
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
google/refine/cli.py
View File

@ -41,7 +41,6 @@ def apply(project_id, history_file):
print('File %s has been successfully applied to project %s' % print('File %s has been successfully applied to project %s' %
(history_file, project_id)) (history_file, project_id))
def create(project_file, def create(project_file,
project_format=None, project_format=None,
columnWidths=None, columnWidths=None,
@ -165,17 +164,19 @@ def download(url, output_file=None):
def export(project_id, encoding=None, output_file=None, export_format=None): def export(project_id, encoding=None, output_file=None, export_format=None):
"""Dump a project to stdout or file.""" """Dump a project to stdout or file."""
project = refine.RefineProject(project_id) project = refine.RefineProject(project_id)
if not output_file:
if not export_format: if not export_format:
export_format = 'tsv' export_format = 'tsv'
if not output_file:
if export_format in ['csv', 'tsv', 'txt']: if export_format in ['csv', 'tsv', 'txt']:
encoding = 'UTF-8' encoding = 'UTF-8'
sys.stdout.write(project.export( sys.stdout.write(project.export(
export_format=export_format, encoding=encoding).read()) export_format=export_format, encoding=encoding).read())
else: else:
ext = os.path.splitext(output_file)[1][1:] ext = os.path.splitext(output_file)[1][1:]
if ext: if ext and not export_format:
export_format = ext.lower() export_format = ext.lower()
if not export_format:
export_format = 'tsv'
if export_format in ['csv', 'tsv', 'txt']: if export_format in ['csv', 'tsv', 'txt']:
encoding = 'UTF-8' encoding = 'UTF-8'
with open(output_file, 'wb') as f: with open(output_file, 'wb') as f: