131 lines
5.1 KiB
Python
131 lines
5.1 KiB
Python
from orm.services.customer_manager.cms_rest.data.sql_alchemy.group_record \
|
|
import GroupRecord
|
|
from orm.services.customer_manager.cms_rest.data.sql_alchemy.models \
|
|
import GroupsCustomerRole
|
|
from orm.services.customer_manager.cms_rest.data.sql_alchemy.region_record \
|
|
import RegionRecord
|
|
from orm.services.customer_manager.cms_rest.logger import get_logger
|
|
|
|
LOG = get_logger(__name__)
|
|
|
|
|
|
class GroupsCustomerRoleRecord:
|
|
def __init__(self, session):
|
|
|
|
# thie model uses for the parameters for any acceess methods - not
|
|
# as instance of record in the table
|
|
self.__groups_customer_role = GroupsCustomerRole()
|
|
self.__TableName = "groups_customer_role"
|
|
|
|
if (session):
|
|
self.session = session
|
|
|
|
def setDBSession(self, session):
|
|
self.session = session
|
|
|
|
@property
|
|
def groups_customer_role(self):
|
|
return self.__groups_customer_role
|
|
|
|
@groups_customer_role.setter
|
|
def groups_customer_role(self):
|
|
self.__groups_customer_role = GroupsCustomerRole()
|
|
|
|
def insert(self, groups_customer_role):
|
|
try:
|
|
self.session.add(groups_customer_role)
|
|
except Exception as exception:
|
|
LOG.log_exception(
|
|
"Failed to insert groups_customer_role" +
|
|
str(groups_customer_role), exception)
|
|
raise
|
|
|
|
def get_group_customer_role_by_keys(self,
|
|
group_uuid,
|
|
role_id,
|
|
region_name,
|
|
customer_uuid):
|
|
region_record = RegionRecord(self.session)
|
|
region_id = region_record.get_region_id_from_name(region_name)
|
|
if region_id is None:
|
|
raise ValueError(
|
|
'region with the region name {0} not found'.format(
|
|
region_name))
|
|
try:
|
|
group = self.session.query(GroupsCustomerRole).filter(
|
|
and_, (
|
|
GroupsCustomerRole.group_id == group_uuid,
|
|
GroupsCustomerRole.customer_id == customer_uuid,
|
|
GroupsCustomerRole.region_id == region_id,
|
|
GroupsCustomerRole.role_id == role_id))
|
|
return group.first()
|
|
|
|
except Exception as exception:
|
|
message = "Failed to get group/project by keys: " \
|
|
" group_uuid:%s customer_uuid:%s region_name:%s " \
|
|
" role_id:%s " \
|
|
% group_uuid, customer_uuid, region_id, role_id
|
|
LOG.log_exception(message, exception)
|
|
raise
|
|
|
|
def get_customer_roles_for_group(self, group_uuid):
|
|
groups_customer_roles = []
|
|
|
|
try:
|
|
query = self.session.query(GroupsCustomerRole).filter(
|
|
GroupsCustomerRole.group_id == group_uuid)
|
|
|
|
for groups_customer_role in query.all():
|
|
groups_customer_roles.append(groups_customer_role)
|
|
return groups_customer_roles
|
|
|
|
except Exception as exception:
|
|
message = "Failed to get projects for group: %s" % (group_uuid)
|
|
LOG.log_exception(message, exception)
|
|
raise
|
|
|
|
def check_groups_customer_role_exist(self, role_id, group_id):
|
|
count = self.session.query(GroupsCustomerRole).filter(
|
|
GroupsCustomerRole.role_id == role_id,
|
|
GroupsCustomerRole.group_id == group_id).count()
|
|
return count >= 1
|
|
|
|
def remove_customer_role_from_group(self,
|
|
group_uuid,
|
|
region_id,
|
|
customer_id,
|
|
role_id):
|
|
|
|
cmd = 'DELETE FROM groups_customer_role WHERE group_id = %s and \
|
|
region_id = %s and customer_id = %s and role_id = %s'
|
|
result = self.session.connection().execute(cmd,
|
|
(group_uuid,
|
|
region_id,
|
|
customer_id,
|
|
role_id))
|
|
|
|
self.session.flush()
|
|
|
|
if result.rowcount == 0:
|
|
LOG.warn('customer with the customer id {0} not found'.format(
|
|
customer_id))
|
|
raise ValueError(
|
|
'customer with the customer id {0} not found'.format(
|
|
customer_id))
|
|
|
|
LOG.debug("num records deleted: " + str(result.rowcount))
|
|
return result
|
|
|
|
def delete_all_customers_from_group(self, group_id):
|
|
# group_id can be a uuid (type of string) or id (type of int).
|
|
# If group_id is uuid, then get id from uuid and use the id in the
|
|
# next sql command
|
|
if isinstance(group_id, basestring):
|
|
group_record = GroupRecord(self.session)
|
|
group_id = group_record.get_group_id_from_uuid(group_id)
|
|
|
|
# not including default region which is -1
|
|
cmd = 'DELETE FROM groups_customer_role WHERE group_id = %s'
|
|
result = self.session.connection().execute(cmd, (group_id,))
|
|
return result
|