def ModifyDepartment(dep_id): form = DepartmentForm() if form.validate_on_submit(): department_name = form.department_name.data.strip() if dep_id is None: department = Department.query.filter( Department.name == department_name).first() if form.create.data is True: if department is not None: flash('Отдел с таким названием уже существует.') else: department = Department(name=department_name) db.session.add(department) db.session.commit() flash('Отдел успешно создан.') elif form.delete.data is True: if department is None: flash('Такого отдела не существует.') else: db.session.delete(department) db.session.commit() flash('Отдел успешно удалён.') else: department = Department.query.filter( Department.id == dep_id).first() if department is None: flash('Такого отдела не существует.') else: department.name = department_name db.session.commit() flash('Отдел успешно изменён.') else: for error in form.department_name.errors: flash(error) return redirect(url_for('main.ShowUsers'))
def import_data(request): rb = xlrd.open_workbook('D:\pydata\prepod_goda.xlsx') sheet = rb.sheet_by_index(0) ins = Institute.objects.all()[0] n = sheet.nrows s = 'sdads' for rownum in range(sheet.nrows): row = sheet.row_values(rownum) d = row[0].strip() t = row[1].strip() g = row[2].strip() dep = Department() tech = Teacher() group = Group() db_d = Department.objects.filter(name=d) if (db_d.count() == 0): dep.name = d dep.institute = ins dep.save() else: dep = db_d[0] db_t = Teacher.objects.filter(last_name=t) if (db_t.count() == 0): tech.last_name = t tech.department = dep tech.save() else: tech = db_t[0] db_g = Group.objects.filter(name=g) if (db_g.count() == 0): group.name = g group.department = dep group.save() else: group = db_g[0] group.teachers.add(tech) group.save() return HttpResponse("OK " + s)