fix quizzes in info

This commit is contained in:
Puyodead1 2023-10-20 16:53:56 -04:00
parent a461f5fc86
commit 935b2a41f3
No known key found for this signature in database
GPG Key ID: A4FA4FEC0DD353FC

29
main.py
View File

@ -368,7 +368,6 @@ class Udemy:
cj = browser_cookie3.vivaldi()
def _get_quiz(self, quiz_id):
print(portal_name)
self.session._headers.update(
{
"Host": "{portal_name}.udemy.com".format(portal_name=portal_name),
@ -376,7 +375,6 @@ class Udemy:
}
)
url = QUIZ_URL.format(portal_name=portal_name, quiz_id=quiz_id)
print(url)
try:
resp = self.session._get(url).json()
except conn_error as error:
@ -989,6 +987,7 @@ class Udemy:
else:
logger.warning(f"Unknown asset type: {asset_type}")
if asset != None:
stream_urls = asset.get("stream_urls")
if stream_urls != None:
# not encrypted
@ -1011,6 +1010,7 @@ class Udemy:
"sources_count": sources_count,
"is_encrypted": False,
"asset_id": asset.get("id"),
"type": asset.get("asset_type")
}
else:
lecture.pop("data") # remove the raw data object after processing
@ -1024,6 +1024,7 @@ class Udemy:
"sources_count": 0,
"is_encrypted": False,
"asset_id": asset.get("id"),
"type": asset.get("asset_type")
}
else:
# encrypted
@ -1047,6 +1048,7 @@ class Udemy:
"sources_count": sources_count,
"is_encrypted": True,
"asset_id": asset.get("id"),
"type": asset.get("asset_type")
}
else:
@ -1061,6 +1063,15 @@ class Udemy:
"sources_count": 0,
"is_encrypted": False,
"asset_id": asset.get("id"),
"type": asset.get("asset_type")
}
else:
lecture = {
**lecture,
"assets": retVal,
"assets_count": len(retVal),
"asset_id": lecture_data.get("id"),
"type": lecture_data.get("type")
}
return lecture
@ -1845,29 +1856,37 @@ def _print_course_info(udemy: Udemy, udemy_object: dict):
parsed_lecture = udemy._parse_lecture(lecture)
lecture_sources = parsed_lecture.get("sources")
lecture_is_encrypted = parsed_lecture.get("is_encrypted")
lecture_is_encrypted = parsed_lecture.get("is_encrypted", None)
lecture_extension = parsed_lecture.get("extension")
lecture_asset_count = parsed_lecture.get("assets_count")
lecture_subtitles = parsed_lecture.get("subtitles")
lecture_video_sources = parsed_lecture.get("video_sources")
lecture_type = parsed_lecture.get("type")
lecture_qualities = []
if lecture_sources:
lecture_sources = sorted(lecture_sources, key=lambda x: int(x.get("height")), reverse=True)
if lecture_video_sources:
lecture_video_sources = sorted(lecture_video_sources, key=lambda x: int(x.get("height")), reverse=True)
if lecture_is_encrypted:
if lecture_is_encrypted and lecture_video_sources != None:
lecture_qualities = ["{}@{}x{}".format(x.get("type"), x.get("width"), x.get("height")) for x in lecture_video_sources]
elif not lecture_is_encrypted and lecture_sources:
elif lecture_is_encrypted == False and lecture_sources != None:
lecture_qualities = ["{}@{}x{}".format(x.get("type"), x.get("height"), x.get("width")) for x in lecture_sources]
if lecture_extension:
continue
logger.info(" > Lecture: {} ({} of {})".format(lecture_title, lecture_index, chapter_lecture_count))
logger.info(" > Type: {}".format(lecture_type))
if lecture_is_encrypted != None:
logger.info(" > DRM: {}".format(lecture_is_encrypted))
if lecture_asset_count:
logger.info(" > Asset Count: {}".format(lecture_asset_count))
if lecture_subtitles:
logger.info(" > Captions: {}".format(", ".join([x.get("language") for x in lecture_subtitles])))
if lecture_qualities:
logger.info(" > Qualities: {}".format(lecture_qualities))
if chapter_index != chapter_count: