def run(self):
log = Logger('RolePolicyGetHandlerThread')
TAG = 'run'
if self.data and self.company_id:
# do stuff here
role = RoleDBHelper()
url_data = self.data.split('/')
role_id = url_data[0]
plugin_name = url_data[1]
role_detail = role.get_role(str(role_id),
company_id=self.company_id)
if role_detail:
role_policy_id = role_detail.get('policy_id')
if plugin_name == 'actions':
action_command = True
else:
action_command = False
if role_policy_id and not action_command:
plugin_data = get_individual_plugin(role_policy_id,
plugin_name)
elif not action_command:
role_policy_id, plugin_data = setup_default_policy()
if role_policy_id:
role.set_role_policy(str(role_id), str(role_policy_id))
plugin_data = plugin_data.get(plugin_name)
else:
log.e(TAG, 'Role Policy ID setup failed.')
opJson = json.dumps({'pass': False, 'count':0,
'message': 'Role policy creation failed.'})
self.request.write(opJson)
tornado.ioloop.IOLoop.instance().add_callback(
self.callback)
else:
plugin_data = {}
if isinstance(plugin_data, str):
plugin_data = json.loads(plugin_data)
plugin_data['_id'] = role_policy_id
plugin_data['object_type'] = 'Role'
plugin_data['name'] = role_detail.get(c.ROLE_TABLE_NAME)
opJson = json.dumps({'count':1, 'message': 'Successfull',
'data':plugin_data, 'pass': True})
self.request.write(opJson)
tornado.ioloop.IOLoop.instance().add_callback(self.callback)
else:
log.e(TAG, 'No valid role id is sent in request')
opJson = json.dumps({'pass': False,
'message': 'No valid role id is sent in request'})
self.request.write(opJson)
tornado.ioloop.IOLoop.instance().add_callback(self.callback)
else:
log.e(TAG, 'UnAuthorized Access for role policy ')
self.request.set_status(401)
tornado.ioloop.IOLoop.instance().add_callback(self.callback)
def run(self):
log = Logger('RolePolicyPutHandlerThread')
TAG = 'run'
# Flag to track status of policy
status = False
if self.data:
# do stuff here
role = RoleDBHelper()
url_data = self.data.split('/')
print self.request.request.body
request_data = json.loads(self.request.request.body)
role_id = url_data[0]
plugin_name = url_data[1]
role_detail = role.get_role(str(role_id),
company_id=self.company_id)
if role_detail:
role_policy_id = role_detail.get('policy_id')
command_handler_json = {'to': 'role', 'id': str(role_id),
'company_id': self.company_id}
if plugin_name != 'actions':
status = put_individual_plugin(role_policy_id, plugin_name,
request_data)
else:
status = True
command_handler_json['action'] = request_data.get('action')
command_handler_json['passcode'] = request_data.get('lock_key')
if status:
print "\nprinting the json output will be send to command\
handler\n", command_handler_json
create_command_handler_task.delay(command_handler_json)
request_data['_id'] = role_policy_id
request_data['object_type'] = 'Role'
request_data['name'] = role_detail.get(c.ROLE_TABLE_NAME)
opJson = json.dumps({'data': request_data, 'pass': True,
'count': 1, 'message': 'Everything fine'})
self.request.write(opJson)
tornado.ioloop.IOLoop.instance().add_callback(self.callback)
else:
opJson = json.dumps({'pass': False, 'count':0,
'message': 'Update operation at policy table failed'})
self.request.write(opJson)
tornado.ioloop.IOLoop.instance().add_callback(self.callback)
else:
log.e(TAG, 'No valid role id is sent in request')
opJson = json.dumps({'pass': False,
'message': 'No valid role id is sent in request'})
self.request.write(opJson)
tornado.ioloop.IOLoop.instance().add_callback(self.callback)
else:
log.e(TAG, 'UnAuthorized Access for role policy ')
self.request.set_status(401)
tornado.ioloop.IOLoop.instance().add_callback(self.callback)
def run(self):
log = Logger('RolesDeleteHandlerThread')
TAG = 'DELETE'
if self.data is None:
log.e(TAG, 'No Role ID in Request')
self.request.set_status(401)
tornado.ioloop.IOLoop.instance().add_callback(self.callback)
else:
role = RoleDBHelper()
is_deleted = role.delete_role(str(self.data), self.company_id)
if is_deleted:
log.i(TAG, 'Role Successfully deleted')
opJson = json.dumps({'pass': True,
'message': 'Role Successfully deleted'})
self.request.write(opJson)
tornado.ioloop.IOLoop.instance().add_callback(self.callback)
else:
log.e(TAG, 'Role Not Deleted')
opJson = json.dumps({'pass': False,
'message': 'Role Not Deleted'})
self.request.write(opJson)
tornado.ioloop.IOLoop.instance().add_callback(self.callback)
def run(self):
log = Logger('RolesDeleteHandlerThread')
TAG = 'DELETE'
if self.data is None:
log.e(TAG, 'No Role ID in Request')
self.request.set_status(401)
tornado.ioloop.IOLoop.instance().add_callback(self.callback)
else:
role = RoleDBHelper()
is_deleted = role.delete_role(str(self.data), self.company_id)
if is_deleted:
log.i(TAG, 'Role Successfully deleted')
opJson = json.dumps({'pass': True,
'message': 'Role Successfully deleted'})
self.request.write(opJson)
tornado.ioloop.IOLoop.instance().add_callback(self.callback)
else:
log.e(TAG, 'Role Not Deleted')
opJson = json.dumps({'pass': False,
'message': 'Role Not Deleted'})
self.request.write(opJson)
tornado.ioloop.IOLoop.instance().add_callback(self.callback)
def run(self):
#Return All the users in the User table
company_id = self.company_id
log = Logger('RolesPostHandler')
tag = 'run'
request_body = json.loads(self.request.request.body)
self.user_role = request_body.get('name')
if self.user_role is None or self.user_role == '':
log.e(tag, 'sent Role is either None or blank')
opJson = json.dumps({'pass': False,
'error': 'sent Role is either None or blank'})
self.request.write(opJson)
tornado.ioloop.IOLoop.instance().add_callback(self.callback)
role = RoleDBHelper()
insert_dict = {
c.ROLE_TABLE_NAME: self.user_role,
c.ROLE_TABLE_COMPANY: company_id,
c.ROLE_TABLE_DELETED: False
}
self.user_role_id = role.add_role(insert_dict)
if self.user_role_id is None:
log.e(tag,'Role not inserted in table')
opJson = json.dumps({'pass': False,
'error': 'Role not inserted in table'})
self.request.write(opJson)
tornado.ioloop.IOLoop.instance().add_callback(self.callback)
else:
opJson = json.dumps({'pass': True, 'role_id': self.user_role_id})
self.request.write(opJson)
tornado.ioloop.IOLoop.instance().add_callback(self.callback)
def run(self):
# Return All the users in the User table
company_id = self.company_id
log = Logger('RolesPostHandler')
tag = 'run'
request_body = json.loads(self.request.request.body)
self.user_role = request_body.get('name')
if self.user_role is None or self.user_role == '':
log.e(tag, 'sent Role is either None or blank')
opJson = json.dumps({'pass': False,
'error': 'sent Role is either None or blank'})
self.request.write(opJson)
tornado.ioloop.IOLoop.instance().add_callback(self.callback)
role = RoleDBHelper()
insert_dict = {
c.ROLE_TABLE_NAME: self.user_role,
c.ROLE_TABLE_COMPANY: company_id,
c.ROLE_TABLE_DELETED: False
}
self.user_role_id = role.add_role(insert_dict)
if self.user_role_id is None:
log.e(tag, 'Role not inserted in table')
opJson = json.dumps({'pass': False,
'error': 'Role not inserted in table'})
self.request.write(opJson)
tornado.ioloop.IOLoop.instance().add_callback(self.callback)
else:
opJson = json.dumps({'pass': True, 'role_id': self.user_role_id})
self.request.write(opJson)
tornado.ioloop.IOLoop.instance().add_callback(self.callback)
def run(self):
# Return All the users in the User table
log = Logger('TeamsGetHandlerThread')
tag = 'run'
print 'In TeamsGetHandlerThread'
final_dict = {}
role = RoleDBHelper()
team = TeamDBHelper()
user = UserDBHelper()
team_name = self.request.get_argument('team', None)
offset = self.request.get_argument('offset', None)
count = self.request.get_argument('count', None)
sort_by = self.request.get_argument(
'sort_by',
True) # Intentionally done
name_query = self.request.get_argument('name', None)
role_query = self.request.get_argument('role', None)
device_query = self.request.get_argument('device_id', None)
sort_order = self.request.get_argument('sort', None)
filter_key = str(self.request.get_argument('filter_key', None))
filter_value = self.request.get_argument('filter_value', None)
# Find all the roles in the Roles Table
if filter_key == 'role':
role_name = filter_value
role_id = role.get_role_by_name(str(role_name), self.company_id)
else:
role_name = None
role_id = None
if name_query:
query = name_query
query_type = 'name'
elif role_query:
query = role_query
query_type = 'role'
elif device_query:
query = device_query
query_type = 'device'
else:
query = None
query_type = None
if team_name:
team_id = team.get_team_by_name(str(team_name), self.company_id)
else:
team_id = None
teams_list = team.get_teams(self.company_id, [c.TEAM_TABLE_NAME,
c.TEAM_TABLE_ID])
if team_id:
result_list, total_count = user.get_users_for_team(
team_name=team_name, team_id=team_id,
role_name=role_name, role_id=role_id,
offset=offset, count=count, sort_by=sort_by,
query=query, query_type=query_type,
sort_order=sort_order)
elif teams_list:
for _team in teams_list:
_team['team_type'] = _team.get('name')
_team['team_name'] = _team.get('name')
_team['team_id'] = _team.get('id')
del _team['name']
del _team['id']
result_list = teams_list
total_count = len(result_list)
else:
result_list = None
if result_list:
print "\n total", total_count
final_dict['data'] = result_list
final_dict['count'] = total_count
final_dict['pass'] = True
final_dict['message'] = 'Success ...'
else:
final_dict['pass'] = True
final_dict['data'] = []
log.e(tag, 'No Team in Team Table')
final_dict['message'] = 'Failed ...'
final_dict['count'] = 0
opJson = json.dumps(final_dict)
self.request.write(opJson)
tornado.ioloop.IOLoop.instance().add_callback(self.callback)
def run(self):
#Return All the users in the User table
log = Logger('UsersGetHandlerThreadWithPage')
tag = 'run'
company_id = self.company_id
final_dict = {}
user = UserDBHelper()
team = TeamDBHelper()
role = RoleDBHelper()
violation = ViolationsDBHelper()
device = DeviceDBHelper()
offset = self.request.get_argument('offset', None)
count = self.request.get_argument('count', None)
name_query = self.request.get_argument('name', None)
team_query = self.request.get_argument('team', None)
role_query = self.request.get_argument('role', None)
device_query = self.request.get_argument('device_id', None)
sort_by = self.request.get_argument('sort_by', True)#Intentionally done
sort_order = self.request.get_argument('sort', None)
filter_key = self.request.get_argument('filter_key', None)
filter_value = self.request.get_argument('filter_value', None)
if filter_key == 'role':
role_name = filter_value
role_id = role.get_role_by_name(str(role_name), company_id)
else:
role_name = None
role_id = None
if filter_key == 'team':
team_name = filter_value
team_id = team.get_team_by_name(str(team_name), company_id)
else:
team_name = None
team_id = None
if filter_key == 'os':
os_mapper = {'Android': 'samsung', 'iOS': 'ios'}
os_type = os_mapper.get(str(filter_value))
else:
os_type = None
if name_query:
query = name_query
query_type = 'name'
elif role_query:
query = role_query
query_type = 'role'
elif device_query:
query = device_query
query_type = 'device'
elif team_query:
query = team_query
query_type = 'team'
else:
query = None
query_type = None
if offset:
result_list, total_count = user.get_users_for_user(
company_id=company_id, offset=offset, count=count,
role_id=role_id, team_id=team_id, query=query,
query_type=query_type, os_type=os_type,
sort_by=sort_by, sort_order=sort_order,
filter_key=filter_key, filter_value=filter_value)
else:
result_list, total_count = user.get_users(
{c.USER_TABLE_COMPANY: company_id})
if result_list:
for user_dict in result_list:
device_deleted = False
device_info = device.get_device_with_udid(user_dict.get(
'user_device'))
if not device_info:
device_info = device.get_device_with_udid(user_dict.get(
'user_device'), status=True)
device_deleted = True
if device_info:
device_id = device_info[0].get(c.DEVICE_TABLE_ID)
else:
device_id = None
if device_deleted:
user_dict['user_device'] = None
user_dict['user_device_os'] = None
if device_id:
print "printing \n device id", device_id
violation_count = violation.get_violation_count(
company_id=company_id, device_id=str(device_id))
else:
violation_count = 0
user_dict['user_violation'] = violation_count
final_dict['data'] = result_list
final_dict['count'] = total_count
final_dict['pass'] = True
final_dict['message'] = 'Seems like things are working ...'
else:
final_dict['pass'] = True
log.e(tag,'No User in User Table')
final_dict['message'] = 'Seems like things are not working ...'
final_dict['count'] = 0
## add all the data into dictionary and create output json
opJson = json.dumps(final_dict)
self.request.write(opJson)
tornado.ioloop.IOLoop.instance().add_callback(self.callback)
def merge(self, device_id):
user_helper = UserDBHelper()
device_helper = DeviceDBHelper()
roles_helper = RoleDBHelper()
teams_helper = TeamDBHelper()
company_helper = CompanyDBHelper()
policy_helper = PolicyDBHelper()
if device_id is not None:
device_details = device_helper.get_device(device_id)
if device_details is not None and 'user_id' in device_details:
user_details = user_helper.get_user(
str(device_details['user_id']))
team_id = user_details['team_id']
role_id = user_details['role_id']
company_id = user_details['company_id']
team_details = teams_helper.get_team(str(team_id))
role_details = roles_helper.get_role(str(role_id))
company_details = company_helper.get_company(str(company_id))
if user_details is not None and 'policy_id' in user_details:
policy_id_user = user_details['policy_id']
else:
print 'No user details found'
if team_details is not None and 'policy_id' in team_details:
policy_id_team = team_details['policy_id']
else:
print 'no team details found'
if role_details is not None and 'policy_id' in role_details:
policy_id_role = role_details['policy_id']
else:
print 'no role details found'
if (company_details is not None
and 'policy_id' in company_details):
policy_id_company = company_details['policy_id']
else:
print 'no company details found'
if policy_id_company is not None:
print 'company policy id=', policy_id_company
policy_company = policy_helper.get_policy(
str(policy_id_company))
else:
policy_company = None
if policy_id_role is not None:
print 'role policy id=', policy_id_role
policy_role = policy_helper.get_policy(str(policy_id_role))
else:
policy_role = None
if policy_id_team is not None:
print 'team policy id=', policy_id_team
policy_team = policy_helper.get_policy(str(policy_id_team))
else:
policy_team = None
if policy_id_user is not None:
print 'user policy id=', policy_id_user
policy_user = policy_helper.get_policy(str(policy_id_user))
else:
policy_user = None
return self.merge_policies(
policy_company,
policy_role,
policy_team,
policy_user)
else:
print 'Invalid device id'
def run(self):
log = Logger('RolePolicyPutHandlerThread')
TAG = 'run'
# Flag to track status of policy
status = False
if self.data:
# do stuff here
role = RoleDBHelper()
url_data = self.data.split('/')
print self.request.request.body
request_data = json.loads(self.request.request.body)
role_id = url_data[0]
plugin_name = url_data[1]
role_detail = role.get_role(str(role_id),
company_id=self.company_id)
if role_detail:
role_policy_id = role_detail.get('policy_id')
command_handler_json = {'to': 'role', 'id': str(role_id),
'company_id': self.company_id}
if plugin_name != 'actions':
status = put_individual_plugin(role_policy_id, plugin_name,
request_data)
else:
status = True
command_handler_json['action'] = request_data.get('action')
command_handler_json[
'passcode'] = request_data.get('lock_key')
if status:
print "\nprinting the json output will be send to command\
handler\n", command_handler_json
create_command_handler_task.delay(command_handler_json)
request_data['_id'] = role_policy_id
request_data['object_type'] = 'Role'
request_data['name'] = role_detail.get(c.ROLE_TABLE_NAME)
opJson = json.dumps(
{'data': request_data, 'pass': True,
'count': 1, 'message': 'Everything fine'})
self.request.write(opJson)
tornado.ioloop.IOLoop.instance().add_callback(
self.callback)
else:
opJson = json.dumps(
{'pass': False, 'count': 0,
'message': 'Update operation at policy table \
failed'})
self.request.write(opJson)
tornado.ioloop.IOLoop.instance().add_callback(
self.callback)
else:
log.e(TAG, 'No valid role id is sent in request')
opJson = json.dumps(
{'pass': False,
'message': 'No valid role id is sent in request'})
self.request.write(opJson)
tornado.ioloop.IOLoop.instance().add_callback(self.callback)
else:
log.e(TAG, 'UnAuthorized Access for role policy ')
self.request.set_status(401)
tornado.ioloop.IOLoop.instance().add_callback(self.callback)
def run(self):
log = Logger('RolePolicyGetHandlerThread')
TAG = 'run'
if self.data and self.company_id:
# do stuff here
role = RoleDBHelper()
url_data = self.data.split('/')
role_id = url_data[0]
plugin_name = url_data[1]
role_detail = role.get_role(str(role_id),
company_id=self.company_id)
if role_detail:
role_policy_id = role_detail.get('policy_id')
if plugin_name == 'actions':
action_command = True
else:
action_command = False
if role_policy_id and not action_command:
plugin_data = get_individual_plugin(role_policy_id,
plugin_name)
elif not action_command:
role_policy_id, plugin_data = setup_default_policy()
if role_policy_id:
role.set_role_policy(str(role_id), str(role_policy_id))
plugin_data = plugin_data.get(plugin_name)
else:
log.e(TAG, 'Role Policy ID setup failed.')
opJson = json.dumps(
{'pass': False, 'count': 0,
'message': 'Role policy creation failed.'})
self.request.write(opJson)
tornado.ioloop.IOLoop.instance().add_callback(
self.callback)
else:
plugin_data = {}
if isinstance(plugin_data, str):
plugin_data = json.loads(plugin_data)
plugin_data['_id'] = role_policy_id
plugin_data['object_type'] = 'Role'
plugin_data['name'] = role_detail.get(c.ROLE_TABLE_NAME)
opJson = json.dumps({'count': 1, 'message': 'Successfull',
'data': plugin_data, 'pass': True})
self.request.write(opJson)
tornado.ioloop.IOLoop.instance().add_callback(self.callback)
else:
log.e(TAG, 'No valid role id is sent in request')
opJson = json.dumps(
{'pass': False,
'message': 'No valid role id is sent in request'})
self.request.write(opJson)
tornado.ioloop.IOLoop.instance().add_callback(self.callback)
else:
log.e(TAG, 'UnAuthorized Access for role policy ')
self.request.set_status(401)
tornado.ioloop.IOLoop.instance().add_callback(self.callback)
def run(self):
log = Logger('RolesPutHandlerThread')
TAG = 'PUT'
input_dict = json.loads(self.request.request.body)
self.role_id = input_dict.get('role_id')
if self.role_id is None:
log.e(TAG, 'No role_id sent aborting update')
opJson = json.dumps({'pass': False, 'message': 'No role_id sent'})
self.request.set_status(401)
self.request.write(opJson)
tornado.ioloop.IOLoop.instance().add_callback(self.callback)
else:
if input_dict.has_key('name'):
print 'Name found'
self.role_name = input_dict['name']
if self.role_name is None:
log.e(TAG, 'No role_name sent aborting update')
opJson = json.dumps({'pass': False, 'message': 'No role_name sent'})
self.request.write(opJson)
tornado.ioloop.IOLoop.instance().add_callback(self.callback)
else:
role = RoleDBHelper()
update_dict = {
c.ROLE_TABLE_NAME: self.role_name
}
result = role.update_role(str(self.role_id),
self.company_id, update_dict)
if result == False:
log.e(TAG, 'Not able to update the role_name at DB')
opJson = json.dumps({'pass': False, 'message': 'Not able to update the role_name at DB'})
self.request.write(opJson)
tornado.ioloop.IOLoop.instance().add_callback(self.callback)
else:
log.i(TAG, 'role_name Updated successfully')
opJson = json.dumps({'pass': True, 'message': 'role_name updated for role.'})
self.request.write(opJson)
tornado.ioloop.IOLoop.instance().add_callback(self.callback)
elif input_dict.has_key('policy_id'):
print 'Policy found'
self.role_policy_id = input_dict['policy_id']
if self.role_policy_id is None:
log.e(TAG, 'No Policy ID sent aborting update')
opJson = json.dumps({'pass': False, 'message': 'No policy_id sent'})
self.request.write(opJson)
tornado.ioloop.IOLoop.instance().add_callback(self.callback)
else:
role = RoleDBHelper()
update_dict = {
c.ROLE_TABLE_POLICY: self.role_policy_id
}
result = role.update_role(str(self.role_id),
self.company_id, update_dict)
if result == False:
log.e(TAG, 'Not able to update the policy_id at DB')
opJson = json.dumps({'pass': False, 'message': 'Not able to update the policy_id at DB'})
self.request.write(opJson)
tornado.ioloop.IOLoop.instance().add_callback(self.callback)
else:
log.i(TAG, 'policy_id Updated successfully')
opJson = json.dumps({'pass': True, 'message': 'policy_id updated for role.'})
self.request.write(opJson)
tornado.ioloop.IOLoop.instance().add_callback(self.callback)
def run(self):
log = Logger('TeamGetHandlerThread')
TAG = 'GET'
page = self.request.get_argument('page', None)
count = self.request.get_argument('count', None)
sort_by = self.request.get_argument('sort_by', None)
if self.data is None:
log.e(TAG, 'No Team ID in Request')
opJson = json.dumps({
'pass':
False,
'message':
'No team found corresponding to the team \
id ' + str(self.data)
})
self.request.write(opJson)
tornado.ioloop.IOLoop.instance().add_callback(self.callback)
else:
team = TeamDBHelper()
teams = team.get_team(str(self.data), company_id=self.company_id)
outer_dict = {}
outer_array = []
print teams
if teams is None:
log.e(
TAG, 'No team found corresponding to the team id ' +
str(self.data))
opJson = json.dumps({
'pass':
False,
'message':
'No team found corresponding to the team \
id ' + str(self.data)
})
self.request.write(opJson)
tornado.ioloop.IOLoop.instance().add_callback(self.callback)
else:
self.team_name = teams.get(c.TEAM_TABLE_NAME)
user = UserDBHelper()
filter_dict = {c.USER_TABLE_TEAM: str(teams.get('id'))}
user_list = user.get_users_with_pages(filter_dict, int(page),
int(count), str(sort_by))
inner_array = []
if user_list is not None:
for users in user_list:
inner_dict = {}
self.user_id = str(users[c.USER_TABLE_ID])
self.user_name = str(users[c.USER_TABLE_NAME])
self.user_email = str(users[c.USER_TABLE_EMAIL])
self.user_role_id = str(users[c.USER_TABLE_ROLE])
# Find out the role for user
role = RoleDBHelper()
roles = role.get_role(self.user_role_id,
[c.ROLE_TABLE_NAME])
if roles is None:
self.user_role = None
else:
self.user_role = str(roles[c.ROLE_TABLE_NAME])
# Find out the device for user
device = DeviceDBHelper()
device_list = device.get_devices_of_user(self.user_id)
if device_list is None:
self.user_device = None
else:
devices = device_list[0]
self.user_device = str(
devices[c.DEVICE_TABLE_UDID])
self.user_device_os = str(
devices[c.DEVICE_TABLE_OS])
# Find out user violations
violation = ViolationsDBHelper()
violation_count = violation.get_violation_count(
str(devices.get('id')))
inner_dict = {
'user_id': self.user_id,
'user_name': self.user_name,
'user_role': self.user_role,
'user_team': self.team_name,
'user_device': self.user_device,
'user_violations': violation_count,
'user_device_os': self.user_device_os,
'user_email': self.user_email
}
inner_array.append(inner_dict)
outer_dict['name'] = self.team_name
outer_dict['users'] = inner_array
outer_dict['team_id'] = self.data
print '\nouter_dict'
print outer_dict
outer_array.append(outer_dict)
self.final_dict['pass'] = True
self.final_dict['teams'] = outer_array
opJson = json.dumps(self.final_dict)
self.request.write(opJson)
tornado.ioloop.IOLoop.instance().add_callback(self.callback)
def perform(self):
TAG = 'run'
print 'In RuN'
self.insert = True
json_data = self.data
print json_data['to']
#Find out the category of the Device on the basis of 'to' field
# Case 1 : Command is sent to the USER
if str(json_data['to']) == 'user':
to_id = str(json_data['id'])
company_id = json_data.get('company_id')
user = UserDBHelper()
user_dict = user.get_user(str(to_id), company_id=company_id,
pluck=[c.USER_TABLE_NAME])
if user_dict is not None:
user_name = str(user_dict[c.USER_TABLE_NAME])
message = "Command sent to " + user_name +\
" having ID = " + str(json_data['id'])
logs = LogsDBHelper()
logs_id = logs.add_log(to_id, str(json_data['to']), 'info',
None, message, raw=None, company=str(company_id))
if logs_id is None:
self.log.e(TAG, 'Not able to insert the logs')
self.command_to_user(json_data)
else:
self.log.e(TAG, 'No details corresponding to user found')
#Case 2: Command is sent to the Teams
elif str(json_data['to']) == 'team':
print 'sending to teams'
team_id = str(json_data['id'])
company_id = json_data.get('company_id')
team = TeamDBHelper()
team_dict = team.get_team(str(team_id), company_id=company_id,
pluck=[c.TEAM_TABLE_NAME])
if team_dict is not None:
team_name = str(team_dict[c.TEAM_TABLE_NAME])
message = "Command sent to " + team_name +\
" having ID = " + str(json_data['id'])
logs = LogsDBHelper()
logs_id = logs.add_log(team_id, str(json_data['to']), 'info',
None, message, raw=None, company=str(company_id))
if logs_id is None:
self.log.e(TAG, 'Not able to insert the logs')
self.command_to_team(json_data)
else:
self.log.e(TAG, "No details corresponding to team_id found. ")
#Case 3: Command is sent to the Role
elif str(json_data['to']) == 'role':
role_id = str(json_data['id'])
company_id = json_data.get('company_id')
role = RoleDBHelper()
role_dict = role.get_role(str(role_id), company_id=company_id,
pluck=[c.ROLE_TABLE_NAME])
if role_dict is not None:
role_name = str(role_dict[c.ROLE_TABLE_NAME])
message = "Command sent to " + role_name +\
" having ID = " + str(json_data['id'])
logs = LogsDBHelper()
logs_id = logs.add_log(role_id, str(json_data['to']), 'info',
None, message, raw=None, company=str(company_id))
if logs_id is None:
self.log.e(TAG, 'Not able to insert the logs')
self.command_to_role(json_data)
else:
self.log.e(TAG, 'No role corresponding to given role_id found')
elif str(json_data['to']) == 'company':
company_id = str(json_data['id'])
company = CompanyDBHelper()
company_dict = company.get_company(str(company_id))
if company_dict is not None:
company_name = str(company_dict[c.COMPANY_TABLE_NAME])
message = "Command sent to " + company_name\
+" having ID = " + str(json_data['id'])
logs = LogsDBHelper()
logs_id = logs.add_log(company_id, str(json_data['to']),
'info', None, message, raw=None, company=company_id)
if logs_id is None:
self.log.e(TAG, 'Not able to insert the logs')
self.command_to_company(json_data)
else:
self.log.e(TAG, 'No data corresponding to team id given found')
#Case 5: Some other parameter sent in 'to' field
else:
self.log.e(TAG, 'Somthing wrong with \'to\' field of POST data')
##Create the O/P JSON
opJson = json.dumps({'pass':False, 'error':'Correct TO field'})
self.log.e(TAG, str(opJson))
def parse(self,
excel_path,
company,
company_name=None,
callback=None,
user_added_callback=None):
COMPANY = company
company_name = company_name
book = open_workbook(filename=excel_path)
sheet = book.sheet_by_index(0)
users = list()
for row_index in range(sheet.nrows):
user = dict()
for col_index in range(sheet.ncols):
cell = sheet.cell(row_index, col_index).value
if col_index == 0:
user[C.USER_TABLE_EMAIL] = cell
elif col_index == 1:
user[C.USER_TABLE_NAME] = cell
elif col_index == 2:
user[C.USER_TABLE_TEAM] = cell
elif col_index == 3:
user[C.USER_TABLE_ROLE] = cell
users.append(user)
roles_list = list()
teams_list = list()
for user in users:
if C.USER_TABLE_ROLE in user:
roles_list.append(user[C.USER_TABLE_ROLE])
if C.USER_TABLE_TEAM in user:
teams_list.append(user[C.USER_TABLE_TEAM])
# get a set to get the unique elements.
roles = dict()
teams = dict()
roles_helper = RoleDBHelper()
teams_helper = TeamDBHelper()
users_helper = UserDBHelper()
enrollment_helper = EnrollmentDBHelper()
for role_name in set(roles_list):
role = dict()
role[C.ROLE_TABLE_COMPANY] = COMPANY
role[C.ROLE_TABLE_NAME] = role_name
roles[role_name] = roles_helper.add_role(role)
for team_name in set(teams_list):
team = dict()
team[C.TEAM_TABLE_COMPANY] = COMPANY
team[C.TEAM_TABLE_NAME] = team_name
team[C.TEAM_TABLE_DELETED] = False
teams[team_name] = teams_helper.add_team(team)
# Now we have id for all teams and roles... Insert the users now.
for user in users:
user_obj = dict()
user_obj[C.USER_TABLE_COMPANY] = COMPANY
user_obj[C.USER_TABLE_EMAIL] = user[C.USER_TABLE_EMAIL]
user_obj[C.USER_TABLE_NAME] = user[C.USER_TABLE_NAME]
user_obj[C.USER_TABLE_ROLE] = roles[user[C.USER_TABLE_ROLE]]
user_obj[C.USER_TABLE_TEAM] = teams[user[C.USER_TABLE_TEAM]]
user['id'] = users_helper.add_user_if_not_exists(user_obj)[0]
if user_added_callback is not None:
user_added_callback(user_obj)
for user in users:
enrollment = dict()
enrollment[C.ENROLLMENT_TABLE_USER] = user['id']
enrollment[C.ENROLLMENT_TABLE_PASSWORD] = generate_password()
enrollment[C.ENROLLMENT_TABLE_IS_ENROLLED] = False
user['passcode'] = enrollment[C.ENROLLMENT_TABLE_PASSWORD]
print 'adding enrollment ', enrollment
user['enrollment_id'] = enrollment_helper.add_enrollment(
enrollment)
user['company_name'] = company_name
if callback is not None:
callback(users)
return users
def merge(self, device_id):
user_helper = UserDBHelper()
device_helper = DeviceDBHelper()
roles_helper = RoleDBHelper()
teams_helper = TeamDBHelper()
company_helper = CompanyDBHelper()
policy_helper = PolicyDBHelper()
if device_id is not None:
device_details = device_helper.get_device(device_id)
if device_details is not None and 'user_id' in device_details:
user_details = user_helper.get_user(
str(device_details['user_id']))
team_id = user_details['team_id']
role_id = user_details['role_id']
company_id = user_details['company_id']
team_details = teams_helper.get_team(str(team_id))
role_details = roles_helper.get_role(str(role_id))
company_details = company_helper.get_company(str(company_id))
if user_details is not None and 'policy_id' in user_details:
policy_id_user = user_details['policy_id']
else:
print 'No user details found'
if team_details is not None and 'policy_id' in team_details:
policy_id_team = team_details['policy_id']
else:
print 'no team details found'
if role_details is not None and 'policy_id' in role_details:
policy_id_role = role_details['policy_id']
else:
print 'no role details found'
if (company_details is not None
and 'policy_id' in company_details):
policy_id_company = company_details['policy_id']
else:
print 'no company details found'
if policy_id_company is not None:
print 'company policy id=', policy_id_company
policy_company = policy_helper.get_policy(
str(policy_id_company))
else:
policy_company = None
if policy_id_role is not None:
print 'role policy id=', policy_id_role
policy_role = policy_helper.get_policy(str(policy_id_role))
else:
policy_role = None
if policy_id_team is not None:
print 'team policy id=', policy_id_team
policy_team = policy_helper.get_policy(str(policy_id_team))
else:
policy_team = None
if policy_id_user is not None:
print 'user policy id=', policy_id_user
policy_user = policy_helper.get_policy(str(policy_id_user))
else:
policy_user = None
return self.merge_policies(policy_company, policy_role,
policy_team, policy_user)
else:
print 'Invalid device id'
def perform(self):
TAG = 'run'
print 'In RuN'
self.insert = True
json_data = self.data
print json_data['to']
# Find out the category of the Device on the basis of 'to' field
# Case 1 : Command is sent to the USER
if str(json_data['to']) == 'user':
to_id = str(json_data['id'])
company_id = json_data.get('company_id')
user = UserDBHelper()
user_dict = user.get_user(str(to_id), company_id=company_id,
pluck=[c.USER_TABLE_NAME])
if user_dict is not None:
user_name = str(user_dict[c.USER_TABLE_NAME])
message = "Command sent to " + user_name +\
" having ID = " + str(json_data['id'])
logs = LogsDBHelper()
logs_id = logs.add_log(to_id,
str(json_data['to']),
'info',
None,
message,
raw=None,
company=str(company_id))
if logs_id is None:
self.log.e(TAG, 'Not able to insert the logs')
self.command_to_user(json_data)
else:
self.log.e(TAG, 'No details corresponding to user found')
# Case 2: Command is sent to the Teams
elif str(json_data['to']) == 'team':
print 'sending to teams'
team_id = str(json_data['id'])
company_id = json_data.get('company_id')
team = TeamDBHelper()
team_dict = team.get_team(str(team_id), company_id=company_id,
pluck=[c.TEAM_TABLE_NAME])
if team_dict is not None:
team_name = str(team_dict[c.TEAM_TABLE_NAME])
message = "Command sent to " + team_name +\
" having ID = " + str(json_data['id'])
logs = LogsDBHelper()
logs_id = logs.add_log(team_id,
str(json_data['to']),
'info',
None,
message,
raw=None,
company=str(company_id))
if logs_id is None:
self.log.e(TAG, 'Not able to insert the logs')
self.command_to_team(json_data)
else:
self.log.e(TAG, "No details corresponding to team_id found. ")
# Case 3: Command is sent to the Role
elif str(json_data['to']) == 'role':
role_id = str(json_data['id'])
company_id = json_data.get('company_id')
role = RoleDBHelper()
role_dict = role.get_role(str(role_id), company_id=company_id,
pluck=[c.ROLE_TABLE_NAME])
if role_dict is not None:
role_name = str(role_dict[c.ROLE_TABLE_NAME])
message = "Command sent to " + role_name +\
" having ID = " + str(json_data['id'])
logs = LogsDBHelper()
logs_id = logs.add_log(role_id,
str(json_data['to']),
'info',
None,
message,
raw=None,
company=str(company_id))
if logs_id is None:
self.log.e(TAG, 'Not able to insert the logs')
self.command_to_role(json_data)
else:
self.log.e(TAG, 'No role corresponding to given role_id found')
elif str(json_data['to']) == 'company':
company_id = str(json_data['id'])
company = CompanyDBHelper()
company_dict = company.get_company(str(company_id))
if company_dict is not None:
company_name = str(company_dict[c.COMPANY_TABLE_NAME])
message = "Command sent to " + company_name\
+ " having ID = " + str(json_data['id'])
logs = LogsDBHelper()
logs_id = logs.add_log(company_id,
str(json_data['to']),
'info',
None,
message,
raw=None,
company=company_id)
if logs_id is None:
self.log.e(TAG, 'Not able to insert the logs')
self.command_to_company(json_data)
else:
self.log.e(TAG, 'No data corresponding to team id given found')
# Case 5: Some other parameter sent in 'to' field
else:
self.log.e(TAG, 'Somthing wrong with \'to\' field of POST data')
# Create the O/P JSON
opJson = json.dumps({'pass': False, 'error': 'Correct TO field'})
self.log.e(TAG, str(opJson))
def run(self):
# Return All the users in the User table
log = Logger('RolesGetHandler')
tag = 'run'
final_dict = {}
role = RoleDBHelper()
team = TeamDBHelper()
user = UserDBHelper()
role_name = self.request.get_argument('role', None)
offset = self.request.get_argument('offset', None)
count = self.request.get_argument('count', None)
name_query = self.request.get_argument('name', None)
team_query = self.request.get_argument('team', None)
device_query = self.request.get_argument('device_id', None)
sort_by = self.request.get_argument(
'sort_by',
True) # Intentionally done
sort_order = self.request.get_argument('sort', None)
filter_key = str(self.request.get_argument('filter_key', None))
filter_value = self.request.get_argument('filter_value', None)
print role_name
print "\nUI sorting order \n", sort_order
# Find all the roles in the Roles Table
# outer_array = []
if name_query:
query = name_query
query_type = 'name'
elif team_query:
query = team_query
query_type = 'team'
elif device_query:
query = device_query
query_type = 'device'
else:
query = None
query_type = None
if filter_key == 'team':
team_name = filter_value
team_id = team.get_team_by_name(str(team_name), self.company_id)
else:
team_name = None
team_id = None
print "printing team id here ..", team_id
if role_name:
role_id = role.get_role_by_name(str(role_name), self.company_id)
else:
roles_list = role.get_roles(self.company_id, [c.ROLE_TABLE_NAME,
c.ROLE_TABLE_ID])
role_id = None
if role_id:
result_list, total_count = user.get_users_for_role(
role_name=role_name, role_id=role_id,
team_name=team_name, team_id=team_id, offset=offset,
count=count, sort_by=sort_by, query=query,
query_type=query_type, sort_order=sort_order)
elif roles_list:
for _role in roles_list:
_role['role_type'] = _role.get('name')
_role['role_name'] = _role.get('name')
_role['role_id'] = _role.get('id')
del _role['name']
del _role['id']
result_list = roles_list
total_count = len(result_list)
else:
result_list = None
if result_list:
final_dict['data'] = result_list
final_dict['count'] = total_count
final_dict['pass'] = True
final_dict['message'] = 'Seems like things are working ...'
else:
final_dict['data'] = []
final_dict['pass'] = True
log.e(tag, 'No Role in Role Table')
final_dict['message'] = 'Seems like things are not working ...'
final_dict['count'] = 0
opJson = json.dumps(final_dict)
self.request.write(opJson)
tornado.ioloop.IOLoop.instance().add_callback(self.callback)
def run(self):
log = Logger('RoleGetHandlerThread')
TAG = 'GET'
page = self.request.get_argument('offset', None)
count = self.request.get_argument('count', None)
sort_by = self.request.get_argument('sort_by', None)
print page
print count
print sort_by
if self.data and self.company_id:
role = RoleDBHelper()
roles = role.get_role(str(self.data), company_id=self.company_id)
outer_dict = {}
outer_array = []
if roles is None:
log.e(TAG, 'No role found corresponding to the role id ' + \
str(self.data))
opJson = json.dumps({'pass': False, 'message':'No role found \
corresponding to the role id ' + str(self.data)})
self.request.write(opJson)
tornado.ioloop.IOLoop.instance().add_callback(self.callback)
else:
self.role_name = roles[c.ROLE_TABLE_NAME]
user = UserDBHelper()
filter_dict = {
c.USER_TABLE_ROLE: str(roles.get('id'))
}
user_list = user.get_users_with_pages(filter_dict, int(page), int(count), str(sort_by))
inner_array = []
if user_list is not None:
for users in user_list:
inner_dict = {}
self.user_id = str(users[c.USER_TABLE_ID])
self.user_name = str(users[c.USER_TABLE_NAME])
self.user_email = str(users[c.USER_TABLE_EMAIL])
self.user_team_id = str(users[c.USER_TABLE_TEAM])
## Find out the team for user
team = TeamDBHelper()
teams = team.get_team(self.user_team_id,
[c.TEAM_TABLE_NAME])
if teams is None:
self.user_team = None
else:
self.user_team = str(teams[c.TEAM_TABLE_NAME])
## Find out the device for user
device = DeviceDBHelper()
device_list = device.get_devices_of_user(self.user_id)
if device_list is None:
self.user_device = None
else:
devices = device_list[0]
self.user_device = str(
devices[c.DEVICE_TABLE_UDID])
self.user_device_os = str(
devices[c.DEVICE_TABLE_OS])
# Find out user violations
violation = ViolationsDBHelper()
violation_count = violation.get_violation_count(
str(devices.get('id')))
inner_dict = {
'user_id': self.user_id,
'user_name': self.user_name,
'user_role': self.role_name,
'user_team': self.user_team,
'user_device': self.user_device,
'user_violations': violation_count,
'user_device_os': self.user_device_os
}
inner_array.append(inner_dict)
outer_dict['name'] = self.role_name
outer_dict['users'] = inner_array
outer_dict['roleID'] = self.data
outer_array.append(outer_dict)
self.final_dict['pass'] = True
self.final_dict['roles'] = outer_array
opJson = json.dumps(self.final_dict)
self.request.write(opJson)
tornado.ioloop.IOLoop.instance().add_callback(self.callback)
else:
log.e(TAG, 'UnAuthorized Access for Roles')
self.request.set_status(401)
tornado.ioloop.IOLoop.instance().add_callback(self.callback)
def run(self):
#Return All the users in the User table
log = Logger('RolesGetHandler')
tag = 'run'
final_dict = {}
role = RoleDBHelper()
team = TeamDBHelper()
user = UserDBHelper()
role_name = self.request.get_argument('role', None)
offset = self.request.get_argument('offset', None)
count = self.request.get_argument('count', None)
name_query = self.request.get_argument('name', None)
team_query = self.request.get_argument('team', None)
device_query = self.request.get_argument('device_id', None)
sort_by = self.request.get_argument('sort_by', True)#Intentionally done
sort_order = self.request.get_argument('sort', None)
filter_key = str(self.request.get_argument('filter_key', None))
filter_value = self.request.get_argument('filter_value', None)
print role_name
print "\nUI sorting order \n", sort_order
## Find all the roles in the Roles Table
# outer_array = []
if name_query:
query = name_query
query_type = 'name'
elif team_query:
query = team_query
query_type = 'team'
elif device_query:
query = device_query
query_type = 'device'
else:
query = None
query_type = None
if filter_key == 'team':
team_name = filter_value
team_id = team.get_team_by_name(str(team_name), self.company_id)
else:
team_name = None
team_id=None
print "printing team id here ..", team_id
if role_name:
role_id = role.get_role_by_name(str(role_name), self.company_id)
else:
roles_list = role.get_roles(self.company_id, [c.ROLE_TABLE_NAME,
c.ROLE_TABLE_ID])
role_id = None
if role_id:
result_list, total_count = user.get_users_for_role(
role_name=role_name, role_id=role_id,
team_name=team_name, team_id=team_id, offset=offset,
count=count, sort_by=sort_by, query=query,
query_type=query_type, sort_order=sort_order)
elif roles_list:
for _role in roles_list:
_role['role_type'] = _role.get('name')
_role['role_name'] = _role.get('name')
_role['role_id'] = _role.get('id')
del _role['name']
del _role['id']
result_list = roles_list
total_count = len(result_list)
else:
result_list = None
if result_list:
final_dict['data'] = result_list
final_dict['count'] = total_count
final_dict['pass'] = True
final_dict['message'] = 'Seems like things are working ...'
else:
final_dict['data'] = []
final_dict['pass'] = True
log.e(tag,'No Role in Role Table')
final_dict['message'] = 'Seems like things are not working ...'
final_dict['count'] = 0
opJson = json.dumps(final_dict)
self.request.write(opJson)
tornado.ioloop.IOLoop.instance().add_callback(self.callback)
def run(self):
# Return All the users in the User table
log = Logger('TeamsGetHandlerThread')
tag = 'run'
print 'In TeamsGetHandlerThread'
final_dict = {}
role = RoleDBHelper()
team = TeamDBHelper()
user = UserDBHelper()
team_name = self.request.get_argument('team', None)
offset = self.request.get_argument('offset', None)
count = self.request.get_argument('count', None)
sort_by = self.request.get_argument('sort_by',
True) # Intentionally done
name_query = self.request.get_argument('name', None)
role_query = self.request.get_argument('role', None)
device_query = self.request.get_argument('device_id', None)
sort_order = self.request.get_argument('sort', None)
filter_key = str(self.request.get_argument('filter_key', None))
filter_value = self.request.get_argument('filter_value', None)
# Find all the roles in the Roles Table
if filter_key == 'role':
role_name = filter_value
role_id = role.get_role_by_name(str(role_name), self.company_id)
else:
role_name = None
role_id = None
if name_query:
query = name_query
query_type = 'name'
elif role_query:
query = role_query
query_type = 'role'
elif device_query:
query = device_query
query_type = 'device'
else:
query = None
query_type = None
if team_name:
team_id = team.get_team_by_name(str(team_name), self.company_id)
else:
team_id = None
teams_list = team.get_teams(self.company_id,
[c.TEAM_TABLE_NAME, c.TEAM_TABLE_ID])
if team_id:
result_list, total_count = user.get_users_for_team(
team_name=team_name,
team_id=team_id,
role_name=role_name,
role_id=role_id,
offset=offset,
count=count,
sort_by=sort_by,
query=query,
query_type=query_type,
sort_order=sort_order)
elif teams_list:
for _team in teams_list:
_team['team_type'] = _team.get('name')
_team['team_name'] = _team.get('name')
_team['team_id'] = _team.get('id')
del _team['name']
del _team['id']
result_list = teams_list
total_count = len(result_list)
else:
result_list = None
if result_list:
print "\n total", total_count
final_dict['data'] = result_list
final_dict['count'] = total_count
final_dict['pass'] = True
final_dict['message'] = 'Success ...'
else:
final_dict['pass'] = True
final_dict['data'] = []
log.e(tag, 'No Team in Team Table')
final_dict['message'] = 'Failed ...'
final_dict['count'] = 0
opJson = json.dumps(final_dict)
self.request.write(opJson)
tornado.ioloop.IOLoop.instance().add_callback(self.callback)
def run(self):
log = Logger('RolesPutHandlerThread')
TAG = 'PUT'
input_dict = json.loads(self.request.request.body)
self.role_id = input_dict.get('role_id')
if self.role_id is None:
log.e(TAG, 'No role_id sent aborting update')
opJson = json.dumps({'pass': False, 'message': 'No role_id sent'})
self.request.set_status(401)
self.request.write(opJson)
tornado.ioloop.IOLoop.instance().add_callback(self.callback)
else:
if 'name' in input_dict:
print 'Name found'
self.role_name = input_dict['name']
if self.role_name is None:
log.e(TAG, 'No role_name sent aborting update')
opJson = json.dumps(
{'pass': False, 'message': 'No role_name sent'})
self.request.write(opJson)
tornado.ioloop.IOLoop.instance().add_callback(
self.callback)
else:
role = RoleDBHelper()
update_dict = {
c.ROLE_TABLE_NAME: self.role_name
}
result = role.update_role(str(self.role_id),
self.company_id, update_dict)
if not result:
log.e(TAG, 'Not able to update the role_name at DB')
opJson = json.dumps(
{'pass': False,
'message': 'Not able to update the \
role_name at DB'})
self.request.write(opJson)
tornado.ioloop.IOLoop.instance().add_callback(
self.callback)
else:
log.i(TAG, 'role_name Updated successfully')
opJson = json.dumps(
{'pass': True,
'message': 'role_name updated for role.'})
self.request.write(opJson)
tornado.ioloop.IOLoop.instance().add_callback(
self.callback)
elif 'policy_id' in input_dict:
print 'Policy found'
self.role_policy_id = input_dict['policy_id']
if self.role_policy_id is None:
log.e(TAG, 'No Policy ID sent aborting update')
opJson = json.dumps(
{'pass': False, 'message': 'No policy_id sent'})
self.request.write(opJson)
tornado.ioloop.IOLoop.instance().add_callback(
self.callback)
else:
role = RoleDBHelper()
update_dict = {
c.ROLE_TABLE_POLICY: self.role_policy_id
}
result = role.update_role(str(self.role_id),
self.company_id, update_dict)
if not result:
log.e(TAG, 'Not able to update the policy_id at DB')
opJson = json.dumps(
{'pass': False,
'message': 'Not able to update the \
policy_id at DB'})
self.request.write(opJson)
tornado.ioloop.IOLoop.instance().add_callback(
self.callback)
else:
log.i(TAG, 'policy_id Updated successfully')
opJson = json.dumps(
{'pass': True,
'message': 'policy_id updated for role.'})
self.request.write(opJson)
tornado.ioloop.IOLoop.instance().add_callback(
self.callback)
def parse(self, excel_path, company, company_name=None, callback=None,
user_added_callback=None):
COMPANY = company
company_name = company_name
book = open_workbook(filename=excel_path)
sheet = book.sheet_by_index(0)
users = list()
for row_index in range(sheet.nrows):
user = dict()
for col_index in range(sheet.ncols):
cell = sheet.cell(row_index, col_index).value
if col_index == 0:
user[C.USER_TABLE_EMAIL] = cell
elif col_index == 1:
user[C.USER_TABLE_NAME] = cell
elif col_index == 2:
user[C.USER_TABLE_TEAM] = cell
elif col_index == 3:
user[C.USER_TABLE_ROLE] = cell
users.append(user)
roles_list = list()
teams_list = list()
for user in users:
if C.USER_TABLE_ROLE in user:
roles_list.append(user[C.USER_TABLE_ROLE])
if C.USER_TABLE_TEAM in user:
teams_list.append(user[C.USER_TABLE_TEAM])
# get a set to get the unique elements.
roles = dict()
teams = dict()
roles_helper = RoleDBHelper()
teams_helper = TeamDBHelper()
users_helper = UserDBHelper()
enrollment_helper = EnrollmentDBHelper()
for role_name in set(roles_list):
role = dict()
role[C.ROLE_TABLE_COMPANY] = COMPANY
role[C.ROLE_TABLE_NAME] = role_name
roles[role_name] = roles_helper.add_role(role)
for team_name in set(teams_list):
team = dict()
team[C.TEAM_TABLE_COMPANY] = COMPANY
team[C.TEAM_TABLE_NAME] = team_name
team[C.TEAM_TABLE_DELETED] = False
teams[team_name] = teams_helper.add_team(team)
# Now we have id for all teams and roles... Insert the users now.
for user in users:
user_obj = dict()
user_obj[C.USER_TABLE_COMPANY] = COMPANY
user_obj[C.USER_TABLE_EMAIL] = user[C.USER_TABLE_EMAIL]
user_obj[C.USER_TABLE_NAME] = user[C.USER_TABLE_NAME]
user_obj[C.USER_TABLE_ROLE] = roles[user[C.USER_TABLE_ROLE]]
user_obj[C.USER_TABLE_TEAM] = teams[user[C.USER_TABLE_TEAM]]
user['id'] = users_helper.add_user_if_not_exists(user_obj)[0]
if user_added_callback is not None:
user_added_callback(user_obj)
for user in users:
enrollment = dict()
enrollment[C.ENROLLMENT_TABLE_USER] = user['id']
enrollment[C.ENROLLMENT_TABLE_PASSWORD] = generate_password()
enrollment[C.ENROLLMENT_TABLE_IS_ENROLLED] = False
user['passcode'] = enrollment[C.ENROLLMENT_TABLE_PASSWORD]
print 'adding enrollment ', enrollment
user['enrollment_id'] = enrollment_helper.add_enrollment(
enrollment)
user['company_name'] = company_name
if callback is not None:
callback(users)
return users
def run(self):
# Return All the users in the User table
log = Logger('UsersGetHandlerThreadWithPage')
tag = 'run'
company_id = self.company_id
final_dict = {}
user = UserDBHelper()
team = TeamDBHelper()
role = RoleDBHelper()
violation = ViolationsDBHelper()
device = DeviceDBHelper()
offset = self.request.get_argument('offset', None)
count = self.request.get_argument('count', None)
name_query = self.request.get_argument('name', None)
team_query = self.request.get_argument('team', None)
role_query = self.request.get_argument('role', None)
device_query = self.request.get_argument('device_id', None)
sort_by = self.request.get_argument('sort_by',
True) # Intentionally done
sort_order = self.request.get_argument('sort', None)
filter_key = self.request.get_argument('filter_key', None)
filter_value = self.request.get_argument('filter_value', None)
if filter_key == 'role':
role_name = filter_value
role_id = role.get_role_by_name(str(role_name), company_id)
else:
role_name = None
role_id = None
if filter_key == 'team':
team_name = filter_value
team_id = team.get_team_by_name(str(team_name), company_id)
else:
team_name = None
team_id = None
if filter_key == 'os':
os_mapper = {'Android': 'samsung', 'iOS': 'ios'}
os_type = os_mapper.get(str(filter_value))
else:
os_type = None
if name_query:
query = name_query
query_type = 'name'
elif role_query:
query = role_query
query_type = 'role'
elif device_query:
query = device_query
query_type = 'device'
elif team_query:
query = team_query
query_type = 'team'
else:
query = None
query_type = None
if offset:
result_list, total_count = user.get_users_for_user(
company_id=company_id,
offset=offset,
count=count,
role_id=role_id,
team_id=team_id,
query=query,
query_type=query_type,
os_type=os_type,
sort_by=sort_by,
sort_order=sort_order,
filter_key=filter_key,
filter_value=filter_value)
else:
result_list, total_count = user.get_users(
{c.USER_TABLE_COMPANY: company_id})
if result_list:
for user_dict in result_list:
device_deleted = False
device_info = device.get_device_with_udid(
user_dict.get('user_device'))
if not device_info:
device_info = device.get_device_with_udid(
user_dict.get('user_device'), status=True)
device_deleted = True
if device_info:
device_id = device_info[0].get(c.DEVICE_TABLE_ID)
else:
device_id = None
if device_deleted:
user_dict['user_device'] = None
user_dict['user_device_os'] = None
if device_id:
print "printing \n device id", device_id
violation_count = violation.get_violation_count(
company_id=company_id, device_id=str(device_id))
else:
violation_count = 0
user_dict['user_violation'] = violation_count
final_dict['data'] = result_list
final_dict['count'] = total_count
final_dict['pass'] = True
final_dict['message'] = 'Seems like things are working ...'
else:
final_dict['pass'] = True
log.e(tag, 'No User in User Table')
final_dict['message'] = 'Seems like things are not working ...'
final_dict['count'] = 0
# add all the data into dictionary and create output json
opJson = json.dumps(final_dict)
self.request.write(opJson)
tornado.ioloop.IOLoop.instance().add_callback(self.callback)