Mirrors/udemy-downloader
1
0
Fork 0
mirror of https://cdm-project.com/Download-Tools/udemy-downloader.git synced 2025-04-30 06:34:33 +02:00
Code Issues Packages Projects Releases Wiki Activity

bug fix (#176)

Browse Source
This commit is contained in:
Puyodead1 2023-08-16 02:43:24 -04:00
parent f4f472de81
commit 23e7e94f16
No known key found for this signature in database
GPG Key ID: A4FA4FEC0DD353FC
1 changed files with 5 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
main.py
View File

@ -1886,11 +1886,6 @@ def main():
clazz = entry.get("_class") clazz = entry.get("_class")
if clazz == "chapter": if clazz == "chapter":
# add all lectures for the previous chapter
if len(lectures) > 0:
udemy_object["chapters"][chapter_index_counter]["lectures"] = lectures
udemy_object["chapters"][chapter_index_counter]["lecture_count"] = len(lectures)
# reset lecture tracking # reset lecture tracking
lecture_counter = 0 lecture_counter = 0
lectures = [] lectures = []
@ -1906,15 +1901,13 @@ def main():
lecture_id = entry.get("id") lecture_id = entry.get("id")
if len(udemy_object["chapters"]) == 0: if len(udemy_object["chapters"]) == 0:
# dummy chapters to handle lectures without chapters # dummy chapters to handle lectures without chapters
lectures = []
chapter_index = entry.get("object_index") chapter_index = entry.get("object_index")
chapter_title = "{0:02d} - ".format(chapter_index) + sanitize_filename(entry.get("title")) chapter_title = "{0:02d} - ".format(chapter_index) + sanitize_filename(entry.get("title"))
if chapter_title not in udemy_object["chapters"]: if chapter_title not in udemy_object["chapters"]:
udemy_object["chapters"].append({"chapter_title": chapter_title, "chapter_id": lecture_id, "chapter_index": chapter_index, "lectures": []}) udemy_object["chapters"].append({"chapter_title": chapter_title, "chapter_id": lecture_id, "chapter_index": chapter_index, "lectures": []})
chapter_index_counter += 1 chapter_index_counter += 1
if lecture_id: if lecture_id:
logger.info(f"Processing {course.index(entry)} of {len(course)}") logger.info(f"Processing {course.index(entry) + 1} of {len(course)}")
lecture_index = entry.get("object_index") lecture_index = entry.get("object_index")
lecture_title = "{0:03d} ".format(lecture_counter) + sanitize_filename(entry.get("title")) lecture_title = "{0:03d} ".format(lecture_counter) + sanitize_filename(entry.get("title"))
@ -1927,7 +1920,6 @@ def main():
lecture_id = entry.get("id") lecture_id = entry.get("id")
if len(udemy_object["chapters"]) == 0: if len(udemy_object["chapters"]) == 0:
# dummy chapters to handle lectures without chapters # dummy chapters to handle lectures without chapters
lectures = []
chapter_index = entry.get("object_index") chapter_index = entry.get("object_index")
chapter_title = "{0:02d} - ".format(chapter_index) + sanitize_filename(entry.get("title")) chapter_title = "{0:02d} - ".format(chapter_index) + sanitize_filename(entry.get("title"))
if chapter_title not in udemy_object["chapters"]: if chapter_title not in udemy_object["chapters"]:
@ -1935,7 +1927,7 @@ def main():
chapter_index_counter += 1 chapter_index_counter += 1
if lecture_id: if lecture_id:
logger.info(f"Processing {course.index(entry)} of {len(course)}") logger.info(f"Processing {course.index(entry) + 1} of {len(course)}")
lecture_index = entry.get("object_index") lecture_index = entry.get("object_index")
lecture_title = "{0:03d} ".format(lecture_counter) + sanitize_filename(entry.get("title")) lecture_title = "{0:03d} ".format(lecture_counter) + sanitize_filename(entry.get("title"))
@ -1944,6 +1936,9 @@ def main():
else: else:
logger.debug("Quiz: ID is None, skipping") logger.debug("Quiz: ID is None, skipping")
udemy_object["chapters"][chapter_index_counter]["lectures"] = lectures
udemy_object["chapters"][chapter_index_counter]["lecture_count"] = len(lectures)
udemy_object["total_chapters"] = len(udemy_object["chapters"]) udemy_object["total_chapters"] = len(udemy_object["chapters"])
udemy_object["total_lectures"] = sum([entry.get("lecture_count", 0) for entry in udemy_object["chapters"] if entry]) udemy_object["total_lectures"] = sum([entry.get("lecture_count", 0) for entry in udemy_object["chapters"] if entry])