address issue 8

[ncclient] / ncclient / operations / __init__.py

diff --git a/ncclient/operations/__init__.py b/ncclient/operations/__init__.py
index 91ee2be..df50258 100644 (file)
--- a/ncclient/operations/__init__.py
+++ b/ncclient/operations/__init__.py
@@ -14,32 +14,19 @@
 
 from errors import OperationError, TimeoutExpiredError, MissingCapabilityError
 from rpc import RPC, RPCReply, RPCError
+
+# rfc4741 ops
 from retrieve import Get, GetConfig, GetReply
 from edit import EditConfig, CopyConfig, DeleteConfig, Validate, Commit, DiscardChanges
 from session import CloseSession, KillSession
 from lock import Lock, Unlock, LockContext
-#from subscribe import CreateSubscription
-
-OPERATIONS = {
-    'get': Get,
-    'get-config': GetConfig,
-    'edit-config': EditConfig,
-    'copy-config': CopyConfig,
-    'validate': Validate,
-    'commit': Commit,
-    'discard-changes': DiscardChanges,
-    'delete-config': DeleteConfig,
-    'lock': Lock,
-    'unlock': Unlock,
-    'close-session': CloseSession,
-    'kill-session': KillSession,
-}
+# others...
+from flowmon import PoweroffMachine, RebootMachine
 
 __all__ = [
     'RPC',
     'RPCReply',
     'RPCError',
-    'OPERATIONS',
     'Get',
     'GetConfig',
     'GetReply',
@@ -51,6 +38,8 @@ __all__ = [
     'DeleteConfig',
     'Lock',
     'Unlock',
+    'PoweroffMachine',
+    'RebootMachine',
     'LockContext',
     'CloseSession',
     'KillSession',