def get_het_id(hetname, chain: Chain):
residues = chain.get_residues()
for residue in residues:
residue_id = residue.get_id()
hetfield = residue_id[0]
if hetfield != " " and hetfield != "W":
if residue_id[0] == hetname:
return residue_id
else:
return None
def get_het_ids(chain: Chain):
residues = chain.get_residues()
residue_list = []
for residue in residues:
residue_id = residue.get_id()
het_field = residue_id[0]
het_name = residue.get_resname()
if het_field != " " and het_field != "W":
residue_list.append(het_name)
return residue_list
