rna_pdb_fetch_header.py: handle pdbid_chain

mmagnus · Dec 4, 2024 · 4bfdbc8 · 4bfdbc8
1 parent 3924a56
commit 4bfdbc8
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/rna_tools/rna_pdb_fetch_header.py b/rna_tools/rna_pdb_fetch_header.py
@@ -85,6 +85,8 @@ def fetch_pdb_header(pdb_id):
         args.file = args.file
 
     for pdb_id in [f.replace('.pdb', '').replace('.cif', '') for f in args.file]:
+        if '_' in pdb_id:
+            pdb_id, chain = pdb_id.split('_')
         header_info = fetch_pdb_header(pdb_id)
         title = header_info.get('entry', {}).get('struct', {}).get('title', 'Title not found')
         # Print the extracted title