export: Export UserHotspot.

This commit is contained in:
Rhea Parekh 2018-07-12 19:56:03 +05:30 committed by Tim Abbott
parent 6f7b7e143f
commit 8897e187c0
1 changed files with 9 additions and 2 deletions

View File

@ -19,7 +19,7 @@ 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
RealmAuditLog, UserHotspot
from zerver.lib.parallel import run_parallel
from typing import Any, Callable, Dict, List, Optional, Set, Tuple, \
Iterable, Union
@ -172,7 +172,6 @@ NON_EXPORTED_TABLES = {
'zerver_service',
'zerver_usergroup',
'zerver_usergroupmembership',
'zerver_userhotspot',
}
IMPLICIT_TABLES = {
@ -202,6 +201,7 @@ DATE_FIELDS = {
'zerver_userpresence': ['timestamp'],
'zerver_userprofile': ['date_joined', 'last_login', 'last_reminder'],
'zerver_realmauditlog': ['event_time'],
'zerver_userhotspot': ['timestamp'],
} # type: Dict[TableName, List[Field]]
def sanity_check_output(data: TableData) -> None:
@ -586,6 +586,13 @@ def get_realm_config() -> Config:
parent_key='modified_user__in',
)
Config(
table='zerver_userhotspot',
model=UserHotspot,
normal_parent=user_profile_config,
parent_key='user__in',
)
# Some of these tables are intermediate "tables" that we
# create only for the export. Think of them as similar to views.