diff --git a/barrOs_library.py b/barrOs_library.py index 07a1c901d31d6bab0a3712d73bce063a983bfb82..05dde8efb3f1454be8937a36e9cea1c45986c77d 100644 --- a/barrOs_library.py +++ b/barrOs_library.py @@ -2283,12 +2283,16 @@ def run_barros(arguments, offset = 1, step = 2, local_angle_threshold = 25, dist isfile = True pdb_file = pdbID chains_inpdb = get_chains_in_pdb(pdb_file, source_pdb=False)[0] - try: - pdbID, chainID = pdb_file[:-4].split('/')[-1].split('_') - except: - pdbID, chainID = pdb_file[:-4], chains_inpdb[0] - if len(pdbID) > 4: - pdbID, chainID = pdb_file[:-4], chains_inpdb[0] + + if 'AF-' in pdbID: + pdbID, chainID = pdb_file, chains_inpdb[0] + else: + try: + pdbID, chainID = pdb_file[:-4].split('/')[-1].split('_') + except: + pdbID, chainID = pdb_file[:-4], chains_inpdb[0] + if len(pdbID) > 4: + pdbID, chainID = pdb_file[:-4], chains_inpdb[0] # check if this is a OPM structure membrane_thickness = parse_membrane_thickness_from_OMP(pdb_file)