Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Added CREATE_RELATIVE_SYMLINKS option and functionality #8

Open
wants to merge 1 commit into
base: main
Choose a base branch
from
Open

Added CREATE_RELATIVE_SYMLINKS option and functionality #8

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 6 additions & 0 deletions .example-env
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -56,3 +56,9 @@ TA_HELPER_SCRIPT="/home/me/projects/ta-helper/ta-helper.py"
# to video "x.mp4" becomes bad then we should delete the x.NFO file and x.vtt
# and x.mp4 symlinks.
CLEANUP_DELETED_VIDEOS = "False"


# In some cases, e.g when your emby server is mounted over NFS and TA runs
# on a local filesystem it might be useful to create relative links for your
# movies.
CREATE_RELATIVE_SYMLINKS = "False"
2 changes: 2 additions & 0 deletions .gitignore
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,3 +2,5 @@
sendEmail_service_key.json
*.log
__pycache__
.venv

33 changes: 30 additions & 3 deletions ta-helper.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,6 +1,8 @@
import apprise
from distutils.util import strtobool
from dotenv import load_dotenv
from typing import Union
from pathlib import Path
import html2text
import logging
import os
Expand Down Expand Up @@ -32,7 +34,7 @@
APPRISE_LINK = str(os.environ.get("APPRISE_LINK"))
QUICK = bool(strtobool(os.environ.get("QUICK", 'True')))
CLEANUP_DELETED_VIDEOS = bool(strtobool(str(os.environ.get("CLEANUP_DELETED_VIDEOS"))))

CREATE_RELATIVE_SYMLINKS= bool(strtobool(str(os.environ.get("CREATE_RELATIVE_SYMLINKS"))))
logger.setLevel(os.environ.get("LOGLEVEL", "INFO"))

def setup_new_channel_resources(chan_name, chan_data):
Expand Down Expand Up @@ -62,11 +64,33 @@ def setup_new_channel_resources(chan_name, chan_data):
"<premiered>" + chan_data['channel_last_refresh'] + "</premiered>\n</episodedetails>")
f.close()


def relative_symlink(target: Union[Path, str], destination: Union[Path, str]):
"""Create a symlink pointing to ``target`` from ``location``.
Args:
target: The target of the symlink (the file/directory that is pointed to)
destination: The location of the symlink itself.
"""
target = Path(target)
destination = Path(destination)
target_dir = destination.parent
target_dir.mkdir(exist_ok=True, parents=True)
relative_source = os.path.relpath(target, target_dir)
dir_fd = os.open(str(target_dir.absolute()), os.O_RDONLY)
logger.info(f"{relative_source} -> {destination.name} in {target_dir}")
try:
os.symlink(relative_source, destination.name, dir_fd=dir_fd)
finally:
os.close(dir_fd)

def generate_new_video_nfo(chan_name, title, video_meta_data):
logger.info("Generating NFO file and subtitle symlink for %s video: %s", video_meta_data['channel']['channel_name'], video_meta_data['title'])
# TA has added a new video. Create a symlink to subtitles and an NFO file for media managers.
video_basename = os.path.splitext(video_meta_data['media_url'])[0]
os.symlink(TA_MEDIA_FOLDER + video_basename + ".en.vtt", TARGET_FOLDER + "/" + chan_name + "/" + title.replace(".mp4",".en.vtt"))
if (CREATE_RELATIVE_SYMLINKS):
relative_symlink(TA_MEDIA_FOLDER + video_basename + ".en.vtt", TARGET_FOLDER + "/" + chan_name + "/" + title.replace(".mp4",".en.vtt"))
else:
os.symlink(TA_MEDIA_FOLDER + video_basename + ".en.vtt", TARGET_FOLDER + "/" + chan_name + "/" + title.replace(".mp4",".en.vtt"))
title = title.replace('.mp4','.nfo')
f= open(TARGET_FOLDER + "/" + chan_name + "/" + title,"w+")
f.write('<?xml version="1.0" ?>\n<episodedetails>\n\t' +
Expand Down Expand Up @@ -207,7 +231,10 @@ def urlify(s):
logger.debug(video['published'] + "_" + video['youtube_id'] + "_" + urlify(video['title']) + ", " + video['media_url'])
title=video['published'] + "_" + video['youtube_id'] + "_" + urlify(video['title']) + ".mp4"
try:
os.symlink(TA_MEDIA_FOLDER + video['media_url'], TARGET_FOLDER + "/" + chan_name + "/" + title)
if (CREATE_RELATIVE_SYMLINKS):
relative_symlink(TA_MEDIA_FOLDER + video['media_url'], TARGET_FOLDER + "/" + chan_name + "/" + title)
else:
os.symlink(TA_MEDIA_FOLDER + video['media_url'], TARGET_FOLDER + "/" + chan_name + "/" + title)
# Getting here means a new video.
logger.info("Processing new video from %s: %s", chan_name, title)
if NOTIFICATIONS_ENABLED:
Expand Down