def Transform(self, client, viewpoint, callback):
from activity import Activity
from viewpoint import Viewpoint
def _OnQuery(followers_activities):
(follower_ids, last_key), activities = followers_activities
activities = [
activity for activity in activities if activity.name == 'share'
]
if len(activities) > 0:
# Find the share activity with the lowest timestamp.
oldest_activity = None
for activity in activities:
if oldest_activity is None or activity.timestamp < oldest_activity.timestamp:
oldest_activity = activity
activity.name = 'share_existing'
# Override oldest activity as share_new and add followers.
oldest_activity.name = 'share_new'
act_dict = json.loads(activities[-1].json)
act_dict['follower_ids'] = [
f_id for f_id in follower_ids if f_id != viewpoint.user_id
]
oldest_activity.json = json.dumps(act_dict)
# Update all activities.
with util.Barrier(partial(callback, viewpoint)) as b:
for activity in activities:
self._LogUpdate(activity)
if Version._mutate_items:
activity.Update(client, b.Callback())
else:
b.Callback()()
else:
callback(viewpoint)
with util.ArrayBarrier(_OnQuery) as b:
Viewpoint.QueryFollowerIds(client, viewpoint.viewpoint_id,
b.Callback())
Activity.RangeQuery(client, viewpoint.viewpoint_id, None, None,
None, b.Callback())
def Transform(self, client, viewpoint, callback):
from activity import Activity
from followed import Followed
from viewpoint import Viewpoint
def _OnQuery(activities_followers):
activities, (follower_ids, last_key) = activities_followers
with util.Barrier(partial(callback, viewpoint)) as b:
old_timestamp = viewpoint.last_updated
if len(activities) > 0:
new_timestamp = max(a.timestamp for a in activities)
else:
# Viewpoint has no activities.
new_timestamp = 0
viewpoint.last_updated = new_timestamp
self._LogUpdate(viewpoint)
for follower_id in follower_ids:
logging.info(
'Followed (user_id=%s, viewpoint_id=%s): %s => %s' %
(follower_id, viewpoint.viewpoint_id,
Followed._TruncateToDay(old_timestamp),
Followed._TruncateToDay(new_timestamp)))
if Version._mutate_items:
Followed.UpdateDateUpdated(client, follower_id,
viewpoint.viewpoint_id,
old_timestamp,
new_timestamp, b.Callback())
with util.ArrayBarrier(_OnQuery) as b:
Activity.RangeQuery(client, viewpoint.viewpoint_id, None, None,
None, b.Callback())
Viewpoint.QueryFollowerIds(client, viewpoint.viewpoint_id,
b.Callback())
