파이썬으로 PyLucene을 활용한 검색 결과 테스트하기

18 Oct 2023

pylucene

PyLucene은 Lucene 검색 엔진을 파이썬에서 사용할 수 있도록 만든 라이브러리입니다. 이를 활용하면 파이썬으로 강력한 검색 기능을 구현할 수 있습니다. 이번 포스트에서는 PyLucene을 사용하여 검색 결과를 테스트하는 방법을 알아보겠습니다.

PyLucene 설치하기

PyLucene은 Java 환경에서 동작하기 때문에 먼저 Java Development Kit (JDK)를 설치해야 합니다. JDK를 설치한 후에는 PyLucene 라이브러리를 설치할 수 있습니다. 다음 명령어를 사용하여 PyLucene을 설치해주세요.

pip install PyLucene

검색 인덱스 생성하기

PyLucene을 사용하여 검색하기 위해서는 먼저 검색 인덱스를 생성해야 합니다. 검색 인덱스는 검색 대상이 되는 문서들을 색인화하여 저장하는 데이터 구조입니다. 인덱스를 생성하기 위해 다음과 같은 단계를 따릅니다.

필요한 문서들을 로드합니다.
문서들의 내용과 메타데이터를 추출합니다.
추출한 내용과 메타데이터를 인덱스에 추가합니다.

아래는 예시 코드입니다.

import lucene

from lucene import Document, Field, StandardAnalyzer, IndexWriter, IndexWriterConfig

# PyLucene 초기화
lucene.initVM()

# 인덱스 디렉토리 지정
index_dir = "index"

# 인덱스 생성을 위한 필요한 정보를 파일에서 로드
documents = load_documents()

# IndexWriter 설정
analyzer = StandardAnalyzer()
config = IndexWriterConfig(analyzer)
config.setOpenMode(IndexWriterConfig.OpenMode.CREATE)

with IndexWriter(index_dir, config) as writer:
    for doc in documents:
        # Document 생성
        d = Document()
        d.add(Field("title", doc["title"], Field.Store.YES, Field.Index.ANALYZED))
        d.add(Field("content", doc["content"], Field.Store.YES, Field.Index.ANALYZED))
        
        # 인덱스에 추가
        writer.addDocument(d)

print("인덱스 생성 완료")

검색 결과 가져오기

인덱스가 생성되었다면 이제 검색 결과를 가져오는 방법을 알아보겠습니다. 아래는 예시 코드입니다.

import lucene

from lucene import IndexSearcher, QueryParser, StandardAnalyzer

# PyLucene 초기화
lucene.initVM()

# 인덱스 디렉토리 지정
index_dir = "index"

# 검색어 입력
query_text = "파이썬 검색"

# IndexSearcher 설정
analyzer = StandardAnalyzer()
searcher = IndexSearcher(index_dir)

# Query 객체 생성
query_parser = QueryParser("content", analyzer)
query = query_parser.parse(query_text)

# 검색 결과 가져오기
hits = searcher.search(query, 10)

# 검색 결과 출력
for hit in hits.scoreDocs:
    doc_id = hit.doc
    doc = searcher.doc(doc_id)
    print("Title:", doc.get("title"))
    print("Content:", doc.get("content"))
    print("---")

# 리소스 해제
searcher.close()

마무리

이번 포스트에서는 PyLucene을 사용하여 검색 결과를 테스트하는 방법에 대해 알아보았습니다. PyLucene은 파이썬 개발자들에게 강력한 검색 기능을 제공하며, 빠르고 효율적인 검색 애플리케이션 개발에 도움이 됩니다.

더 자세한 내용은 PyLucene 공식 문서를 참고해주세요.