class Register:
def __init__(self):
self.db = DatabaseManager()
def register_unit(self, unit, ws):
unit.ws = ws
unit.state = 'STARTED'
self.db.update(unit)
def update(cls, service_args, id):
db = DatabaseManager()
try:
updated_service = db.get_by_id(Service, id)
except NotFoundException as e:
raise e
updated_service.config = service_args.get('config') or updated_service.config
updated_service.flavor = service_args.get('flavor') or updated_service.flavor
updated_service.image = service_args.get('image') or updated_service.image
updated_service.service_type = service_args.get('service_type') or updated_service.service_type
updated_service.size = service_args.get('size') or updated_service.size
conf = sys_util().get_sys_conf()
checker = FactoryAgent().get_agent(conf['checker'])
checker.check(service=updated_service)
db.update(updated_service)
return updated_service
def update(cls, secgroup_args, id):
db = DatabaseManager()
try:
updated_secgroup = db.get_by_id(SecurityGroup, id)
except NotFoundException as e:
raise e
_sec_rules = secgroup_args.get('rules')
_new_sec_rules = []
for _sec_rule_args in _sec_rules:
_new_sec_rule = Rule(**_sec_rule_args)
_new_sec_rules.append(_new_sec_rule)
updated_secgroup.rules = _new_sec_rules or updated_secgroup.rules
updated_secgroup.name = secgroup_args.get('name') or updated_secgroup.name
check(security_group=updated_secgroup)
db.update(updated_secgroup)
return updated_secgroup
def update(cls, service_args, id):
db = DatabaseManager()
try:
updated_service = db.get_by_id(Service, id)
except NotFoundException as e:
raise e
updated_service.config = service_args.get("config") or updated_service.config
updated_service.flavor = service_args.get("flavor") or updated_service.flavor
updated_service.image = service_args.get("image") or updated_service.image
updated_service.service_type = service_args.get("service_type") or updated_service.service_type
updated_service.size = service_args.get("size") or updated_service.size
conf = sys_util().get_sys_conf()
checker = FactoryAgent().get_agent(conf["checker"])
checker.check(service=updated_service)
db.update(updated_service)
return updated_service
def update(cls, secgroup_args, id):
db = DatabaseManager()
try:
updated_secgroup = db.get_by_id(SecurityGroup, id)
except NotFoundException as e:
raise e
_sec_rules = secgroup_args.get('rules')
_new_sec_rules = []
for _sec_rule_args in _sec_rules:
_new_sec_rule = Rule(**_sec_rule_args)
_new_sec_rules.append(_new_sec_rule)
updated_secgroup.rules = _new_sec_rules or updated_secgroup.rules
updated_secgroup.name = secgroup_args.get(
'name') or updated_secgroup.name
check(security_group=updated_secgroup)
db.update(updated_secgroup)
return updated_secgroup