code.grnet.gr Git - ncclient/blob - ncclient/operations/session.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 'Session-related NETCONF operations'
  16
  17 from ncclient.rpc import RPC
  18 from copy import deepcopy
  19
  20 class CloseSession(RPC):
  21
  22     'CloseSession is always synchronous'
  23
  24     SPEC = { 'tag': 'close-session' }
  25
  26     def _delivery_hook(self):
  27         if self.reply.ok:
  28             self.session.expect_close()
  29         self.session.close()
  30
  31     def request(self):
  32         return self._request(CloseSession.SPEC)
  33
  34
  35 class KillSession(RPC):
  36
  37     SPEC = {
  38         'tag': 'kill-session',
  39         'children': [ { 'tag': 'session-id', 'text': None} ]
  40     }
  41
  42     def request(self, session_id):
  43         if not isinstance(session_id, basestring): # just making sure...
  44             session_id = str(session_id)
  45         spec = deepcopy(SPEC)
  46         spec['children'][0]['text'] = session_id
  47         return self._request(spec)