Read from stdin or pipe

http://stackoverflow.com/a/31973024/188963

This is how I ended up doing this. I didn't really like any of the other solutions, they didn't seem very pythonic.

This will make a container for any open file input to iterate over all the lines. This will also take care of closing the file at the end of the context manager.

I feel like this is probably how the for line in sys.stdin: block should operate by default.

class FileInput(object):                                                        
    def __init__(self, file):                                                   
        self.file = file                                                       
                                                                            
    def __enter__(self):                                                        
        return self                                                             
                                                                            
    def __exit__(self, *args, **kwargs):                                        
        self.file.close()                                                       
                                                                            
    def __iter__(self):                                                         
        return self                                                             
                                                                            
    def next(self):                                                             
        line = self.file.readline()                                             
                                                                            
        if line == None or line == "":                                          
            raise StopIteration                                                 
                                                                            
        return line  

with FileInput(sys.stdin) as f:
    for line in f:
        print f

with FileInput(open('tmpfile') as f:
    for line in f:
        print f

From the command line both of the following should work:

tail -f /var/log/debug.log | python fileinput.py
cat /var/log/debug.log | python fileinput.py