mirror of
https://github.com/zulip/zulip.git
synced 2025-11-04 22:13:26 +00:00
393afc9781
Since we include internal realms while creating system groups in "0382_create_role_based_system_groups.py", we should do it when creating new internal realms as well to be consistent. Tests are changed accordingly as UserGroup objects are created. We also change the user group ids used in api docs examples such that user groups are of correct realm.
68 lines
2.2 KiB
Python
68 lines
2.2 KiB
Python
from typing import Iterable, Optional, Tuple
|
|
|
|
from django.conf import settings
|
|
|
|
from zerver.lib.bulk_create import bulk_create_users
|
|
from zerver.lib.user_groups import create_system_user_groups_for_realm
|
|
from zerver.models import (
|
|
Realm,
|
|
RealmAuditLog,
|
|
RealmUserDefault,
|
|
UserProfile,
|
|
get_client,
|
|
get_system_bot,
|
|
)
|
|
|
|
|
|
def server_initialized() -> bool:
|
|
return Realm.objects.exists()
|
|
|
|
|
|
def create_internal_realm() -> None:
|
|
from zerver.actions.users import do_change_can_forge_sender
|
|
|
|
realm = Realm.objects.create(string_id=settings.SYSTEM_BOT_REALM, name="System bot realm")
|
|
RealmAuditLog.objects.create(
|
|
realm=realm, event_type=RealmAuditLog.REALM_CREATED, event_time=realm.date_created
|
|
)
|
|
RealmUserDefault.objects.create(realm=realm)
|
|
create_system_user_groups_for_realm(realm)
|
|
|
|
# Create some client objects for common requests. Not required;
|
|
# just ensures these get low IDs in production, and in development
|
|
# avoids an extra database write for the first HTTP request in
|
|
# most tests.
|
|
get_client("website")
|
|
get_client("ZulipMobile")
|
|
get_client("ZulipElectron")
|
|
|
|
internal_bots = [
|
|
(bot["name"], bot["email_template"] % (settings.INTERNAL_BOT_DOMAIN,))
|
|
for bot in settings.INTERNAL_BOTS
|
|
]
|
|
create_users(realm, internal_bots, bot_type=UserProfile.DEFAULT_BOT)
|
|
# Set the owners for these bots to the bots themselves
|
|
bots = UserProfile.objects.filter(email__in=[bot_info[1] for bot_info in internal_bots])
|
|
for bot in bots:
|
|
bot.bot_owner = bot
|
|
bot.save()
|
|
|
|
# Initialize the email gateway bot as able to forge senders.
|
|
email_gateway_bot = get_system_bot(settings.EMAIL_GATEWAY_BOT, realm.id)
|
|
do_change_can_forge_sender(email_gateway_bot, True)
|
|
|
|
|
|
def create_users(
|
|
realm: Realm,
|
|
name_list: Iterable[Tuple[str, str]],
|
|
tos_version: Optional[str] = None,
|
|
bot_type: Optional[int] = None,
|
|
bot_owner: Optional[UserProfile] = None,
|
|
) -> None:
|
|
user_set = set()
|
|
for full_name, email in name_list:
|
|
user_set.add((email, full_name, True))
|
|
bulk_create_users(
|
|
realm, user_set, bot_type=bot_type, bot_owner=bot_owner, tos_version=tos_version
|
|
)
|