def test_group(request):
title = "Add New Group"
group_add_form = groupAddForm(request.POST or None)
cmdformset = CommandAliasFormSet(request.POST or None,
prefix="AddCmdAliasForm")
context = {
'title': title,
'form': group_add_form,
'cmdformset': cmdformset
}
if group_add_form.is_valid():
current_user = '******'
if request.user.is_authenticated():
current_user = request.user.username
group_obj = GroupModel(
gname=group_add_form.cleaned_data['gname'],
display_name=group_add_form.cleaned_data['display_name'],
mail_pdl=group_add_form.cleaned_data['mail_pdl'],
update_by=current_user)
group_obj.save()
for _each_selection in group_add_form.cleaned_data['cmd_aliases']:
cmd_alias_model = CmdAliasModel.objects.get(
cmd_alias=_each_selection)
cmd_alias_model.update_by = current_user
cmd_alias_model.update_date = datetime.now()
cmd_alias_model.gid = group_obj
cmd_alias_model.save()
message = "%s added successfully" % (
group_add_form.cleaned_data['display_name'])
return render_to_response('confirm.html', {'message': message})
return render(request, "add_group.html", context)
from models import GroupModel
if __name__ == '__main__':
gr_manager = GroupModel()
while True:
try:
choise = int(input("""
---------------------------------------------------
Выберите действие:
1. Вывод списка всех групп.
2. Добавление новой группы в указанный факультет.
3. Удаление группы.
4. Завершение работы.
---------------------------------------------------
Ваш выбор: """))
except ValueError:
print('Вводите только числа!')
continue
if choise == 1:
print("Список всех груп: \n", gr_manager.get_all_groups())
elif choise == 2:
gr_manager.add_group(input('Введите название групы: '), input("Введите название факультета: "))
elif choise == 3:
gr_manager.dell_group(input('Введите название группы: '))
elif choise == 4:
break
else:
print("Неправильный вариант действия!")
