[Kimchi-devel] [PATCH] [Kimchi 4/9] Virt-Viewer launcher: virtviewerfile module

[Daniel Henrique Barboza]

On 07/07/2016 11:32 AM, Lucio Correia wrote:
> On 07-07-2016 09:57, dhbarboza82 at gmail.com wrote:
>> From: Daniel Henrique Barboza <danielhb at 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 at 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..baccc8a
>> --- /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})
> Why 2 dicts here? Shouldn't it be only one dict?
Why 2? Because I messed up! hehehe

Will fix in v2
>
>
>> +
>> +
>> +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)
>>
>
>