diff --git a/distlib/scripts.py b/distlib/scripts.py index b1fc705..195dc3f 100644 --- a/distlib/scripts.py +++ b/distlib/scripts.py @@ -42,8 +42,8 @@ SCRIPT_TEMPLATE = r'''# -*- coding: utf-8 -*- import re import sys -from %(module)s import %(import_name)s if __name__ == '__main__': + from %(module)s import %(import_name)s sys.argv[0] = re.sub(r'(-script\.pyw|\.exe)?$', '', sys.argv[0]) sys.exit(%(func)s()) '''