Statistics
| Branch: | Tag: | Revision:

root / ncclient / operations / session.py @ 1d540e60

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, should be ok to parse in callback
29
        if self._reply.ok: # (implicitly parse)
30
            self._session.expect_close()
31
        self._session.close()
32
    
33
    def request(self):
34
        self._request(self.spec)
35

    
36

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