remove now unused thumbnail methods from searching module

This commit is contained in:
simon 2021-10-14 14:42:35 +07:00
parent ce5f8b6a38
commit 044b12d5a5
2 changed files with 3 additions and 56 deletions

View File

@ -7,15 +7,12 @@ Functionality:
""" """
import math import math
import os
import urllib.parse import urllib.parse
from datetime import datetime from datetime import datetime
import requests import requests
from home.src.config import AppConfig from home.src.config import AppConfig
from home.src.helper import ignore_filelist
from home.src.thumbnails import ThumbManager from home.src.thumbnails import ThumbManager
from PIL import Image
class SearchHandler: class SearchHandler:
@ -24,11 +21,10 @@ class SearchHandler:
CONFIG = AppConfig().config CONFIG = AppConfig().config
CACHE_DIR = CONFIG["application"]["cache_dir"] CACHE_DIR = CONFIG["application"]["cache_dir"]
def __init__(self, url, data, cache=True): def __init__(self, url, data):
self.max_hits = None self.max_hits = None
self.url = url self.url = url
self.data = data self.data = data
self.cache = cache
def get_data(self): def get_data(self):
"""get the data""" """get the data"""
@ -62,11 +58,6 @@ class SearchHandler:
channel_dict = self.channel_cache_link(hit) channel_dict = self.channel_cache_link(hit)
if channel_dict not in all_channels: if channel_dict not in all_channels:
all_channels.append(channel_dict) all_channels.append(channel_dict)
if self.cache:
# validate cache
pass
# self.cache_dl_vids(all_videos)
# self.cache_dl_chan(all_channels)
return return_value return return_value
@ -105,50 +96,6 @@ class SearchHandler:
} }
return channel_dict return channel_dict
def cache_dl_vids(self, all_videos):
"""video thumbs links for cache"""
vid_cache = os.path.join(self.CACHE_DIR, "videos")
vid_cached = os.listdir(vid_cache)
all_vid_cached = ignore_filelist(vid_cached)
# videos
for video_dict in all_videos:
youtube_id = video_dict["youtube_id"]
if not youtube_id + ".jpg" in all_vid_cached:
cache_path = os.path.join(vid_cache, youtube_id + ".jpg")
thumb_url = video_dict["vid_thumb"]
img_raw = requests.get(thumb_url, stream=True).raw
img = Image.open(img_raw)
width, height = img.size
if not width / height == 16 / 9:
new_height = width / 16 * 9
offset = (height - new_height) / 2
img = img.crop((0, offset, width, height - offset))
img.convert("RGB").save(cache_path)
def cache_dl_chan(self, all_channels):
"""download channel thumbs"""
chan_cache = os.path.join(self.CACHE_DIR, "channels")
chan_cached = os.listdir(chan_cache)
all_chan_cached = ignore_filelist(chan_cached)
for channel_dict in all_channels:
channel_id_cache = channel_dict["channel_id"]
channel_banner_url = channel_dict["chan_banner"]
channel_banner = channel_id_cache + "_banner.jpg"
channel_thumb_url = channel_dict["chan_thumb"]
channel_thumb = channel_id_cache + "_thumb.jpg"
# thumb
if channel_thumb_url and channel_thumb not in all_chan_cached:
cache_path = os.path.join(chan_cache, channel_thumb)
img_raw = requests.get(channel_thumb_url, stream=True).content
with open(cache_path, "wb") as f:
f.write(img_raw)
# banner
if channel_banner_url and channel_banner not in all_chan_cached:
cache_path = os.path.join(chan_cache, channel_banner)
img_raw = requests.get(channel_banner_url, stream=True).content
with open(cache_path, "wb") as f:
f.write(img_raw)
@staticmethod @staticmethod
def hit_cleanup(hit): def hit_cleanup(hit):
"""clean up and parse data from a single hit""" """clean up and parse data from a single hit"""
@ -215,7 +162,7 @@ class SearchForm:
} }
}, },
} }
look_up = SearchHandler(url, data, cache=False) look_up = SearchHandler(url, data)
search_results = look_up.get_data() search_results = look_up.get_data()
return {"results": search_results} return {"results": search_results}

View File

@ -158,7 +158,7 @@ class DownloadView(View):
url = config["application"]["es_url"] + "/ta_download/_search" url = config["application"]["es_url"] + "/ta_download/_search"
data = self.build_data(pagination_handler, show_ignored_only) data = self.build_data(pagination_handler, show_ignored_only)
search = SearchHandler(url, data, cache=False) search = SearchHandler(url, data)
videos_hits = search.get_data() videos_hits = search.get_data()
max_hits = search.max_hits max_hits = search.max_hits