Esempio n. 1
0
class Register:

    def __init__(self):
        self.db = DatabaseManager()

    def register_unit(self, unit, ws):
        unit.ws = ws
        unit.state = 'STARTED'
        self.db.update(unit)
Esempio n. 2
0
 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
Esempio n. 5
0
 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