From 34daf6ebe6dd7fdca83c1cd05e1e288cf511a60b Mon Sep 17 00:00:00 2001 From: Ilya Siamionau Date: Fri, 8 Sep 2023 10:19:42 +0200 Subject: [PATCH] Add reposts support to the firehose process commits example (#140) --- examples/firehose/process_commits.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/examples/firehose/process_commits.py b/examples/firehose/process_commits.py index ea75bd58..b3cbaf72 100644 --- a/examples/firehose/process_commits.py +++ b/examples/firehose/process_commits.py @@ -38,15 +38,19 @@ def _get_ops_by_type(commit: models.ComAtprotoSyncSubscribeRepos.Commit) -> dict operation_by_type['likes']['created'].append({'record': record, **create_info}) elif uri.collection == ids.AppBskyFeedPost and is_record_type(record, ids.AppBskyFeedPost): operation_by_type['posts']['created'].append({'record': record, **create_info}) + elif uri.collection == ids.AppBskyFeedRepost and is_record_type(record, ids.AppBskyFeedRepost): + operation_by_type['reposts']['created'].append({'record': record, **create_info}) elif uri.collection == ids.AppBskyGraphFollow and is_record_type(record, ids.AppBskyGraphFollow): operation_by_type['follows']['created'].append({'record': record, **create_info}) if op.action == 'delete': if uri.collection == ids.AppBskyFeedLike: operation_by_type['likes']['deleted'].append({'uri': str(uri)}) - if uri.collection == ids.AppBskyFeedPost: + elif uri.collection == ids.AppBskyFeedPost: operation_by_type['posts']['deleted'].append({'uri': str(uri)}) - if uri.collection == ids.AppBskyGraphFollow: + elif uri.collection == ids.AppBskyFeedRepost: + operation_by_type['reposts']['deleted'].append({'uri': str(uri)}) + elif uri.collection == ids.AppBskyGraphFollow: operation_by_type['follows']['deleted'].append({'uri': str(uri)}) return operation_by_type