mirror of
https://github.com/zulip/zulip.git
synced 2025-11-04 14:03:30 +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>
60 lines
2.0 KiB
Python
60 lines
2.0 KiB
Python
import cProfile
|
|
import logging
|
|
import tempfile
|
|
from typing import Any, Dict
|
|
|
|
from django.core.management.base import CommandParser
|
|
from django.http import HttpRequest, HttpResponse
|
|
|
|
from zerver.lib.management import ZulipBaseCommand
|
|
from zerver.middleware import LogRequests
|
|
from zerver.models import UserMessage, UserProfile
|
|
from zerver.views.messages import get_messages_backend
|
|
|
|
request_logger = LogRequests()
|
|
|
|
class MockSession:
|
|
def __init__(self) -> None:
|
|
self.modified = False
|
|
|
|
class MockRequest(HttpRequest):
|
|
def __init__(self, user: UserProfile) -> None:
|
|
self.user = user
|
|
self.path = '/'
|
|
self.method = "POST"
|
|
self.META = {"REMOTE_ADDR": "127.0.0.1"}
|
|
anchor = UserMessage.objects.filter(user_profile=self.user).order_by("-message")[200].message_id
|
|
self.REQUEST = {
|
|
"anchor": anchor,
|
|
"num_before": 1200,
|
|
"num_after": 200,
|
|
}
|
|
self.GET: Dict[Any, Any] = {}
|
|
self.session = MockSession()
|
|
|
|
def get_full_path(self) -> str:
|
|
return self.path
|
|
|
|
def profile_request(request: HttpRequest) -> HttpResponse:
|
|
request_logger.process_request(request)
|
|
prof = cProfile.Profile()
|
|
prof.enable()
|
|
ret = get_messages_backend(request, request.user,
|
|
apply_markdown=True)
|
|
prof.disable()
|
|
with tempfile.NamedTemporaryFile(prefix='profile.data.', delete=False) as stats_file:
|
|
prof.dump_stats(stats_file.name)
|
|
request_logger.process_response(request, ret)
|
|
logging.info("Profiling data written to %s", stats_file.name)
|
|
return ret
|
|
|
|
class Command(ZulipBaseCommand):
|
|
def add_arguments(self, parser: CommandParser) -> None:
|
|
parser.add_argument("email", metavar="<email>", type=str, help="Email address of the user")
|
|
self.add_realm_args(parser)
|
|
|
|
def handle(self, *args: Any, **options: Any) -> None:
|
|
realm = self.get_realm(options)
|
|
user = self.get_user(options["email"], realm)
|
|
profile_request(MockRequest(user))
|