libvirt.py

Go to the documentation of this file.

#!/usr/bin/python
# -*- coding: utf-8 -*-
# @cond LICENSE
#
# Copyright [2010-2013] Institute of Nuclear Physics PAN, Krakow, Poland
#
# Licensed under the Apache License, Version 2.0 (the "License");
#    you may not use this file except in compliance with the License.
#    You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
#    distributed under the License is distributed on an "AS IS" BASIS,
#    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#    See the License for the specific language governing permissions and
#    limitations under the License.
#
# @endcond LICENSE

##
# @package src.node.tools.libvirt
# @author Maciej Nabozny <mn@mnabozny.pl>
# 

import os
import sys
import stat
import subprocess

sys.path.append('/usr/lib/cc1/')

from common.tools import config


def configure(address, username, transport, driver, suffix):
    print "NODE: Updating libvirt config"
    qemu = open('/etc/libvirt/qemu.conf', 'r')
    qemu_conf = qemu.readlines()
    qemu.close()

    qemu = open('/etc/libvirt/qemu.conf', 'w')
    for line in qemu_conf:
        if '#user = "root"' in line:
            qemu.write('user = "root"\n')
        elif '#group = "root"' in line:
            qemu.write('group = "root"\n')
        elif '#dynamic_ownership = 1' in line:
            qemu.write('dynamic_ownership = 0')
        else:
            qemu.write(line)
    qemu.close()
    os.chmod('/etc/libvirt/qemu.conf', stat.S_IROTH | stat.S_IRUSR | stat.S_IRGRP | stat.S_IWUSR)

    config.update('node', 'lv_conn_string', '%s+%s://%s@%s/%s' % (driver, transport, username, address, suffix))

    print "NODE: Adding cc1 user to libvirt groups"
    subprocess.call("usermod -a -G libvirt,kvm cc1", shell=True)

    print "NODE: Restarting libvirt"
    subprocess.call('/etc/init.d/libvirt-bin restart', shell=True)

    return 0


def remove():
    print "NODE: Removing package changes from libvirt config"
    qemu = open('/etc/libvirt/qemu.conf', 'r')
    qemu_conf = qemu.readlines()
    qemu.close()

    qemu = open('/etc/libvirt/qemu.conf', 'w')
    for line in qemu_conf:
        if 'user = "root"' in line:
            qemu.write('#user = "root"\n')
        elif 'group = "root"' in line:
            qemu.write('#group = "root"\n')
        elif 'dynamic_ownership = 0' in line:
            qemu.write('#dynamic_ownership = 1')
        else:
            qemu.write(line)
    qemu.close()


def purge():
    remove()
    return 0