[已解决]30讲，查找文件中视频格式的文件，问题

Arcticfoxer

print('===视频查询程序===')
import os

def finvid(pth):
    os.chdir(pth)
    str1 = ''

    for each_file in os.listdir(os.curdir):
        name,ext = os.path.splitext(each_file)
        if ext == '.mp4' or '.rmvb' or '.avi':
            str1 += (os.getcwd() + os.sep + each_file + '\n\n')
        if os.path.isdir(each_file):
            finvid(each_file)
            os.chdir(os.pardir)
    a = open('D:\\vedioList.txt','w',encoding='utf-8')
    a.write(str1)
    a.close()


pth = input('请输入待查找的初始目录：')
finvid(pth)

如代码，运行后txt文件中写进去的是文件夹中的每个文件的路径，并没有查找视频文件

最佳答案

月排行榜 / 总排行榜

逃兵

2020-12-22 15:22:08

print('===视频查询程序===')
import os

def finvid(pth):
    os.chdir(pth)
    str1 = ''

    for each_file in os.listdir(os.curdir):
        name,ext = os.path.splitext(each_file)
        if ext == '.mp4' or ext == '.rmvb' or ext == '.avi':
            str1 += (os.getcwd() + os.sep + each_file + '\n\n')
        if os.path.isdir(each_file):
            finvid(each_file)
            os.chdir(os.pardir)
    a = open('D:\\vedioList.txt','w',encoding='utf-8')
    a.write(str1)
    a.close()


pth = input('请输入待查找的初始目录：')
finvid(pth)

跳转到最佳答案楼层

逃兵

print('===视频查询程序===')
import os

def finvid(pth):
    os.chdir(pth)
    str1 = ''

    for each_file in os.listdir(os.curdir):
        name,ext = os.path.splitext(each_file)
        if ext == '.mp4' or ext == '.rmvb' or ext == '.avi':
            str1 += (os.getcwd() + os.sep + each_file + '\n\n')
        if os.path.isdir(each_file):
            finvid(each_file)
            os.chdir(os.pardir)
    a = open('D:\\vedioList.txt','w',encoding='utf-8')
    a.write(str1)
    a.close()


pth = input('请输入待查找的初始目录：')
finvid(pth)

Arcticfoxer

逃兵 发表于 2020-12-22 15:22

再问一句，为何这样不行，只找到MP4
if  ext == ('.mp4' or '.rmvb' or '.avi'):

逃兵

Arcticfoxer 发表于 2020-12-22 17:07
再问一句，为何这样不行，只找到MP4
if  ext == ('.mp4' or '.rmvb' or '.avi'):

可以改成

if ext in ('.mp4' , '.rmvb' , '.avi'):

==是判断是否相等，左边是字符串，右边是元组，肯定不相等

Arcticfoxer

逃兵 发表于 2020-12-22 17:20
可以改成

==是判断是否相等，左边是字符串，右边是元组，肯定不相等

哦哦，感觉自己瓦特了