123 lines
5.2 KiB
Python
123 lines
5.2 KiB
Python
|
from __future__ import unicode_literals
|
||
|
|
|
||
|
|
from datetime import datetime
|
||
|
|
import json
|
||
|
|
|
||
|
|
from .common import InfoExtractor
|
||
|
|
from ..compat import compat_str
|
||
|
|
from ..utils import ExtractorError
|
||
|
|
from ..utils import int_or_none
|
||
|
|
|
||
|
|
|
||
|
|
class SkillshareBaseIE(InfoExtractor):
|
||
|
|
_NETRC_MACHINE = 'udemy'
|
||
|
|
|
||
|
|
_TN_RE = r"uploads/video/thumbnails/[0-9a-f]+/(?P<width>[0-9]+)-(?P<height>[0-9]+)"
|
||
|
|
_LOGIN_URL = "https://api.skillshare.com/login"
|
||
|
|
_VIDEO_URL = "https://api.skillshare.com/sessions/%s/download"
|
||
|
|
|
||
|
|
def _real_initialize(self):
|
||
|
|
self._login()
|
||
|
|
|
||
|
|
def _login(self):
|
||
|
|
username, password = self._get_login_info()
|
||
|
|
if username is None or password is None:
|
||
|
|
self.raise_login_required("An email and password is needed to download any video (even non-premium ones)")
|
||
|
|
|
||
|
|
data = {
|
||
|
|
"email": username,
|
||
|
|
"password": password
|
||
|
|
}
|
||
|
|
headers = {
|
||
|
|
"Content-Type": "application/json"
|
||
|
|
}
|
||
|
|
user_json = self._download_json(self._LOGIN_URL,
|
||
|
|
None,
|
||
|
|
note="Logging in",
|
||
|
|
errnote="Error logging in, make sure the email and password is correct",
|
||
|
|
data=json.dumps(data).encode(),
|
||
|
|
headers=headers)
|
||
|
|
|
||
|
|
user_type = user_json.get("membership_label", "Premium Member")
|
||
|
|
if user_type == "Basic Member":
|
||
|
|
self._user_type = 0
|
||
|
|
elif user_type == "Premium Member":
|
||
|
|
self._user_type = 2
|
||
|
|
else:
|
||
|
|
raise ExtractorError("User type %s unknown" % user_json["membership_label"])
|
||
|
|
|
||
|
|
|
||
|
|
# I can find no way of linking to a specific video so only entire course downloads are available.
|
||
|
|
class SkillshareCourseIE(SkillshareBaseIE):
|
||
|
|
IE_NAME = 'skillshare:course'
|
||
|
|
IE_DESC = 'skillshare.com classes'
|
||
|
|
_VALID_URL = r'https?://(?:www\.)?skillshare\.com/classes/[^/]+/(?P<id>[0-9]+)'
|
||
|
|
|
||
|
|
_CLASS_URL = "https://api.skillshare.com/classes/%s"
|
||
|
|
|
||
|
|
_TEST = {
|
||
|
|
"url": "https://www.skillshare.com/classes/Blender-3D-Fire-Smoke-Simulation-Guide/1850126092",
|
||
|
|
"only_matching": True
|
||
|
|
}
|
||
|
|
|
||
|
|
def _real_extract(self, url):
|
||
|
|
# Technically the SKU, not ID but the SKU is a more universal identifier.
|
||
|
|
class_id = self._match_id(url)
|
||
|
|
class_json = self._download_json(self._CLASS_URL % class_id,
|
||
|
|
None,
|
||
|
|
note="Getting class details",
|
||
|
|
errnote="Error getting class details")
|
||
|
|
|
||
|
|
if class_json.get("enrollment_type", 0) > self._user_type:
|
||
|
|
raise ExtractorError("This course requires a premium account and thus can't be downloaded")
|
||
|
|
|
||
|
|
lessons_json = []
|
||
|
|
# Pretty sure all classes only have one unit but flattening just in case.
|
||
|
|
for unit_json in class_json["_embedded"]["units"]["_embedded"]["units"]:
|
||
|
|
lessons_json += (unit_json["_embedded"]["sessions"]["_embedded"]["sessions"])
|
||
|
|
|
||
|
|
videos = []
|
||
|
|
for lesson_json in lessons_json:
|
||
|
|
lesson_thumbnail_urls = [
|
||
|
|
lesson_json.get("video_thumbnail_url", ""),
|
||
|
|
lesson_json.get("video_thumbnail_url", ""),
|
||
|
|
lesson_json.get("image_thumbnail", "")
|
||
|
|
]
|
||
|
|
lesson_thumbnails_json = []
|
||
|
|
for lesson_thumbnail_url in lesson_thumbnail_urls:
|
||
|
|
lesson_thumbnails_json.append({
|
||
|
|
"url": lesson_thumbnail_url,
|
||
|
|
"width": int_or_none(self._search_regex(self._TN_RE, lesson_thumbnail_url, "width", fatal=False)),
|
||
|
|
"height": int_or_none(self._search_regex(self._TN_RE, lesson_thumbnail_url, "height", fatal=False)),
|
||
|
|
})
|
||
|
|
|
||
|
|
try:
|
||
|
|
lesson_timestamp_dt = datetime.strptime(lesson_json.get("create_time", ""), "%Y-%m-%d %H:%M:%S")
|
||
|
|
lesson_timestamp = int(lesson_timestamp_dt.strftime("%s"))
|
||
|
|
except ValueError:
|
||
|
|
lesson_timestamp = None
|
||
|
|
|
||
|
|
videos.append({
|
||
|
|
"id": str(lesson_json["id"]),
|
||
|
|
"title": lesson_json.get("title"),
|
||
|
|
"url": self._VIDEO_URL % str(lesson_json["id"]),
|
||
|
|
"ext": "mp4",
|
||
|
|
"thumbnails": lesson_thumbnails_json,
|
||
|
|
"uploader": class_json["_embedded"].get("teacher", {}).get("full_name"),
|
||
|
|
"creator": class_json["_embedded"].get("teacher", {}).get("full_name"),
|
||
|
|
"timestamp": lesson_timestamp,
|
||
|
|
"uploader_id": str(class_json["_embedded"].get("teacher", {}).get("username", 0)),
|
||
|
|
"categories": [class_json.get("category")],
|
||
|
|
"chapter": lesson_json.get("_links", {}).get("unit", {}).get("title"),
|
||
|
|
"chapter_id": compat_str(lesson_json.get("unit_id"))
|
||
|
|
})
|
||
|
|
|
||
|
|
return {
|
||
|
|
"id": class_id,
|
||
|
|
"title": class_json.get("title"),
|
||
|
|
"uploader": class_json["_embedded"].get("teacher", {}).get("full_name"),
|
||
|
|
"uploader_id": str(class_json["_embedded"].get("teacher", {}).get("username", 0)),
|
||
|
|
"_type": "playlist",
|
||
|
|
"entries": videos
|
||
|
|
}
|