paulmataruso/zulip
1
0
Fork 0
mirror of https://github.com/zulip/zulip.git synced 2025-11-12 18:06:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

export: Extract fetch_user_profile_cross_realm().

Browse Source
This commit is contained in:
Steve Howell
2016-08-10 17:39:21 -07:00
committed by Tim Abbott
parent fee2106c6f
commit b383f5ca5d
1 changed files with 18 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
zerver/lib/export.py
View File

@@ -180,16 +180,12 @@ def export_with_admin_auth(realm, response, include_invite_only=True):
# type: (Any, **Any) -> Any # type: (Any, **Any) -> Any
return model.objects.filter(realm=realm, **kwargs) return model.objects.filter(realm=realm, **kwargs)
cross_realm_context = {'realm': realm}
response['zerver_userprofile'] = [model_to_dict(x, exclude=["password", "api_key"]) response['zerver_userprofile'] = [model_to_dict(x, exclude=["password", "api_key"])
for x in filter_by_realm(UserProfile)] for x in filter_by_realm(UserProfile)]
if realm.domain == "zulip.com": fetch_user_profile_cross_realm(response, cross_realm_context)
response['zerver_userprofile_crossrealm'] = []
else:
response['zerver_userprofile_crossrealm'] = [dict(email=x.email, id=x.id) for x in [
get_user_profile_by_email(settings.NOTIFICATION_BOT),
get_user_profile_by_email(settings.EMAIL_GATEWAY_BOT),
get_user_profile_by_email(settings.WELCOME_BOT),
]]
floatify_datetime_fields(response, 'zerver_userprofile') floatify_datetime_fields(response, 'zerver_userprofile')
user_profile_ids = get_primary_ids(response['zerver_userprofile']) user_profile_ids = get_primary_ids(response['zerver_userprofile'])
@@ -248,6 +244,20 @@ def export_with_admin_auth(realm, response, include_invite_only=True):
response["zerver_attachment"] = make_raw(attachment_query) response["zerver_attachment"] = make_raw(attachment_query)
floatify_datetime_fields(response, 'zerver_attachment') floatify_datetime_fields(response, 'zerver_attachment')
def fetch_user_profile_cross_realm(response, context):
# type: (TableData, Context) -> None
realm = context['realm']
if realm.domain == "zulip.com":
response['zerver_userprofile_crossrealm'] = []
else:
response['zerver_userprofile_crossrealm'] = [dict(email=x.email, id=x.id) for x in [
get_user_profile_by_email(settings.NOTIFICATION_BOT),
get_user_profile_by_email(settings.EMAIL_GATEWAY_BOT),
get_user_profile_by_email(settings.WELCOME_BOT),
]]
def fetch_huddle_objects(response, context): def fetch_huddle_objects(response, context):
# type: (TableData, Context) -> Tuple[List[Record], List[Record]] # type: (TableData, Context) -> Tuple[List[Record], List[Record]]