PDF-Lote-Scanner/main.py at main · pedromts2/PDF-Lote-Scanner · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
import fitz
import os

def obter_info_lotes(diretorio_pdfs):
    info_lotes = {}

    # Itera sobre todos os arquivos no diretório
    for arquivo_pdf in os.listdir(diretorio_pdfs):
        if arquivo_pdf.endswith('.pdf'):
            arquivo_pdf_path = os.path.join(diretorio_pdfs, arquivo_pdf)

            pdf_doc = fitz.open(arquivo_pdf_path)

            # Itera sobre todas as páginas para identificar os lotes
            for pagina_num in range(pdf_doc.page_count):
                pagina = pdf_doc[pagina_num]
                texto = pagina.get_text()

                # Identifica o lote com base em um padrão (por exemplo, "Lote:")
                if 'Lote:' in texto:
                    # Obtém o número do lote (ajuste conforme necessário)
                    numero_lote = texto.split('Lote:')[1].split()[0]

                    # Adiciona o número do lote e a quantidade de páginas ao dicionário
                    if numero_lote not in info_lotes:
                        info_lotes[numero_lote] = 0

                    info_lotes[numero_lote] += 1

            pdf_doc.close()

    return info_lotes

if __name__ == "__main__":
    # Caminho para o diretório contendo os arquivos PDF
    diretorio_pdfs = "C:/Importar Drive/pdf"

    # Obtém informações sobre os lotes nos arquivos PDF
    info_lotes = obter_info_lotes(diretorio_pdfs)

    # Exibe as informações obtidas
    print("Informações sobre lotes:")
    for numero_lote, num_paginas in info_lotes.items():
        print(f"Lote {numero_lote}: {num_paginas} página(s)")