python遍歷文件

#file_list.py  test file menu
import os
import re
def list_cwd():
    return os.listdir(os.getcwd())




def files_cwd():
   return [p for p in list_cwd() if os.path.isfile(p) ]


def folders_ces():
    return [p for p in list_cwd() if os.path.isdir(p)]


path1 = 'C:\\Users'


def list_py(path):
    """    
    if path ==None:
        path = os.getcwd()
        return [fname for fname in os.listdir(path) if os.path.isfile(fname) if fname.endswith('.html')]


    else:
    """
    #path = path1
    print(path)
    print()
    #print("lsitdir = " + str(os.listdir(path)))
    print('printf txt files')
    print()
    
    for f in os.listdir(path):
        
       #print("isfile  = " + str(os.path.isfile(path + '\\' +f)))
       # print()
        if   os.path.isfile(path + '\\' +f):               #isfile(p):此處的P是路徑。該路徑下是一個文件的名稱時，返回True
            if  f.endswith('.txt'):                        #endswith用於判斷字符串是否以指定的後綴結尾。如果是則返回True
                print(f)
    print()            
    print('printf doc|docx files')
    print()
    for f in os.listdir(path):
        
        #print("isfile  = " + str(os.path.isfile(path + '\\' +f)))
        #print()
        if   os.path.isfile(path + '\\' +f):
            if  f.endswith('.docx') or f.endswith('.doc'):
                print(f)       


                
    #return [fname for fname in os.listdir(path) if os.path.isfile(fname) if fname.endswith('.html')]




def size_in_bytes(fname):
    return os.stat(fname).st_size


def cwd_size_in_bytes():
    total = 0
    for name in files_cwd():
        total = total + size_in_bytes(name)
    return total