asn1.py

import zlib
from pyasn1.codec.ber import decoder

def decode_residues(res):
    residues = []
    try:
        code, rest = decoder.decode(zlib.decompress(res, 16 + zlib.MAX_WBITS))
    except:
        pass
    for i in range(len(code)):
        c = code[i]
        range_from, range_to, gi = tuple([c[j] for j in range(len(c))])
        for x in range(range_from, range_to + 1):
            residues.append(x)
    return residues


def get_residues(obs_int_id, mol_int="mol"):
    global MolResFace, IntResFace

    if mol_int == "mol":
        qres = db_ibis.query(MolResFace).filter(MolResFace.obs_int_id == obs_int_id).one()
    else:
        qres = db_ibis.query(IntResFace).filter(IntResFace.obs_int_id == obs_int_id).one()

    residues = decode_residues(qres.seqloc)
    return residues