def GatherPageData(self, mr):
"""Build up a dictionary of data values to use when rendering the page.
Args:
mr: commonly used info parsed from the request.
Returns:
Dict of values used by EZT for rendering the page.
"""
can_create_project = permissions.CanCreateProject(mr.perms)
# Kick off the search pipeline, it has its own promises for parallelism.
pipeline = projectsearch.ProjectSearchPipeline(mr, self.services)
# Meanwhile, determine which projects the signed-in user has starred.
with work_env.WorkEnv(mr, self.services) as we:
starred_projects = we.ListStarredProjects()
starred_project_ids = {p.project_id for p in starred_projects}
# A dict of project id to the user's membership status.
project_memberships = {}
if mr.auth.user_id:
with work_env.WorkEnv(mr, self.services) as we:
owned, _archive_owned, member_of, contrib_of = (
we.GetUserProjects(mr.auth.effective_ids))
project_memberships.update(
{proj.project_id: 'Owner'
for proj in owned})
project_memberships.update(
{proj.project_id: 'Member'
for proj in member_of})
project_memberships.update(
{proj.project_id: 'Contributor'
for proj in contrib_of})
# Finish the project search pipeline.
pipeline.SearchForIDs()
pipeline.GetProjectsAndPaginate(mr.cnxn, urls.HOSTING_HOME)
project_ids = [p.project_id for p in pipeline.visible_results]
star_count_dict = self.services.project_star.CountItemsStars(
mr.cnxn, project_ids)
# Make ProjectView objects
project_view_list = [
project_views.ProjectView(
p,
starred=p.project_id in starred_project_ids,
num_stars=star_count_dict.get(p.project_id),
membership_desc=project_memberships.get(p.project_id))
for p in pipeline.visible_results
]
return {
'can_create_project': ezt.boolean(can_create_project),
'learn_more_link': settings.learn_more_link,
'projects': project_view_list,
'pagination': pipeline.pagination,
}
def testCachedContentTimestamp(self):
project = self.services.project.GetProjectByName('fake cnxn', 'test')
# Project was never updated since we added cached_content_timestamp.
project.cached_content_timestamp = 0
view = project_views.ProjectView(project, now=1 * 60 * 60 + 234)
self.assertEqual(1 * 60 * 60, view.cached_content_timestamp)
# Project was updated within the last hour, use that timestamp.
project.cached_content_timestamp = 1 * 60 * 60 + 123
view = project_views.ProjectView(project, now=1 * 60 * 60 + 234)
self.assertEqual(1 * 60 * 60 + 123, view.cached_content_timestamp)
# Project was not updated within the last hour, but user groups
# could have been updated on groups.google.com without any
# notification to us, so the client will ask for an updated feed
# at least once an hour.
project.cached_content_timestamp = 1 * 60 * 60 + 123
view = project_views.ProjectView(project, now=2 * 60 * 60 + 234)
self.assertEqual(2 * 60 * 60, view.cached_content_timestamp)
def GatherPageData(self, mr):
"""Build up a dictionary of data values to use when rendering the page."""
viewed_user = mr.viewed_user_auth.user_pb
if self.services.usergroup.GetGroupSettings(
mr.cnxn, mr.viewed_user_auth.user_id):
url = framework_helpers.FormatAbsoluteURL(mr,
'/g/%s/' %
viewed_user.email,
include_project=False)
self.redirect(url, abort=True) # Show group page instead.
with work_env.WorkEnv(mr, self.services) as we:
project_lists = we.GetUserProjects(
mr.viewed_user_auth.effective_ids)
(visible_ownership, visible_archived, visible_membership,
visible_contrib) = project_lists
with mr.profiler.Phase('Getting user groups'):
group_settings = self.services.usergroup.GetAllGroupSettings(
mr.cnxn, mr.viewed_user_auth.effective_ids)
member_ids, owner_ids = self.services.usergroup.LookupAllMembers(
mr.cnxn, list(group_settings.keys()))
friend_project_ids = [] # TODO(issue 4202): implement this.
visible_group_ids = []
for group_id in group_settings:
if permissions.CanViewGroupMembers(mr.perms,
mr.auth.effective_ids,
group_settings[group_id],
member_ids[group_id],
owner_ids[group_id],
friend_project_ids):
visible_group_ids.append(group_id)
user_group_views = framework_views.MakeAllUserViews(
mr.cnxn, self.services.user, visible_group_ids)
user_group_views = sorted(list(user_group_views.values()),
key=lambda ugv: ugv.email)
with mr.profiler.Phase('Getting linked accounts'):
linked_parent = None
linked_children = []
linked_views = framework_views.MakeAllUserViews(
mr.cnxn, self.services.user, [viewed_user.linked_parent_id],
viewed_user.linked_child_ids)
if viewed_user.linked_parent_id:
linked_parent = linked_views[viewed_user.linked_parent_id]
if viewed_user.linked_child_ids:
linked_children = [
linked_views[child_id]
for child_id in viewed_user.linked_child_ids
]
offer_unlink = (mr.auth.user_id == viewed_user.user_id
or mr.auth.user_id in linked_views)
incoming_invite_users = []
outgoing_invite_users = []
possible_parent_accounts = []
can_edit_invites = mr.auth.user_id == mr.viewed_user_auth.user_id
display_link_invites = can_edit_invites or mr.auth.user_pb.is_site_admin
# TODO(jrobbins): allow site admin to edit invites for other users.
if display_link_invites:
with work_env.WorkEnv(mr,
self.services,
phase='Getting link invites'):
incoming_invite_ids, outgoing_invite_ids = we.GetPendingLinkedInvites(
user_id=viewed_user.user_id)
invite_views = framework_views.MakeAllUserViews(
mr.cnxn, self.services.user, incoming_invite_ids,
outgoing_invite_ids)
incoming_invite_users = [
invite_views[uid] for uid in incoming_invite_ids
]
outgoing_invite_users = [
invite_views[uid] for uid in outgoing_invite_ids
]
possible_parent_accounts = _ComputePossibleParentAccounts(
we, mr.viewed_user_auth.user_view, linked_parent,
linked_children)
viewed_user_display_name = framework_views.GetViewedUserDisplayName(mr)
with work_env.WorkEnv(mr, self.services) as we:
starred_projects = we.ListStarredProjects(
viewed_user_id=mr.viewed_user_auth.user_id)
logged_in_starred = we.ListStarredProjects()
logged_in_starred_pids = {p.project_id for p in logged_in_starred}
starred_user_ids = self.services.user_star.LookupStarredItemIDs(
mr.cnxn, mr.viewed_user_auth.user_id)
starred_user_dict = framework_views.MakeAllUserViews(
mr.cnxn, self.services.user, starred_user_ids)
starred_users = list(starred_user_dict.values())
starred_users_json = json.dumps(
[uv.display_name for uv in starred_users])
is_user_starred = self._IsUserStarred(mr.cnxn, mr.auth.user_id,
mr.viewed_user_auth.user_id)
if viewed_user.last_visit_timestamp:
last_visit_str = timestr.FormatRelativeDate(
viewed_user.last_visit_timestamp, days_only=True)
last_visit_str = last_visit_str or 'Less than 2 days ago'
else:
last_visit_str = 'Never'
if viewed_user.email_bounce_timestamp:
last_bounce_str = timestr.FormatRelativeDate(
viewed_user.email_bounce_timestamp, days_only=True)
last_bounce_str = last_bounce_str or 'Less than 2 days ago'
else:
last_bounce_str = None
can_ban = permissions.CanBan(mr, self.services)
viewed_user_is_spammer = viewed_user.banned.lower() == 'spam'
viewed_user_may_be_spammer = not viewed_user_is_spammer
all_projects = self.services.project.GetAllProjects(mr.cnxn)
for project_id in all_projects:
project = all_projects[project_id]
viewed_user_perms = permissions.GetPermissions(
viewed_user, mr.viewed_user_auth.effective_ids, project)
if (viewed_user_perms != permissions.EMPTY_PERMISSIONSET
and viewed_user_perms != permissions.USER_PERMISSIONSET):
viewed_user_may_be_spammer = False
ban_token = None
ban_spammer_token = None
if mr.auth.user_id and can_ban:
form_token_path = mr.request.path + 'ban.do'
ban_token = xsrf.GenerateToken(mr.auth.user_id, form_token_path)
form_token_path = mr.request.path + 'banSpammer.do'
ban_spammer_token = xsrf.GenerateToken(mr.auth.user_id,
form_token_path)
page_data = {
'user_tab_mode':
'st2',
'viewed_user_display_name':
viewed_user_display_name,
'viewed_user_may_be_spammer':
ezt.boolean(viewed_user_may_be_spammer),
'viewed_user_is_spammer':
ezt.boolean(viewed_user_is_spammer),
'viewed_user_is_banned':
ezt.boolean(viewed_user.banned),
'owner_of_projects': [
project_views.ProjectView(p,
starred=p.project_id
in logged_in_starred_pids)
for p in visible_ownership
],
'committer_of_projects': [
project_views.ProjectView(p,
starred=p.project_id
in logged_in_starred_pids)
for p in visible_membership
],
'contributor_to_projects': [
project_views.ProjectView(p,
starred=p.project_id
in logged_in_starred_pids)
for p in visible_contrib
],
'owner_of_archived_projects':
[project_views.ProjectView(p) for p in visible_archived],
'starred_projects': [
project_views.ProjectView(p,
starred=p.project_id
in logged_in_starred_pids)
for p in starred_projects
],
'starred_users':
starred_users,
'starred_users_json':
starred_users_json,
'is_user_starred':
ezt.boolean(is_user_starred),
'viewing_user_page':
ezt.boolean(True),
'last_visit_str':
last_visit_str,
'last_bounce_str':
last_bounce_str,
'vacation_message':
viewed_user.vacation_message,
'can_ban':
ezt.boolean(can_ban),
'ban_token':
ban_token,
'ban_spammer_token':
ban_spammer_token,
'user_groups':
user_group_views,
'linked_parent':
linked_parent,
'linked_children':
linked_children,
'incoming_invite_users':
incoming_invite_users,
'outgoing_invite_users':
outgoing_invite_users,
'possible_parent_accounts':
possible_parent_accounts,
'can_edit_invites':
ezt.boolean(can_edit_invites),
'offer_unlink':
ezt.boolean(offer_unlink),
}
viewed_user_prefs = None
if mr.perms.HasPerm(permissions.EDIT_OTHER_USERS, None, None):
with work_env.WorkEnv(mr, self.services) as we:
viewed_user_prefs = we.GetUserPrefs(
mr.viewed_user_auth.user_id)
user_settings = (
framework_helpers.UserSettings.GatherUnifiedSettingsPageData(
mr.auth.user_id, mr.viewed_user_auth.user_view, viewed_user,
viewed_user_prefs))
page_data.update(user_settings)
return page_data
def __init__(self,
pb,
mr,
prefetched_issues,
users_by_id,
prefetched_projects,
prefetched_configs,
autolink=None,
all_ref_artifacts=None,
ending=None,
highlight=None):
"""Constructs an ActivityView out of an Activity protocol buffer.
Args:
pb: an IssueComment or Activity protocol buffer.
mr: HTTP request info, used by the artifact autolink.
prefetched_issues: dictionary of the issues for the comments being shown.
users_by_id: dict {user_id: UserView} for all relevant users.
prefetched_projects: dict {project_id: project} including all the projects
that we might need.
prefetched_configs: dict {project_id: config} for those projects.
autolink: Autolink instance.
all_ref_artifacts: list of all artifacts in the activity stream.
ending: ending type for activity titles, 'in_project' or 'by_user'
highlight: what to highlight in the middle column on user updates pages
i.e. 'project', 'user', or None
"""
template_helpers.PBProxy.__init__(self, pb)
activity_type = 'ProjectIssueUpdate' # TODO(jrobbins): more types
self.comment = None
self.issue = None
self.field_changed = None
self.multiple_fields_changed = ezt.boolean(False)
self.project = None
self.user = None
self.timestamp = time.time(
) # Bogus value makes bad ones highly visible.
if isinstance(pb, tracker_pb2.IssueComment):
self.timestamp = pb.timestamp
issue = prefetched_issues[pb.issue_id]
if self.timestamp == issue.opened_timestamp:
issue_change_id = None # This comment is the description.
else:
issue_change_id = pb.timestamp # instead of seq num.
self.comment = tracker_views.IssueCommentView(
mr.project_name, pb, users_by_id, autolink, all_ref_artifacts,
mr, issue)
# TODO(jrobbins): pass effective_ids of the commenter so that he/she
# can be identified as a project member or not.
config = prefetched_configs[issue.project_id]
self.issue = tracker_views.IssueView(issue, users_by_id, config)
self.user = self.comment.creator
project = prefetched_projects[issue.project_id]
self.project_name = project.project_name
self.project = project_views.ProjectView(project)
else:
logging.warn('unknown activity object %r', pb)
nested_page_data = {
'activity_type': activity_type,
'issue_change_id': issue_change_id,
'comment': self.comment,
'issue': self.issue,
'project': self.project,
'user': self.user,
'timestamp': self.timestamp,
'ending_type': ending,
}
self.escaped_title = self._TITLE_TEMPLATE.GetResponse(
nested_page_data).strip()
self.escaped_body = self._BODY_TEMPLATE.GetResponse(
nested_page_data).strip()
if autolink is not None and all_ref_artifacts is not None:
# TODO(jrobbins): actually parse the comment text. Actually render runs.
runs = autolink.MarkupAutolinks(
mr, [template_helpers.TextRun(self.escaped_body)],
all_ref_artifacts)
self.escaped_body = ''.join(run.content for run in runs)
self.date_bucket, self.date_relative = timestr.GetHumanScaleDate(
self.timestamp)
time_tuple = time.localtime(self.timestamp)
self.date_tooltip = time.asctime(time_tuple)
# We always highlight the user for starring activities
if activity_type.startswith('UserStar'):
self.highlight = 'user'
else:
self.highlight = highlight
def testNormalProject(self):
project = self.services.project.GetProjectByName('fake cnxn', 'test')
project_view = project_views.ProjectView(project)
self.assertEqual('test', project_view.project_name)
self.assertEqual('/p/test', project_view.relative_home_url)
self.assertEqual('LIVE', project_view.state_name)
def GatherPageData(self, mr):
"""Build up a dictionary of data values to use when rendering the page."""
viewed_user = mr.viewed_user_auth.user_pb
if self.services.usergroup.GetGroupSettings(
mr.cnxn, mr.viewed_user_auth.user_id):
url = framework_helpers.FormatAbsoluteURL(mr,
'/g/%s/' %
viewed_user.email,
include_project=False)
self.redirect(url, abort=True) # Show group page instead.
with self.profiler.Phase('GetUserProjects'):
project_lists = sitewide_helpers.GetUserProjects(
mr.cnxn, self.services, mr.auth.user_pb, mr.auth.effective_ids,
mr.viewed_user_auth.effective_ids)
(visible_ownership, visible_archived, visible_membership,
visible_contrib) = project_lists
viewed_user_display_name = framework_views.GetViewedUserDisplayName(mr)
with self.profiler.Phase('GetStarredProjects'):
starred_projects = sitewide_helpers.GetViewableStarredProjects(
mr.cnxn, self.services, mr.viewed_user_auth.user_id,
mr.auth.effective_ids, mr.auth.user_pb)
logged_in_starred_pids = []
if mr.auth.user_id:
logged_in_starred_pids = self.services.project_star.LookupStarredItemIDs(
mr.cnxn, mr.auth.user_id)
starred_user_ids = self.services.user_star.LookupStarredItemIDs(
mr.cnxn, mr.viewed_user_auth.user_id)
starred_user_dict = framework_views.MakeAllUserViews(
mr.cnxn, self.services.user, starred_user_ids)
starred_users = starred_user_dict.values()
is_user_starred = self._IsUserStarred(mr.cnxn, mr.auth.user_id,
mr.viewed_user_auth.user_id)
if viewed_user.last_visit_timestamp:
last_visit_str = timestr.FormatRelativeDate(
viewed_user.last_visit_timestamp, days_only=True)
last_visit_str = last_visit_str or 'Less than 2 days ago'
else:
last_visit_str = 'Never'
if viewed_user.email_bounce_timestamp:
last_bounce_str = timestr.FormatRelativeDate(
viewed_user.email_bounce_timestamp, days_only=True)
last_bounce_str = last_bounce_str or 'Less than 2 days ago'
else:
last_bounce_str = None
can_ban = permissions.CanBan(mr, self.services)
viewed_user_is_spammer = viewed_user.banned.lower() == 'spam'
viewed_user_may_be_spammer = not viewed_user_is_spammer
all_projects = self.services.project.GetAllProjects(mr.cnxn)
for project_id in all_projects:
project = all_projects[project_id]
viewed_user_perms = permissions.GetPermissions(
viewed_user, mr.viewed_user_auth.effective_ids, project)
if (viewed_user_perms != permissions.EMPTY_PERMISSIONSET
and viewed_user_perms != permissions.USER_PERMISSIONSET):
viewed_user_may_be_spammer = False
ban_token = None
ban_spammer_token = None
if mr.auth.user_id and can_ban:
form_token_path = mr.request.path + 'ban.do'
ban_token = xsrf.GenerateToken(mr.auth.user_id, form_token_path)
form_token_path = mr.request.path + 'banSpammer.do'
ban_spammer_token = xsrf.GenerateToken(mr.auth.user_id,
form_token_path)
page_data = {
'user_tab_mode':
'st2',
'viewed_user_display_name':
viewed_user_display_name,
'viewed_user_may_be_spammer':
ezt.boolean(viewed_user_may_be_spammer),
'viewed_user_is_spammer':
ezt.boolean(viewed_user_is_spammer),
'viewed_user_is_banned':
ezt.boolean(viewed_user.banned),
'viewed_user_ignore_action_limits':
(ezt.boolean(viewed_user.ignore_action_limits)),
'owner_of_projects': [
project_views.ProjectView(p,
starred=p.project_id
in logged_in_starred_pids)
for p in visible_ownership
],
'committer_of_projects': [
project_views.ProjectView(p,
starred=p.project_id
in logged_in_starred_pids)
for p in visible_membership
],
'contributor_to_projects': [
project_views.ProjectView(p,
starred=p.project_id
in logged_in_starred_pids)
for p in visible_contrib
],
'owner_of_archived_projects':
[project_views.ProjectView(p) for p in visible_archived],
'starred_projects': [
project_views.ProjectView(p,
starred=p.project_id
in logged_in_starred_pids)
for p in starred_projects
],
'starred_users':
starred_users,
'is_user_starred':
ezt.boolean(is_user_starred),
'viewing_user_page':
ezt.boolean(True),
'last_visit_str':
last_visit_str,
'last_bounce_str':
last_bounce_str,
'vacation_message':
viewed_user.vacation_message,
'can_ban':
ezt.boolean(can_ban),
'ban_token':
ban_token,
'ban_spammer_token':
ban_spammer_token
}
settings = framework_helpers.UserSettings.GatherUnifiedSettingsPageData(
mr.auth.user_id, mr.viewed_user_auth.user_view, viewed_user)
page_data.update(settings)
return page_data
