Das deutsche Python-Forum

Code: Alles auswählen

#!/usr/bin/env python
# -*- coding: iso-8859-15 -*-

FILENAME = "testdatei.txt"


def create_file(filename):
    """
    Erstellt die Testdatei mit Text
    """
    
    # Textzeilen (\n steht für einen Zeilenumbruch)
    lines = [
        "Ich bin Zeile 1\n",
        "Ich bin Zeile 2\n",
        "Ich bin Zeile 3\n",
    ]
    
    # Datei zum Schreiben öffnen und Zeilen hineinschreiben
    f = file(filename, "w")
    f.writelines(lines)
    f.close()


def insert_line(filename, insert_pos, line):
    """
    Fügt der angegebenen Datei oberhalb der angegebenen Position die
    Zeile hinzu. Diese Funktion sollte nicht für große Dateien (50 MB)
    verwendet werden.
    """
    
    # Datei zum Lesen öffnen und alle Zeilen in eine Liste einlesen
    f = file(filename, "rU")
    lines = f.readlines()
    f.close()
    
    # Wenn die Zeile nicht mit einem Zeilenumbruch endet, dann wird dieser
    # hinzugefügt.
    if not line.endswith("\n"):
        line += "\n"
    
    # Der Liste die neue Zeile hinzufügen
    lines.insert(insert_pos, line)
    
    # Speichern
    f = file(filename, "w")
    f.writelines(lines)
    f.close()


def main():
    create_file(FILENAME)
    insert_line(FILENAME, 1, "Ich bin gekommen um zu bleiben.")


if __name__ == "__main__":
    main()