mirror of https://github.com/zulip/zulip.git
77 lines
2.3 KiB
Python
77 lines
2.3 KiB
Python
from dataclasses import dataclass
|
|
from typing import Any, Dict, List
|
|
|
|
from django.conf import settings
|
|
|
|
from zerver.models import OnboardingStep, UserProfile
|
|
|
|
|
|
@dataclass
|
|
class OneTimeNotice:
|
|
name: str
|
|
|
|
def to_dict(self) -> Dict[str, str]:
|
|
return {
|
|
"type": "one_time_notice",
|
|
"name": self.name,
|
|
}
|
|
|
|
|
|
ONE_TIME_NOTICES: List[OneTimeNotice] = [
|
|
OneTimeNotice(
|
|
name="visibility_policy_banner",
|
|
),
|
|
OneTimeNotice(
|
|
name="intro_inbox_view_modal",
|
|
),
|
|
OneTimeNotice(
|
|
name="intro_recent_view_modal",
|
|
),
|
|
OneTimeNotice(
|
|
name="first_stream_created_banner",
|
|
),
|
|
OneTimeNotice(
|
|
name="jump_to_conversation_banner",
|
|
),
|
|
]
|
|
|
|
# We may introduce onboarding step of types other than 'one time notice'
|
|
# in future. Earlier, we had 'hotspot' and 'one time notice' as the two
|
|
# types. We can simply do:
|
|
# ALL_ONBOARDING_STEPS: List[Union[OneTimeNotice, OtherType]]
|
|
# to avoid API changes when new type is introduced in the future.
|
|
ALL_ONBOARDING_STEPS: List[OneTimeNotice] = ONE_TIME_NOTICES
|
|
|
|
|
|
def get_next_onboarding_steps(user: UserProfile) -> List[Dict[str, Any]]:
|
|
# If a Zulip server has disabled the tutorial, never send any
|
|
# onboarding steps.
|
|
if not settings.TUTORIAL_ENABLED:
|
|
return []
|
|
|
|
seen_onboarding_steps = frozenset(
|
|
OnboardingStep.objects.filter(user=user).values_list("onboarding_step", flat=True)
|
|
)
|
|
|
|
onboarding_steps: List[Dict[str, Any]] = []
|
|
for one_time_notice in ONE_TIME_NOTICES:
|
|
if one_time_notice.name in seen_onboarding_steps:
|
|
continue
|
|
onboarding_steps.append(one_time_notice.to_dict())
|
|
|
|
return onboarding_steps
|
|
|
|
|
|
def copy_onboarding_steps(source_profile: UserProfile, target_profile: UserProfile) -> None:
|
|
for onboarding_step in frozenset(OnboardingStep.objects.filter(user=source_profile)):
|
|
OnboardingStep.objects.create(
|
|
user=target_profile,
|
|
onboarding_step=onboarding_step.onboarding_step,
|
|
timestamp=onboarding_step.timestamp,
|
|
)
|
|
|
|
# TODO: The 'tutorial_status' field of 'UserProfile' model
|
|
# is no longer used. Remove it.
|
|
target_profile.tutorial_status = source_profile.tutorial_status
|
|
target_profile.save(update_fields=["tutorial_status"])
|