mirror of
https://github.com/zulip/zulip.git
synced 2025-11-10 17:07:07 +00:00
69730a78cc
Automatically generated by the following script, based on the output
of lint with flake8-comma:
import re
import sys
last_filename = None
last_row = None
lines = []
for msg in sys.stdin:
m = re.match(
r"\x1b\[35mflake8 \|\x1b\[0m \x1b\[1;31m(.+):(\d+):(\d+): (\w+)", msg
)
if m:
filename, row_str, col_str, err = m.groups()
row, col = int(row_str), int(col_str)
if filename == last_filename:
assert last_row != row
else:
if last_filename is not None:
with open(last_filename, "w") as f:
f.writelines(lines)
with open(filename) as f:
lines = f.readlines()
last_filename = filename
last_row = row
line = lines[row - 1]
if err in ["C812", "C815"]:
lines[row - 1] = line[: col - 1] + "," + line[col - 1 :]
elif err in ["C819"]:
assert line[col - 2] == ","
lines[row - 1] = line[: col - 2] + line[col - 1 :].lstrip(" ")
if last_filename is not None:
with open(last_filename, "w") as f:
f.writelines(lines)
Signed-off-by: Anders Kaseorg <anders@zulipchat.com>
61 lines
2.0 KiB
Python
61 lines
2.0 KiB
Python
import logging
|
|
from argparse import ArgumentParser
|
|
from typing import Any, List, Optional
|
|
|
|
from django.db import connection
|
|
|
|
from zerver.lib.fix_unreads import fix
|
|
from zerver.lib.management import CommandError, ZulipBaseCommand
|
|
from zerver.models import Realm, UserProfile
|
|
|
|
logging.getLogger('zulip.fix_unreads').setLevel(logging.INFO)
|
|
|
|
class Command(ZulipBaseCommand):
|
|
help = """Fix problems related to unread counts."""
|
|
|
|
def add_arguments(self, parser: ArgumentParser) -> None:
|
|
parser.add_argument('emails',
|
|
metavar='<emails>',
|
|
type=str,
|
|
nargs='*',
|
|
help='email address to spelunk')
|
|
parser.add_argument('--all',
|
|
action='store_true',
|
|
dest='all',
|
|
default=False,
|
|
help='fix all users in specified realm')
|
|
self.add_realm_args(parser)
|
|
|
|
def fix_all_users(self, realm: Realm) -> None:
|
|
user_profiles = list(UserProfile.objects.filter(
|
|
realm=realm,
|
|
is_bot=False,
|
|
))
|
|
for user_profile in user_profiles:
|
|
fix(user_profile)
|
|
connection.commit()
|
|
|
|
def fix_emails(self, realm: Optional[Realm], emails: List[str]) -> None:
|
|
|
|
for email in emails:
|
|
try:
|
|
user_profile = self.get_user(email, realm)
|
|
except CommandError:
|
|
print(f"e-mail {email} doesn't exist in the realm {realm}, skipping")
|
|
return
|
|
|
|
fix(user_profile)
|
|
connection.commit()
|
|
|
|
def handle(self, *args: Any, **options: Any) -> None:
|
|
realm = self.get_realm(options)
|
|
|
|
if options['all']:
|
|
if realm is None:
|
|
raise CommandError('You must specify a realm if you choose the --all option.')
|
|
|
|
self.fix_all_users(realm)
|
|
return
|
|
|
|
self.fix_emails(realm, options['emails'])
|