#!/usr/bin/env python

# A DiGIR client

import os.path

import sys

import urllib

import urllib2

sys.path.append(os.path.dirname(__file__)+"/../lib")

import opts

import streams

import util

# Config

timeout = 20 # sec

request_xml_template = '''\

<?xml version="1.0" encoding="UTF-8"?>

<request

    xmlns="http://digir.net/schema/protocol/2003/1.0"

    xmlns:xsd="http://www.w3.org/2001/XMLSchema"

    xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"

    xmlns:digir="http://digir.net/schema/protocol/2003/1.0"

    xmlns:darwin="http://digir.net/schema/conceptual/darwin/2003/1.0"

    xmlns:dwc="http://digir.net/schema/conceptual/darwin/2003/1.0"

    xsi:schemaLocation="http://digir.net/schema/protocol/2003/1.0 

      http://digir.sourceforge.net/schema/protocol/2003/1.0/digir.xsd 

      http://digir.net/schema/conceptual/darwin/2003/1.0 

      http://digir.sourceforge.net/schema/conceptual/darwin/2003/1.0/darwin2.xsd">

    <header>

        <version>1.0</version>

        <sendTime>2012-03-29 01:01:33.00Z</sendTime>

        <source>128.111.220.180</source>

        <destination resource="[resource]">[url]</destination>

        <type>search</type>

    </header>

    <search>

        <filter>

            <equals>

                <darwin:Kingdom>PLANTAE</darwin:Kingdom>

            </equals>

        </filter>

        <records limit="[count]" start="[start]">

            <structure schemaLocation="http://digir.sourceforge.net/schema/conceptual/darwin/full/2003/1.0/darwin2full.xsd"/>

        </records>

        <count>true</count>

    </search>

</request>

'''

def main():

    env_names = []

    def usage_err():

        raise SystemExit('Usage: '+opts.env_usage(env_names, True)+' '

            +sys.argv[0]+' 2>>log')

    # Get config from env vars

    url = opts.get_env_var('url', None, env_names)

    resource = opts.get_env_var('resource', None, env_names)

    start = util.cast(int, opts.get_env_var('start', 0, env_names))

    count = util.cast(int, opts.get_env_var('n', 1, env_names))

    if url == None or resource == None: usage_err()

    this_request_xml_template = (request_xml_template.replace('[url]', url)

        .replace('[resource]', resource).replace('[count]', str(count)))

    request_xml = this_request_xml_template.replace('[start]', str(start))

    this_url = url+'?'+urllib.urlencode({'request': request_xml})

    stream = streams.StreamIter(streams.TimeoutInputStream(

        urllib2.urlopen(this_url), timeout))

    try:

        # Copy lines

        for line in stream:

            sys.stdout.write(line)

    finally: # still run if break is called

        stream.close()

main()