Индексирование и поиск в текстовом файле

Question

У меня есть текстовый файл, содержащий содержимое книги. Я хочу взять этот файл и создать индекс, который позволяет пользователю выполнять поиск по файлу для выполнения поиска.

Поиск будет состоять из ввода слова. Затем программа вернет следующее:

• Каждая глава, которая включает это слово.
• Номер строки строки , содержащей слово.
• Вся строка, на которой написано слово.

Я попытался следующий код:

infile = open(file) 

Dict = {} 

word = input("Enter a word to search: ") 

linenum = 0 
line = infile.readline() 
for line in infile 
    linenum += 1 
    for word in wordList: 
     if word in line: 
      Dict[word] = Dict.setdefault(word, []) + [linenum] 
      print(count, word) 
    line = infile.readline() 

return Dict 

Что-то вроде этого не работает, и, кажется, слишком неудобно для обработки других модулей, которые потребовались бы:

• «или» оператору поиск по одному слову
• Оператор «и» для поиска одного слова и другого в той же главе

Любые предложения были бы замечательными.

Answer 1

def classify_lines_on_chapter(book_contents): 
    lines_vs_chapter = [] 
    for line in book_contents: 
     if line.isupper(): 
      current_chapter = line.strip() 
     lines_vs_chapter.append(current_chapter) 
    return lines_vs_chapter 


def classify_words_on_lines(book_contents): 
    words_vs_lines = {} 
    for i, line in enumerate(book_contents): 
     for word in set([word.strip(string.punctuation) for word in line.split()]): 
      if word: 
       words_vs_lines.setdefault(word, []).append(i) 
    return words_vs_lines 


def main(): 
    skip_lines = 93 

    with open('book.txt') as book: 
     book_contents = book.readlines()[skip_lines:] 

    lines_vs_chapter = classify_lines_on_chapter(book_contents) 
    words_vs_lines = classify_words_on_lines(book_contents) 

    while True: 
     word = input("Enter word to search - ") 
     # Enter a blank input to exit 
     if not word: 
      break 

     line_numbers = words_vs_lines.get(word, None) 
     if not line_numbers: 
      print("Word not found!!\n") 
      continue 

     for line_number in line_numbers: 
      line = book_contents[line_number] 
      chapter = lines_vs_chapter[line_number] 
      print("Line " + str(line_number + 1 + skip_lines)) 
      print("Chapter '" + str(chapter) + "'") 
      print(line) 


if __name__ == '__main__': 
    main() 

Попробуйте на this input file. Переименуйте его как book.txt перед запуском.