mirror of
https://github.com/kennethreitz/tablib.git
synced 2026-06-05 23:10:17 +00:00
Kenneth Reitz
84 lines
1.5 KiB
Python
84 lines
1.5 KiB
Python
#!/usr/bin/env python
|
|
# encoding: utf-8
|
|
|
|
""" Tabbed CLI Inteface Application
|
|
"""
|
|
|
|
import io
|
|
import sys
|
|
|
|
import argue
|
|
|
|
import tablib
|
|
from helpers import Struct, piped
|
|
|
|
|
|
|
|
FORMATS = [fmt.title for fmt in tablib.formats.FORMATS]
|
|
|
|
opts = []
|
|
|
|
opts.append(('v', 'version', False, 'Report tabbed version'))
|
|
|
|
for format in FORMATS:
|
|
opts.append(('', format, False, 'Output to %s' % (format.upper())))
|
|
|
|
|
|
|
|
@argue.command(options=opts, usage='[FILE] [--FORMAT | FILE]')
|
|
def start(in_file=None, out_file=None, **opts):
|
|
"""Covertly convert dataset formats"""
|
|
|
|
opts = Struct(**opts)
|
|
|
|
if opts.version:
|
|
print('Tabbed, Ver. %s' % tablib.core.__version__)
|
|
sys.exit(0)
|
|
|
|
stdin = piped()
|
|
|
|
if stdin:
|
|
data = tablib.import_set(stdin)
|
|
|
|
elif in_file:
|
|
|
|
try:
|
|
in_stream =- io.open(in_file, 'r').read()
|
|
except Exception, e:
|
|
print(' %s cannot be read.' % in_file)
|
|
sys.exit(65)
|
|
|
|
try:
|
|
tablib.import_set(in_stream)
|
|
except Exception, e:
|
|
raise e
|
|
print('Import format not supported.')
|
|
sys.exit(65)
|
|
else:
|
|
print('Please provide input.')
|
|
sys.exit(65)
|
|
|
|
|
|
_formats_sum = sum(opts[f] for f in FORMATS)
|
|
|
|
# Multiple output formats given
|
|
if _formats_sum > 1:
|
|
print('Please specify a single output format.')
|
|
sys.exit(64)
|
|
|
|
# No output formats given
|
|
elif _formats_sum < 1:
|
|
print('Please specify an output format.')
|
|
sys.exit(64)
|
|
|
|
|
|
# fetch options.formats list
|
|
# if sum(()) > 1
|
|
# log only one data format please
|
|
# if sum of formats == 0, specity format
|
|
|
|
# look for filename
|
|
|
|
# print opts.__dict__
|
|
# print in_file
|
|
# print out_file |