How to read input to python in bash

Question

I have code from here and I would like to use this python script in bash like:

python conversion.py sample.tsv sample.xlsx

python script:

import csv
from xlsxwriter.workbook import Workbook

# Add some command-line logic to read the file names.
tsv_file = 'sample.tsv'
xlsx_file = 'sample.xlsx'

# Create an XlsxWriter workbook object and add a worksheet.
workbook = Workbook(xlsx_file)
worksheet = workbook.add_worksheet()

# Create a TSV file reader.
tsv_reader = csv.reader(open(tsv_file, 'rb'), delimiter='\t')

# Read the row data from the TSV file and write it to the XLSX file.
for row, data in enumerate(tsv_reader):
    worksheet.write_row(row, 0, data)

# Close the XLSX file.
workbook.close()

How can I modify this python script. Sorry for the maybe stupid question, but I do not work with python.

import sys, tsv_file = sys.argv[1], xlsx_file = sys.argv[2]. — Aran-Fey
– Aran-Fey, Commented Aug 19, 2016 at 9:48

Luffy Cyliu · Accepted Answer · 2016-08-19 09:52:16Z

3

You can use sys.argv.

....

tsv_file = sys.argv[1]

xlsx_file = sys.argv[2]

...

You can run like:

main.py sample.tsv sample.xlsx

You also can check argparse package. It provides much fancy functions. It is basically a wraper of sys.argv.

answered Aug 19, 2016 at 9:52

Luffy Cyliu

8532 gold badges9 silver badges11 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

Geroge Over a year ago

Hi, thank you for reply - I have this error: Traceback (most recent call last): File "conversion_xlsx.py", line 5, in <module> tsv_file = sys.argv[1] NameError: name 'sys' is not defined, Any idea? Thank you so much.

Geroge Over a year ago

I forgot import sys.. Thank you so much.

Collectives™ on Stack Overflow

How to read input to python in bash

1 Answer 1

2 Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related