Skip to content

Commit c6b626d

Browse files
committed
Fix: Generates output in wrong place if parent directory name contains a dot
Change to splitext instead.
1 parent f2a45d5 commit c6b626d

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

pseudo_python/main.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
#!/usr/bin/env python3
2-
2+
import os
33
import sys
44
import pseudo_python
55
import pseudo_python.errors
@@ -38,7 +38,7 @@ def main():
3838
filename = sys.argv[1]
3939
with open(filename, 'r') as f:
4040
source = f.read()
41-
base = filename.partition('.')[0]
41+
base, _ = os.path.splitext(filename)
4242
try:
4343
if len(sys.argv) == 2:
4444
yaml.Dumper.ignore_aliases = lambda *args : True
@@ -49,7 +49,8 @@ def main():
4949
else:
5050
arg = sys.argv[2]
5151
if '.' in arg:
52-
base, _, language = arg.rpartition('.')
52+
base, language = os.path.splitext(arg)
53+
language = language[1:]
5354
else:
5455
language = arg
5556
if language not in pseudo.SUPPORTED_FORMATS:

0 commit comments

Comments
 (0)