#!/usr/bin/env python

import sys
import getopt

from Bio.Medline import PubMed

USAGE = \
"""query_pubmed <query>

Do a query on PubMed and print a list of the PMID's that are found.

"""

def print_pmid(id):
    print id
    sys.stdout.flush()

if __name__ == '__main__':
    try:
        optlist, args = getopt.getopt(sys.argv[1:], "h")
    except getopt.error, x:
        print USAGE
        print >>sys.stderr, x
        sys.exit(0)

    if optlist:   # only option is '-h'
        print USAGE
        sys.exit(0)

    query = sys.argv[1]
    PubMed.search_for(query, callback_fn=print_pmid)