Skip to content

Commit 374ddd2

Browse files
author
Fabian Pedregosa
committed
Merge pull request #5 from octavo/master
Revert back to optparse
2 parents 88bb527 + 027243c commit 374ddd2

File tree

1 file changed

+5
-6
lines changed

1 file changed

+5
-6
lines changed

memory_profiler.py

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -248,18 +248,17 @@ def show_results(prof, stream=None):
248248
stream.write(template.format(l, mem, inc, line))
249249

250250
if __name__ == '__main__':
251-
from argparse import ArgumentParser
252-
parser = ArgumentParser(usage=_CMD_USAGE)
253-
parser.add_argument('filename', help='The file to profile')
251+
from optparse import OptionParser
252+
parser = OptionParser(usage=_CMD_USAGE)
254253

255-
if not sys.argv[1:] or sys.argv[1] in ("--help", "-h"):
254+
if not sys.argv[1:]:
256255
parser.print_help()
257256
sys.exit(2)
258257

259-
args = parser.parse_args()
258+
(options, args) = parser.parse_args()
260259

261260
prof = LineProfiler()
262-
__file__ = _find_script(args.filename)
261+
__file__ = _find_script(args[0])
263262
if sys.version_info[0] < 3:
264263
import __builtin__
265264
__builtin__.__dict__['profile'] = prof

0 commit comments

Comments
 (0)