mirror of
https://github.com/zulip/zulip.git
synced 2025-11-05 06:23:38 +00:00
export: Fix 'OnboardingUserMessage' table not being exported.
Earlier, the export tool was logging a warning: "??? NO DATA EXPORTED FOR TABLE zerver_onboardingusermessage!!!" This bug was due to not configuring a Config object for 'OnboardingUserMessage' in 'get_realm_config()'. This commit fixes the bug to export the table properly.
This commit is contained in:
committed by
Tim Abbott
parent
e2623f3937
commit
7d379e00b0
@@ -47,6 +47,7 @@ from zerver.models import (
|
|||||||
MutedUser,
|
MutedUser,
|
||||||
NamedUserGroup,
|
NamedUserGroup,
|
||||||
OnboardingStep,
|
OnboardingStep,
|
||||||
|
OnboardingUserMessage,
|
||||||
Reaction,
|
Reaction,
|
||||||
Realm,
|
Realm,
|
||||||
RealmAuditLog,
|
RealmAuditLog,
|
||||||
@@ -769,6 +770,13 @@ def get_realm_config() -> Config:
|
|||||||
include_rows="realm_id__in",
|
include_rows="realm_id__in",
|
||||||
)
|
)
|
||||||
|
|
||||||
|
Config(
|
||||||
|
table="zerver_onboardingusermessage",
|
||||||
|
model=OnboardingUserMessage,
|
||||||
|
virtual_parent=realm_config,
|
||||||
|
custom_fetch=custom_fetch_onboarding_usermessage,
|
||||||
|
)
|
||||||
|
|
||||||
user_profile_config = Config(
|
user_profile_config = Config(
|
||||||
custom_tables=[
|
custom_tables=[
|
||||||
"zerver_userprofile",
|
"zerver_userprofile",
|
||||||
@@ -1197,6 +1205,18 @@ def custom_fetch_realm_audit_logs_for_realm(response: TableData, context: Contex
|
|||||||
response["zerver_realmauditlog"] = rows
|
response["zerver_realmauditlog"] = rows
|
||||||
|
|
||||||
|
|
||||||
|
def custom_fetch_onboarding_usermessage(response: TableData, context: Context) -> None:
|
||||||
|
realm = context["realm"]
|
||||||
|
response["zerver_onboardingusermessage"] = []
|
||||||
|
|
||||||
|
onboarding_usermessage_query = OnboardingUserMessage.objects.filter(realm=realm)
|
||||||
|
for onboarding_usermessage in onboarding_usermessage_query:
|
||||||
|
onboarding_usermessage_obj = model_to_dict(onboarding_usermessage)
|
||||||
|
onboarding_usermessage_obj["flags_mask"] = onboarding_usermessage.flags.mask
|
||||||
|
del onboarding_usermessage_obj["flags"]
|
||||||
|
response["zerver_onboardingusermessage"].append(onboarding_usermessage_obj)
|
||||||
|
|
||||||
|
|
||||||
def fetch_usermessages(
|
def fetch_usermessages(
|
||||||
realm: Realm,
|
realm: Realm,
|
||||||
message_ids: set[int],
|
message_ids: set[int],
|
||||||
|
|||||||
Reference in New Issue
Block a user