from sfa.rspecs.rspec_elements import RSpecElement, RSpecElements

class PGv2:
    format = 'ProtoGENI'
    type = '*'
    version = '2'
    schema = 'http://www.protogeni.net/resources/rspec/2/ad.xsd' 
    namespaces = { 
        'default' : 'http://www.protogeni.net/resources/rspec/2',
        'flack': "http://www.protogeni.net/resources/rspec/ext/flack/1",
        'planetlab': "http://www.planet-lab.org/resources/sfa/ext/planetlab/1", 
    }
    elements = [
        RSpecElement(RSpecElements.NETWORK, 'network', '//default:node[@component_manager_id][1]'),
        RSpecElement(RSpecElements.NODE, 'node', '//default:node | //node'),
        RSpecElement(RSpecElements.SLIVER, 'sliver', '//default:node/default:sliver_type | //node/sliver_type'),
    ]


class PGv2Ad(PGv2):
    schema = 'http://www.protogeni.net/resources/rspec/2/ad.xsd'

class PGv2Manifest(PGv2):
    schema = 'http://www.protogeni.net/resources/rspec/2/manifest.xsd'

if __name__ == '__main__':
    from sfa.rspecs.rspec import RSpec
    from sfa.rspecs.rspec_elements import *
    r = RSpec('/tmp/pg.rspec')
    r.load_rspec_elements(PGv2.elements)
    r.namespaces = PGv2.namespaces
    print r.get(RSpecElements.NODE)