Fixing problem reading data for some IdenticalChains...

setup.py

@@ -6,7 +6,7 @@
 
 setup(
     name="spyprot",
-    version='0.7.10',
+    version='0.7.12',
     author="INTERDISCIPLINARY LABORATORY of BIOLOGICAL SYSTEMS MODELLING, University of Warsaw, Warsaw, Poland",
     author_email="[email protected], [email protected]",
     description="This package provides a set of tools for accessing protein databases and manipulating PDB/CIF files.",

spyprot/fetchChainInfo.py

@@ -713,6 +713,9 @@ def __init__(self, pdbcode, chain):
         documents = self.exec_query("pdb_id,entity_id,chain_id,assembly_composition", [('pdb_id', pdbcode)])
 
         for i in range(len(documents)):
+            if 'chain_id' not in documents[i] or 'entity_id' not in documents[i]:
+                self.results = (None, None)
+                continue
             chain_id = documents[i]['chain_id']
             ent_id = documents[i]['entity_id']
             if chain in chain_id:

test_fetch_chain.py

@@ -11,6 +11,10 @@ def test_fetchChainInfo_IdenticalChains():
 
 
 def test_fetchChainInfo_IdenticalChainsAndEntityId():
+    f = IdenticalChainsAndEntityId('7dzp', 'A')
+    ent_id, chains = f.get()
+    assert ent_id == None
+    assert chains == None
     f = IdenticalChainsAndEntityId('6wm4', 'Y')
     ent_id, chains = f.get()
     assert int(ent_id) == 9