G01_0374/venv_win32/Lib/site-packages/pythonwin/pywin/framework/stdin.py

# Copyright (c) 2000 David Abrahams. Permission to copy, use, modify, sell
# and distribute this software is granted provided this copyright
# notice appears in all copies. This software is provided "as is" without
# express or implied warranty, and with no claim as to its suitability for
# any purpose.
"""Provides a class Stdin which can be used to emulate the regular old
sys.stdin for the PythonWin interactive window. Right now it just pops
up a raw_input() dialog. With luck, someone will integrate it into the
actual PythonWin interactive window someday.

WARNING: Importing this file automatically replaces sys.stdin with an
instance of Stdin (below). This is useful because you can just open
Stdin.py in PythonWin and hit the import button to get it set up right
if you don't feel like changing PythonWin's source. To put things back
the way they were, simply use this magic incantation:
    import sys
    sys.stdin = sys.stdin.real_file
"""
import sys

try:
    get_input_line = raw_input # py2x
except NameError:
    get_input_line = input # py3k

class Stdin:
    def __init__(self):
        self.real_file = sys.stdin # NOTE: Likely to be None in py3k
        self.buffer = ""
        self.closed = False

    def __getattr__(self, name):
        """Forward most functions to the real sys.stdin for absolute realism.
        """
        if self.real_file is None:
            raise AttributeError(name)
        return getattr(self.real_file, name)
    
    def isatty(self):
        """Return 1 if the file is connected to a tty(-like) device, else 0. 
        """
        return 1

    def read(self, size = -1):
        """Read at most size bytes from the file (less if the read
        hits EOF or no more data is immediately available on a pipe,
        tty or similar device). If the size argument is negative or
        omitted, read all data until EOF is reached. The bytes are
        returned as a string object. An empty string is returned when
        EOF is encountered immediately. (For certain files, like ttys,
        it makes sense to continue reading after an EOF is hit.)"""
        result_size = self.__get_lines(size)
        return self.__extract_from_buffer(result_size)

    def readline(self, size = -1):
        """Read one entire line from the file. A trailing newline
        character is kept in the string2.6 (but may be absent when a file ends
        with an incomplete line). If the size argument is present and
        non-negative, it is a maximum byte count (including the trailing
        newline) and an incomplete line may be returned. An empty string is
        returned when EOF is hit immediately. Note: unlike stdio's fgets(),
        the returned string contains null characters ('\0') if they occurred
        in the input.
        """
        maximum_result_size = self.__get_lines(size, lambda buffer: '\n' in buffer)

        if '\n' in self.buffer[:maximum_result_size]:
            result_size = self.buffer.find('\n', 0, maximum_result_size) + 1
            assert(result_size > 0)
        else:
            result_size = maximum_result_size
            
        return self.__extract_from_buffer(result_size)

    def __extract_from_buffer(self, character_count):
        """Remove the first character_count characters from the internal buffer and
        return them.
        """
        result = self.buffer[:character_count]
        self.buffer = self.buffer[character_count:]
        return result
    
    def __get_lines(self, desired_size, done_reading = lambda buffer: False):
        """Keep adding lines to our internal buffer until done_reading(self.buffer)
        is true or EOF has been reached or we have desired_size bytes in the buffer.
        If desired_size < 0, we are never satisfied until we reach EOF. If done_reading
        is not supplied, it is not consulted.

        If desired_size < 0, returns the length of the internal buffer. Otherwise,
        returns desired_size.
        """
        while not done_reading(self.buffer) and (desired_size < 0
                                                 or len(self.buffer) < desired_size):
            try:
                self.__get_line()
            except (EOFError, KeyboardInterrupt): # deal with cancellation of get_input_line dialog
                desired_size = len(self.buffer) # Be satisfied!

        if desired_size < 0:
            return len(self.buffer)
        else:
            return desired_size
        
    def __get_line(self):
        """Grab one line from get_input_line() and append it to the buffer.
        """
        line = get_input_line()
        print('>>>',line)  # echo input to console
        self.buffer = self.buffer + line + '\n'

    def readlines(self, *sizehint): 
        """Read until EOF using readline() and return a list containing the lines
        thus read. If the optional sizehint argument is present, instead of
        reading up to EOF, whole lines totalling approximately sizehint bytes
        (possibly after rounding up to an internal buffer size) are read.
        """
        result = []
        total_read = 0
        while sizehint == () or total_read < sizehint[0]:
            line = self.readline()
            if line == '':
                break
            total_read = total_read + len(line)
            result.append(line)
        return result

if __name__ == "__main__":
    test_input = r"""this is some test
input that I am hoping
~
will be very instructive
and when I am done
I will have tested everything.
Twelve and twenty blackbirds
baked in a pie. Patty cake
patty cake so am I.
~
Thirty-five niggling idiots!
Sell you soul to the devil, baby
"""

    def fake_raw_input(prompt=None):
        """Replacement for raw_input() which pulls lines out of global test_input.
        For testing only!
        """
        global test_input
        if '\n' not in test_input:
            end_of_line_pos = len(test_input)
        else:
            end_of_line_pos = test_input.find('\n')
        result = test_input[:end_of_line_pos]
        test_input = test_input[end_of_line_pos + 1:]
        if len(result) == 0 or result[0] == '~':
            raise EOFError()
        return result
    
    get_input_line = fake_raw_input

    # Some completely inadequate tests, just to make sure the code's not totally broken    
    try:    
        x = Stdin()
        print(x.read())
        print(x.readline())
        print(x.read(12))
        print(x.readline(47))
        print(x.readline(3))
        print(x.readlines())
    finally:
        get_input_line = raw_input
else:
    import sys
    sys.stdin = Stdin()