X-Git-Url: https://code.grnet.gr/git/ganeti-local/blobdiff_plain/ec3cc4a81b9b7eb665a7b1d192a29c0b59a8a9a7..fbff213691328562cc4029f3babaa2e72e2012c4:/lib/cmdlib/query.py

diff --git a/lib/cmdlib/query.py b/lib/cmdlib/query.py
index 16a6c29..6cfd207 100644
--- a/lib/cmdlib/query.py
+++ b/lib/cmdlib/query.py
@@ -24,27 +24,27 @@
 from ganeti import constants
 from ganeti import errors
 from ganeti import query
-from ganeti.cmdlib import _ExtStorageQuery
-from ganeti.cmdlib import _OsQuery
-from ganeti.cmdlib.backup import _ExportQuery
+from ganeti.cmdlib.backup import ExportQuery
 from ganeti.cmdlib.base import NoHooksLU
-from ganeti.cmdlib.cluster import _ClusterQuery
-from ganeti.cmdlib.group import _GroupQuery
-from ganeti.cmdlib.instance import _InstanceQuery
-from ganeti.cmdlib.network import _NetworkQuery
-from ganeti.cmdlib.node import _NodeQuery
+from ganeti.cmdlib.cluster import ClusterQuery
+from ganeti.cmdlib.group import GroupQuery
+from ganeti.cmdlib.instance_query import InstanceQuery
+from ganeti.cmdlib.misc import ExtStorageQuery
+from ganeti.cmdlib.network import NetworkQuery
+from ganeti.cmdlib.node import NodeQuery
+from ganeti.cmdlib.operating_system import OsQuery
 
 
 #: Query type implementations
 _QUERY_IMPL = {
-  constants.QR_CLUSTER: _ClusterQuery,
-  constants.QR_INSTANCE: _InstanceQuery,
-  constants.QR_NODE: _NodeQuery,
-  constants.QR_GROUP: _GroupQuery,
-  constants.QR_NETWORK: _NetworkQuery,
-  constants.QR_OS: _OsQuery,
-  constants.QR_EXTSTORAGE: _ExtStorageQuery,
-  constants.QR_EXPORT: _ExportQuery,
+  constants.QR_CLUSTER: ClusterQuery,
+  constants.QR_INSTANCE: InstanceQuery,
+  constants.QR_NODE: NodeQuery,
+  constants.QR_GROUP: GroupQuery,
+  constants.QR_NETWORK: NetworkQuery,
+  constants.QR_OS: OsQuery,
+  constants.QR_EXTSTORAGE: ExtStorageQuery,
+  constants.QR_EXPORT: ExportQuery,
   }
 
 assert set(_QUERY_IMPL.keys()) == constants.QR_VIA_OP