paulmataruso/zulip
1
0
Fork 0
mirror of https://github.com/zulip/zulip.git synced 2025-11-04 22:13:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

python: Reformat with Black, except quotes.

Browse Source 
Signed-off-by: Anders Kaseorg <anders@zulip.com>
This commit is contained in:
Anders Kaseorg
2021-02-11 23:19:30 -08:00
committed by Tim Abbott
parent 5028c081cb
commit 11741543da
817 changed files with 44952 additions and 24860 deletions
Download Patch File Download Diff File Expand all files Collapse all files

43
zerver/management/commands/sync_ldap_user_data.py
View File

@@ -15,7 +15,9 @@ logger = logging.getLogger('zulip.sync_ldap_user_data')
log_to_file(logger, settings.LDAP_SYNC_LOG_PATH)
# Run this on a cronjob to pick up on name changes.
def sync_ldap_user_data(user_profiles: List[UserProfile], deactivation_protection: bool=True) -> None:
def sync_ldap_user_data(
user_profiles: List[UserProfile], deactivation_protection: bool = True
) -> None:
logger.info("Starting update.")
with transaction.atomic():
realms = {u.realm.string_id for u in user_profiles}
@@ -31,30 +33,42 @@ def sync_ldap_user_data(user_profiles: List[UserProfile], deactivation_protectio
if deactivation_protection:
if not UserProfile.objects.filter(is_bot=False, is_active=True).exists():
error_msg = ("Ldap sync would have deactivated all users. This is most likely due " +
"to a misconfiguration of LDAP settings. Rolling back...\n" +
"Use the --force option if the mass deactivation is intended.")
error_msg = (
"Ldap sync would have deactivated all users. This is most likely due "
+ "to a misconfiguration of LDAP settings. Rolling back...\n"
+ "Use the --force option if the mass deactivation is intended."
)
logger.error(error_msg)
# Raising an exception in this atomic block will rollback the transaction.
raise Exception(error_msg)
for string_id in realms:
if not UserProfile.objects.filter(is_bot=False, is_active=True, realm__string_id=string_id,
role__gte=UserProfile.ROLE_REALM_ADMINISTRATOR).exists():
error_msg = ("Ldap sync would have deactivated all administrators of realm %s. " +
"This is most likely due " +
"to a misconfiguration of LDAP settings. Rolling back...\n" +
"Use the --force option if the mass deactivation is intended.")
if not UserProfile.objects.filter(
is_bot=False,
is_active=True,
realm__string_id=string_id,
role__gte=UserProfile.ROLE_REALM_ADMINISTRATOR,
).exists():
error_msg = (
"Ldap sync would have deactivated all administrators of realm %s. "
+ "This is most likely due "
+ "to a misconfiguration of LDAP settings. Rolling back...\n"
+ "Use the --force option if the mass deactivation is intended."
)
error_msg = error_msg % (string_id,)
logger.error(error_msg)
raise Exception(error_msg)
logger.info("Finished update.")
class Command(ZulipBaseCommand):
def add_arguments(self, parser: ArgumentParser) -> None:
parser.add_argument('-f', '--force',
action="store_true",
help='Disable the protection against deactivating all users.')
parser.add_argument(
'-f',
'--force',
action="store_true",
help='Disable the protection against deactivating all users.',
)
self.add_realm_args(parser)
self.add_user_list_args(parser)
@@ -62,8 +76,7 @@ class Command(ZulipBaseCommand):
def handle(self, *args: Any, **options: Any) -> None:
if options.get('realm_id') is not None:
realm = self.get_realm(options)
user_profiles = self.get_users(options, realm, is_bot=False,
include_deactivated=True)
user_profiles = self.get_users(options, realm, is_bot=False, include_deactivated=True)
else:
user_profiles = UserProfile.objects.select_related().filter(is_bot=False)
sync_ldap_user_data(user_profiles, not options['force'])