1
0

youtubedl_wrapper.py 4.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125
  1. #
  2. # This file is part of stov, written by Helmut Pozimski 2012-2017.
  3. #
  4. # stov is free software: you can redistribute it and/or modify
  5. # it under the terms of the GNU General Public License as published by
  6. # the Free Software Foundation, version 2 of the License.
  7. #
  8. # stov is distributed in the hope that it will be useful,
  9. # but WITHOUT ANY WARRANTY; without even the implied warranty of
  10. # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  11. # GNU General Public License for more details.
  12. #
  13. # You should have received a copy of the GNU General Public License
  14. # along with stov. If not, see <http://www.gnu.org/licenses/>.
  15. # -*- coding: utf8 -*-
  16. """ Provides a wrapper around youtube-dl"""
  17. import subprocess
  18. import sys
  19. import logging
  20. from lib_stov import stov_exceptions
  21. LOGGER = logging.getLogger("stov")
  22. def get_ids(conf, url, reverse=False):
  23. """
  24. Retrieves the IDs
  25. :param conf: configuration object
  26. :type conf: lib_stov.configuration.Conf
  27. :param url: URL to pass to youtube-dl
  28. :type url: str
  29. :param reverse: look up a playlist in reverse order to get \
  30. the recent videos first
  31. :type reverse: bool
  32. :return: video IDs
  33. :rtype: list
  34. """
  35. videos_list = []
  36. if conf.outputlevel == "verbose":
  37. stderr = sys.stderr
  38. else:
  39. stderr = open("/dev/null", "w")
  40. LOGGER.debug(_("Executing command: %s %s %s %s %s %s"),
  41. conf.values["youtube-dl"], "--max-downloads",
  42. conf.values["maxvideos"], "-i", "--get-id", url)
  43. try:
  44. if reverse:
  45. video_ids = subprocess.check_output(
  46. [conf.values["youtube-dl"], "--max-downloads",
  47. conf.values["maxvideos"], "-i", "--playlist-reverse",
  48. "--get-id", url], stderr=stderr)
  49. else:
  50. video_ids = subprocess.check_output(
  51. [conf.values["youtube-dl"], "--max-downloads",
  52. conf.values["maxvideos"],
  53. "-i", "--get-id", url], stderr=stderr)
  54. except subprocess.CalledProcessError as error_message:
  55. video_ids = error_message.output
  56. video_ids = video_ids.decode(sys.stdout.encoding).strip()
  57. for video in video_ids.split("\n"):
  58. videos_list.append(video)
  59. LOGGER.debug("generated list: %s", videos_list)
  60. return videos_list
  61. def get_title(conf, url):
  62. """
  63. Retrieves the title of a specified video
  64. :param conf: configuration object
  65. :type conf: lib_stov.configuration.Conf
  66. :param url: URL to pass to youtube-dl
  67. :type url: str
  68. """
  69. if conf.outputlevel == "verbose":
  70. stderr = sys.stderr
  71. else:
  72. stderr = open("/dev/null", "w")
  73. LOGGER.debug(_("Executing command: %s %s %s"),
  74. conf.values["youtube-dl"], "--get-title", url)
  75. video_title = subprocess.check_output([
  76. conf.values["youtube-dl"], "--get-title", url], stderr=stderr)
  77. video_title = video_title.decode(sys.stdout.encoding)
  78. return video_title
  79. def download_video(conf, url, itag_value):
  80. """
  81. Downloads a video from a specified url using youtube-dl.
  82. :param conf: configuration object
  83. :type conf: lib_stov.configuration.Conf
  84. :param itag_value: video and audio parameter
  85. :type itag_value: str
  86. :param url: URL to pass to youtube-dl
  87. :type url: str
  88. """
  89. try:
  90. LOGGER.debug(_("Executing command: %s -f %s %s"),
  91. conf.values["youtube-dl"], itag_value, url)
  92. if conf.outputlevel == "default":
  93. subprocess.check_call([conf.values["youtube-dl"], "-f %s"
  94. % itag_value,
  95. "-o", "%(title)s-%(id)s.%(ext)s", url],
  96. stderr=sys.stderr,
  97. stdout=open("/dev/null", "w"))
  98. elif conf.outputlevel == "verbose":
  99. subprocess.check_call([conf.values["youtube-dl"], "-f %s"
  100. % itag_value,
  101. "-o", "%(title)s-%(id)s.%(ext)s", url],
  102. stderr=sys.stderr, stdout=sys.stdout)
  103. elif conf.outputlevel == "quiet":
  104. subprocess.check_call([conf.values["youtube-dl"], "-f %s/%s"
  105. % itag_value,
  106. "-o", "%(title)s-%(id)s.%(ext)s", url],
  107. stderr=open("/dev/null", "w"),
  108. stdout=open("/dev/null", "w"))
  109. except subprocess.CalledProcessError as error:
  110. LOGGER.debug(_("Error while calling youtube-dl: %s"), error.output)
  111. raise stov_exceptions.YoutubeDlCallFailed()