def init(self, mirrors, mirrors_fallback, package_apis, url2mirrors, url2mirrors_fallback, url2package_apis):
MMAssignAlgo.init(self, mirrors, mirrors_fallback, package_apis, url2mirrors, url2mirrors_fallback, url2package_apis)
self.config = PluginInventoryAAConfig()
self.config.setup(mmcconfdir + '/pulse2/package-server/plugin_terminal_type.ini')
self.database = PluginInventoryAADatabase()
self.database.activate(self.config)
self.populateCache()
self.logger.debug("init done for terminal_type assign algo")
def init(self, mirrors, mirrors_fallback, package_apis, url2mirrors, url2mirrors_fallback, url2package_apis):
MMAssignAlgo.init(self, mirrors, mirrors_fallback, package_apis, url2mirrors, url2mirrors_fallback, url2package_apis)
self.config = PluginInventoryAAConfig()
self.config.setup(mmcconfdir + '/pulse2/package-server/plugin_terminal_type.ini')
self.database = PluginInventoryAADatabase()
self.database.activate(self.config)
self.populateCache()
self.logger.debug("init done for terminal_type assign algo")
class MMUserAssignAlgo(MMAssignAlgo):
name = 'terminal_type'
assign = {}
def init(self, mirrors, mirrors_fallback, package_apis, url2mirrors, url2mirrors_fallback, url2package_apis):
MMAssignAlgo.init(self, mirrors, mirrors_fallback, package_apis, url2mirrors, url2mirrors_fallback, url2package_apis)
self.config = PluginInventoryAAConfig()
self.config.setup(mmcconfdir + '/pulse2/package-server/plugin_terminal_type.ini')
self.database = PluginInventoryAADatabase()
self.database.activate(self.config)
self.populateCache()
self.logger.debug("init done for terminal_type assign algo")
def populateCache(self):
"""
Map machines UUIDs to type
"""
self.logger.info("Populating computer type cache")
self.types = {}
for row in self.database.buildPopulateCacheQuery():
self.types["UUID" + str(row[2])] = row[0].Value
self.logger.info("Populate done (%d computers)" % len(self.types))
def __getMachineType(self, m):
try:
ret = self.types[m['uuid']]
except KeyError:
ret = self.database.getMachineType(m['uuid'])
# Put result in memory cache
self.types[m['uuid']] = ret
return ret
def getMachineMirror(self, m):
if not m['uuid'] in self.assign:
self.assign[m['uuid']] = {}
if not 'getMirror' in self.assign[m['uuid']]:
type = self.__getMachineType(m)
self.assign[m['uuid']]['getMirror'] = []
if type != None:
for u in self.config.type2url[type]['mirror']:
self.assign[m['uuid']]['getMirror'].append(self.url2mirrors[u])
return self.assign[m['uuid']]['getMirror']
def getMachineMirrorFallback(self, m):
if not m['uuid'] in self.assign:
self.assign[m['uuid']] = {}
if not 'getFallbackMirror' in self.assign[m['uuid']]:
type = self.__getMachineType(m)
self.assign[m['uuid']]['getFallbackMirror'] = []
if type != None:
for u in self.config.type2url[type]['mirror']:
self.assign[m['uuid']]['getFallbackMirror'].append(self.url2mirrors_fallback[u])
return self.assign[m['uuid']]['getFallbackMirror']
def getMachinePackageApi(self, m):
if not m['uuid'] in self.assign:
self.assign[m['uuid']] = {}
if not 'getMachinePackageApi' in self.assign[m['uuid']]:
type = self.__getMachineType(m)
self.assign[m['uuid']]['getMachinePackageApi'] = []
if type != None:
for u in self.config.type2url[type]['package_api']:
self.assign[m['uuid']]['getMachinePackageApi'].append(self.url2package_apis[u])
return self.assign[m['uuid']]['getMachinePackageApi']
def getComputersPackageApi(self, machines):
pass
class MMUserAssignAlgo(MMAssignAlgo):
name = 'terminal_type'
assign = {}
def init(self, mirrors, mirrors_fallback, package_apis, url2mirrors, url2mirrors_fallback, url2package_apis):
MMAssignAlgo.init(self, mirrors, mirrors_fallback, package_apis, url2mirrors, url2mirrors_fallback, url2package_apis)
self.config = PluginInventoryAAConfig()
self.config.setup(mmcconfdir + '/pulse2/package-server/plugin_terminal_type.ini')
self.database = PluginInventoryAADatabase()
self.database.activate(self.config)
self.populateCache()
self.logger.debug("init done for terminal_type assign algo")
def populateCache(self):
"""
Map machines UUIDs to type
"""
self.logger.info("Populating computer type cache")
self.types = {}
for row in self.database.buildPopulateCacheQuery():
self.types["UUID" + str(row[2])] = row[0].Value
self.logger.info("Populate done (%d computers)" % len(self.types))
def __getMachineType(self, m):
try:
ret = self.types[m['uuid']]
except KeyError:
ret = self.database.getMachineType(m['uuid'])
# Put result in memory cache
self.types[m['uuid']] = ret
return ret
def getMachineMirror(self, m):
if not self.assign.has_key(m['uuid']):
self.assign[m['uuid']] = {}
if not self.assign[m['uuid']].has_key('getMirror'):
type = self.__getMachineType(m)
self.assign[m['uuid']]['getMirror'] = []
if type != None:
for u in self.config.type2url[type]['mirror']:
self.assign[m['uuid']]['getMirror'].append(self.url2mirrors[u])
return self.assign[m['uuid']]['getMirror']
def getMachineMirrorFallback(self, m):
if not self.assign.has_key(m['uuid']):
self.assign[m['uuid']] = {}
if not self.assign[m['uuid']].has_key('getFallbackMirror'):
type = self.__getMachineType(m)
self.assign[m['uuid']]['getFallbackMirror'] = []
if type != None:
for u in self.config.type2url[type]['mirror']:
self.assign[m['uuid']]['getFallbackMirror'].append(self.url2mirrors_fallback[u])
return self.assign[m['uuid']]['getFallbackMirror']
def getMachinePackageApi(self, m):
if not self.assign.has_key(m['uuid']):
self.assign[m['uuid']] = {}
if not self.assign[m['uuid']].has_key('getMachinePackageApi'):
type = self.__getMachineType(m)
self.assign[m['uuid']]['getMachinePackageApi'] = []
if type != None:
for u in self.config.type2url[type]['package_api']:
self.assign[m['uuid']]['getMachinePackageApi'].append(self.url2package_apis[u])
return self.assign[m['uuid']]['getMachinePackageApi']
def getComputersPackageApi(self, machines):
pass
