diff --git a/u8/program.py b/u8/program.py index 2747c81..65434cd 100644 --- a/u8/program.py +++ b/u8/program.py @@ -41,27 +41,15 @@ def pocet_nodov(node): def main(): - lines = [] - while True: - try: - line = input() - except EOFError: - break - line = line.rstrip("\r") - if line == "": - lines.append("") - break - lines.append(line) - - for line in sys.stdin: - lines.append(line.rstrip("\r\n")) - - print("Expert zo zoo to vie.") - - if not lines: + danii = sys.stdin.read() + if not danii: print("Chybna databaza") return + lines = danii.splitlines() + + print("Expert zo zoo to vie.") + try: prvi = lines.index("") except ValueError: