# a dict with "capability name" as the key, and the version
# as the value.
import glob
import os
from up2date_client.capabilities import parseCap
try: # python2
import UserDict
except ImportError: # python3
import collections as UserDict
class ClientCapabilities(UserDict.UserDict):
def __init__(self):
UserDict.UserDict.__init__(self)
self.populate()
def populate(self, capsToPopulate=None):
# FIXME: at some point, this will be
# intelligently populated...
localcaps = {
"caneatCheese":{'version':1, 'value': 1}
}
if capsToPopulate:
localcaps = capsToPopulate
self.data = localcaps
def headerFormat(self):
headerList = []
for key in self.data.keys():
headerName = "X-RHN-Client-Capability"
value = "%s(%s)=%s" % (key,
self.data[key]['version'],
self.data[key]['value'])
headerList.append((headerName, value))
return headerList
caps = ClientCapabilities()
def loadLocalCaps(capsDir = "/etc/sysconfig/rhn/clientCaps.d"):
capsFiles = glob.glob("%s/*" % capsDir)
for capsFile in capsFiles:
if os.path.isdir(capsFile):
continue
if not os.access(capsFile, os.R_OK):
continue
fd = open(capsFile, "r")
for line in fd.readlines():
line = line.strip()
if not line or line[0] == "#":
continue
caplist = parseCap(line)
for (cap,data) in caplist:
caps.data[cap] = data
# print(caps.data)
loadLocalCaps()
# register local caps we require.
def registerCap(cap, data):
caps.data[cap] = data
# figure out something pretty here
registerCap("packages.runTransaction", {'version':'1', 'value':'1'})
registerCap("packages.rollBack", {'version':'1', 'value':'1'})
registerCap("packages.verify", {'version':'1', 'value':'1'})
registerCap("packages.extended_profile", {'version':'2', 'value':'1'})
registerCap("reboot.reboot", {'version':'1', 'value':'1'})
registerCap("packages.update", {'version':'2', 'value':'2'})
