zdf_mediathek.py 5.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155
  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. """ This module implements support for subscriptions from the ZDF Mediathek"""
  17. import json
  18. import logging
  19. import urllib.request
  20. from datetime import datetime, timedelta
  21. from lib_stov import stov_exceptions
  22. LOGGER = logging.getLogger("stov")
  23. class ZDFChannel(object):
  24. """Stores the relevant attributes of a ZDF Mediathek channel"""
  25. def __init__(self, title, videos=None):
  26. if videos is None:
  27. videos = []
  28. self.title = title
  29. self.videos = videos
  30. self.type = "channel"
  31. class ZDFVideo(object):
  32. """Stores the relevant attributes of a single video in the ZDF
  33. Mediathek
  34. """
  35. def __init__(self, title, url):
  36. self.title = title
  37. self.video_id = url
  38. class Connector(object):
  39. """Connector class performing operations against the API"""
  40. def __init__(self, subscription_type, name, conf, search=""):
  41. if subscription_type == "user":
  42. self._type = "channel"
  43. else:
  44. self._type = subscription_type
  45. self._name = name
  46. self._conf = conf
  47. self._search = search
  48. if self._type != "channel":
  49. raise stov_exceptions.TypeNotSupported()
  50. def _fetch_videos(self, existing_videos):
  51. """ Fetches the videos of the day before from the ZDF API and returns
  52. a list of newly added videos.
  53. :param existing_videos: videos that already exist in the database
  54. :type existing_videos: list
  55. :return: List of all newly retrieved videos
  56. :rtype: list
  57. """
  58. videos_list = []
  59. new_videos = []
  60. today = datetime.today()
  61. for i in range(7, -1, -1):
  62. connection = urllib.request.urlopen(
  63. "https://zdf-cdn.live.cellular.de/mediathekV2/broadcast-"
  64. "missed/%s"
  65. % (today-timedelta(days=i)).strftime("%Y-%m-%d"))
  66. data = connection.read().decode("utf-8")
  67. response = json.loads(data)
  68. for cluster in response["broadcastCluster"]:
  69. for broadcast in cluster["teaser"]:
  70. if self._name in broadcast["headline"]:
  71. if self._search:
  72. if self._search in broadcast["titel"]:
  73. new_videos.append((broadcast["sharingUrl"],
  74. broadcast["titel"]))
  75. else:
  76. new_videos.append((broadcast["sharingUrl"],
  77. broadcast["titel"]))
  78. if new_videos:
  79. for broadcast in new_videos:
  80. video_exists = False
  81. if existing_videos:
  82. for existing_video in existing_videos:
  83. if broadcast[0] == existing_video.site_id:
  84. video_exists = True
  85. break
  86. if not video_exists:
  87. videos_list.append(ZDFVideo(broadcast[1], broadcast[0]))
  88. return videos_list
  89. def parse_api_data(self, existing_videos):
  90. """ Takes the existing videos passed to it and wraps them into a
  91. channel object.
  92. :param existing_videos: list of existing_videos
  93. :type existing_videos: list
  94. :return: Channel object
  95. :rtype: ZDFChannel
  96. """
  97. videos = self._fetch_videos(existing_videos)
  98. channel = ZDFChannel(self._name, videos)
  99. return channel
  100. @staticmethod
  101. def construct_video_url(url):
  102. """
  103. Compatibility method, just returns the url
  104. :param url: The url to return
  105. :type url: str
  106. :return: url
  107. :rtype: str
  108. """
  109. return url
  110. @staticmethod
  111. def get_quality_parameter(config):
  112. """
  113. Determines which quality value results from codec and resolution
  114. settings and returns it
  115. :param config: configuration object
  116. :type config: lib_stov.configuration.Conf
  117. :return: itag value
  118. :rtype: str
  119. """
  120. LOGGER.debug(_("Trying to determine the itag value for youtube-dl from"
  121. " your quality and codec settings."))
  122. quality_value = ""
  123. if config.values["videocodec"] == "flv":
  124. if config.values["maxresolution"] == "480p":
  125. quality_value = "hds-1489"
  126. elif config.values["videocodec"] == "mp4":
  127. if config.values["maxresolution"] == "720p":
  128. quality_value = "hls-3286"
  129. if quality_value:
  130. LOGGER.debug(_("Found value: %s."), quality_value)
  131. return quality_value + "/" + config.values["videocodec"]
  132. else:
  133. LOGGER.debug(_("Could not determine an itag value "
  134. "from the configuration"))
  135. return "hls-3286" + "/" + config.values["videocodec"]