mmagnus / rna-tools

    return npath

def fetch_ba(pdb_id, path="."):

    """fetch biological assembly pdb file from RCSB.org

    >>> fetch_ba('1xjr')

    ...

    """

    try:

        import urllib3

    except ImportError:

        print('urllib3 is required')

        return

    http = urllib3.PoolManager()

    # try:

    response = http.request('GET', url='https://files.rcsb.org/download/' +

                            pdb_id.lower() + '.pdb1')

    if not response.status == 200:

        raise PDBFetchError()

    txt = response.data

    npath = path + os.sep + pdb_id + '_ba.pdb'

    print('downloading...' + npath)

    with open(npath, 'wb') as f:

        f.write(txt)

    print('ok')

    return pdb_id + '_ba.pdb'

def fetch_cif_ba(cif_id, path="."):

    return pdb_id + '_ba.pdb'

def fetch_cif_ba(cif_id, path="."):

    """fetch biological assembly cif file from RCSB.org"""

    import urrlib3

    http = urllib3.PoolManager()

    # try:

    response = http.request('GET', url='https://files.rcsb.org/download/' +

                            cif_id.lower() + '-assembly1.cif')

    if not response.status == 200:

        raise PDBFetchError()

    txt = response.data

    npath = path + os.sep + cif_id + '_ba.cif'

    print('downloading...' + npath)

    with open(npath, 'wb') as f:

        f.write(txt)

    print('ok')

    return cif_id + '_ba.cif'

def fetch_cif(cif_id, path="."):

    """fetch biological assembly cif file from RCSB.org"""