def refine(video, **kwargs):
"""
:param video:
:param embedded_subtitles:
:param kwargs:
:return:
"""
client = DroneManager.get_client(video, kwargs)
additional_data = client.get_additional_data(video)
if additional_data:
if "scene_name" in additional_data:
client.update_video(video, additional_data["scene_name"])
elif "original_filepath" in additional_data:
client.update_video(
video,
os.path.splitext(additional_data["original_filepath"])[0])
if "release_group" in additional_data and not video.release_group:
video.release_group = remove_crap_from_fn(
additional_data["release_group"])
def get_guess(self, video, scene_name):
"""
run guessit on scene_name
:param video:
:param scene_name:
:return:
"""
ext = os.path.splitext(video.name)[1]
guess_from = remove_crap_from_fn(scene_name + ext)
# guess
hints = {
"single_value": True,
"type": "movie",
}
return guess_from, guessit(guess_from, options=hints)
def get_guess(self, video, scene_name):
"""
run guessit on scene_name
:param video:
:param scene_name:
:return:
"""
ext = os.path.splitext(video.name)[1]
guess_from = remove_crap_from_fn(scene_name + ext)
# guess
hints = {
"single_value": True,
"type": "movie",
}
return guess_from, guessit(guess_from, options=hints)
def update_video(video, fn):
guess_from = remove_crap_from_fn(fn)
logger.debug(u"Got original filename: %s", guess_from)
# guess
hints = {
"single_value": True,
"type": "episode" if isinstance(video, Episode) else "movie",
}
guess = guessit(guess_from, options=hints)
for attr in ("release_group", "format",):
if attr in guess:
value = guess.get(attr)
logger.debug(u"%s: Filling attribute %s: %s", video.name, attr, value)
setattr(video, attr, value)
video.original_name = os.path.basename(guess_from)
def refine(video, **kwargs):
"""
:param video:
:param embedded_subtitles:
:param kwargs:
:return:
"""
client = DroneManager.get_client(video, kwargs)
additional_data = client.get_additional_data(video)
if additional_data:
if "scene_name" in additional_data:
client.update_video(video, additional_data["scene_name"])
elif "original_filepath" in additional_data:
client.update_video(video, os.path.splitext(additional_data["original_filepath"])[0])
if "release_group" in additional_data and not video.release_group:
video.release_group = remove_crap_from_fn(additional_data["release_group"])
def update_video(video, fn):
guess_from = remove_crap_from_fn(fn)
logger.debug(u"Got original filename: %s", guess_from)
# guess
hints = {
"single_value": True,
"type": "episode" if isinstance(video, Episode) else "movie",
}
guess = guessit(guess_from, options=hints)
for attr in (
"release_group",
"format",
):
if attr in guess:
value = guess.get(attr)
logger.debug(u"%s: Filling attribute %s: %s", video.name, attr,
value)
setattr(video, attr, value)
video.original_name = os.path.basename(guess_from)
