Fix exporting extension when downloading book

1 files changed, 3 insertions, 2 deletions

diff --git a/src/book_detail.py b/src/book_detail.py
index f20d89f..49ee8a7 100644
--- a/src/book_detail.py
+++ b/src/book_detail.py
@@ -70,6 +70,7 @@ def download_book(book_id, data, path):
     home = pathlib.Path.home()
     safe_author = str_format.safe_path(data['authors'][0]) if len(data['authors']) > 0 else book_id
     safe_title = str_format.safe_path(data['title'])
-    basename = str_format.safe_path(os.path.basename(path))
-    dest = f'{home}/downloads/{safe_author}-{safe_title}-{basename}'
+    basename, ext = os.path.splitext(os.path.basename(path))
+    safe_basename = str_format.safe_path(basename)
+    dest = f'{home}/downloads/{safe_author}-{safe_title}-{safe_basename}{ext}'
     shutil.copyfile(path, dest)