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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
|
#!/usr/bin/python -tt
#
# Copyright (c) 2011 Intel, Inc.
#
# This program is free software; you can redistribute it and/or modify it
# under the terms of the GNU General Public License as published by the Free
# Software Foundation; version 2 of the License
#
# This program 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 General Public License
# for more details.
#
# You should have received a copy of the GNU General Public License along
# with this program; if not, write to the Free Software Foundation, Inc., 59
# Temple Place - Suite 330, Boston, MA 02111-1307, USA.
import os
import shutil
from mic import msger
from mic.utils import errors
class _Plugin(object):
class __metaclass__(type):
def __init__(cls, name, bases, attrs):
if not hasattr(cls, 'plugins'):
cls.plugins = {}
elif 'mic_plugin_type' in attrs:
if attrs['mic_plugin_type'] not in cls.plugins:
cls.plugins[attrs['mic_plugin_type']] = {}
elif hasattr(cls, 'mic_plugin_type') and 'name' in attrs:
cls.plugins[cls.mic_plugin_type][attrs['name']] = cls
def show_plugins(cls):
for cls in cls.plugins[cls.mic_plugin_type]:
print cls
def get_plugins(cls):
return cls.plugins
class ImagerPlugin(_Plugin):
mic_plugin_type = "imager"
@classmethod
def check_image_exists(self, destdir, apacking=None,
images=(),
release=None):
# if it's a packing file, reset images
if apacking:
images = [apacking]
# release option will override images
if release is not None:
images = [os.path.basename(destdir.rstrip('/'))]
destdir = os.path.dirname(destdir.rstrip('/'))
for name in images:
if not name:
continue
image = os.path.join(destdir, name)
if not os.path.exists(image):
continue
if msger.ask("Target image/dir: %s already exists, "
"clean up and continue?" % image):
if os.path.isdir(image):
shutil.rmtree(image)
else:
os.unlink(image)
else:
raise errors.Abort("Cancled")
def do_create(self):
pass
def do_chroot(self):
pass
class BackendPlugin(_Plugin):
mic_plugin_type="backend"
# suppress the verbose rpm warnings
if msger.get_loglevel() != 'debug':
import rpm
rpm.setVerbosity(rpm.RPMLOG_ERR)
def addRepository(self):
pass
def get_plugins(typen):
ps = ImagerPlugin.get_plugins()
if typen in ps:
return ps[typen]
else:
return None
__all__ = ['ImagerPlugin', 'BackendPlugin', 'get_plugins']
|
