Read a PDB entry using the pypdb module

readPdbEntry-pypdb.py

from pypdb import *

#testPdb = '4xkl'
#testPdb = '5v4k'
testPdb = '3vtv'
entityExtraInfo = get_all_info(testPdb)

print(entityExtraInfo)
print('\n')

if int(describe_pdb(testPdb)['nr_entities']) == 1:
    print(entityExtraInfo.get('polymer').get('macroMolecule').get('@name'))
elif int(describe_pdb(testPdb)['nr_entities']) > 1:
    for mol in entityExtraInfo['polymer']:
        print(mol.get('macroMolecule').get('@name'))
else:
    print('No molecule name available')