root / ncclient / operations / lock.py @ 0cdb8b3c
History | View | Annotate | Download (1.8 kB)
1 | 179b00d4 | Shikhar Bhushan | # Copyright 2h009 Shikhar Bhushan
|
---|---|---|---|
2 | 3043ae4a | Shikhar Bhushan | #
|
3 | 3043ae4a | Shikhar Bhushan | # Licensed under the Apache License, Version 2.0 (the "License");
|
4 | 3043ae4a | Shikhar Bhushan | # you may not use this file except in compliance with the License.
|
5 | 3043ae4a | Shikhar Bhushan | # You may obtain a copy of the License at
|
6 | 3043ae4a | Shikhar Bhushan | #
|
7 | 3043ae4a | Shikhar Bhushan | # http://www.apache.org/licenses/LICENSE-2.0
|
8 | 3043ae4a | Shikhar Bhushan | #
|
9 | 3043ae4a | Shikhar Bhushan | # Unless required by applicable law or agreed to in writing, software
|
10 | 3043ae4a | Shikhar Bhushan | # distributed under the License is distributed on an "AS IS" BASIS,
|
11 | 3043ae4a | Shikhar Bhushan | # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
12 | 3043ae4a | Shikhar Bhushan | # See the License for the specific language governing permissions and
|
13 | 3043ae4a | Shikhar Bhushan | # limitations under the License.
|
14 | 3043ae4a | Shikhar Bhushan | |
15 | 3043ae4a | Shikhar Bhushan | 'Locking-related NETCONF operations'
|
16 | 3043ae4a | Shikhar Bhushan | |
17 | ebf2bbc6 | Shikhar Bhushan | from rpc import RPC |
18 | 94803aaf | Shikhar Bhushan | |
19 | a6c00291 | Shikhar Bhushan | class Lock(RPC): |
20 | 3043ae4a | Shikhar Bhushan | |
21 | 179b00d4 | Shikhar Bhushan | # tested: no
|
22 | ebf2bbc6 | Shikhar Bhushan | # combed: yes
|
23 | 179b00d4 | Shikhar Bhushan | |
24 | 1fca349b | Shikhar Bhushan | SPEC = { |
25 | 1fca349b | Shikhar Bhushan | 'tag': 'lock', |
26 | 179b00d4 | Shikhar Bhushan | 'subtree': {
|
27 | 1fca349b | Shikhar Bhushan | 'tag': 'target', |
28 | 179b00d4 | Shikhar Bhushan | 'subtree': {'tag': None } |
29 | 1fca349b | Shikhar Bhushan | } |
30 | 1fca349b | Shikhar Bhushan | } |
31 | 3043ae4a | Shikhar Bhushan | |
32 | 179b00d4 | Shikhar Bhushan | def request(self, target): |
33 | a6c00291 | Shikhar Bhushan | spec = Lock.SPEC.copy() |
34 | 179b00d4 | Shikhar Bhushan | spec['subtree']['subtree']['tag'] = target |
35 | 1fca349b | Shikhar Bhushan | return self._request(spec) |
36 | 6c1f8a66 | Shikhar Bhushan | |
37 | 6625258b | Shikhar Bhushan | |
38 | a6c00291 | Shikhar Bhushan | class Unlock(RPC): |
39 | 6c1f8a66 | Shikhar Bhushan | |
40 | 179b00d4 | Shikhar Bhushan | # tested: no
|
41 | ebf2bbc6 | Shikhar Bhushan | # combed: yes
|
42 | 179b00d4 | Shikhar Bhushan | |
43 | 1fca349b | Shikhar Bhushan | SPEC = { |
44 | 1fca349b | Shikhar Bhushan | 'tag': 'unlock', |
45 | 179b00d4 | Shikhar Bhushan | 'subtree': {
|
46 | 1fca349b | Shikhar Bhushan | 'tag': 'target', |
47 | 179b00d4 | Shikhar Bhushan | 'subtree': {'tag': None } |
48 | 1fca349b | Shikhar Bhushan | } |
49 | 1fca349b | Shikhar Bhushan | } |
50 | 6c1f8a66 | Shikhar Bhushan | |
51 | 179b00d4 | Shikhar Bhushan | def request(self, target): |
52 | a6c00291 | Shikhar Bhushan | spec = Unlock.SPEC.copy() |
53 | 179b00d4 | Shikhar Bhushan | spec['subtree']['subtree']['tag'] = target |
54 | a6c00291 | Shikhar Bhushan | return self._request(spec) |
55 | d6688264 | Shikhar Bhushan | |
56 | d6688264 | Shikhar Bhushan | |
57 | d6688264 | Shikhar Bhushan | class LockContext: |
58 | 179b00d4 | Shikhar Bhushan | |
59 | 179b00d4 | Shikhar Bhushan | # tested: no
|
60 | ebf2bbc6 | Shikhar Bhushan | # combed: yes
|
61 | 179b00d4 | Shikhar Bhushan | |
62 | 179b00d4 | Shikhar Bhushan | def __init__(self, session, target): |
63 | d6688264 | Shikhar Bhushan | self.session = session
|
64 | d6688264 | Shikhar Bhushan | self.target = target
|
65 | 179b00d4 | Shikhar Bhushan | |
66 | d6688264 | Shikhar Bhushan | def __enter__(self): |
67 | 179b00d4 | Shikhar Bhushan | reply = Lock(self.session).request(self.target) |
68 | 179b00d4 | Shikhar Bhushan | if not reply.ok: |
69 | 179b00d4 | Shikhar Bhushan | raise reply.error
|
70 | 179b00d4 | Shikhar Bhushan | else:
|
71 | 179b00d4 | Shikhar Bhushan | return self |
72 | d6688264 | Shikhar Bhushan | |
73 | 179b00d4 | Shikhar Bhushan | def __exit__(self, *args): |
74 | 179b00d4 | Shikhar Bhushan | reply = Unlock(session).request(self.target)
|
75 | 179b00d4 | Shikhar Bhushan | if not reply.ok: |
76 | 179b00d4 | Shikhar Bhushan | raise reply.error
|
77 | d6688264 | Shikhar Bhushan | return False |