def insert_user_departments():
users = User.list()
department_A = Department.find(id=60430)
department_B = Department.find(id=60350)
for user in users:
user.departments.append(department_A)
user.departments.append(department_B)
print 'associated users with departments'
def insert_user_departments():
users = User.list()
department_A = Department.find(id=60430)
department_B = Department.find(id=60350)
for user in users:
user.departments.append(department_A)
user.departments.append(department_B)
print 'associated users with departments'
def insert_departments():
reader = csv.reader(open(os.path.join(__location__, 'departments.csv')), delimiter=',', quotechar='"')
for row in reader:
if Department.find(name=row[0]) is None:
department = Department(
name=row[0],
id=row[1]
)
department.save()
parent = Department.find(id=row[2])
if parent is not None:
parent.children.append(department)
print 'inserted departments.'
def put(self, id):
json = request.json['user']
print json
user = UserModel.find(id=id)
user.email = json['email']
user.role = json['role']
user.employee_id = json['employee_id']
user.status = json['status']
user.departments = []
for department in json['departments']:
user.departments.append(Department.find(id=department['id']))
user.save()
departments = []
for department in user.departments:
departments.append({
'id': department.id,
'parent_id': department.parent_id,
'name': department.name,
'description': department.description
})
work_orders = []
for work_order in current_user.work_orders:
work_orders.append({
'id': work_order.id,
'department': work_order.department_id,
'user': work_order.user_id,
'asset': work_order.asset_id
})
model = {
'user': {
'id': user.id,
'user_id': user.id,
'firstname': user.firstname,
'lastname': user.lastname,
'email': user.email,
'username': user.username,
'employee_id': user.employee_id,
'status': user.status,
'role': user.role,
'departments': departments,
'work_orders': work_orders
}
}
return jsonify(model)
def put(self, id):
json = request.json['user']
print json
user = UserModel.find(id=id)
user.email = json['email']
user.role = json['role']
user.employee_id = json['employee_id']
user.status = json['status']
user.departments = []
for department in json['departments']:
user.departments.append(Department.find(id=department['id']))
user.save()
departments = []
for department in user.departments:
departments.append({
'id': department.id,
'parent_id': department.parent_id,
'name': department.name,
'description': department.description
})
work_orders = []
for work_order in current_user.work_orders:
work_orders.append({
'id': work_order.id,
'department': work_order.department_id,
'user': work_order.user_id,
'asset': work_order.asset_id
})
model = {
'user': {
'id': user.id,
'user_id': user.id,
'firstname': user.firstname,
'lastname': user.lastname,
'email': user.email,
'username': user.username,
'employee_id': user.employee_id,
'status': user.status,
'role': user.role,
'departments': departments,
'work_orders': work_orders
}
}
return jsonify(model)
def post(self):
json = request.json['user']
user = UserModel(firstname=json['firstname'],
lastname=json['lastname'],
email=json['email'],
username=json['firstname'] + '.' + json['lastname'],
employee_id=json['employee_id'],
status=json['status'],
password=''.join(
random.choice(string.ascii_uppercase +
string.digits) for _ in range(12)),
role=json['role'])
for department in json['departments']:
user.departments.append(Department.find(id=department['id']))
user.save()
departments = []
for department in user.departments:
departments.append({
'id': department.id,
'parent_id': department.parent_id,
'name': department.name,
'description': department.description
})
model = {
'user': {
'id': user.id,
'user_id': user.id,
'firstname': user.firstname,
'lastname': user.lastname,
'email': user.email,
'username': user.username,
'employee_id': user.employee_id,
'status': user.status,
'role': user.role,
'departments': departments
}
}
print model
return jsonify(model), 201
def post(self):
json = request.json['user']
user = UserModel(
firstname=json['firstname'],
lastname=json['lastname'],
email=json['email'],
username=json['firstname'] + '.' + json['lastname'],
employee_id=json['employee_id'],
status=json['status'],
password=''.join(random.choice(string.ascii_uppercase + string.digits) for _ in range(12)),
role=json['role']
)
for department in json['departments']:
user.departments.append(Department.find(id=department['id']))
user.save()
departments = []
for department in user.departments:
departments.append({
'id': department.id,
'parent_id': department.parent_id,
'name': department.name,
'description': department.description
})
model = {
'user': {
'id': user.id,
'user_id': user.id,
'firstname': user.firstname,
'lastname': user.lastname,
'email': user.email,
'username': user.username,
'employee_id': user.employee_id,
'status': user.status,
'role': user.role,
'departments': departments
}
}
print model
return jsonify(model), 201
def insert_departments():
reader = csv.reader(open(os.path.join(__location__, 'departments.csv')),
delimiter=',',
quotechar='"')
for row in reader:
if Department.find(name=row[0]) is None:
department = Department(name=row[0], id=row[1])
department.save()
parent = Department.find(id=row[2])
if parent is not None:
parent.children.append(department)
print 'inserted departments.'
