poster.py

#!/usr/bin/env python2

import requests
import argparse
import os.path
from datetime import datetime

parser = argparse.ArgumentParser( description='Simple HTTP POST Requester' );
parser.add_argument( 'url', help='URL to send the POST request' );
parser.add_argument( '--auth', nargs=2, help='Provide credential' );
parser.add_argument( '--header', nargs='*', help='Specify required request headers' );
parser.add_argument( '--fheader', help='Specify a file containing the required request headers' );
parser.add_argument( '--ua', help='Specify User-Agent header' );
parser.add_argument( '--ct', help='Specify Content-Type header' );
parser.add_argument( '--data', help='Data to be sent inside the request body' );
parser.add_argument( '--fdata', help='Specify a file containing the data to be sent inside the request body' );
args = parser.parse_args();

#args.url = args.url.lower()  
#url = args.url;
path = 'requests/';     # path to write requests's response

def end( reason ):
    print reason;
    exit();

def check_url_syntax( url ):
    if ((url.find('http://',0,7) == -1) and (url.find('https://',0,8) == -1)):
        end('\nERROR: not starting with http or https.\nCheck your URL and try again.\n');

def verify_url( url ):
    print('\nChecking if URL is available...');
    req = requests.post(url, data='');   
    if (req.status_code == requests.codes.ok):
        return
    else:
        print "HTTP Response Status Code %s - %s" %( req.status_code, req.reason );
        end("Exiting.\n");


def make_request( url, usr_headers, content='None' ):
    if ( args.auth is not None ):
        http = requests.post( url, data=content, headers=usr_headers, auth=(args.auth[0], args.auth[1]) );
        if (http.status_code == requests.codes.ok):
            http.close();
            return http;
        else:
            print('\n%s\n' % http.text);
            end("\nHTTP Response Status Code %s - %s\n" %( http.status_code, http.reason ) );
    else:
        http = requests.post( url, data=content, headers=usr_headers );
    http.close();
    return http;

def get_address( url ):
    if url.startswith( 'http://' ):
        address = url.replace( 'http://', '', 1 );
        address = address.replace( '/', '_' );
        time = datetime.now();
        address = address + '_' + time.strftime( "%Y%m%d_%H%M.%S.%f" );    # www.example.com_webservice.php_20141110_1822.15.059238.txt
        return address;
    elif url.startswith( 'https://' ):
        address = url.replace( 'https://', '', 1 );
        address = address.replace( '/', '_' );
        time = datetime.now();
        address = address + '_' + time.strftime( "%Y%m%d_%H%M.%S.%f" );    # www.example.com_webservice.php_20141110_1822.15.059238.txt
        return address;
    else:
        end('Malformed URL');

def save_response( request_content, response_content ):
    if not os.path.exists(path):
        os.makedirs(path);
    fp_req = open( path + url + '_req.xml', 'w+' );     # requests/www.example.com_webservice.php_20141110_1822.15.059238_req.xml 
    fp_resp = open( path + url + '_resp.xml', 'w+' );    # requests/www.example.com_webservice.php_20141110_1822.15.059238_resp.xml
    fp_req.write( request_content );
    fp_resp.write( response_content );
    fp_req.close();
    fp_resp.close();

def add_default_headers():
    return dict({ 'Content-Type': 'text/html; charset=utf-8', 'User-Agent': 'Poster/1.0' });


def add_header( header_dict, field_value_dict ):
    header_dict.update( field_value_dict );
    return

def add_headers( args ):
    hr = add_default_headers();
    if ( args.header is not None ):
        add_header( hr, list2dict( args.header ) );
    if ( args.fheader is not None ):
        add_header( hr, get_headers_from_file( args.fheader ) );
    if ( args.ua is not None ):
        add_header( hr, list2dict( [ 'User-Agent', args.ua ] ) );
    if ( args.ct is not None ):
        add_header( hr, list2dict( [ 'Content-Type', args.ct ] ) );
    return hr;


def get_headers_from_file( hdr_file ):
    if (os.path.exists( hdr_file )):
        fp_hdr = open( hdr_file, 'r');
        lines = dict();
        for line in fp_hdr:
            line = replace_tabs(line);
            lines.update( list2dict( line ) );
        fp_hdr.close();
        return lines;
    else:
        end("File not found: %s\n" %( hdr_file ));

def list2dict( llist ):
    return dict( llist[i:i+2] for i in range( 0, len( llist ), 2) )


def replace_tabs( string ):
    return string.split();


check_url_syntax( args.url );
#verify_url( args.url );
url = get_address( args.url ); 
hr = add_headers( args );
http_resp = make_request( args.url, hr, args.data );
print( '\n%s \n' %( http_resp.text ) );
end( "HTTP Response Status Code %s - %s\n" %( http_resp.status_code, http_resp.reason ) );