pdb downloader

download_pdb.py

#!/usr/bin/env python

from pathlib import Path

import click
import requests


def download_pdb(accession_code: str, destination: Path):
    accession_code = accession_code.upper()
    url = f'https://files.rcsb.org/download/{accession_code}.pdb'
    request = requests.get(url, allow_redirects=True)
    with open(destination, mode='wb') as f:
        f.write(request.content)


@click.command()
@click.argument('accession_code')
def cli(accession_code: str):
    output_filename = f'{accession_code}.pdb'
    download_pdb(accession_code, output_filename)


if __name__ == '__main__':
    cli()