[Python #14] [Django #7] 내 포스팅 검색 #2

pixabay

지난 시간에 구현한 검색을 좀 더 다듬어 봤다. 사용하다 보니 문제가 보였기 때문이다. tag, title, body 값을 모두 넣으면 조건에 맞는 모든 결과가 검색되는데 생각해보니 내가 의도한 바와는 조금 다르다. 즉 tag or title or body로 연산된 것.

보통 이 세 개값을 동시에 사용하여 검색할 경우, tag가 포함된 글목록에서 title 혹은 body값이 포함되어 있는 글을 추가로 검색하기 위해서다. 즉 tag and (title or body).

그럼 바꿔보자. 바꾸는 김에 python의 클래스로 구현해 보자. 아직까지 클래스가 어떤 점에서 더 좋은지는 모르겠지만 일단 해보자. (#python초보#어렵다어려워)

services.py

검색 클래스를 services.py 내에 생성 후 전역 변수 account, q_titles, q_texts, q_tags, blogs 설정
init 함수로 기본 인자 query를 dict으로 설정 및 전역변수에 값 할당
search_posts 함수에 검색 기능 추가. tag 값이 없거나 혹은 매칭 되는 값이 있을 경우에만 추가로 title 혹은 body 값 체크, 값이 매칭될 경우 새 list에 추가
title과 body 값 체크는 하나의 함수에 넣기에 가독성이 떨어지므로 is_keywords_contains로 분리
글의 태그 값 추출을 위한 get_tags 함수 생성

import json
from steem import Steem


class Search:
    s = Steem()
    account = 'june0620'
    q_titles = []
    q_texts = []
    q_tags = []
    blogs = []

    def __init__(self, query: dict):
        self.q_titles = query['titles']
        self.q_texts = query['texts']
        self.q_tags = query['tags']
        self.blogs = self.s.get_blog(self.account, 0, 400)

    def search_posts(self):
        s_blogs = []
        for blog in self.blogs[:]:
            blog_data = blog['comment']
            title = blog_data['title'].lower()
            body = blog_data['body'].lower()
            tags = self.get_tags(blog_data['json_metadata'])
            if not self.q_tags or set(self.q_tags) & set(tags):
                if self.is_keywords_contains(title, body):
                    s_blogs.append(blog)
        return s_blogs

    def is_keywords_contains(self, title: str = [], body: str = []):
        if any(q_title in title for q_title in self.q_titles):
            return True
        if any(q_text in body for q_text in self.q_texts):
            return True
        return False

    def get_tags(self, json_metadata: str):
        metadata = json.loads(json_metadata)
        if metadata:
            return metadata['tags']
        else:
            return []

views.py

django의 ListView를 상속받아 queryset에 검색된 목록을 넣어주고 html에 뿌려준다.

from django.http import HttpResponse
from django.views.generic import ListView
from .services import Search
from steem import Steem

s = Steem()


def main_view(request):
    data = s.get_account('june0620')
    response = HttpResponse()
    response.write(data)
    return response


class posts(ListView):
    template_name = 'album.html'
    context_object_name = 'all_posts'

    def get(self, request, *args, **kwargs):
        # self.queryset = s.get_blog(kwargs['account'], entry_id=0, limit=100)
        query = {
            'tags': ['kr'],
            'titles': ['python'],
            'texts': ['사랑하겠습니다']
        }
        se = Search(query=query)
        self.queryset = se.search_posts()
        return super().get(request, *args, **kwargs)