mirror of
https://github.com/tubearchivist/tubearchivist-frontend.git
synced 2024-11-22 11:50:14 +00:00
remove now unused thumbnail methods from searching module
This commit is contained in:
parent
ce5f8b6a38
commit
044b12d5a5
@ -7,15 +7,12 @@ Functionality:
|
||||
"""
|
||||
|
||||
import math
|
||||
import os
|
||||
import urllib.parse
|
||||
from datetime import datetime
|
||||
|
||||
import requests
|
||||
from home.src.config import AppConfig
|
||||
from home.src.helper import ignore_filelist
|
||||
from home.src.thumbnails import ThumbManager
|
||||
from PIL import Image
|
||||
|
||||
|
||||
class SearchHandler:
|
||||
@ -24,11 +21,10 @@ class SearchHandler:
|
||||
CONFIG = AppConfig().config
|
||||
CACHE_DIR = CONFIG["application"]["cache_dir"]
|
||||
|
||||
def __init__(self, url, data, cache=True):
|
||||
def __init__(self, url, data):
|
||||
self.max_hits = None
|
||||
self.url = url
|
||||
self.data = data
|
||||
self.cache = cache
|
||||
|
||||
def get_data(self):
|
||||
"""get the data"""
|
||||
@ -62,11 +58,6 @@ class SearchHandler:
|
||||
channel_dict = self.channel_cache_link(hit)
|
||||
if channel_dict not in all_channels:
|
||||
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
|
||||
|
||||
@ -105,50 +96,6 @@ class SearchHandler:
|
||||
}
|
||||
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
|
||||
def hit_cleanup(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()
|
||||
return {"results": search_results}
|
||||
|
||||
|
@ -158,7 +158,7 @@ class DownloadView(View):
|
||||
|
||||
url = config["application"]["es_url"] + "/ta_download/_search"
|
||||
data = self.build_data(pagination_handler, show_ignored_only)
|
||||
search = SearchHandler(url, data, cache=False)
|
||||
search = SearchHandler(url, data)
|
||||
|
||||
videos_hits = search.get_data()
|
||||
max_hits = search.max_hits
|
||||
|
Loading…
Reference in New Issue
Block a user