code.grnet.gr Git - ncclient/blob - ncclient/operations/retrieve.py

   1 # Copyright 2009 Shikhar Bhushan
   2 #
   3 # Licensed under the Apache License, Version 2.0 (the "License");
   4 # you may not use this file except in compliance with the License.
   5 # You may obtain a copy of the License at
   6 #
   7 #    http://www.apache.org/licenses/LICENSE-2.0
   8 #
   9 # Unless required by applicable law or agreed to in writing, software
  10 # distributed under the License is distributed on an "AS IS" BASIS,
  11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12 # See the License for the specific language governing permissions and
  13 # limitations under the License.
  14
  15 from ncclient.rpc import RPC, RPCReply
  16
  17 def build_filter(spec, type, criteria):
  18     filter = {
  19         'tag': 'filter',
  20         'attributes': {'type': type}
  21     }
  22     if type == 'subtree':
  23         filter['children'] = [criteria]
  24     elif type == 'xpath':
  25         filter['attributes']['select'] = criteria
  26     return filter
  27
  28 class GetReply(RPCReply):
  29
  30     def parse(self):
  31         RPCReply.parse(self)
  32
  33 class Get(RPC):
  34
  35     SPEC = {
  36         'tag': 'get',
  37         'children': []
  38     }
  39
  40     REPLY_CLS = GetReply
  41
  42     def request(self, filter=None):
  43         spec = Get.SPEC.copy()
  44         if filter is not None:
  45             #if filter[0] == 'xpath':
  46             #    self._assert(':xpath')
  47             spec['children'].append(build_filter(*filter))
  48         return self._request(spec)
  49
  50 class GetReply(RPCReply):
  51
  52     def parse(self):
  53         RPCReply.parse(self)
  54
  55 class GetConfig(RPC): # xx
  56
  57     SPEC = {
  58         'tag': 'get-config',
  59         'children': [ { 'tag': 'source', 'children': {'tag': None } } ]
  60     }
  61
  62     REPLY_CLS = GetReply
  63
  64     def request(self, source=None, source_url=None, filter=None):
  65         self._one_of(source, source_url)
  66         spec = GetConfig.SPEC.copy()
  67         if source is not None:
  68             spec['children'][0]['children']['tag'] = source
  69         if source_url is not None:
  70             #self._assert(':url')
  71             spec['children'][0]['children']['tag'] = 'url'
  72             spec['children'][0]['children']['text'] = source_url
  73         if filter is not None:
  74             #if filter[0] == 'xpath':
  75             #    self._assert(':xpath')
  76             spec['children'].append(build_filter(*filter))
  77         return self._request(spec)