50 lines
1.7 KiB
Python
50 lines
1.7 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import logging
|
|
|
|
import praw.models
|
|
|
|
from bdfr.archive_entry.base_archive_entry import BaseArchiveEntry
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class SubmissionArchiveEntry(BaseArchiveEntry):
|
|
def __init__(self, submission: praw.models.Submission):
|
|
super(SubmissionArchiveEntry, self).__init__(submission)
|
|
|
|
def compile(self) -> dict:
|
|
comments = self._get_comments()
|
|
self._get_post_details()
|
|
out = self.post_details
|
|
out["comments"] = comments
|
|
return out
|
|
|
|
def _get_post_details(self):
|
|
self.post_details = {
|
|
"title": self.source.title,
|
|
"name": self.source.name,
|
|
"url": self.source.url,
|
|
"selftext": self.source.selftext,
|
|
"score": self.source.score,
|
|
"upvote_ratio": self.source.upvote_ratio,
|
|
"permalink": self.source.permalink,
|
|
"id": self.source.id,
|
|
"author": self.source.author.name if self.source.author else "DELETED",
|
|
"link_flair_text": self.source.link_flair_text,
|
|
"num_comments": self.source.num_comments,
|
|
"over_18": self.source.over_18,
|
|
"spoiler": self.source.spoiler,
|
|
"pinned": self.source.pinned,
|
|
"locked": self.source.locked,
|
|
"distinguished": self.source.distinguished,
|
|
"created_utc": self.source.created_utc,
|
|
}
|
|
|
|
def _get_comments(self) -> list[dict]:
|
|
logger.debug(f"Retrieving full comment tree for submission {self.source.id}")
|
|
comments = []
|
|
self.source.comments.replace_more(limit=None)
|
|
for top_level_comment in self.source.comments:
|
|
comments.append(self._convert_comment_to_dict(top_level_comment))
|
|
return comments
|