From df37f2479ce9a816e63f30e06649a0137f89d4af Mon Sep 17 00:00:00 2001
From: Joana Pereira <joanapereira@4414.bioz.unibas.ch>
Date: Sat, 22 Jul 2023 09:18:10 +0200
Subject: [PATCH] fixed issue with plotting when the parameter was not possible
 to compute

---
 barrOs_library.py | 19 +++++++------------
 1 file changed, 7 insertions(+), 12 deletions(-)

diff --git a/barrOs_library.py b/barrOs_library.py
index 35c4ab9..23db097 100644
--- a/barrOs_library.py
+++ b/barrOs_library.py
@@ -2284,18 +2284,12 @@ def run_barros(arguments, offset = 1, step = 2, local_angle_threshold = 25, dist
             pdb_file = pdbID
             chains_inpdb = get_chains_in_pdb(pdb_file, source_pdb=False)[0]
 
-            print(pdb_file, 'model_' in pdbID)
-            print(pdbID, 'model_' in pdbID)
-
-            if 'model_' in pdbID:
-                pdbID, chainID = pdbID, 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]
+            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)
@@ -2309,6 +2303,7 @@ def run_barros(arguments, offset = 1, step = 2, local_angle_threshold = 25, dist
             pdb_file, protein_type, membrane_thickness = download_pdb(pdbID)
             pdbID, chainID = pdbID.split('_')
 
+        print(pdbID)
         if complexes_only and pdb_file != 'not available':
             chains_inpdb, pdb_file = get_chains_in_pdb(pdb_file)
             target_chains = []
-- 
GitLab