Create a Local Search Engine to Instantly Find Files on Your Computer in Python

import os
import sys
import time
from pathlib import Path
import fnmatch

class LocalSearchEngine:
    def __init__(self, root_directory="."):
        self.root_directory = Path(root_directory)
        self.file_index = {}
        
    def build_index(self):
        """Build a complete index of files in the root directory."""
        start_time = time.time()
        indexed_count = 0
        
        for root, dirs, files in os.walk(self.root_directory):
            for file in files:
                file_path = Path(root) / file
                file_lower = file.lower()
                
                # Index by exact filename
                self.file_index[file_lower] = file_path
                
                # Index by extension
                ext = file_path.suffix.lower()
                if ext not in self.file_index:
                    self.file_index[ext] = []
                self.file_index[ext].append(file_path)
                
                indexed_count += 1
                
        elapsed_time = time.time() - start_time
        print(f"Indexed {indexed_count} files in {elapsed_time:.2f} seconds")
        
    def search_by_name(self, query):
        """Search for files by exact name (case-insensitive)."""
        query_lower = query.lower()
        if query_lower in self.file_index:
            return [self.file_index[query_lower]]
        return []
    
    def search_by_pattern(self, pattern):
        """Search for files matching a glob pattern."""
        results = []
        pattern_lower = pattern.lower()
        
        for file_lower, file_path in self.file_index.items():
            if isinstance(file_path, Path) and fnmatch.fnmatch(file_lower, pattern_lower):
                results.append(file_path)
                
        return results
    
    def search_by_extension(self, extension):
        """Search for files with a specific extension."""
        ext = extension.lower() if extension.startswith('.') else f".{extension}"
        return self.file_index.get(ext, [])

if __name__ == "__main__":
    # Demo with current directory
    engine = LocalSearchEngine(".")
    engine.build_index()
    
    # Example searches
    print("\n--- Search Examples ---")
    query_name = "test.py"
    results = engine.search_by_name(query_name)
    print(f"Files named '{query_name}': {results}")
    
    pattern = "*.txt"
    results = engine.search_by_pattern(pattern)
    print(f"Files matching '{pattern}': {results}")
    
    extension = ".py"
    results = engine.search_by_extension(extension)
    print(f"Files with extension '{extension}': {results[:5]}...")