diff --git a/scripts/init.py b/scripts/init.py index e4b1ee14b8dc9270484b9443edf006ddbd726d56..08f538045a05d3f143c6d18e20431650507170a8 100644 --- a/scripts/init.py +++ b/scripts/init.py @@ -69,8 +69,12 @@ def _load_files(): graphical_objects=[] try: for f in input_files: - e=os.path.splitext(f[0])[1] - if e in ['.pdb', '.ent', '.ent.gz', '.pdb.gz']: + is_pdb_file=False + for ext in ['.pdb', '.ent', '.ent.gz', '.pdb.gz']: + if f[0].endswith(ext): + is_pdb_file=True + break + if is_pdb_file: es=io.LoadPDB(f[0], load_multi=True) for i, e in enumerate(es): index+=1