Add option for archiver full context

2021-06-11 15:31:11 +10:00
parent 4be75fd48a
commit 6eeadc8821
4 changed files with 20 additions and 1 deletions
--- a/bdfr/archiver.py
+++ b/bdfr/archiver.py
@@ -61,6 +61,9 @@ class Archiver(RedditConnector):
            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)):
+        if self.args.full_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
        archive_entry = self._pull_lever_entry_factory(praw_item)
        if self.args.format == 'json':
            self._write_entry_json(archive_entry)