Das deutsche Python-Forum

Code: Alles auswählen

#!/usr/bin/env python3
from itertools import islice
from sys import argv

ENCODING = "utf-8"
MAX_OUPUT_LINES = 10


def ask_positive_integer(prompt):
    while True:
        try:
            result = int(input(prompt))
        except ValueError:
            print("Bitte eine Zahl eingeben!")
        else:
            if result >= 0:
                return result

            print("Die Zahl muss positiv sein!")


def create_file(filename):
    with open(filename, "w", encoding=ENCODING) as file:
        line_count = ask_positive_integer(
            "Wie viele Zeilen sollen in die Datei geschrieben werden? "
        )
        file.writelines(
            input(f"Bitte nächste Zeile für {filename} eingeben: ") + "\n"
            for _ in range(line_count)
        )


def process_file(filename):
    with open(filename, "r", encoding=ENCODING) as file:
        lines = list(islice(file, MAX_OUPUT_LINES + 1))

    if not lines:
        print("Die Datei ist leider leer.")
    elif len(lines) <= MAX_OUPUT_LINES:
        print(f"Die Datei hat {len(lines)} Zeilen.")
        print("Hier ist der Inhalt:", lines)
    else:
        print("Die Datei ist zu gross, um hier ausgegeben zu werden.")


def main():
    if len(argv) != 2:
        print(f"Verwendung: {argv[0]} <dateiname>")
        return

    create_file(argv[1])
    process_file(argv[1])


if __name__ == "__main__":
    main()