A lot of people use Python as a replacement for shell scripts, using it to automate common system tasks, such as manipulating files, configuring systems, and so forth. This article aims to describe accepting Script Input via Redirection, Pipes, or Input Files.
Problem – To have a script to be able to accept input using whatever mechanism is easiest for the user. This should include piping output from a command to the script, redirecting a file into the script, or just passing a filename, or list of filenames, to the script on the command line.
Python’s built-in fileinput module makes this very simple and concise if the script looks like this.
Code #1 :
import fileinput
with fileinput.input() as f_input:
for line in f_input:
print(line, end ='')
|
Then input to the script can already be accepted in all of the previously mentioned ways. If the script is saved and make it executable then the one can get the expected output using all of the following:
Code #2 :
$ ls | ./filein.py
$ ./filein.py/etc/passwd
$ ./filein.py < /etc/passwd
|
The fileinput.input() function creates and returns an instance of the FileInput class. In addition to containing a few handy helper methods, the instance can also be used as a context manager. So, to put all of this together, if one wrote a script that expected to be printing output from several files at once, one might have it include the filename and line number in the output, as shown in the code given below –
Code #3 :
import fileinput
with fileinput.input('/etc/passwd') as f:
for line in f:
print(f.filename(), f.lineno(), line, end ='')
|
/etc/passwd1
/etc/passwd2
/etc/passwd3
<other output omitted>
Using it as a context manager ensures that the file is closed when it’s no longer being used, and one leveraged a few handy FileInput helper methods here to get some extra information in the output.
Don't miss your chance to ride the wave of the data revolution! Every industry is scaling new heights by tapping into the power of data. Sharpen your skills, become a part of the hottest trend in the 21st century.Dive into the future of technology - explore the
Complete Machine Learning and Data Science Program by GeeksforGeeks and stay ahead of the curve.
Last Updated :
12 Jun, 2019
Like Article
Save Article
Share your thoughts in the comments
Please Login to comment...
