1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
|
#
# Copyright 2019 Red Hat, Inc.
#
# This work is licensed under the GNU GPLv2 or later.
# See the COPYING file in the top-level directory.
#
import libvirt
def check_libvirt_collision(collision_cb, val):
"""
Run the passed collision function with val as the only argument:
If libvirtError is raised, return False
If no libvirtError raised, return True
"""
check = False
if val is not None:
try:
if collision_cb(val) is not None:
check = True
except libvirt.libvirtError:
pass
return check
def generate_name(base, collision_cb, suffix="",
start_num=1, sep="-", force_num=False):
"""
Generate a new name from the passed base string, verifying it doesn't
collide with the collision callback.
This can be used to generate disk path names from the parent VM or pool
name. Names generated look like 'base-#suffix', ex:
If foobar, and foobar-1.img already exist, and:
base = "foobar"
suffix = ".img"
output = "foobar-2.img"
:param base: The base string to use for the name (e.g. "my-orig-vm-clone")
:param collision_cb: A callback function to check for collision,
receives the generated name as its only arg
:param start_num: The number to start at for generating non colliding names
:param sep: The separator to use between the basename and the
generated number (default is "-")
:param force_num: Force the generated name to always end with a number
"""
base = str(base)
numrange = list(range(start_num, start_num + 100000))
if not force_num:
numrange = [None] + numrange
ret = None
for i in numrange:
tryname = base
if i is not None:
tryname += ("%s%d" % (sep, i))
tryname += suffix
if not collision_cb(tryname):
ret = tryname
break
assert ret
return ret
|
