diff --git a/snscrape/modules/reddit.py b/snscrape/modules/reddit.py index 55af939c..17d1b512 100644 --- a/snscrape/modules/reddit.py +++ b/snscrape/modules/reddit.py @@ -20,11 +20,14 @@ @dataclasses.dataclass class Submission(snscrape.base.Item): author: typing.Optional[str] # E.g. submission hf7k6 + author_id: typing.Optional[str] date: datetime.datetime id: str link: typing.Optional[str] selftext: typing.Optional[str] subreddit: typing.Optional[str] # E.g. submission 617p51 + subreddit_id: typing.Optional[str] + score: int title: str url: str @@ -37,11 +40,14 @@ def __str__(self): @dataclasses.dataclass class Comment(snscrape.base.Item): author: typing.Optional[str] + author_id: typing.Optional[str] body: str date: datetime.datetime id: str parentId: typing.Optional[str] subreddit: typing.Optional[str] + subreddit_id: typing.Optional[str] + score: int url: str created = snscrape.base._DeprecatedProperty('created', lambda self: self.date, 'date') @@ -115,9 +121,12 @@ def _api_obj_to_item(self, d): kwargs = { 'author': d.get('author'), + 'author_id': d.get('author_fullname'), 'date': datetime.datetime.fromtimestamp(d['created_utc'], datetime.timezone.utc), 'url': f'https://old.reddit.com{permalink}', 'subreddit': d.get('subreddit'), + 'subreddit_id': d.get('subreddit_id'), + 'score': int(d.get('score')) } if cls is Submission: kwargs['selftext'] = d.get('selftext') or None