change input overload to return input var

[Almenon]

areplInputIterator = None

def input(prompt=None):
    global areplInputIterator

    if prompt is not None: print(prompt)
    try:
        if areplInputIterator is not None:
            try:
                return next(areplInputIterator)
            except StopIteration:
                raise StopIteration("there is no more input")
            return next(areplInputIterator)
        else:
            if type(stdin) is str:
                areplInputIterator = iter([line for line in stdin.split()])
            elif callable(stdin):
                return next(stdin())
            else:
                areplInputIterator = iter([line for line in stdin])

            return next(areplInputIterator)
    except NameError:
        print("AREPL requires stdin to be hardcoded, like so: stdin = 'hello world'; print(input())")

stdin = "fe\nhey"
x=input()
y = input()
#z = input()

[Almenon]

fixed a bug where I did next(stdin()) instead of saving the iterator
Made StopIteration more user friendly
got rid of useless line
Added test case for generator
changed naming of mock stdin to match python naming convention

areplInputIterator = None

def input(prompt=None):
    global areplInputIterator

    if prompt is not None: print(prompt)
    try:
        if areplInputIterator is not None:
            try:
                return next(areplInputIterator)
            except StopIteration:
                # give simple user-friendly error, we dont want users worry about error in arepl source code
                raise StopIteration("There is no more input") from None
        else:
            if type(standard_input) is str:
                areplInputIterator = iter([line for line in standard_input.split()])
            elif callable(standard_input):
                areplInputIterator = standard_input()
            else:
                areplInputIterator = iter([line for line in standard_input])

            return next(areplInputIterator)
    except NameError:
        print("AREPL requires standard_input to be hardcoded, like so: standard_input = 'hello world'; print(input())")

# user can define standard input with three ways:
def standard_input():
    yield 'hello'
    yield 'world'
standard_input = ["hello", "world"]
standard_input = "hello\nworld"

x=input() # should be hello
y=input() # should be world
z=input() # should result in StopIteration error saying no more input