mirror of
https://github.com/yt-dlp/yt-dlp.git
synced 2024-12-27 05:35:52 +01:00
592e97e855
We could only know if we had to delete the original file, but this system allows to specify us more files (like subtitles).
113 lines
4.1 KiB
Python
113 lines
4.1 KiB
Python
from __future__ import unicode_literals
|
|
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
|
|
from .common import PostProcessor
|
|
from ..compat import (
|
|
subprocess_check_output
|
|
)
|
|
from ..utils import (
|
|
check_executable,
|
|
hyphenate_date,
|
|
)
|
|
|
|
|
|
class XAttrMetadataPP(PostProcessor):
|
|
|
|
#
|
|
# More info about extended attributes for media:
|
|
# http://freedesktop.org/wiki/CommonExtendedAttributes/
|
|
# http://www.freedesktop.org/wiki/PhreedomDraft/
|
|
# http://dublincore.org/documents/usageguide/elements.shtml
|
|
#
|
|
# TODO:
|
|
# * capture youtube keywords and put them in 'user.dublincore.subject' (comma-separated)
|
|
# * figure out which xattrs can be used for 'duration', 'thumbnail', 'resolution'
|
|
#
|
|
|
|
def run(self, info):
|
|
""" Set extended attributes on downloaded file (if xattr support is found). """
|
|
|
|
# This mess below finds the best xattr tool for the job and creates a
|
|
# "write_xattr" function.
|
|
try:
|
|
# try the pyxattr module...
|
|
import xattr
|
|
|
|
def write_xattr(path, key, value):
|
|
return xattr.setxattr(path, key, value)
|
|
|
|
except ImportError:
|
|
if os.name == 'nt':
|
|
# Write xattrs to NTFS Alternate Data Streams:
|
|
# http://en.wikipedia.org/wiki/NTFS#Alternate_data_streams_.28ADS.29
|
|
def write_xattr(path, key, value):
|
|
assert ':' not in key
|
|
assert os.path.exists(path)
|
|
|
|
ads_fn = path + ":" + key
|
|
with open(ads_fn, "wb") as f:
|
|
f.write(value)
|
|
else:
|
|
user_has_setfattr = check_executable("setfattr", ['--version'])
|
|
user_has_xattr = check_executable("xattr", ['-h'])
|
|
|
|
if user_has_setfattr or user_has_xattr:
|
|
|
|
def write_xattr(path, key, value):
|
|
if user_has_setfattr:
|
|
cmd = ['setfattr', '-n', key, '-v', value, path]
|
|
elif user_has_xattr:
|
|
cmd = ['xattr', '-w', key, value, path]
|
|
|
|
subprocess_check_output(cmd)
|
|
|
|
else:
|
|
# On Unix, and can't find pyxattr, setfattr, or xattr.
|
|
if sys.platform.startswith('linux'):
|
|
self._downloader.report_error(
|
|
"Couldn't find a tool to set the xattrs. "
|
|
"Install either the python 'pyxattr' or 'xattr' "
|
|
"modules, or the GNU 'attr' package "
|
|
"(which contains the 'setfattr' tool).")
|
|
else:
|
|
self._downloader.report_error(
|
|
"Couldn't find a tool to set the xattrs. "
|
|
"Install either the python 'xattr' module, "
|
|
"or the 'xattr' binary.")
|
|
|
|
# Write the metadata to the file's xattrs
|
|
self._downloader.to_screen('[metadata] Writing metadata to file\'s xattrs')
|
|
|
|
filename = info['filepath']
|
|
|
|
try:
|
|
xattr_mapping = {
|
|
'user.xdg.referrer.url': 'webpage_url',
|
|
# 'user.xdg.comment': 'description',
|
|
'user.dublincore.title': 'title',
|
|
'user.dublincore.date': 'upload_date',
|
|
'user.dublincore.description': 'description',
|
|
'user.dublincore.contributor': 'uploader',
|
|
'user.dublincore.format': 'format',
|
|
}
|
|
|
|
for xattrname, infoname in xattr_mapping.items():
|
|
|
|
value = info.get(infoname)
|
|
|
|
if value:
|
|
if infoname == "upload_date":
|
|
value = hyphenate_date(value)
|
|
|
|
byte_value = value.encode('utf-8')
|
|
write_xattr(filename, xattrname, byte_value)
|
|
|
|
return [], info
|
|
|
|
except (subprocess.CalledProcessError, OSError):
|
|
self._downloader.report_error("This filesystem doesn't support extended attributes. (You may have to enable them in your /etc/fstab)")
|
|
return [], info
|