如何确定Python中打开文件的大小？

Question

有一个文件，我想确保不增长大于2 GB（因为它必须运行在使用ext 2的系统上）。考虑到我将在两次检查之间写入这个文件，检查文件大小的好方法是什么？特别是，我是否需要担心尚未写入磁盘的缓冲，未刷新的更改？

Answer 1

你可以开始是这样的：如果你不写文件

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

显然，这不工作：

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) 

然后，你可以使用它像这样从头开始，但你可以调整你的方法来处理初始数据。您可能还需要重写一些其他方法：例如，writelines。

无论编码如何，它都可以工作，因为字符串只是字节序列。

>>> 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

也许不是你想要的，但我会建议它。

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

另外一个打开的文件，你可以使用fstat功能，它可以在一个打开的文件中。它需要一个整数的文件句柄，不是一个文件对象，所以你必须使用的fileno方法的文件对象：

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

Answer 3

最可靠的是创建一个包装类，当你打开它会检查文件的大小，跟踪写入和查找操作，根据这些操作计算当前大小并防止超出大小限制。

Answer 4

或者，如果文件已经打开：

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

这是该文件的字节数是。

Answer 5

os.fstat(file_obj.fileno()).st_size应该做的伎俩。我认为它会返回写入的字节。如果你担心缓冲，你总是可以在手边冲水。

Answer 6

我不熟悉python，但是没有流对象（或打开文件时得到的任何东西）有一个包含流的当前位置的属性？

与您使用C函数ftell()或.NET中的Stream.Position类似。

显然，只有当您位于流的末尾时，才能正常工作。

这种方法的好处是您不必关闭文件或担心未刷新的数据。

Answer 7

虽然这是一个老问题，但我认为Isak有最简单的解决方案。以下是如何做到这一点在Python：

# 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