fix: video_note and voice download error

tangyoha · Apr 7, 2023 · c237633 · c237633
1 parent c4e9833
commit c237633
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 7 deletions.
diff --git a/media_downloader.py b/media_downloader.py
@@ -207,7 +207,7 @@ async def _get_media_meta(
             media_obj.date.isoformat(),  # type: ignore
             file_format,
         )
-
+        file_name = validate_title(file_name)
         temp_file_name = os.path.join(app.temp_save_path, dirname, file_name)
 
         file_name = os.path.join(file_save_path, file_name)

diff --git a/tests/test_media_downloader.py b/tests/test_media_downloader.py
@@ -440,11 +440,11 @@ def test_get_media_meta(self):
         self.assertEqual(
             (
                 platform_generic_path(
-                    "/root/project/test1/2019_07/1 - voice_2019-07-25T14:53:50.ogg"
+                    "/root/project/test1/2019_07/1 - voice_2019-07-25T14_53_50.ogg"
                 ),
                 platform_generic_path(
                     os.path.join(
-                        app.temp_save_path, "test1/1 - voice_2019-07-25T14:53:50.ogg"
+                        app.temp_save_path, "test1/1 - voice_2019-07-25T14_53_50.ogg"
                     )
                 ),
                 "ogg",
@@ -679,12 +679,12 @@ def test_get_media_meta(self):
         self.assertEqual(
             (
                 platform_generic_path(
-                    "/root/project/test2/2019_07/6 - video_note_2019-07-25T14:53:50.mp4"
+                    "/root/project/test2/2019_07/6 - video_note_2019-07-25T14_53_50.mp4"
                 ),
                 platform_generic_path(
                     os.path.join(
                         app.temp_save_path,
-                        "test2/6 - video_note_2019-07-25T14:53:50.mp4",
+                        "test2/6 - video_note_2019-07-25T14_53_50.mp4",
                     )
                 ),
                 "mp4",

diff --git a/tests/utils/test_updates.py b/tests/utils/test_updates.py
@@ -1,10 +1,8 @@
 """Unittest module for update checker."""
-import os
 import sys
 import unittest
 
 import mock
-from rich.markdown import Markdown
 
 sys.path.append("..")  # Adds higher directory to python modules path.
 from utils.updates import check_for_updates