forked from username/flaskpaste
195752fe75
- Remove non-existent test_mime_detection.py from unit tests - Use relative paths in security tests for container compatibility
82 lines
2.2 KiB
Python
82 lines
2.2 KiB
Python
#!/usr/bin/env python3
|
|
"""FlaskPaste server with cProfile profiling enabled."""
|
|
|
|
import atexit
|
|
import cProfile
|
|
import io
|
|
import pstats
|
|
import signal
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
sys.path.insert(0, str(Path(__file__).parent.parent.parent))
|
|
|
|
from app import create_app
|
|
|
|
# Global profiler
|
|
profiler = cProfile.Profile()
|
|
profile_output = "/tmp/flaskpaste_profile.prof" # noqa: S108
|
|
stats_output = "/tmp/flaskpaste_profile_stats.txt" # noqa: S108
|
|
|
|
|
|
def save_profile():
|
|
"""Save profiling results on exit."""
|
|
profiler.disable()
|
|
|
|
# Save raw profile data
|
|
profiler.dump_stats(profile_output)
|
|
print(f"\nProfile data saved to: {profile_output}", file=sys.stderr)
|
|
|
|
# Save human-readable stats
|
|
s = io.StringIO()
|
|
ps = pstats.Stats(profiler, stream=s)
|
|
ps.strip_dirs()
|
|
ps.sort_stats("cumulative")
|
|
ps.print_stats(50)
|
|
|
|
with open(stats_output, "w") as f:
|
|
f.write("=" * 80 + "\n")
|
|
f.write("FlaskPaste Profiling Results\n")
|
|
f.write("=" * 80 + "\n\n")
|
|
f.write(s.getvalue())
|
|
|
|
# Also get callers for top functions
|
|
s2 = io.StringIO()
|
|
ps2 = pstats.Stats(profiler, stream=s2)
|
|
ps2.strip_dirs()
|
|
ps2.sort_stats("cumulative")
|
|
ps2.print_callers(20)
|
|
f.write("\n\n" + "=" * 80 + "\n")
|
|
f.write("Top Function Callers\n")
|
|
f.write("=" * 80 + "\n\n")
|
|
f.write(s2.getvalue())
|
|
|
|
print(f"Stats saved to: {stats_output}", file=sys.stderr)
|
|
|
|
|
|
def signal_handler(signum, frame):
|
|
"""Handle shutdown signals."""
|
|
print(f"\nReceived signal {signum}, saving profile...", file=sys.stderr)
|
|
save_profile()
|
|
sys.exit(0)
|
|
|
|
|
|
# Register cleanup handlers
|
|
atexit.register(save_profile)
|
|
signal.signal(signal.SIGTERM, signal_handler)
|
|
signal.signal(signal.SIGINT, signal_handler)
|
|
|
|
if __name__ == "__main__":
|
|
print("Starting FlaskPaste with profiling enabled...", file=sys.stderr)
|
|
print(f"Profile will be saved to: {profile_output}", file=sys.stderr)
|
|
print(f"Stats will be saved to: {stats_output}", file=sys.stderr)
|
|
print("Press Ctrl+C to stop and save profile.\n", file=sys.stderr)
|
|
|
|
app = create_app("development")
|
|
|
|
# Start profiling
|
|
profiler.enable()
|
|
|
|
# Run the server
|
|
app.run(host="127.0.0.1", port=5099, threaded=True, use_reloader=False)
|