Statistics
| Branch: | Tag: | Revision:

root / ncclient / operations / session.py @ 11d9e642

History | View | Annotate | Download (1.6 kB)

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 rpc import RPC
18

    
19

    
20
class CloseSession(RPC):
21
    
22
    def __init__(self, *args, **kwds):
23
        RPC.__init__(self, *args, **kwds)
24
        self.spec = { 'tag': 'close-session' }
25
    
26
    def deliver(self, reply):
27
        RPC.deliver(self, reply)
28
        # can't be too huge a reply, should be ok to parse in callback
29
        self._reply.parse()
30
        if self._reply.ok:
31
            self._session.expect_close()
32
        self._session.close()
33
    
34
    def request(self, reply_event=None):
35
        self._request(self.spec, reply_event)
36

    
37

    
38
class KillSession(RPC):
39
    
40
    def __init__(self, *args, **kwds):
41
        RPC.__init__(self, *args, **kwds)
42
        self.spec = {
43
            'tag': 'kill-session',
44
            'children': [ { 'tag': 'session-id', 'text': None} ]
45
            }
46
    
47
    def request(self, session_id):
48
        if not isinstance(session_id, basestring): # just make sure...
49
            session_id = str(session_id)
50
        self.spec['children'][0]['text'] = session_id
51
        self._request(self.spec)