diff options
Diffstat (limited to 'Modules/stdwinmodule.c')
| -rw-r--r-- | Modules/stdwinmodule.c | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/Modules/stdwinmodule.c b/Modules/stdwinmodule.c index 1fff814a39..fc81cffded 100644 --- a/Modules/stdwinmodule.c +++ b/Modules/stdwinmodule.c @@ -2598,6 +2598,7 @@ initstdwin() { object *m, *d; static int inited = 0; + char buf[1000]; if (!inited) { int argc = 0; @@ -2607,6 +2608,18 @@ initstdwin() if (!checkstringlist(sys_argv, &argv, &argc)) err_clear(); } + if (argc > 0) { + /* If argv[0] has a ".py" suffix, remove the suffix */ + char *p = strrchr(argv[0], '.'); + if (p != NULL && strcmp(p, ".py") == 0) { + int n = p - argv[0]; + if (n >= sizeof(buf)) + n = sizeof(buf)-1; + strncpy(buf, argv[0], n); + buf[n] = '\0'; + argv[0] = buf; + } + } winitargs(&argc, &argv); if (argv != NULL) { if (!putbackstringlist(sys_argv, argv, argc)) |