Wie ermittle ich die Größe einer geöffneten Datei in Python?

Question

Es gibt eine Datei, die ich sicherstellen möchte, dass sie nicht größer als 2 GB wird (da sie auf einem System laufen muss, das ext 2 verwendet). Wie kann man die Größe einer Datei überprüfen, wenn man bedenkt, dass ich zwischen den Überprüfungen in diese Datei schreiben werde? Muss ich mich insbesondere über gepufferte, nicht abgeschlossene Änderungen Gedanken machen, die noch nicht auf die Festplatte geschrieben wurden?

Answer 1

Sie mit so etwas wie dies beginnen könnte:

class TrackedFile(file): 
    def __init__(self, filename, mode): 
     self.size = 0 
     super(TrackedFile, self).__init__(filename, mode) 
    def write(self, s): 
     self.size += len(s) 
     super(TrackedFile, self).write(s) 

Dann könnten Sie es wie folgt verwendet werden:

>>> f = TrackedFile('palindrome.txt', 'w') 
>>> f.size 
0 
>>> f.write('A man a plan a canal ') 
>>> f.size 
21 
>>> f.write('Panama') 
27 

Offensichtlich ist diese Umsetzung funktioniert nicht, wenn Sie die Datei nicht zu schreiben sind, von Grund auf, aber Sie könnten Ihre __init__ Methode anpassen, um anfängliche Daten zu behandeln. Sie müssen möglicherweise auch einige andere Methoden überschreiben: writelines, zum Beispiel.

Dies funktioniert unabhängig von der Codierung, da Strings nur Sequenzen von Bytes sind.

>>> f2 = TrackedFile('palindrome-latin1.txt', 'w') 
>>> f2.write(u'A man a plan a canál '.encode('latin1') 
>>> f3 = TrackedFile('palindrome-utf8.txt', 'w') 
>>> f3.write(u'A man a plan a canál '.encode('utf-8')) 
>>> f2.size 
21 
>>> f3.size 
22 

Answer 2

Vielleicht nicht, was Sie wollen, aber ich werde es trotzdem vorschlagen.

import os 
a = os.path.getsize("C:/TestFolder/Input/1.avi") 

Alternativ zu einer geöffneten Datei können Sie die fstat Funktion verwenden, die auf einer geöffneten Datei verwendet werden kann. Es dauert eine ganze Datei-Handle, nicht ein Dateiobjekt, so müssen Sie die fileno Methode auf dem Dateiobjekt verwenden:

a = open("C:/TestFolder/Input/1.avi") 
b = os.fstat(a.fileno()).st_size 

Answer 3

Zuverlässigstes eine Verpackungsklasse schaffen würde, die Datei die Größe überprüfen würde, wenn Sie es öffnen, Verfolgen Sie Schreib- und Suchvorgänge, zählen Sie die aktuelle Größe basierend auf diesen Vorgängen und verhindern Sie, dass die Größenbeschränkung überschritten wird.

Answer 4

Oder, wenn die Datei bereits geöffnet ist:

>>> fsock = open('/etc/hosts', 'rb').read() 
>>> len(fsock) 
444 

Das ist, wie viele Bytes die Datei.

Answer 5

os.fstat(file_obj.fileno()).st_size sollte den Trick tun. Ich denke, dass es die geschriebenen Bytes zurückgibt. Sie können immer einen Flush vor der Hand machen, wenn Sie sich Gedanken über die Pufferung machen.

Answer 6

Ich bin nicht vertraut mit Python, aber hat das Stream-Objekt (oder was auch immer Sie beim Öffnen einer Datei bekommen) eine Eigenschaft, die die aktuelle Position des Streams enthält?

Ähnlich wie Sie mit der ftell() C-Funktion oder Stream.Position in .NET erhalten.

Offensichtlich funktioniert das nur, wenn Sie am Ende des Streams positioniert sind, was Sie sind, wenn Sie gerade schreiben.

Der Vorteil dieses Ansatzes besteht darin, dass Sie die Datei nicht schließen müssen und sich nicht um nicht gelöschte Daten sorgen müssen.

Answer 7

Obwohl dies eine alte Frage ist, denke ich, dass Isak die einfachste Lösung hat. Hier ist, wie es in Python zu tun:

# Assuming f is an open file 
>>> pos = f.tell() # Save the current position 
>>> f.seek(0, 2) # Seek to the end of the file 
>>> length = f.tell() # The current position is the length 
>>> f.seek(pos) # Return to the saved position 
>>> print length 
1024