From b635a434139439537163b0f05bdad3a2ce4ba112 Mon Sep 17 00:00:00 2001 From: H4dr1en Date: Mon, 16 Sep 2019 11:29:59 +0200 Subject: [PATCH] Added windows support --- bin/flameprof | 2 +- flameprof.py | 8 +++++++- setup.py | 3 +++ 3 files changed, 11 insertions(+), 2 deletions(-) diff --git a/bin/flameprof b/bin/flameprof index 7e83116..c2554d8 100755 --- a/bin/flameprof +++ b/bin/flameprof @@ -1,2 +1,2 @@ #!/bin/sh -exec python -m flameprof "$@" +exec python -m flameprof "$@" \ No newline at end of file diff --git a/flameprof.py b/flameprof.py index ba9c3a8..bb5bc17 100644 --- a/flameprof.py +++ b/flameprof.py @@ -371,7 +371,9 @@ def get_arg_parser(): return parser -if __name__ == '__main__': + +def main(): + parser = get_arg_parser() args, rest = parser.parse_known_args() @@ -409,6 +411,10 @@ def get_arg_parser(): render(s.stats, get_out(args.out), args.format, args.threshold / 100, args.width, args.row_height, args.font_size, args.log_mult) + + +if __name__ == '__main__': + main() else: try: import pytest diff --git a/setup.py b/setup.py index ced6890..3d30bd7 100644 --- a/setup.py +++ b/setup.py @@ -12,6 +12,9 @@ long_description=open('README.rst').read(), py_modules=['flameprof'], scripts=['bin/flameprof'], + entry_points = { + 'console_scripts': ['flameprof=flameprof:main'], + }, include_package_data=True, zip_safe=False, platforms='any',