python - python：特定のファイルサイズを超える場合は、別のテキストファイルにアウトファイルします

Question

私はubuntuでPythonでscapyを使用しています。誰かが例をコーディングする方法を知っているかどうか尋ねたいと思います：

スクリプトの実行中に書き込みを行っている2つのテキストファイルがあるとします。ファイルが例500バイトを超えていることを確認したい場合は、ファイルに何でも保存し、書き込む新しいテキストファイルを作成します。（output1.txt、output2.txtなど。）

専門知識がお役に立てば幸いです。ありがとう

---

私のコードの一部は次のとおりです。

file = open("output.txt","w")

def example(p):
    if p.haslayer(Dot11Beacon):
        if p.addr2 not in uniqueAP:
            file.writelines(p.addr2 + "\n")

---

したがって、スクリプトがターミナルで実行されている間、スクリプトはoutput.txtというファイルに書き込まれますが、スクリプトを改善してテキストファイルのファイルサイズを確認し、それを超えると現在のファイルへの書き込みを停止して、たとえば、新しいoutput2.txtを選択して続行します。

Answer 1

RotatingFileHandlerを使用したくない場合の簡単な例。

os.stat('filename').st_sizeファイルサイズの確認に使用する必要があります。

import os
import sys

class RotatingFile(object):
    def __init__(self, directory='', filename='foo', max_files=sys.maxint,
        max_file_size=50000):
        self.ii = 1
        self.directory, self.filename      = directory, filename
        self.max_file_size, self.max_files = max_file_size, max_files
        self.finished, self.fh             = False, None
        self.open()

    def rotate(self):
        """Rotate the file, if necessary"""
        if (os.stat(self.filename_template).st_size>self.max_file_size):
            self.close()
            self.ii += 1
            if (self.ii<=self.max_files):
                self.open()
            else:
                self.close()
                self.finished = True

    def open(self):
        self.fh = open(self.filename_template, 'w')

    def write(self, text=""):
        self.fh.write(text)
        self.fh.flush()
        self.rotate()

    def close(self):
        self.fh.close()

    @property
    def filename_template(self):
        return self.directory + self.filename + "_%0.2d" % self.ii

if __name__=='__main__':
    myfile = RotatingFile(max_files=9)
    while not myfile.finished:
        myfile.write('this is a test')

これを実行した後...

[mpenning@Bucksnort ~]$ ls -la | grep foo_
-rw-r--r--  1 mpenning mpenning    50008 Jun  5 06:51 foo_01
-rw-r--r--  1 mpenning mpenning    50008 Jun  5 06:51 foo_02
-rw-r--r--  1 mpenning mpenning    50008 Jun  5 06:51 foo_03
-rw-r--r--  1 mpenning mpenning    50008 Jun  5 06:51 foo_04
-rw-r--r--  1 mpenning mpenning    50008 Jun  5 06:51 foo_05
-rw-r--r--  1 mpenning mpenning    50008 Jun  5 06:51 foo_06
-rw-r--r--  1 mpenning mpenning    50008 Jun  5 06:51 foo_07
-rw-r--r--  1 mpenning mpenning    50008 Jun  5 06:51 foo_08
-rw-r--r--  1 mpenning mpenning    50008 Jun  5 06:51 foo_09
[mpenning@Bucksnort ~]$

Answer 2

別の方法（少し短い）は、ファイルサイズをチェックwithする条件を持つ句を使用するifことです：-

def RotateFile(count, file_size, file_name):
    indX = 0
    while indX < count:  
        with open(file_name,"a") as fd:  
            ##
            fd.write("DUMMY TEXT\n")      
            ##  
            if int(os.path.getsize(file_name)) > file_size:
                indX += 1
                file_name = "new_file"+str(indX)+".txt"

withステートメントはコンテキストマネージャーを作成するため、if条件の終わりに達するまでに、のファイルオブジェクトはfd自動的に閉じられます。

次に、必要なオプションを指定して、メソッドに渡します。

if __name__ == "__main__":
    count = 10                           ## 10 number of files
    file_size = 10000                    ## 10KB in size
    file_name = "new_file.txt"           ## with this name
    RotateFile(count, file_size, file_name)

実行後、それぞれ10KBのサイズのファイルが10個（現在のディレクトリに）与えられます...