mirror of
https://github.com/yt-dlp/yt-dlp.git
synced 2024-11-10 03:10:38 +00:00
168da92b9a
In particular, explicitly require NT before trying ADS, and do not try to parse process output that may be localized.
111 lines
4.1 KiB
Python
111 lines
4.1 KiB
Python
import os
|
|
import subprocess
|
|
import sys
|
|
|
|
from .common import PostProcessor
|
|
from ..utils import (
|
|
check_executable,
|
|
hyphenate_date,
|
|
preferredencoding,
|
|
)
|
|
|
|
|
|
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(key.find(":") < 0)
|
|
assert(path.find(":") < 0)
|
|
assert(os.path.exists(path))
|
|
|
|
ads_fn = path + ":" + key
|
|
with open(ads_fn, "w") 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(preferredencoding())
|
|
write_xattr(filename, xattrname, byte_value)
|
|
|
|
return True, 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 False, info
|
|
|