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

15
main.py
View File

@ -1886,11 +1886,6 @@ def main():
clazz = entry.get("_class")
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
lecture_counter = 0
lectures = []
@ -1906,15 +1901,13 @@ def main():
lecture_id = entry.get("id")
if len(udemy_object["chapters"]) == 0:
# dummy chapters to handle lectures without chapters
lectures = []
chapter_index = entry.get("object_index")
chapter_title = "{0:02d} - ".format(chapter_index) + sanitize_filename(entry.get("title"))
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": []})
chapter_index_counter += 1
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_title = "{0:03d} ".format(lecture_counter) + sanitize_filename(entry.get("title"))
@ -1927,7 +1920,6 @@ def main():
lecture_id = entry.get("id")
if len(udemy_object["chapters"]) == 0:
# dummy chapters to handle lectures without chapters
lectures = []
chapter_index = entry.get("object_index")
chapter_title = "{0:02d} - ".format(chapter_index) + sanitize_filename(entry.get("title"))
if chapter_title not in udemy_object["chapters"]:
@ -1935,7 +1927,7 @@ def main():
chapter_index_counter += 1
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_title = "{0:03d} ".format(lecture_counter) + sanitize_filename(entry.get("title"))
@ -1944,6 +1936,9 @@ def main():
else:
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_lectures"] = sum([entry.get("lecture_count", 0) for entry in udemy_object["chapters"] if entry])