def add_permission():
"""添加权限"""
if request.method == 'GET':
pers = Permission.query.all()
return render_template('addpermission.html', pers=pers)
if request.method == 'POST':
p_name = request.form.get('p_name')
p_er = request.form.get('p_er')
p_name_test_repeat = Permission.query.filter(
Permission.p_name == p_name).first()
if p_name_test_repeat:
msg = '*权限名称重复'
return render_template('addpermission.html', msg=msg)
p_er_test_repeat = Permission.query.filter(
Permission.p_er == p_er).first()
if p_er_test_repeat:
msg1 = '*权限简写名重复'
return render_template('addpermission.html', msg1=msg1)
permission = Permission(p_name=p_name, p_er=p_er)
permission.save()
return redirect(url_for('user.permission_list'))
def add_permission(request, groupname):
login = login_verify(request, None, True)
if login:
return login
if request.method == 'POST':
value = request.POST.get('value', '')
description = request.POST.get('description', '')
group = Group.objects.get(name=groupname)
permission = Permission(value=value, description=description)
permission.save()
group.permissions.add(permission)
group.save()
return HttpResponseRedirect('/view_group/' + groupname + '/')
else:
form = AddPermissionForm()
context = {'form': form, 'groupname': groupname}
return render(request, 'app/add_permission.html', context)
def test_saving_and_retrieving_models(self):
company = Company()
company.name = 'PSL'
company.email = '*****@*****.**'
company.telephone = 7777777
company.save()
first_room = Room()
first_room.name = 'Peter Santamaria'
first_room.company = company
first_room.security_level = 2
first_room.description = 'Conference room'
first_room.id = 'r00m1'
first_room.save()
second_room = Room()
second_room.name = '201'
second_room.company = company
second_room.security_level = 1
second_room.description = 'Class room'
second_room.id = 'r00m2'
second_room.save()
third_room = Room()
third_room.name = 'rest'
third_room.company = company
third_room.security_level = 1
third_room.description = 'Rest room'
third_room.id = 'r00m3'
third_room.save()
event = Event()
event.name = 'My event'
event.company = company
event.start_date = datetime(2017, 3, 16)
event.end_date = datetime(2017, 3, 21)
event.event_id = '3v3nt'
event.rooms.add(first_room)
event.rooms.add(second_room)
event.rooms.add(third_room)
event.save()
enduser = EndUser()
enduser.id = 'u53r'
enduser.name = 'Sebastian'
enduser.last_name = 'Villegas'
enduser.email = '*****@*****.**'
enduser.save()
first_permission = Permission()
first_permission.user_id = enduser
first_permission.event = event
first_permission.id = '93rm151'
first_permission.save()
second_permission = Permission()
second_permission.user_id = enduser
second_permission.event = event
second_permission.id = '93rm152'
second_permission.save()
saved_company = Company.objects.first()
self.assertEqual(saved_company, company)
saved_rooms = Room.objects.all()
self.assertEqual(saved_rooms.count(), 3)
self.assertEqual(saved_rooms[0], first_room)
self.assertEqual(saved_rooms[1], second_room)
self.assertEqual(saved_rooms[2], third_room)
saved_event = Event.objects.first()
self.assertEqual(saved_event, event)
saved_enduser = EndUser.objects.first()
self.assertEqual(saved_enduser, enduser)
saved_permissions = Permission.objects.all()
self.assertEqual(saved_permissions.count(), 2)
self.assertEqual(saved_permissions[0], first_permission)
self.assertEqual(saved_permissions[1], second_permission)
try:
group = Group.objects.get(name='ICanHaz')
except:
group = Group.objects.create(name='ICanHaz')
group.save()
permission_created = False
if group.permissions:
for permission in group.permissions.all():
if permission.value == "ICanHaz.access":
permission_created = True
break
if not permission_created:
p = Permission(value="ICanHaz.access", description="Access to the permission system")
p.save()
group.permissions.add(p)
group.save()
names = []
uids = []
for line in codecs.open('admins_to_add.txt', 'r', 'utf-8'):
names.append(" ".join(line.split()[:-1]))
uids.append(line.split()[-1])
for g, i, j, index in [[v,v,v,v] for v in range(0,len(names))]:
try:
i = User(name=names[index], uid=uids[index])
i.save()
year = str(datetime.now().year)
