def update(self, id, format='html'):
ignore = ['Submit',
'username']
params = [
(p, request.params[p]) for p in request.params \
if p not in ignore]
data = {}
for key, value in params:
if key == 'cpu_speed' and not value:
continue
elif key == 'game' and not value:
continue
data[key] = value
activity_engine = ActivityEngine(Session,
session['active_user']['user_id'])
status = activity_engine.update('node', id, data)
if format =='html':
session['flash'] = status
session.save()
return redirect(url(controller='nodes', action='show', id=id))
elif format == 'json':
request.content_type = 'application/json'
return status
def updatebyname(self, name, format='html'):
id = Session.query(Node).filter(Node.hostname == name).first()
if id:
id = id.id
else:
return 'Host: %s is not in nodetraq' % name
ignore = ['Submit',
'username']
params = [
(p, request.params[p]) for p in request.params \
if p not in ignore]
data = {}
for key, value in params:
if value:
if key == 'cpu_speed' and not value:
continue
data[key] = value
activity_engine = ActivityEngine(Session,
session['active_user']['user_id'])
status = activity_engine.update('node', id, data)
if format =='html':
session['flash'] = status
session.save()
return redirect(url(controller='nodes', action='show', id=id))
elif format == 'json':
request.content_type = 'application/json'
return status
def updatebyname(self, name, format='html'):
id = Session.query(Node).filter(Node.hostname == name).first()
if id:
id = id.id
else:
return 'Host: %s is not in nodetraq' % name
ignore = ['Submit', 'username']
params = [
(p, request.params[p]) for p in request.params \
if p not in ignore]
data = {}
for key, value in params:
if value:
if key == 'cpu_speed' and not value:
continue
data[key] = value
activity_engine = ActivityEngine(Session,
session['active_user']['user_id'])
status = activity_engine.update('node', id, data)
if format == 'html':
session['flash'] = status
session.save()
return redirect(url(controller='nodes', action='show', id=id))
elif format == 'json':
request.content_type = 'application/json'
return status
def update(self, id, format='html'):
ignore = ['Submit', 'username']
params = [
(p, request.params[p]) for p in request.params \
if p not in ignore]
data = {}
for key, value in params:
if key == 'cpu_speed' and not value:
continue
elif key == 'game' and not value:
continue
data[key] = value
activity_engine = ActivityEngine(Session,
session['active_user']['user_id'])
status = activity_engine.update('node', id, data)
if format == 'html':
session['flash'] = status
session.save()
return redirect(url(controller='nodes', action='show', id=id))
elif format == 'json':
request.content_type = 'application/json'
return status
def enablehost(self, format=None):
valid = False
response.content_type = "application/json"
ids = None
ips = None
data = {}
ports = []
if 'ids[]' in request.params:
ids = [v for k, v in request.params.items() if 'ids[]' in k]
elif 'ips[]' in request.params:
ips = [v for k, v in request.params.items() if 'ips[]' in k]
if 'pool' in request.params:
pool = request.params['pool']
if 'ports[]' in request.params:
ports = [v for k,v in request.params.items()\
if 'ports[]' in k]
if ips or ids:
valid = True
if not valid:
session['flash'] = 'No ids were sent'
session.save()
if format == 'json':
return '{"success": false}'
else:
return redirect(url(controller='loadbalancer', action='index'))
lb = LoadBalancer(ip='your_lb_ip')
if ids:
data['hosts'] = [
Session.query(Node.hostname)\
.filter(Node.id == id)\
.first()[0] for id in ids ]
members = [
':'.join([Session.query(Node.primary_ip)\
.filter(Node.id == id)\
.first()[0], ports[i]])\
for i,id in enumerate(ids)]
elif ips:
data['hosts'] = [
Session.query(Node.hostname)\
.filter(or_(Node.primary_ip == ip,
Node.secondary_ip == ip))\
.first()[0] for ip in ips ]
members = [':'.join([ip, ports[i]])\
for i,ip in enumerate(ips)]
lb.enable_members(pool, members)
data['pool'] = pool
data['type'] = 'enable'
ae = ActivityEngine(Session, session['active_user']['user_id'])
status = ae.update('loadbalancer', None, data)
session['flash'] = status
session.save()
if format == 'json':
return '{"success": true}'
else:
return redirect(url(controller='loadbalancer', action='index'))
def enable_user(self, account):
dh = DirectoryHelper(
directoryCN, directoryPassword, directoryDomain)
result = dh.enable_user(account)
if result:
session['flash'] = 'Successfully enabled %s' % account
session.save()
else:
session['flash'] = 'Error disabling %s' % account
session.save()
ae = ActivityEngine(Session, session['active_user']['user_id'])
ae.update('user', None,
{ 'nsAccountLock': 'false', 'account': account })
return redirect(url(
controller='users', action='edit', account=account))
def disablehost(self, format=None):
valid = False
response.content_type = "application/json"
ids = None
ips = None
data = {}
if 'ids[]' in request.params:
ids = [v for k,v in request.params.items() if 'ids[]' in k]
elif 'ips[]' in request.params:
ips = [v for k,v in reuqest.params.items() if 'ips[]' in k]
if 'pool' in request.params:
pool = request.params['pool']
if 'ports[]' in request.params:
ports = [v for k,v in request.params.items()\
if 'ports[]' in k]
if ids or ips:
valid = True
if not valid:
session['flash'] = 'No ids were sent'
session.save()
if format == 'json':
return '{"success": false}'
else:
return redirect(url(
controller='loadbalancer', action='index'))
lb = LoadBalancer(ip='your_lb_ip')
if ids:
data['hosts'] = [ q[0] for q in\
Session.query(Node.hostname)\
.filter(Node.id.in_(ids)).all() ]
members = [
':'.join([Session.query(Node.primary_ip)\
.filter(Node.id == id)\
.first()[0], ports[i]])\
for i,id in enumerate(ids)]
elif ips:
data['hosts'] = [q[0] for q in Session.query(Node.hostname)\
.filter(or_(Node.primary_ip.in_(ips),
Node.secondary_ip.in_(ips)))\
.all() ]
members = [
':'.join([ip, ports[i]]) for i,ip in enumerate(ips)]
lb.disable_members(pool, members)
data['type'] = 'disable'
data['pool'] = pool
ae = ActivityEngine(
Session, session['active_user']['user_id'])
status = ae.update('loadbalancer', None, data)
session['flash'] = status
if format == 'json':
return '{"success": true}'
else:
return redirect(url(
controller='loadbalancer', action='index'))
def sendbatchedit(self):
response.content_type = 'application/json'
user_id = int(request.params['user_id'])
field = request.params['field']
info = [json.loads(d[1]) for d in request.params.items()\
if 'items[]' in d]
if field == 'hostname':
for item in info:
data = {}
data['hostname'] = item['hostname']
activity_engine = ActivityEngine(Session,
user_id)
status = activity_engine.update(
'node', item['node_id'], data)
session['flash'] = "Success"
session.save()
elif field == 'groups':
for item in info:
node = Session.query(Node)\
.filter(Node.id == item['node_id']).first()
group_list = item['groups']\
.replace(' ', '').split(',')
node.groups = Session.query(Group)\
.filter(Group.name.in_(group_list)).all()
Session.add(node)
Session.commit()
session['flash'] = "Success"
session.save()
elif field == 'db_backups':
for item in info:
backup = Session.query(NodeDatabaseBackup)\
.filter(NodeDatabaseBackup.id == item['backup_id'])\
.first()
node = Session.query(Node)\
.filter(Node.id == item['node_id']).first()
backup.server_id = node.id
if item['data_type'] == 'storage':
storage = Session.query(Node)\
.filter(Node.hostname == item['value']).first()
backup.storage_id = storage.id
elif item['data_type'] == 'directory':
backup.directory = item['value']
Session.add(backup)
Session.commit()
session['flash'] = "Success"
session.save()
else:
return '{"success": false}'
return '{"success": true}'
def update(self):
id = request.params['group_id']
data = {}
data['name'] = request.params['name']
data['description'] = request.params['description']
activity_engine = ActivityEngine(Session,
session['active_user']['user_id'])
status = activity_engine.update('group', id, data)
session['flash'] = status
session.save()
return redirect(url(controller='groups', action='index'))
def sendbatchedit(self):
response.content_type = 'application/json'
user_id = int(request.params['user_id'])
field = request.params['field']
info = [json.loads(d[1]) for d in request.params.items()\
if 'items[]' in d]
if field == 'hostname':
for item in info:
data = {}
data['hostname'] = item['hostname']
activity_engine = ActivityEngine(Session, user_id)
status = activity_engine.update('node', item['node_id'], data)
session['flash'] = "Success"
session.save()
elif field == 'groups':
for item in info:
node = Session.query(Node)\
.filter(Node.id == item['node_id']).first()
group_list = item['groups']\
.replace(' ', '').split(',')
node.groups = Session.query(Group)\
.filter(Group.name.in_(group_list)).all()
Session.add(node)
Session.commit()
session['flash'] = "Success"
session.save()
elif field == 'db_backups':
for item in info:
backup = Session.query(NodeDatabaseBackup)\
.filter(NodeDatabaseBackup.id == item['backup_id'])\
.first()
node = Session.query(Node)\
.filter(Node.id == item['node_id']).first()
backup.server_id = node.id
if item['data_type'] == 'storage':
storage = Session.query(Node)\
.filter(Node.hostname == item['value']).first()
backup.storage_id = storage.id
elif item['data_type'] == 'directory':
backup.directory = item['value']
Session.add(backup)
Session.commit()
session['flash'] = "Success"
session.save()
else:
return '{"success": false}'
return '{"success": true}'
def update(self):
id = request.params['group_id']
data = {}
data['name'] = request.params['name']
data['description'] = request.params['description']
activity_engine = ActivityEngine(Session,
session['active_user']['user_id'])
status = activity_engine.update('group', id, data)
session['flash'] = status
session.save()
return redirect(url(controller='groups', action='index'))
def update(self, account):
update_items = (
'cn', 'uidNumber', 'mail',
'loginShell', 'title', 'manager',
'departmentNumber', 'roomNumber',
'deploycode', 'orgchartmanager', 'utilityaccount')
dh = DirectoryHelper(
directoryCN, directoryPassword, directoryDomain)
updates = {}
groups = [v for k,v in request.params.items() if k == 'group']
for item in update_items:
if item in request.params:
if item == 'cn':
updates['cn'] = request.params[item]
updates['gecos'] = request.params[item]
splitname = request.params[item].split()
updates['givenName'] = splitname[0]
if len(splitname) > 1:
updates['sn'] = splitname[1]
updates[item] = request.params[item]
# for key in updates, update ldap and ad
for k, v in updates.iteritems():
dh.update_attribute(account, k, v)
# for group in groups, update ldap groups
dh.set_groups(account, groups)
session['flash'] = 'Successfully updated %s' % account
session.save()
updates['account'] = account
ae = ActivityEngine(Session, session['active_user']['user_id'])
ae.update('user', None, updates)
return redirect(url(controller='users', action='edit', account=account))
def removehost(self, format=None):
valid = False
response.content_type = "application/json"
ids = []
ips = []
data = {}
ports = []
if 'ids[]' in request.params:
ids = [v for k, v in request.params.items() if 'ids[]' in k]
if 'ips[]' in request.params:
ips = [v for k, v in request.params.items() if 'ips[]' in k]
if 'pool' in request.params:
pool = request.params['pool']
if 'ports[]' in request.params:
ports = [v for k,v in request.params.items()\
if 'ports[]' in k]
if ids or ips:
if ports:
valid = True
if not valid:
session['flash'] = 'No ids were sent'
session.save()
if format == 'json':
return '{"success": false}'
else:
return redirect(url(controller='loadbalancer', action='index'))
lb = LoadBalancer(ip='your_lb_ip')
members = lb.b.LocalLB.Pool.typefactory.create(
'Common.IPPortDefinitionSequence')
if ids:
data['hosts'] = [
q[0] for q in\
Session.query(Node.hostname)\
.filter(Node.id.in_(ids)).all() ]
members.item = [lb._member_factory(
':'.join([Session.query(Node.primary_ip)\
.filter(Node.id == id).first()[0],
ports[i]])) for i,id in enumerate(ids)]
if ips:
data['hosts'] = [ q[0] for q in \
Session.query(Node.hostname)\
.filter(or_(Node.primary_ip.in_(ips),
Node.secondary_ip.in_(ips)))\
.all() ]
members.item = [
lb._member_factory(':'.join(ip, ports[i]))
for i, ip in enumerate(ips)
]
lb.remove_members(pool, members)
data['type'] = 'remove'
data['pool'] = pool
ae = ActivityEngine(Session, session['active_user']['user_id'])
status = ae.update('loadbalancer', None, data)
session['flash'] = status
session.save()
if format == 'json':
return '{"success": true}'
else:
return redirect(url(controller='loadbalancer', action='index'))
def addhost(self, format=None):
valid = False
response.content_type = "application/json"
ids = []
ips = []
ports = []
if 'ids[]' in request.params:
ids = [v for k,v in request.params.items() if 'ids[]' in k]
if 'ips[]' in request.params:
ips = [v for k,v in request.params.items() if 'ips[]' in k]
if 'pool' in request.params:
pool = request.params['pool']
if 'ports[]' in request.params:
ports = [v for k,v in request.params.items()\
if 'ports[]' in k]
if ids or ips:
if ports:
valid = True
if not valid:
session['flash'] = 'No ids were sent'
session.save()
if format == 'json':
return '{"success": false}'
else:
return redirect(url(
controller='loadbalancer', action='index'))
lb = LoadBalancer(ip='your_lb_ip')
members = lb.b.LocalLB.Pool.typefactory.create(
'Common.IPPortDefinitionSequence')
data = {}
if ids:
data['hosts'] = [
Session.query(Node.hostname)\
.filter(Node.id == id)\
.first()[0] for id in ids ]
iplist = [ip[0] for ip in Session.query(Node.primary_ip)\
.filter(Node.id.in_(ids)).all()]
members.item = [lb._member_factory(
':'.join([ip, ports[i]]))\
for i,ip in enumerate(iplist)]
if ips:
data['hosts'] = [ q[0] for q in\
Session.query(Node.hostname)\
.filter(or_(Node.primary_ip == ip,
Node.secondary_ip == ip))\
.all() for ip in ips ]
members.item = [lb._member_factory(
':'.join([ip, port]))\
for i,ip in enumerate(ips)]
lb.add_members(pool, members)
data['pool'] = pool
data['type'] = 'add'
ae = ActivityEngine(
Session, session['active_user']['user_id'])
if len(ids) > 2 or len(ips) > 2:
status = ae.update_many('loadbalancer', data)
else:
status = ae.update('loadbalancer', None, data)
session['flash'] = status
session.save()
if format == 'json':
return '{"success": true}'
else:
return redirect(url(
controller='loadbalancer', action='index'))
