Я пытаюсь написать программу, которая сможет искать все файлы (как имена файлов, так и их содержимое) в указанном пользователем каталоге для определенной строки, а затем перемещать эти файлы в новый указанный пользователемкаталог.
РЕДАКТИРОВАТЬ: Хорошо, поэтому я сделал несколько изменений в моем коде.Теперь это работает так: список файлов получается с помощью os.path.walk ().Затем в каждом файле списка ищется строка, указанная пользователем.Сначала проверяются только имена файлов на строку, а все положительные совпадения перемещаются в отдельный список.Затем мы начинаем заглядывать в файлы, используя расширение файла, чтобы определить, как открыть файл через win32com.client.Наконец, любые файлы, все еще находящиеся в исходном списке, считаются простыми текстовыми файлами, и их открывают и ищут соответственно.
Однако по какой-либо причине программа перемещает только простые текстовые файлы.Если кто-нибудь может понять, почему это так, это было бы огромной помощью.:)
################
#Import required modules
import fileinput
from shutil import move
from os.path import abspath, join, splitext, split
from os import mkdir, walk, remove
import win32com.client
################
#Create lists to hold file names
file_list = list()
file_move_list = list()
#Define file extensions which need to be converted
excel_set = [".xls", ".xlsx", ".xlsm", ".xlsb"]
msword_set = [".doc", ".docx"]
################
#Define functions
def getFileList( searchdirectory ):
#Get a list of all items in the directory to search
for (dirpath, dirnames, filenames) in walk( searchdirectory ):
for path in [ abspath( join( dirpath, filename ) ) for filename in filenames ]:
file_list.append( path )
def searchFiles( readfilelist, movefilelist, searchstring ):
#Get plain text from each file and search for searchstring
for filename in readfilelist:
ext = splitext( filename )[1]
#Check filenames
if searchstring in filename:
movefilelist.append( filename )
readfilelist.remove( filename )
#Check if file is a pdf
elif ext == ".pdf":
content = getPDFContent( filename )
if searchstring in content:
movefilelist.append( filename )
#Check if file is a word document
elif ext in msword_set:
app = win32com.client.Dispatch('Word.Application')
doc = app.Documents.Open( filename )
if searchstring in doc.Content.Text:
movefilelist.append( filename )
app.Quit()
#Check if file is an excel workbook/spreadsheet
elif ext in excel_set:
app = win32com.client.Dispatch( 'Excel.Application' )
fileDir, fileName = split( filename )
nameOnly = splitext( fileName )
newName = nameOnly[0] + ".csv"
outCSV = join( fileDir, newName )
workbook = app.Workbooks.Open( filename )
workbook.SaveAs(outCSV, FileFormat=24) # 24 is csv format
workbook.Close(False)
for line in open( outCSV, mode='r' ):
if searchstring in line:
movefilelist.append( filename )
app.Quit()
remove( outCSV )
#Assume all other files are plain text
else:
for line in open( filename, mode='r' ):
if searchstring in line:
movefilelist.append( filename )
readfilelist.remove( filename )
def moveFiles( movelist, destinationdirectory ):
mkdir( destinationdirectory )
for path in movelist:
#Move the files to the destination folder
move( path, destinationdirectory )
print( 'Done' )
def getPDFContent( filename ):
content = ""
pdf = pyPdf.PdfFileReader( file( filename, "rb" ) )
# Extract text from each page and add to content
for i in range( 0, pdf.getNumPages() ):
content += pdf.getPage(i).extractText() + " \n"
return content
################
#Run as main
if __name__=='__main__':
search_directory = input( 'Enter the path of the directory you wish to search through: ' )
search_string = input( 'Enter the search term: ' )
destination_directory = input( 'Enter the name of the new directory which will contain the moved files: ' )
getFileList( search_directory )
searchFiles( file_list, file_move_list, search_string )
moveFiles( file_move_list, destination_directory )
Любая помощь, которую я могу получить по этому вопросу, высоко ценится.(К вашему сведению, я использую Python 3.2.1)