From 263a429bcb56b4a2754b5f7c2414b15f20635e3b Mon Sep 17 00:00:00 2001
From: Ovear <ovearj@gmail.com>
Date: Thu, 29 Sep 2022 16:10:59 +0800
Subject: [PATCH] Feat: save html description in case we need it

---
 Modules/config.py   | 8 ++++++++
 Modules/download.py | 5 ++++-
 2 files changed, 12 insertions(+), 1 deletion(-)

diff --git a/Modules/config.py b/Modules/config.py
index ef8959d..4f32cb0 100644
--- a/Modules/config.py
+++ b/Modules/config.py
@@ -127,6 +127,13 @@ parser.add_argument(
     action="store_true",
     help="download description as a JSON list",
 )
+parser.add_argument(
+    "--html-description",
+    "-hd",
+    dest="html_description",
+    action="store_true",
+    help="save description in original html format",
+)
 parser.add_argument(
     "--login",
     action="store_true",
@@ -167,6 +174,7 @@ login = args.login
 check = args.check
 index = args.index
 submissions = args.submissions
+html_description = args.html_description
 json_description = args.json_description
 metadata = args.metadata
 dont_redownload = args.redownload
diff --git a/Modules/download.py b/Modules/download.py
index 4b8e1e5..b2b26ec 100644
--- a/Modules/download.py
+++ b/Modules/download.py
@@ -76,7 +76,10 @@ def download(path):
             idx.write(f"({view_id})\n")
 
     if config.metadata is True:
-        dsc = s.find(class_="submission-description").text.strip().replace("\r\n", "\n")
+        if config.html_description is True:
+            dsc = s.find(class_="submission-description").prettify()
+        else:
+            dsc = s.find(class_="submission-description").text.strip().replace("\r\n", "\n")
         if config.json_description is True:
             dsc = []
         data = {