Pylance typing

Fix Pylance warnings for typing
This commit is contained in:
OMEGARAZER
2022-11-30 21:12:03 -05:00
parent b438f81a43
commit 69fa1f3f09
4 changed files with 7 additions and 7 deletions

View File

@@ -7,7 +7,7 @@ from praw.models import Comment, Submission
class BaseArchiveEntry(ABC):
def __init__(self, source: (Comment, Submission)):
def __init__(self, source: Comment | Submission):
self.source = source
self.post_details: dict = {}

View File

@@ -65,7 +65,7 @@ class Archiver(RedditConnector):
return results
@staticmethod
def _pull_lever_entry_factory(praw_item: (praw.models.Submission, praw.models.Comment)) -> BaseArchiveEntry:
def _pull_lever_entry_factory(praw_item: praw.models.Submission | praw.models.Comment) -> BaseArchiveEntry:
if isinstance(praw_item, praw.models.Submission):
return SubmissionArchiveEntry(praw_item)
elif isinstance(praw_item, praw.models.Comment):
@@ -73,7 +73,7 @@ class Archiver(RedditConnector):
else:
raise ArchiverError(f'Factory failed to classify item of type {type(praw_item).__name__}')
def write_entry(self, praw_item: (praw.models.Submission, praw.models.Comment)):
def write_entry(self, praw_item: praw.models.Submission | praw.models.Comment):
if self.args.comment_context and isinstance(praw_item, praw.models.Comment):
logger.debug(f'Converting comment {praw_item.id} to submission {praw_item.submission.id}')
praw_item = praw_item.submission

View File

@@ -34,7 +34,7 @@ class FileNameFormatter:
self.directory_format_string: list[str] = directory_format_string.split('/')
self.time_format_string = time_format_string
def _format_name(self, submission: (Comment, Submission), format_string: str) -> str:
def _format_name(self, submission: Comment | Submission, format_string: str) -> str:
if isinstance(submission, Submission):
attributes = self._generate_name_dict_from_submission(submission)
elif isinstance(submission, Comment):