保存文件夹下面的所有.txt文件全路径到文本文件中 yolo

# -*- coding: utf-8 -*-
import time
import os
import shutil
import string
 
def readFilename(path, allfile):
    filelist = os.listdir(path)
 
    for filename in filelist:
        filepath = os.path.join(path, filename)
        if os.path.isdir(filepath):
            readFilename(filepath, allfile)
        else:
            allfile.append(filepath)
    return allfile
 
 
if __name__ == '__main__':
    path1 = "/home/zjh/pzy/ship_detect/datasets/train2019"
    allfile1 = []
    allfile1 = readFilename(path1, allfile1)
    allname1 = []
    txtpath = "/home/zjh/pzy/ship_detect/train_coco_2019.txt"
    for name in allfile1:
        file_cls = name.split("/")[-1].split(".")[-1]
        if file_cls == 'txt':
 
            with open(txtpath, 'a+') as fp:
                fp.write("".join(name) + "\n")