summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rwxr-xr-xParser/asdl_c.py5
1 files changed, 2 insertions, 3 deletions
diff --git a/Parser/asdl_c.py b/Parser/asdl_c.py
index 2167c30..f45bee4 100755
--- a/Parser/asdl_c.py
+++ b/Parser/asdl_c.py
@@ -639,9 +639,8 @@ class ChainOfVisitors:
def main(srcfile):
argv0 = sys.argv[0]
- index = argv0.find('/')
- if index > 0:
- argv0 = argv0[index + 1:]
+ components = argv0.split(os.sep)
+ argv0 = os.sep.join(components[-2:])
auto_gen_msg = '/* File automatically generated by %s */\n' % argv0
mod = asdl.parse(srcfile)
if not asdl.check(mod):