6:10 p.m.
From: Daniel Henrique Barboza <danielhb(a)linux.vnet.ibm.com>
This new module contains functions to create virt
viewer files and a new model called VMVirtViewerFileModel
that will return a link to the new generated script
file in the lookup() method.
Signed-off-by: Daniel Henrique Barboza <danielhb(a)linux.vnet.ibm.com>
---
model/virtviewerfile.py | 100 ++++++++++++++++++++++++++++++++++++++++++++++++
1 file changed, 100 insertions(+)
create mode 100644 model/virtviewerfile.py
diff --git a/model/virtviewerfile.py b/model/virtviewerfile.py
new file mode 100644
index 0000000..9b2a9bd
--- /dev/null
+++ b/model/virtviewerfile.py
@@ -0,0 +1,100 @@
+#
+# Project Kimchi
+#
+# Copyright IBM Corp, 2016
+#
+# This library is free software; you can redistribute it and/or
+# modify it under the terms of the GNU Lesser General Public
+# License as published by the Free Software Foundation; either
+# version 2.1 of the License, or (at your option) any later version.
+#
+# This library is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+# Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public
+# License along with this library; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
+#
+
+import cherrypy
+import libvirt
+import os
+
+from wok.config import config as wok_config
+from wok.exception import NotFoundError, OperationFailed
+from wok.plugins.kimchi import config as kimchi_config
+from wok.plugins.kimchi.model.vms import VMModel
+
+
+def write_virt_viewer_file(params):
+ file_template = """\
+[virt-viewer]
+type=%(type)s
+host=%(host)s
+port=%(graphics_port)s
+"""
+ file_contents = file_template % params
+
+ if params.get('graphics_passwd'):
+ file_contents += 'password=%s\n' % params['graphics_passwd']
+
+ try:
+ with open(params.get('path'), 'w') as vv_file:
+ vv_file.write(file_contents)
+ except Exception:
+ raise
+
+
+def _get_request_host():
+ host = cherrypy.request.headers.get('Host')
+ if not host:
+ host = wok_config.get("server", "host")
+ host = host.split(':')[0]
+ return host
+
+
+def create_virt_viewer_file(vm_name, graphics_info):
+ graphics_type = graphics_info[0]
+ graphics_port = graphics_info[2]
+ graphics_passwd = graphics_info[3]
+
+ try:
+ host = _get_request_host()
+
+ default_dir = kimchi_config.get_virtviewerfiles_path()
+ file_path = os.path.join(default_dir, '%s-access.vv' % vm_name)
+
+ file_params = {
+ 'type': graphics_type,
+ 'graphics_port': graphics_port,
+ 'graphics_passwd': graphics_passwd,
+ 'host': host,
+ 'path': file_path
+ }
+ write_virt_viewer_file(file_params)
+ return file_path
+
+ except Exception as e:
+ raise OperationFailed("KCHVM0084E",
+ {'name': vm_name, 'err': e.message})
+
+
+class VMVirtViewerFileModel(object):
+ def __init__(self, **kargs):
+ self.conn = kargs['conn']
+
+ def _check_if_vm_running(self, name):
+ dom = VMModel.get_vm(name, self.conn)
+ d_info = dom.info()
+ if d_info[0] != libvirt.VIR_DOMAIN_RUNNING:
+ raise NotFoundError("KCHVM0083E", {'name': name})
+
+ def lookup(self, name):
+ self._check_if_vm_running(name)
+ graphics_info = VMModel.get_graphics(name, self.conn)
+ file_path = create_virt_viewer_file(name, graphics_info)
+
+ return 'plugins/kimchi/data/virtviewerfiles/%s' %\
+ os.path.basename(file_path)
--
2.5.5