refactor: mypy

vzhd1701 · Oct 18, 2023 · 3e925ec · 3e925ec
1 parent 052d10a
commit 3e925ec
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 8 deletions.
diff --git a/evernote_backup/cli.py b/evernote_backup/cli.py
@@ -89,13 +89,13 @@
 
 @click.group(cls=NaturalOrderGroup)
 @optgroup.group("Verbosity", cls=MutuallyExclusiveOptionGroup)  # type: ignore
-@optgroup.option(  # type: ignore
+@optgroup.option(
     "--quiet",
     "-q",
     is_flag=True,
     help="Quiet mode, output only critical errors.",
 )
-@optgroup.option(  # type: ignore
+@optgroup.option(
     "--verbose",
     "-v",
     is_flag=True,

diff --git a/evernote_backup/evernote_client.py b/evernote_backup/evernote_client.py
@@ -81,7 +81,7 @@ def user(self) -> str:
             self._user = self.user_store.getUser().username
         return self._user
 
-    def get_note_store(self, shard_id: str = None) -> "Store":
+    def get_note_store(self, shard_id: Optional[str] = None) -> "Store":
         if shard_id is None:
             shard_id = self.shard
 

diff --git a/evernote_backup/note_storage.py b/evernote_backup/note_storage.py
@@ -363,7 +363,7 @@ def _get_notes_by_notebook(self, notebook_guid: str) -> List[str]:
                 (notebook_guid,),
             )
 
-            sorted_notes = sorted(cur, key=lambda x: x["title"])  # type: ignore
+            sorted_notes = sorted(cur, key=lambda x: x["title"])
 
             return [r["guid"] for r in sorted_notes]
 

diff --git a/evernote_backup/note_synchronizer.py b/evernote_backup/note_synchronizer.py
@@ -4,7 +4,7 @@
 import struct
 import threading
 from concurrent.futures import FIRST_EXCEPTION, ThreadPoolExecutor, as_completed, wait
-from typing import Any, Dict, Iterable, List, Tuple
+from typing import Any, Dict, Iterable, List, Optional, Tuple
 
 from click import progressbar
 from evernote.edam.notestore.ttypes import SyncChunk
@@ -106,7 +106,7 @@ def __init__(  # noqa: WPS211
         self._thread_data = threading.local()
         self._note_client: EvernoteClientSync
 
-    def __call__(self, note_id: str, auth_data: NotebookAuth = None) -> Note:
+    def __call__(self, note_id: str, auth_data: Optional[NotebookAuth] = None) -> Note:
         self.memory_manager.wait_till_enough_memory()
 
         if self.stop:
@@ -263,7 +263,7 @@ def _sync_chunks(self) -> None:
             return
 
         last_usn = current_usn
-        with progressbar(
+        with progressbar(  # type: ignore
             length=remote_usn - current_usn,
             show_pos=True,
             file=get_progress_output(),
@@ -350,7 +350,7 @@ def _download_scheduled_notes(self, notes_to_sync: Tuple[NoteForSync, ...]) -> N
         logger.debug(f"Sync worker threads: {self.max_download_workers}")
 
         with ThreadPoolExecutor(max_workers=self.max_download_workers) as executor:
-            with progressbar(
+            with progressbar(  # type: ignore
                 length=len(notes_to_sync),
                 show_pos=True,
                 file=get_progress_output(),