From 98a7762a5102b3bc3591fc90b7766af71df28bfc Mon Sep 17 00:00:00 2001
From: Rhea Parekh <rheaparekh12@gmail.com>
Date: Wed, 11 Jul 2018 23:27:15 +0530
Subject: [PATCH] export: Export user groups.

---
 zerver/lib/export.py | 19 ++++++++++++++++---
 1 file changed, 16 insertions(+), 3 deletions(-)

diff --git a/zerver/lib/export.py b/zerver/lib/export.py
index b9e8a01666..91357557e3 100644
--- a/zerver/lib/export.py
+++ b/zerver/lib/export.py
@@ -19,7 +19,8 @@ from zerver.models import UserProfile, Realm, Client, Huddle, Stream, \
     RealmDomain, Recipient, DefaultStream, get_user_profile_by_id, \
     UserPresence, UserActivity, UserActivityInterval, CustomProfileField, \
     CustomProfileFieldValue, get_display_recipient, Attachment, get_system_bot, \
-    RealmAuditLog, UserHotspot, MutedTopic, Service
+    RealmAuditLog, UserHotspot, MutedTopic, Service, UserGroup, \
+    UserGroupMembership
 from zerver.lib.parallel import run_parallel
 from typing import Any, Callable, Dict, List, Optional, Set, Tuple, \
     Iterable, Union
@@ -168,8 +169,6 @@ NON_EXPORTED_TABLES = {
     'zerver_botstoragedata',
     'zerver_botconfigdata',
     'zerver_pushdevicetoken',
-    'zerver_usergroup',
-    'zerver_usergroupmembership',
 }
 
 IMPLICIT_TABLES = {
@@ -541,6 +540,20 @@ def get_realm_config() -> Config:
         custom_fetch=fetch_user_profile,
     )
 
+    Config(
+        table='zerver_usergroup',
+        model=UserGroup,
+        normal_parent=realm_config,
+        parent_key='realm__in',
+    )
+
+    Config(
+        table='zerver_usergroupmembership',
+        model=UserGroupMembership,
+        normal_parent=user_profile_config,
+        parent_key='user_group__in',
+    )
+
     Config(
         custom_tables=[
             'zerver_userprofile_crossrealm',