drive_search.py

from googleapiclient.discovery import build
from google.oauth2.service_account import Credentials
from fuzzywuzzy import process  # Importando a biblioteca fuzzywuzzy
import os

SCOPES = ["https://www.googleapis.com/auth/drive.readonly"]
FOLDER_ID = "1hqfPQnsVL2Ld8hu0GRIqcuOp-eDz-CAX"  # ID da pasta que você quer buscar

SERVICE_ACCOUNT_FILE = os.path.join(os.getcwd(), "./files/credenciais.json")
def authenticate_drive():
    """Autentica no Google Drive usando uma conta de serviço."""
    credentials = Credentials.from_service_account_file(SERVICE_ACCOUNT_FILE, scopes=SCOPES)
    service = build("drive", "v3", credentials=credentials)
    return service

def list_files_recursive(service, folder_id, path=""):
    """Lista todos os arquivos e subpastas no Google Drive de forma recursiva."""
    query = f"'{folder_id}' in parents and trashed = false"
    response = service.files().list(
        q=query,
        spaces="drive",
        fields="files(id, name, mimeType, parents)",
    ).execute()

    files = response.get("files", [])
    all_files = []

    for file in files:
        # caminho completo
        current_path = f"{path}/{file['name']}"
        all_files.append({"id": file["id"], "name": file["name"], "path": current_path})

        # buscar recursivamente
        if file["mimeType"] == "application/vnd.google-apps.folder":
            all_files.extend(list_files_recursive(service, file["id"], current_path))

    return all_files

def find_file_by_name(files, search_name):
    """Encontra um arquivo com nome aproximado utilizando fuzzy matching."""
    # Usando fuzzywuzzy para encontrar o arquivo mais próximo
    file_names = [file["name"] for file in files]
    best_match, score = process.extractOne(search_name, file_names)
    
    if score >= 80:  # Ajuste a pontuação mínima de correspondência (por exemplo, 80%)
        matching_files = [file for file in files if file["name"] == best_match]
        return matching_files
    else:
        return []

def search_file_in_drive(search_name):
    """Procura o arquivo mais relevante no Google Drive e retorna o link."""
    service = authenticate_drive()
    print("Autenticado com sucesso no Google Drive!")

    # Listar arquivos na pasta e subpastas
    files = list_files_recursive(service, FOLDER_ID)

    if not files:
        print("Nenhum arquivo encontrado na pasta!")
        return None  # Retorna None se nenhum arquivo for encontrado

    print(f"Total de arquivos encontrados: {len(files)}")

    # Encontrar o arquivo com nome aproximado
    matching_files = find_file_by_name(files, search_name)

    if matching_files:
        best_file = matching_files[0]  # Pega o arquivo mais relevante (primeiro da lista)
        link = f"https://drive.google.com/file/d/{best_file['id']}/view"
        print(f"Arquivo encontrado: {best_file['name']}")
        print(f"Link: {link}")
        return link  # Retorna o link do arquivo encontrado
    else:
        print(f"Nenhum arquivo encontrado com nome aproximado '{search_name}'.")
        return None  # Retorna None se nenhum arquivo for encontrado