Skip to content

Commit 2027f00

Browse files
authored
Merge pull request #149 from jkuhl-uni/feat/read_ms5_xsf
fix for the case that there are multiple dots in file names
2 parents 3ca7958 + ea1c89c commit 2027f00

File tree

1 file changed

+14
-9
lines changed

1 file changed

+14
-9
lines changed

pyerrors/input/openQCD.py

Lines changed: 14 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1061,23 +1061,28 @@ def read_ms5_xsf(path, prefix, qc, corr, sep="r", **kwargs):
10611061
found = []
10621062
files = []
10631063
names = []
1064+
1065+
if "names" in kwargs:
1066+
names = kwargs.get("names")
1067+
10641068
for (dirpath, dirnames, filenames) in os.walk(path + "/"):
10651069
found.extend(filenames)
10661070
break
10671071

10681072
for f in found:
10691073
if fnmatch.fnmatch(f, prefix + "*.ms5_xsf_" + qc + ".dat"):
10701074
files.append(f)
1071-
if not sep == "":
1072-
names.append(prefix + "|r" + f.split(".")[0].split(sep)[1])
1073-
else:
1074-
names.append(prefix)
1075-
files = sorted(files)
1075+
if "names" not in kwargs:
1076+
if not sep == "":
1077+
se = f.split(".")[0]
1078+
for s in f.split(".")[1:-1]:
1079+
se += "." + s
1080+
names.append(se.split(sep)[0] + "|r" + se.split(sep)[1])
1081+
else:
1082+
names.append(prefix)
10761083

1077-
if "names" in kwargs:
1078-
names = kwargs.get("names")
1079-
else:
1080-
names = sorted(names)
1084+
names = sorted(names)
1085+
files = sorted(files)
10811086

10821087
cnfgs = []
10831088
realsamples = []

0 commit comments

Comments
 (0)