group_create.py

#!/usr/bin/env python
from argparse import ArgumentParser
import os
import pycurl
import base64
import json

from Common import getConfig
from CurlCallback import Test


###########################################################################################
# Note this script is created via reverse-engineering the following java codes in iRODS-rest
#  - src/test/java/org/irods/jargon/rest/commands/user/UserGroupServiceTest.java
#  - src/main/java/org/irods/jargon/rest/commands/user/UserGroupRequest.java
#
# as the documentation is missing for this part of operation. The interface may be changed
# in the future.  Always consult the latest documentation for an update
###########################################################################################
if __name__ == "__main__":
    cfg = getConfig( os.path.dirname(os.path.abspath(__file__)) + '/etc/config.ini' )

    parg = ArgumentParser(description='create iRODS groups')

    ## positional arguments
    parg.add_argument('group',
                      metavar = 'group',
                      nargs   = '+',
                      help    = 'iRODS user groups')

    ## optional arguments
    parg.add_argument('-z','--zone',
                      action  = 'store',
                      dest    = 'zone',
                      default = 'rdm-tst',
                      help    = 'specify the iRODS zone')

    args = parg.parse_args()

    for g in args.group:
        accept_frsp      = 'application/json'
        auth             = 'Authorization: Basic %s' % base64.b64encode("%s:%s" % (cfg.get('iRODS','username'), cfg.get('iRODS','password')))
        base_url         = cfg.get('RESTFUL','http_endpt')
        resource         = 'user_group'
        http_request     = 'PUT'

        ## new user group data
        ugroup_data  = {'userGroupName':g, 'zone':args.zone}
        content_type = 'application/json'

        ## several listing options
        params = []
        dest_url = '?'.join([os.path.join(base_url, resource), '&'.join(params)])

        print 'creating user group %s' % ugroup_data['userGroupName']
        print 'sending %s to %s' % (http_request, dest_url)

        t = Test()
        c = pycurl.Curl()
        c.setopt(c.URL, dest_url)
        c.setopt(c.HEADERFUNCTION, t.header_callback)
        c.setopt(c.WRITEFUNCTION , t.body_callback)
        c.setopt(c.CUSTOMREQUEST , http_request)
        c.setopt(c.HTTPHEADER    , ['ACCEPT: %s' % accept_frsp, 'Content-Type: %s' % content_type, auth])
        c.setopt(c.POSTFIELDS    , json.dumps(ugroup_data))
        c.perform()
        c.close()

        print t.header

        print json.dumps( json.loads(t.contents), indent=4, sort_keys=True, separators=(',',':') )