Python Files and Directories Operation

In this article, I have implemented the following two functions.

Read all files from a folder.
Read the specific files from a folder.

Read all files from a folder

Show you the code directly.

import os
import sys

global fileNameList

def showAllFileName(filePath):

    fileNames = os.listdir(filePath)
    for fileName in fileNames:
        addPrefixFileName = filePath + '/' + fileName
        #print(addPrefixFileName)
        if os.path.isdir(addPrefixFileName):           
            showAllFileName(addPrefixFileName)
        else:
            print(fileName)

if __name__ == '__main__':
    if len(sys.argv) < 2 :
        print("Please enter a directory\n")
        exit()

    checkFolder = os.path.isdir(sys.argv[1])
    if not checkFolder:
        print("Please enter a directory\n")
        exit()
    filePath = (sys.argv)[1]
    showAllFileName(filePath)

The following is my test folder structure.

After run “python main.py folder1” command, the following shows the result.

Read the specific files from a folder

First, show you the code.

import os
import sys

global fileNameList

def showAllFileName(filePath):

    fileNames = os.listdir(filePath)
    for fileName in fileNames:
        addPrefixFileName = filePath + '/' + fileName
        #print(addPrefixFileName)
        if os.path.isdir(addPrefixFileName):           
            showAllFileName(addPrefixFileName)
        else:
            if fileName.endswith('.vm'):
                print(fileName)

if __name__ == '__main__':
    if len(sys.argv) < 2 :
        print("Please enter a directory\n")
        exit()

    checkFolder = os.path.isdir(sys.argv[1])
    if not checkFolder:
        print("Please enter a directory\n")
        exit()
    filePath = (sys.argv)[1]
    showAllFileName(filePath)

Then, The following is the folder structure.