Python 搜索部分文件名

Question

I am using Python 3 and want to check for 2 things. Firstly I want to ensure there are 3 files in a directory, then I want to check that the 3 files match certain rules....

• 3 files in directory
• A file called testfile1.txt
• A file that matches myfile-*.txt (for example myfile-473spain.txt)
• A file that matches newlog-*.log (for example newlog-55.log)

I have this so far...

list = os.listdir('myfiles') # dir is your directory path
number_files = len(list)

if number_files !=3 :
    print('Incorrect Number Of Files')
else:
    print('Correct Number Of Files')

if os.path.isfile('myfiles/testfile1.txt'):
    print ("myfiles/testfile1.txt - FOUND")
else:
    print ("myfiles/testfile1.txt - NOT FOUND")

But I am now stuck on how to search for the 2 partial match files. What can I try next?

Answer 1

Use glob.glob :

import glob

if glob.glob('myfiles/myfile-*.txt'):
    print('match for "myfile-*.txt" found')
else:
    print('match for "myfile-*.txt" not found')

Answer 2

You could loop over the list you created in your first rule. Then you could you the "in" statement: Something like:

for fileName in list:
   if ("myfile-" in fileName and ".txt" in fileName):
      #OK

Answer 3

To match the beginning and end of a string, simply use the str.endswith and str.startswith. If you want to do more complicated matching, look into the re module.

file = 'myfile-something.txt'

if file.startswith('myfile-') and file.endswith('.txt'):
    print('success')