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
102
103
104
105
106
107
108
109
|
"""
Serializer code for cobbler
Copyright 2007, Red Hat, Inc
Michael DeHaan <mdehaan@redhat.com>
NOTE: as it stands, the performance of this serializer is not great
nor has it been throughly tested. It is, however, about 4x faster
than the YAML version. It could be optimized further.
This software may be freely redistributed under the terms of the GNU
general public license.
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., 675 Mass Ave, Cambridge, MA 02139, USA.
"""
import distutils.sysconfig
import os
import sys
import glob
import traceback
plib = distutils.sysconfig.get_python_lib()
mod_path="%s/cobbler" % plib
sys.path.insert(0, mod_path)
from rhpl.translate import _, N_, textdomain, utf8
from cexceptions import *
import os
import shelve
def register():
"""
The mandatory cobbler module registration hook.
"""
return True
def serialize(obj):
"""
Save an object to disk. Object must "implement" Serializable.
Will create intermediate paths if it can. Returns True on Success,
False on permission errors.
"""
fd = shelve.open(obj.filename() + ".shelve","c")
# FIXME: this needs to understand deletes
# FIXME: create partial serializer and don't use this
for entry in obj:
fd[entry.name] = entry.to_datastruct()
fd.sync()
return True
def serialize_item(obj, item):
fd = shelve.open(obj.filename() + ".shelve","w")
fd[item.name] = item.to_datastruct()
fd.sync()
return True
# NOTE: not heavily tested
def serialize_item(obj, item):
fd = shelve.open(obj.filename() + ".shelve","w")
del fd[item.name]
fd.sync()
return True
def deserialize(obj,topological=False):
"""
Populate an existing object with the contents of datastruct.
Object must "implement" Serializable. Returns True assuming
files could be read and contained decent YAML. Otherwise returns
False.
"""
filename = obj.filename() + ".shelve"
try:
fd = shelve.open(filename, "r")
except:
if not os.path.exists(filename):
return True
else:
traceback.print_exc()
raise CX(_("Can't access storage file"))
datastruct = []
for (key,value) in fd.iteritems():
datastruct.append(value)
fd.close()
if topological and type(datastruct) == list:
# in order to build the graph links from the flat list, sort by the
# depth of items in the graph. If an object doesn't have a depth, sort it as
# if the depth were 0. It will be assigned a proper depth at serialization
# time. This is a bit cleaner implementation wise than a topological sort,
# though that would make a shiny upgrade.
datastruct.sort(__depth_cmp)
obj.from_datastruct(datastruct)
return True
def __depth_cmp(item1, item2):
if not item1.has_key("depth"):
return 1
if not item2.has_key("depth"):
return -1
return cmp(item1["depth"],item2["depth"])
|