def handle_delete(self, request, user, *args, **kwargs):
"""Treat requests DELETE to remove Model.
URL: model/<id_model>/
"""
try:
self.log.info('Remove Model')
# User permission
if not has_perm(user, AdminPermission.BRAND_MANAGEMENT,
AdminPermission.WRITE_OPERATION):
self.log.error(
u'User does not have permission to perform the operation.')
raise UserNotAuthorizedError(None)
id_model = kwargs.get('id_model')
# Valid ID Model
if not is_valid_int_greater_zero_param(id_model):
self.log.error(
u'The id_model parameter is not a valid value: %s.',
id_model)
raise InvalidValueError(None, 'id_model', id_model)
# Find Model by ID to check if it exist
model = Modelo.get_by_pk(id_model)
with distributedlock(LOCK_MODEL % id_model):
try:
if model.equipamento_set.count() > 0:
raise ModeloUsedByEquipamentoError(
None, u'O modelo %s tem equipamento associado.' %
model.id)
# remove Model
model.delete()
except ModeloUsedByEquipamentoError, e:
raise e
except Exception, e:
self.log.error(u'Failed to remove the Model.')
raise EquipamentoError(e, u'Failed to remove the Model.')
def handle_delete(self, request, user, *args, **kwargs):
"""Treat requests DELETE to remove Model.
URL: model/<id_model>/
"""
try:
self.log.info("Remove Model")
# User permission
if not has_perm(user, AdminPermission.BRAND_MANAGEMENT, AdminPermission.WRITE_OPERATION):
self.log.error(
u'User does not have permission to perform the operation.')
raise UserNotAuthorizedError(None)
id_model = kwargs.get('id_model')
# Valid ID Model
if not is_valid_int_greater_zero_param(id_model):
self.log.error(
u'The id_model parameter is not a valid value: %s.', id_model)
raise InvalidValueError(None, 'id_model', id_model)
# Find Model by ID to check if it exist
model = Modelo.get_by_pk(id_model)
with distributedlock(LOCK_MODEL % id_model):
try:
if model.equipamento_set.count() > 0:
raise ModeloUsedByEquipamentoError(
None, u"O modelo %s tem equipamento associado." % model.id)
# remove Model
model.delete()
except ModeloUsedByEquipamentoError, e:
raise e
except Exception, e:
self.log.error(u'Failed to remove the Model.')
raise EquipamentoError(e, u'Failed to remove the Model.')
def handle_get(self, request, user, *args, **kwargs):
"""Treat requests GET to list all the Model by Brand.
URL: model/brand/<id_brand>/
"""
try:
self.log.info("GET to list all the Model by Brand")
# User permission
if not has_perm(user, AdminPermission.BRAND_MANAGEMENT, AdminPermission.READ_OPERATION):
self.log.error(
u'User does not have permission to perform the operation.')
raise UserNotAuthorizedError(None)
id_brand = kwargs.get('id_brand')
# Valid ID Brand
if not is_valid_int_greater_zero_param(id_brand):
self.log.error(
u'The id_brand parameter is not a valid value: %s.', id_brand)
raise InvalidValueError(None, 'id_groupl3', id_brand)
# Find Brand by ID to check if it exist
Marca.get_by_pk(id_brand)
model_list = []
for model in Modelo.get_by_brand(id_brand):
model_map = dict()
model_map['id'] = model.id
model_map['nome'] = model.nome
model_map['id_marca'] = model.marca.id
model_map['nome_marca'] = model.marca.nome
model_list.append(model_map)
return self.response(dumps_networkapi({'model': model_list}))
except InvalidValueError, e:
return self.response_error(269, e.param, e.value)
def handle_get(self, request, user, *args, **kwargs):
"""Treat requests GET to list all the Model by Brand.
URL: model/brand/<id_brand>/
"""
try:
self.log.info('GET to list all the Model by Brand')
# User permission
if not has_perm(user, AdminPermission.BRAND_MANAGEMENT, AdminPermission.READ_OPERATION):
self.log.error(
u'User does not have permission to perform the operation.')
raise UserNotAuthorizedError(None)
id_brand = kwargs.get('id_brand')
# Valid ID Brand
if not is_valid_int_greater_zero_param(id_brand):
self.log.error(
u'The id_brand parameter is not a valid value: %s.', id_brand)
raise InvalidValueError(None, 'id_groupl3', id_brand)
# Find Brand by ID to check if it exist
Marca.get_by_pk(id_brand)
model_list = []
for model in Modelo.get_by_brand(id_brand):
model_map = dict()
model_map['id'] = model.id
model_map['nome'] = model.nome
model_map['id_marca'] = model.marca.id
model_map['nome_marca'] = model.marca.nome
model_list.append(model_map)
return self.response(dumps_networkapi({'model': model_list}))
except InvalidValueError, e:
return self.response_error(269, e.param, e.value)
def handle_post(self, request, user, *args, **kwargs):
"""Treat requests POST to add Model.
URL: model/
"""
try:
self.log.info("Add Model")
# User permission
if not has_perm(user, AdminPermission.BRAND_MANAGEMENT, AdminPermission.WRITE_OPERATION):
self.log.error(
u'User does not have permission to perform the operation.')
raise UserNotAuthorizedError(None)
# Load XML data
xml_map, attrs_map = loads(request.raw_post_data)
# XML data format
networkapi_map = xml_map.get('networkapi')
if networkapi_map is None:
return self.response_error(3, u'There is no value to the networkapi tag of XML request.')
model_map = networkapi_map.get('model')
if model_map is None:
return self.response_error(3, u'There is no value to the model tag of XML request.')
# Get XML data
name = model_map.get('name')
id_brand = model_map.get('id_brand')
# Valid name
if not is_valid_string_minsize(name, 3) or not is_valid_string_maxsize(name, 100):
self.log.error(u'Parameter name is invalid. Value: %s', name)
raise InvalidValueError(None, 'name', name)
# Valid ID Brand
if not is_valid_int_greater_zero_param(id_brand):
self.log.error(
u'The id_brand parameter is not a valid value: %s.', id_brand)
raise InvalidValueError(None, 'id_brand', id_brand)
# Find Brand by ID to check if it exist
brand = Marca.get_by_pk(id_brand)
try:
Modelo.get_by_name_brand(name, id_brand)
raise MarcaModeloNameDuplicatedError(
None, u'Já existe um modelo com o nome %s com marca %s.' % (name, brand.nome))
except ModeloNotFoundError:
pass
model = Modelo()
# set variables
model.nome = name
model.marca = brand
try:
# save Model
model.save()
except Exception, e:
self.log.error(u'Failed to save the Model.')
raise EquipamentoError(e, u'Failed to save the Model.')
model_map = dict()
model_map['model'] = model_to_dict(
model, exclude=["nome", "marca"])
return self.response(dumps_networkapi(model_map))
def handle_put(self, request, user, *args, **kwargs):
"""Treat requests PUT to edit Model.
URL: model/<id_model>/
"""
try:
self.log.info('Edit Model')
# User permission
if not has_perm(user, AdminPermission.BRAND_MANAGEMENT,
AdminPermission.WRITE_OPERATION):
self.log.error(
u'User does not have permission to perform the operation.')
raise UserNotAuthorizedError(None)
id_model = kwargs.get('id_model')
# Load XML data
xml_map, attrs_map = loads(request.raw_post_data)
# XML data format
networkapi_map = xml_map.get('networkapi')
if networkapi_map is None:
return self.response_error(
3,
u'There is no value to the networkapi tag of XML request.'
)
model_map = networkapi_map.get('model')
if model_map is None:
return self.response_error(
3, u'There is no value to the model tag of XML request.')
# Get XML data
name = model_map.get('name')
id_brand = model_map.get('id_brand')
# Valid ID Model
if not is_valid_int_greater_zero_param(id_model):
self.log.error(
u'The id_model parameter is not a valid value: %s.',
id_model)
raise InvalidValueError(None, 'id_model', id_model)
# Valid name
if not is_valid_string_minsize(
name, 3) or not is_valid_string_maxsize(name, 100):
self.log.error(u'Parameter name is invalid. Value: %s', name)
raise InvalidValueError(None, 'name', name)
# Valid ID Brand
if not is_valid_int_greater_zero_param(id_brand):
self.log.error(
u'The id_brand parameter is not a valid value: %s.',
id_brand)
raise InvalidValueError(None, 'id_brand', id_brand)
# Find Brand by ID to check if it exist
brand = Marca.get_by_pk(id_brand)
# Find Model by ID to check if it exist
model = Modelo.get_by_pk(id_model)
with distributedlock(LOCK_MODEL % id_model):
try:
if not (model.nome.lower() == name.lower()
and model.marca.id == id_brand):
Modelo.get_by_name_brand(name, id_brand)
raise MarcaModeloNameDuplicatedError(
None,
u'Já existe um modelo com o nome %s com marca %s.'
% (name, brand.nome))
except ModeloNotFoundError:
pass
# set variables
model.nome = name
model.marca = brand
try:
# update Model
model.save()
except Exception, e:
self.log.error(u'Failed to update the Model.')
raise EquipamentoError(e, u'Failed to update the Model.')
return self.response(dumps_networkapi({}))
except InvalidValueError, e:
return self.response_error(269, e.param, e.value)
def handle_put(self, request, user, *args, **kwargs):
"""Treat requests PUT to edit Model.
URL: model/<id_model>/
"""
try:
self.log.info("Edit Model")
# User permission
if not has_perm(user, AdminPermission.BRAND_MANAGEMENT, AdminPermission.WRITE_OPERATION):
self.log.error(
u'User does not have permission to perform the operation.')
raise UserNotAuthorizedError(None)
id_model = kwargs.get('id_model')
# Load XML data
xml_map, attrs_map = loads(request.raw_post_data)
# XML data format
networkapi_map = xml_map.get('networkapi')
if networkapi_map is None:
return self.response_error(3, u'There is no value to the networkapi tag of XML request.')
model_map = networkapi_map.get('model')
if model_map is None:
return self.response_error(3, u'There is no value to the model tag of XML request.')
# Get XML data
name = model_map.get('name')
id_brand = model_map.get('id_brand')
# Valid ID Model
if not is_valid_int_greater_zero_param(id_model):
self.log.error(
u'The id_model parameter is not a valid value: %s.', id_model)
raise InvalidValueError(None, 'id_model', id_model)
# Valid name
if not is_valid_string_minsize(name, 3) or not is_valid_string_maxsize(name, 100):
self.log.error(u'Parameter name is invalid. Value: %s', name)
raise InvalidValueError(None, 'name', name)
# Valid ID Brand
if not is_valid_int_greater_zero_param(id_brand):
self.log.error(
u'The id_brand parameter is not a valid value: %s.', id_brand)
raise InvalidValueError(None, 'id_brand', id_brand)
# Find Brand by ID to check if it exist
brand = Marca.get_by_pk(id_brand)
# Find Model by ID to check if it exist
model = Modelo.get_by_pk(id_model)
with distributedlock(LOCK_MODEL % id_model):
try:
if not (model.nome.lower() == name.lower() and model.marca.id == id_brand):
Modelo.get_by_name_brand(name, id_brand)
raise MarcaModeloNameDuplicatedError(
None, u'Já existe um modelo com o nome %s com marca %s.' % (name, brand.nome))
except ModeloNotFoundError:
pass
# set variables
model.nome = name
model.marca = brand
try:
# update Model
model.save()
except Exception, e:
self.log.error(u'Failed to update the Model.')
raise EquipamentoError(e, u'Failed to update the Model.')
return self.response(dumps_networkapi({}))
except InvalidValueError, e:
return self.response_error(269, e.param, e.value)
def handle_put(self, request, user, *args, **kwargs):
"""Treat requests PUT to edit Script.
URL: script/<id_script>/
"""
try:
self.log.info("Edit Script")
# User permission
if not has_perm(user, AdminPermission.SCRIPT_MANAGEMENT, AdminPermission.WRITE_OPERATION):
self.log.error(u"User does not have permission to perform the operation.")
raise UserNotAuthorizedError(None)
id_script = kwargs.get("id_script")
# Load XML data
xml_map, attrs_map = loads(request.raw_post_data)
# XML data format
networkapi_map = xml_map.get("networkapi")
if networkapi_map is None:
return self.response_error(3, u"There is no value to the networkapi tag of XML request.")
script_map = networkapi_map.get("script")
if script_map is None:
return self.response_error(3, u"There is no value to the script tag of XML request.")
# Get XML data
script = script_map.get("script")
id_script_type = script_map.get("id_script_type")
models = script_map.get("model")
description = script_map.get("description")
# Valid ID Script
if not is_valid_int_greater_zero_param(id_script):
self.log.error(u"The id_script parameter is not a valid value: %s.", id_script)
raise InvalidValueError(None, "id_script", id_script)
# Valid Script
if not is_valid_string_minsize(script, 3) or not is_valid_string_maxsize(script, 40):
self.log.error(u"Parameter script is invalid. Value: %s", script)
raise InvalidValueError(None, "script", script)
# Valid ID Script Type
if not is_valid_int_greater_zero_param(id_script_type):
self.log.error(u"The id_script_type parameter is not a valid value: %s.", id_script_type)
raise InvalidValueError(None, "id_script_type", id_script_type)
# Valid description
if not is_valid_string_minsize(description, 3) or not is_valid_string_maxsize(description, 100):
self.log.error(u"Parameter description is invalid. Value: %s", description)
raise InvalidValueError(None, "description", description)
# Find Script by ID to check if it exist
scr = Roteiro.get_by_pk(id_script)
# Find Script Type by ID to check if it exist
script_type = TipoRoteiro.get_by_pk(id_script_type)
models_old = []
scr_models = ModeloRoteiro.objects.all().filter(roteiro__id=scr.id)
for i in scr_models:
models_old.append(int(i.modelo.id))
if models is not None and type(models) is not list:
var = int(models)
models = []
models.append(var)
else:
models = [int(x) for x in models]
desassociar = set(models_old) - set(models)
for i in desassociar:
scr_model = ModeloRoteiro()
scr_model.remover(user, int(i), int(scr.id))
associar = set(models) - set(models_old)
for i in associar:
scr_models = ModeloRoteiro()
scr_models.roteiro = scr
scr_models.modelo = Modelo.get_by_pk(i)
scr_models.create(user)
# verificar se há equipamento daquele modelo que não está associado a um roteiro
for ids in models:
equipamentos = Equipamento.objects.filter(modelo__id=int(ids))
for equip in equipamentos:
try:
equip_roteiro = EquipamentoRoteiro.objects.filter(
equipamento__id=equip.id, roteiro__tipo_roteiro__id=scr.tipo_roteiro.id
).uniqueResult()
equip_roteiro.id
except:
equip_rot = EquipamentoRoteiro()
equip_rot.equipamento = equip
equip_rot.roteiro = scr
equip_rot.create(user)
pass
with distributedlock(LOCK_SCRIPT % id_script):
try:
if not scr.roteiro.lower() == script.lower() and not scr.tipo_roteiro.id == id_script_type:
Roteiro.get_by_name_script(script, id_script_type)
raise RoteiroNameDuplicatedError(
None,
u"Já existe um roteiro com o nome %s com tipo de roteiro %s." % (script, script_type.tipo),
)
except RoteiroNotFoundError:
pass
# set variables
scr.roteiro = script
scr.tipo_roteiro = script_type
scr.descricao = description
try:
# update Script
scr.save()
except Exception, e:
self.log.error(u"Failed to update the Script.")
raise RoteiroError(e, u"Failed to update the Script.")
return self.response(dumps_networkapi({}))
except InvalidValueError, e:
return self.response_error(269, e.param, e.value)
def handle_post(self, request, user, *args, **kwargs):
"""Trata uma requisicao POST para editar um equipamento.
URL: equipmento/edit/
"""
try:
# Load XML data
xml_map, attrs_map = loads(request.raw_post_data)
# XML data format
networkapi_map = xml_map.get('networkapi')
if networkapi_map is None:
msg = u'There is no value to the networkapi tag of XML request.'
self.log.error(msg)
return self.response_error(3, msg)
equip_map = networkapi_map.get('equipamento')
if equip_map is None:
msg = u'There is no value to the ip tag of XML request.'
self.log.error(msg)
return self.response_error(3, msg)
# Get XML data
equip_id = equip_map.get('id_equip')
id_modelo = equip_map.get('id_modelo')
nome = equip_map.get('nome')
id_tipo_equipamento = equip_map.get('id_tipo_equipamento')
maintenance = equip_map.get('maintenance')
# Valid equip_id
if not is_valid_int_greater_zero_param(equip_id):
self.log.error(
u'Parameter equip_id is invalid. Value: %s.', equip_id)
raise InvalidValueError(None, 'equip_id', equip_id)
# Valid id_modelo
if not is_valid_int_greater_zero_param(id_modelo):
self.log.error(
u'Parameter id_modelo is invalid. Value: %s.', id_modelo)
raise InvalidValueError(None, 'id_modelo', id_modelo)
# Valid id_tipo_equipamento
if not is_valid_int_greater_zero_param(id_tipo_equipamento):
self.log.error(
u'Parameter id_tipo_equipamento is invalid. Value: %s.', id_tipo_equipamento)
raise InvalidValueError(
None, 'id_tipo_equipamento', id_tipo_equipamento)
# Valid nome
if not is_valid_string_minsize(nome, 3) or not is_valid_string_maxsize(nome, 80) or not is_valid_regex(nome, "^[A-Z0-9-_]+$"):
self.log.error(u'Parameter nome is invalid. Value: %s', nome)
raise InvalidValueError(None, 'nome', nome)
# Business Rules
# New equipment
equip = Equipamento()
equip = equip.get_by_pk(equip_id)
#maintenance is a new feature. Check existing value if not defined in request
#Old calls does not send this field
if maintenance is None:
maintenance = equip.maintenance
if not is_valid_boolean_param(maintenance):
self.log.error(u'The maintenance parameter is not a valid value: %s.', maintenance)
raise InvalidValueError(None, 'maintenance', maintenance)
# User permission
if not has_perm(user, AdminPermission.EQUIPMENT_MANAGEMENT, AdminPermission.WRITE_OPERATION, None, equip_id, AdminPermission.EQUIP_WRITE_OPERATION):
raise UserNotAuthorizedError(
None, u'User does not have permission to perform the operation.')
with distributedlock(LOCK_EQUIPMENT % equip_id):
tipo_equip = TipoEquipamento.get_by_pk(id_tipo_equipamento)
if equip.tipo_equipamento != tipo_equip:
# Environments with filters using current equip type, with
# equipment associated
envs = [eq_env.ambiente.id for eq_env in equip.equipamentoambiente_set.filter(
ambiente__filter__filterequiptype__equiptype=equip.tipo_equipamento)]
# Filters case 1 and 2
filters_ok = True
# Networks in environments with same ip range
nets_same_range = NetworkIPv4.objects.filter(vlan__ambiente__in=envs).values(
'oct1', 'oct2', 'oct3', 'oct4', 'block').annotate(count=Count('id')).filter(count__gt=1)
if len(nets_same_range) > 0:
for net_gp in nets_same_range:
nets_current_range = NetworkIPv4.objects.filter(vlan__ambiente__in=envs, oct1=net_gp[
'oct1'], oct2=net_gp['oct2'], oct3=net_gp['oct3'], oct4=net_gp['oct4'], block=net_gp['block'])
filters_of_envs = [
net.vlan.ambiente.filter.id for net in nets_current_range]
for fil_ in filters_of_envs:
if TipoEquipamento.objects.filter(id=id_tipo_equipamento, filterequiptype__filter=fil_).count() == 0:
filters_ok = False
break
if not filters_ok:
raise EquipTypeCantBeChangedError(
None, 'O tipo de equipamento não pode ser modificado pois existe um filtro em uso que não possui o novo tipo de equipamento informado.')
# Networks ipv6 in environments with same ipv6 range
nets_v6_same_range = NetworkIPv6.objects.filter(vlan__ambiente__in=envs).values(
'block1', 'block2', 'block3', 'block4', 'block5', 'block6', 'block7', 'block8', 'block').annotate(count=Count('id')).filter(count__gt=1)
if len(nets_v6_same_range) > 0:
for net_gp in nets_v6_same_range:
nets_current_range = NetworkIPv6.objects.filter(vlan__ambiente__in=envs, block1=net_gp['block1'], block2=net_gp['block2'], block3=net_gp[
'block3'], block4=net_gp['block4'], block5=net_gp['block5'], block6=net_gp['block6'], block7=net_gp['block7'], block8=net_gp['block8'], block=net_gp['block'])
filters_of_envs = [
net.vlan.ambiente.filter.id for net in nets_current_range]
for fil_ in filters_of_envs:
if TipoEquipamento.objects.filter(id=id_tipo_equipamento, filterequiptype__filter=fil_).count() == 0:
filters_ok = False
break
if not filters_ok:
raise EquipTypeCantBeChangedError(
None, 'O tipo de equipamento não pode ser modificado pois existe um filtro em uso que não possui o novo tipo de equipamento informado.')
# Filters case 1 and 2 end
# Filter case 3
# Get vlans with same number
vlans_same_number = Vlan.objects.filter(ambiente__in=envs).values(
'num_vlan').annotate(count=Count('id')).filter(count__gt=1)
if len(vlans_same_number) > 0:
for vlan_gp in vlans_same_number:
vlans_current_number = Vlan.objects.filter(
ambiente__in=envs, num_vlan=vlan_gp['num_vlan'])
filters_of_envs = [
vlan.ambiente.filter.id for vlan in vlans_current_number]
for fil_ in filters_of_envs:
if TipoEquipamento.objects.filter(id=id_tipo_equipamento, filterequiptype__filter=fil_).count() == 0:
filters_ok = False
break
if not filters_ok:
raise EquipTypeCantBeChangedError(
None, 'O tipo de equipamento não pode ser modificado pois existe um filtro em uso que não possui o novo tipo de equipamento informado.')
# Filter case 3 end
# Test all vip requests if equip.tipo_equipamento is
# balancing
if equip.tipo_equipamento == TipoEquipamento.get_tipo_balanceador():
vips = RequisicaoVips.objects.all()
vip_ips = []
vip_ipsv6 = []
for vip in vips:
if vip.vip_criado:
if vip.ip is not None:
if vip.ip.ipequipamento_set.filter(equipamento=equip.id).count() > 0:
raise EquipTypeCantBeChangedError(
None, 'O tipo de equipamento não pode ser modificado pois este equipamento é o balanceador associado com o vip criado %s.' % vip.id)
if vip.ipv6 is not None:
if vip.ipv6.ipv6equipament_set.filter(equipamento=equip.id).count() > 0:
raise EquipTypeCantBeChangedError(
None, 'O tipo de equipamento não pode ser modificado pois este equipamento é o balanceador associado com o vip criado %s.' % vip.id)
else:
if vip.ip is not None:
vip_ips.append(vip.ip.id)
if vip.ipv6 is not None:
vip_ipsv6.append(vip.ipv6.id)
nets_using_balancer_in_vips_ = [
ip_.networkipv4 for ip_ in Ip.objects.filter(id__in=vip_ips)]
nets_using_balancer_in_vips = [ip_.networkipv4 for ip_ in Ip.objects.filter(
networkipv4__in=nets_using_balancer_in_vips_, ipequipamento__equipamento=equip.id)]
nets_v6_using_balancer_in_vips_ = [
ip_.networkipv6 for ip_ in Ipv6.objects.filter(id__in=vip_ipsv6)]
nets_v6_using_balancer_in_vips = [ip_.networkipv6 for ip_ in Ipv6.objects.filter(
networkipv6__in=nets_v6_using_balancer_in_vips_, ipv6equipament__equipamento=equip.id)]
for net in nets_using_balancer_in_vips:
net_str = str(net.oct1) + '.' + str(net.oct2) + '.' + \
str(net.oct3) + '.' + str(net.oct4) + \
'/' + str(net.block)
if IpEquipamento.objects.filter(ip__networkipv4=net, equipamento__tipo_equipamento=TipoEquipamento.get_tipo_balanceador()).exclude(equipamento=equip).count() == 0:
raise EquipTypeCantBeChangedError(
None, 'O tipo de equipamento não pode ser modificado pois este equipamento é o único balanceador disponível na rede %s da vlan %s.' % (net_str, net.vlan.nome))
for net in nets_v6_using_balancer_in_vips:
net_str = str(net.block1) + ':' + str(net.block2) + ':' + str(net.block3) + ':' + str(net.block4) + ':' + str(
net.block5) + ':' + str(net.block6) + ':' + str(net.block7) + ':' + str(net.block8) + '/' + str(net.block)
if Ipv6Equipament.objects.filter(ip__networkipv6=net, equipamento__tipo_equipamento=TipoEquipamento.get_tipo_balanceador()).exclude(equipamento=equip).count() == 0:
raise EquipTypeCantBeChangedError(
None, 'O tipo de equipamento não pode ser modificado pois este equipamento é o único balanceador disponível na rede %s da vlan %s.' % (net_str, net.vlan.nome))
ip_equipamento_list = IpEquipamento.objects.filter(
equipamento=equip_id)
ip6_equipamento_list = Ipv6Equipament.objects.filter(
equipamento=equip_id)
# Delete vlan's cache
key_list = []
for eq in ip_equipamento_list:
vlan = eq.ip.networkipv4.vlan
vlan_id = vlan.id
key_list.append(vlan_id)
for eq in ip6_equipamento_list:
vlan = eq.ip.networkipv6.vlan
vlan_id = vlan.id
key_list.append(vlan_id)
destroy_cache_function(key_list)
# Delete equipment's cache
destroy_cache_function([equip_id], True)
modelo = Modelo.get_by_pk(id_modelo)
equip.edit(user, nome, tipo_equip, modelo)
return self.response(dumps_networkapi({}))
except EquipTypeCantBeChangedError, e:
return self.response_error(150, e.message)
def handle_post(self, request, user, *args, **kwargs):
"""Treat requests POST to add Script.
URL: script/
"""
try:
self.log.info("Add Script")
# User permission
if not has_perm(user, AdminPermission.SCRIPT_MANAGEMENT,
AdminPermission.WRITE_OPERATION):
self.log.error(
u'User does not have permission to perform the operation.')
raise UserNotAuthorizedError(None)
# Load XML data
xml_map, attrs_map = loads(request.raw_post_data)
# XML data format
networkapi_map = xml_map.get('networkapi')
if networkapi_map is None:
return self.response_error(
3,
u'There is no value to the networkapi tag of XML request.'
)
script_map = networkapi_map.get('script')
if script_map is None:
return self.response_error(
3, u'There is no value to the script tag of XML request.')
# Get XML data
script = script_map.get('script')
id_script_type = script_map.get('id_script_type')
model = script_map.get('model')
description = script_map.get('description')
# Valid Script
if not is_valid_string_minsize(
script, 3) or not is_valid_string_maxsize(script, 40):
self.log.error(u'Parameter script is invalid. Value: %s',
script)
raise InvalidValueError(None, 'script', script)
# Valid ID Script Type
if not is_valid_int_greater_zero_param(id_script_type):
self.log.error(
u'The id_script_type parameter is not a valid value: %s.',
id_script_type)
raise InvalidValueError(None, 'id_script_type', id_script_type)
# Valid description
if not is_valid_string_minsize(description,
3) or not is_valid_string_maxsize(
description, 100):
self.log.error(u'Parameter description is invalid. Value: %s',
description)
raise InvalidValueError(None, 'description', description)
# Find Script Type by ID to check if it exist
script_type = TipoRoteiro.get_by_pk(id_script_type)
try:
Roteiro.get_by_name_script(script, id_script_type)
raise RoteiroNameDuplicatedError(
None,
u'Já existe um roteiro com o nome %s com tipo de roteiro %s.'
% (script, script_type.tipo))
except RoteiroNotFoundError:
pass
scr = Roteiro()
# set variables
scr.roteiro = script
scr.tipo_roteiro = script_type
scr.descricao = description
modelo_list = []
try:
# save Script
scr.save()
except Exception, e:
self.log.error(u'Failed to save the Script.')
raise RoteiroError(e, u'Failed to save the Script.')
#associar o modelo ao roteiro
try:
if type(model) is unicode:
item = model
model = []
model.append(item)
for ids in model:
modelos = ModeloRoteiro()
modelos.roteiro = scr
modelo = Modelo().get_by_pk(int(ids))
modelos.modelo = modelo
modelos.create(user)
modelo_list.append(modelos.modelo)
except Exception, e:
raise RoteiroError(e, u"Failed to save modelo_roteiro.")
def handle_put(self, request, user, *args, **kwargs):
"""Treat requests PUT to edit Script.
URL: script/<id_script>/
"""
try:
self.log.info("Edit Script")
# User permission
if not has_perm(user, AdminPermission.SCRIPT_MANAGEMENT,
AdminPermission.WRITE_OPERATION):
self.log.error(
u'User does not have permission to perform the operation.')
raise UserNotAuthorizedError(None)
id_script = kwargs.get('id_script')
# Load XML data
xml_map, attrs_map = loads(request.raw_post_data)
# XML data format
networkapi_map = xml_map.get('networkapi')
if networkapi_map is None:
return self.response_error(
3,
u'There is no value to the networkapi tag of XML request.'
)
script_map = networkapi_map.get('script')
if script_map is None:
return self.response_error(
3, u'There is no value to the script tag of XML request.')
# Get XML data
script = script_map.get('script')
id_script_type = script_map.get('id_script_type')
models = script_map.get('model')
description = script_map.get('description')
# Valid ID Script
if not is_valid_int_greater_zero_param(id_script):
self.log.error(
u'The id_script parameter is not a valid value: %s.',
id_script)
raise InvalidValueError(None, 'id_script', id_script)
# Valid Script
if not is_valid_string_minsize(
script, 3) or not is_valid_string_maxsize(script, 40):
self.log.error(u'Parameter script is invalid. Value: %s',
script)
raise InvalidValueError(None, 'script', script)
# Valid ID Script Type
if not is_valid_int_greater_zero_param(id_script_type):
self.log.error(
u'The id_script_type parameter is not a valid value: %s.',
id_script_type)
raise InvalidValueError(None, 'id_script_type', id_script_type)
# Valid description
if not is_valid_string_minsize(description,
3) or not is_valid_string_maxsize(
description, 100):
self.log.error(u'Parameter description is invalid. Value: %s',
description)
raise InvalidValueError(None, 'description', description)
# Find Script by ID to check if it exist
scr = Roteiro.get_by_pk(id_script)
# Find Script Type by ID to check if it exist
script_type = TipoRoteiro.get_by_pk(id_script_type)
models_old = []
scr_models = ModeloRoteiro.objects.all().filter(roteiro__id=scr.id)
for i in scr_models:
models_old.append(int(i.modelo.id))
if models is not None and type(models) is not list:
var = int(models)
models = []
models.append(var)
else:
models = [int(x) for x in models]
desassociar = set(models_old) - set(models)
for i in desassociar:
scr_model = ModeloRoteiro()
scr_model.remover(user, int(i), int(scr.id))
associar = set(models) - set(models_old)
for i in associar:
scr_models = ModeloRoteiro()
scr_models.roteiro = scr
scr_models.modelo = Modelo.get_by_pk(i)
scr_models.create(user)
#verificar se há equipamento daquele modelo que não está associado a um roteiro
for ids in models:
equipamentos = Equipamento.objects.filter(modelo__id=int(ids))
for equip in equipamentos:
try:
equip_roteiro = EquipamentoRoteiro.objects.filter(
equipamento__id=equip.id,
roteiro__tipo_roteiro__id=scr.tipo_roteiro.id
).uniqueResult()
equip_roteiro.id
except:
equip_rot = EquipamentoRoteiro()
equip_rot.equipamento = equip
equip_rot.roteiro = scr
equip_rot.create(user)
pass
with distributedlock(LOCK_SCRIPT % id_script):
try:
if not scr.roteiro.lower() == script.lower(
) and not scr.tipo_roteiro.id == id_script_type:
Roteiro.get_by_name_script(script, id_script_type)
raise RoteiroNameDuplicatedError(
None,
u'Já existe um roteiro com o nome %s com tipo de roteiro %s.'
% (script, script_type.tipo))
except RoteiroNotFoundError:
pass
# set variables
scr.roteiro = script
scr.tipo_roteiro = script_type
scr.descricao = description
try:
# update Script
scr.save()
except Exception, e:
self.log.error(u'Failed to update the Script.')
raise RoteiroError(e, u'Failed to update the Script.')
return self.response(dumps_networkapi({}))
except InvalidValueError, e:
return self.response_error(269, e.param, e.value)
def handle_post(self, request, user, *args, **kwargs):
"""Trata uma requisicao POST para editar um equipamento.
URL: equipmento/edit/
"""
try:
# Load XML data
xml_map, attrs_map = loads(request.raw_post_data)
# XML data format
networkapi_map = xml_map.get('networkapi')
if networkapi_map is None:
msg = u'There is no value to the networkapi tag of XML request.'
self.log.error(msg)
return self.response_error(3, msg)
equip_map = networkapi_map.get('equipamento')
if equip_map is None:
msg = u'There is no value to the ip tag of XML request.'
self.log.error(msg)
return self.response_error(3, msg)
# Get XML data
equip_id = equip_map.get('id_equip')
id_modelo = equip_map.get('id_modelo')
nome = equip_map.get('nome')
id_tipo_equipamento = equip_map.get('id_tipo_equipamento')
maintenance = equip_map.get('maintenance')
# Valid equip_id
if not is_valid_int_greater_zero_param(equip_id):
self.log.error(u'Parameter equip_id is invalid. Value: %s.',
equip_id)
raise InvalidValueError(None, 'equip_id', equip_id)
# Valid id_modelo
if not is_valid_int_greater_zero_param(id_modelo):
self.log.error(u'Parameter id_modelo is invalid. Value: %s.',
id_modelo)
raise InvalidValueError(None, 'id_modelo', id_modelo)
# Valid id_tipo_equipamento
if not is_valid_int_greater_zero_param(id_tipo_equipamento):
self.log.error(
u'Parameter id_tipo_equipamento is invalid. Value: %s.',
id_tipo_equipamento)
raise InvalidValueError(None, 'id_tipo_equipamento',
id_tipo_equipamento)
# Valid nome
if not is_valid_string_minsize(
nome, 3) or not is_valid_string_maxsize(
nome, 80) or not is_valid_regex(nome, '^[A-Z0-9-_]+$'):
self.log.error(u'Parameter nome is invalid. Value: %s', nome)
raise InvalidValueError(None, 'nome', nome)
# Business Rules
# New equipment
equip = Equipamento()
equip = equip.get_by_pk(equip_id)
# maintenance is a new feature. Check existing value if not defined in request
# Old calls does not send this field
if maintenance is None:
maintenance = equip.maintenance
if not is_valid_boolean_param(maintenance):
self.log.error(
u'The maintenance parameter is not a valid value: %s.',
maintenance)
raise InvalidValueError(None, 'maintenance', maintenance)
if maintenance in ['1', 'True', True]:
maintenance = True
else:
maintenance = False
# User permission
if not has_perm(user, AdminPermission.EQUIPMENT_MANAGEMENT,
AdminPermission.WRITE_OPERATION, None, equip_id,
AdminPermission.EQUIP_WRITE_OPERATION):
raise UserNotAuthorizedError(
None,
u'User does not have permission to perform the operation.')
with distributedlock(LOCK_EQUIPMENT % equip_id):
tipo_equip = TipoEquipamento.get_by_pk(id_tipo_equipamento)
if equip.tipo_equipamento != tipo_equip:
# Environments with filters using current equip type, with
# equipment associated
envs = [
eq_env.ambiente.id
for eq_env in equip.equipamentoambiente_set.filter(
ambiente__filter__filterequiptype__equiptype=equip.
tipo_equipamento)
]
# Filters case 1 and 2
filters_ok = True
# Networks in environments with same ip range
nets_same_range = NetworkIPv4.objects.filter(
vlan__ambiente__in=envs).values(
'oct1', 'oct2', 'oct3', 'oct4', 'block').annotate(
count=Count('id')).filter(count__gt=1)
if len(nets_same_range) > 0:
for net_gp in nets_same_range:
nets_current_range = NetworkIPv4.objects.filter(
vlan__ambiente__in=envs,
oct1=net_gp['oct1'],
oct2=net_gp['oct2'],
oct3=net_gp['oct3'],
oct4=net_gp['oct4'],
block=net_gp['block'])
filters_of_envs = [
net.vlan.ambiente.filter.id
for net in nets_current_range
]
for fil_ in filters_of_envs:
if TipoEquipamento.objects.filter(
id=id_tipo_equipamento,
filterequiptype__filter=fil_).count(
) == 0:
filters_ok = False
break
if not filters_ok:
raise EquipTypeCantBeChangedError(
None,
'O tipo de equipamento não pode ser modificado pois existe um filtro em uso que não possui o novo tipo de equipamento informado.'
)
# Networks ipv6 in environments with same ipv6 range
nets_v6_same_range = NetworkIPv6.objects.filter(
vlan__ambiente__in=envs).values(
'block1', 'block2', 'block3', 'block4', 'block5',
'block6', 'block7', 'block8', 'block').annotate(
count=Count('id')).filter(count__gt=1)
if len(nets_v6_same_range) > 0:
for net_gp in nets_v6_same_range:
nets_current_range = NetworkIPv6.objects.filter(
vlan__ambiente__in=envs,
block1=net_gp['block1'],
block2=net_gp['block2'],
block3=net_gp['block3'],
block4=net_gp['block4'],
block5=net_gp['block5'],
block6=net_gp['block6'],
block7=net_gp['block7'],
block8=net_gp['block8'],
block=net_gp['block'])
filters_of_envs = [
net.vlan.ambiente.filter.id
for net in nets_current_range
]
for fil_ in filters_of_envs:
if TipoEquipamento.objects.filter(
id=id_tipo_equipamento,
filterequiptype__filter=fil_).count(
) == 0:
filters_ok = False
break
if not filters_ok:
raise EquipTypeCantBeChangedError(
None,
'O tipo de equipamento não pode ser modificado pois existe um filtro em uso que não possui o novo tipo de equipamento informado.'
)
# Filters case 1 and 2 end
# Filter case 3
# Get vlans with same number
vlans_same_number = Vlan.objects.filter(
ambiente__in=envs).values('num_vlan').annotate(
count=Count('id')).filter(count__gt=1)
if len(vlans_same_number) > 0:
for vlan_gp in vlans_same_number:
vlans_current_number = Vlan.objects.filter(
ambiente__in=envs,
num_vlan=vlan_gp['num_vlan'])
filters_of_envs = [
vlan.ambiente.filter.id
for vlan in vlans_current_number
]
for fil_ in filters_of_envs:
if TipoEquipamento.objects.filter(
id=id_tipo_equipamento,
filterequiptype__filter=fil_).count(
) == 0:
filters_ok = False
break
if not filters_ok:
raise EquipTypeCantBeChangedError(
None,
'O tipo de equipamento não pode ser modificado pois existe um filtro em uso que não possui o novo tipo de equipamento informado.'
)
# Filter case 3 end
# Test all vip requests if equip.tipo_equipamento is
# balancing
if equip.tipo_equipamento == TipoEquipamento.get_tipo_balanceador(
):
vips = RequisicaoVips.objects.all()
vip_ips = []
vip_ipsv6 = []
for vip in vips:
if vip.vip_criado:
if vip.ip is not None:
if vip.ip.ipequipamento_set.filter(
equipamento=equip.id).count() > 0:
raise EquipTypeCantBeChangedError(
None,
'O tipo de equipamento não pode ser modificado pois este equipamento é o balanceador associado com o vip criado %s.'
% vip.id)
if vip.ipv6 is not None:
if vip.ipv6.ipv6equipament_set.filter(
equipamento=equip.id).count() > 0:
raise EquipTypeCantBeChangedError(
None,
'O tipo de equipamento não pode ser modificado pois este equipamento é o balanceador associado com o vip criado %s.'
% vip.id)
else:
if vip.ip is not None:
vip_ips.append(vip.ip.id)
if vip.ipv6 is not None:
vip_ipsv6.append(vip.ipv6.id)
nets_using_balancer_in_vips_ = [
ip_.networkipv4
for ip_ in Ip.objects.filter(id__in=vip_ips)
]
nets_using_balancer_in_vips = [
ip_.networkipv4 for ip_ in Ip.objects.filter(
networkipv4__in=nets_using_balancer_in_vips_,
ipequipamento__equipamento=equip.id)
]
nets_v6_using_balancer_in_vips_ = [
ip_.networkipv6
for ip_ in Ipv6.objects.filter(id__in=vip_ipsv6)
]
nets_v6_using_balancer_in_vips = [
ip_.networkipv6 for ip_ in Ipv6.objects.filter(
networkipv6__in=nets_v6_using_balancer_in_vips_,
ipv6equipament__equipamento=equip.id)
]
for net in nets_using_balancer_in_vips:
net_str = str(net.oct1) + '.' + str(net.oct2) + '.' + \
str(net.oct3) + '.' + str(net.oct4) + \
'/' + str(net.block)
if IpEquipamento.objects.filter(
ip__networkipv4=net,
equipamento__tipo_equipamento=TipoEquipamento
.get_tipo_balanceador()).exclude(
equipamento=equip).count() == 0:
raise EquipTypeCantBeChangedError(
None,
'O tipo de equipamento não pode ser modificado pois este equipamento é o único balanceador disponível na rede %s da vlan %s.'
% (net_str, net.vlan.nome))
for net in nets_v6_using_balancer_in_vips:
net_str = str(net.block1) + ':' + str(
net.block2
) + ':' + str(net.block3) + ':' + str(
net.block4) + ':' + str(
net.block5) + ':' + str(
net.block6) + ':' + str(
net.block7) + ':' + str(
net.block8) + '/' + str(
net.block)
if Ipv6Equipament.objects.filter(
ip__networkipv6=net,
equipamento__tipo_equipamento=TipoEquipamento
.get_tipo_balanceador()).exclude(
equipamento=equip).count() == 0:
raise EquipTypeCantBeChangedError(
None,
'O tipo de equipamento não pode ser modificado pois este equipamento é o único balanceador disponível na rede %s da vlan %s.'
% (net_str, net.vlan.nome))
ip_equipamento_list = IpEquipamento.objects.filter(
equipamento=equip_id)
ip6_equipamento_list = Ipv6Equipament.objects.filter(
equipamento=equip_id)
# Delete vlan's cache
key_list = []
for eq in ip_equipamento_list:
vlan = eq.ip.networkipv4.vlan
vlan_id = vlan.id
key_list.append(vlan_id)
for eq in ip6_equipamento_list:
vlan = eq.ip.networkipv6.vlan
vlan_id = vlan.id
key_list.append(vlan_id)
destroy_cache_function(key_list)
# Delete equipment's cache
destroy_cache_function([equip_id], True)
modelo = Modelo.get_by_pk(id_modelo)
equip.edit(user, nome, tipo_equip, modelo, maintenance)
return self.response(dumps_networkapi({}))
except EquipTypeCantBeChangedError, e:
return self.response_error(150, e.message)
def handle_post(self, request, user, *args, **kwargs):
"""Treat requests POST to add Model.
URL: model/
"""
try:
self.log.info("Add Model")
# User permission
if not has_perm(user, AdminPermission.BRAND_MANAGEMENT,
AdminPermission.WRITE_OPERATION):
self.log.error(
u'User does not have permission to perform the operation.')
raise UserNotAuthorizedError(None)
# Load XML data
xml_map, attrs_map = loads(request.raw_post_data)
# XML data format
networkapi_map = xml_map.get('networkapi')
if networkapi_map is None:
return self.response_error(
3,
u'There is no value to the networkapi tag of XML request.'
)
model_map = networkapi_map.get('model')
if model_map is None:
return self.response_error(
3, u'There is no value to the model tag of XML request.')
# Get XML data
name = model_map.get('name')
id_brand = model_map.get('id_brand')
# Valid name
if not is_valid_string_minsize(
name, 3) or not is_valid_string_maxsize(name, 100):
self.log.error(u'Parameter name is invalid. Value: %s', name)
raise InvalidValueError(None, 'name', name)
# Valid ID Brand
if not is_valid_int_greater_zero_param(id_brand):
self.log.error(
u'The id_brand parameter is not a valid value: %s.',
id_brand)
raise InvalidValueError(None, 'id_brand', id_brand)
# Find Brand by ID to check if it exist
brand = Marca.get_by_pk(id_brand)
try:
Modelo.get_by_name_brand(name, id_brand)
raise MarcaModeloNameDuplicatedError(
None, u'Já existe um modelo com o nome %s com marca %s.' %
(name, brand.nome))
except ModeloNotFoundError:
pass
model = Modelo()
# set variables
model.nome = name
model.marca = brand
try:
# save Model
model.save()
except Exception, e:
self.log.error(u'Failed to save the Model.')
raise EquipamentoError(e, u'Failed to save the Model.')
model_map = dict()
model_map['model'] = model_to_dict(model,
exclude=["nome", "marca"])
return self.response(dumps_networkapi(model_map))
