From 3541d2a55aca69391613a84d02a2798115c6ef4a Mon Sep 17 00:00:00 2001 From: Dima Gerasimov Date: Sat, 16 Mar 2019 11:01:03 +0000 Subject: [PATCH] add support for getting all saves --- reddit/__init__.py | 17 +++++++++-------- 1 file changed, 9 insertions(+), 8 deletions(-) diff --git a/reddit/__init__.py b/reddit/__init__.py index bfd112c..67b2977 100755 --- a/reddit/__init__.py +++ b/reddit/__init__.py @@ -165,14 +165,15 @@ def get_events(all_=True) -> List[Event]: return list(sorted(events, key=lambda e: e.cmp_key)) def get_saves(all_=True) -> List[Save]: - # TODO hmm.... do we want ALL reddit saves I ever had? - # TODO for now even last ones would be ok - assert all_ is False, 'all saves are not supported yet...' - backups = _get_backups(all_=all_) - [backup] = backups - - saves = get_state(backup) - return list(saves.values()) + events = get_events(all_=all_) + saves = [] + for e in events: + if e.text.startswith('favorited'): + ss = e.kind + assert isinstance(ss, Save) + saves.append(ss) + assert len(saves) > 0 + return saves def test():