tmis 0.1.3

data/lib/tmis/interface/models/cabinet_table_model.rb

@@ -0,0 +1,133 @@
+# encoding: UTF-8
+#~~~~~~~~~~~~~~~~~~~~~~~~~~
+require_relative '../../engine/models/cabinet'
+#~~~~~~~~~~~~~~~~~~~~~~~~~~
+class CabinetTableModel < Qt::AbstractTableModel
+
+  def initialize(cabinets, parent)
+    super()
+    @cabinets = cabinets
+    @view = parent
+    @view.setItemDelegateForColumn(1, RadioButtonDelegate.new(self))
+  end
+
+  def refresh
+    @cabinets = Cabinet.all
+    emit layoutChanged()
+  end
+
+  def rowCount(parent)
+    @cabinets.size
+  end
+
+  def columnCount(parent)
+    2
+  end
+
+  def data(index, role = Qt::DisplayRole)
+    cabinet = @cabinets[index.row]
+    default = Qt::Variant.new
+    case role
+    when Qt::DisplayRole
+      case index.column
+      when 0
+        cabinet.title.to_v
+      when 1
+        cabinet.with_computers.to_s.to_v
+      else
+        raise "invalid column #{index.column}"
+      end
+    when Qt::EditRole
+      case index.column
+      when 0
+        cabinet.title.to_v
+      when 1
+        cabinet.with_computers.to_v
+      else
+        raise "invalid column #{index.column}"
+      end
+    else
+      default
+    end
+  end
+
+  def headerData(section, orientation, role = Qt::DisplayRole)
+    invalid = Qt::Variant.new
+    return invalid unless role == Qt::DisplayRole
+    v = case orientation
+        when Qt::Horizontal
+          %w(Название Компьютерный)[section]
+        else
+          ''
+        end
+    Qt::Variant.new(v)
+  end
+
+  def flags(index)
+    Qt::ItemIsEditable | super(index)
+  end
+
+  def setData(index, variant, role = Qt::EditRole)
+    if index.valid? and role == Qt::EditRole
+      cabinet = @cabinets[index.row]
+      case index.column
+      when 0
+        cabinet.title = variant.toString.force_encoding('UTF-8')
+      when 1
+        cabinet.with_computers = variant.toBool
+      else
+        raise "invalid column #{index.column}"
+      end
+      cabinet.save
+      emit dataChanged(index, index)
+      true
+    else
+      false
+    end
+  end
+
+  def insert_new
+    @cabinets.prepend(Cabinet.new)
+    emit layoutChanged()
+  end
+
+  def remove_current
+    if @view.currentIndex.valid?
+      cabinet = @cabinets[@view.currentIndex.row]
+      unless cabinet.stub
+        cabinet.try(:destroy)
+        @cabinets.delete_at(@view.currentIndex.row)
+        emit layoutChanged()
+        @view.currentIndex = createIndex(-1, -1)
+      end
+    end
+  end
+
+end
+
+class RadioButtonDelegate < Qt::ItemDelegate
+
+  def initialize(parent)
+    super
+  end
+
+  def createEditor(parent, option, index)
+    Qt::CheckBox.new(parent)
+  end
+
+  def setEditorData(editor, index)
+    value = index.data.toBool #index.model.data(index, Qt::EditRole)
+    button = editor
+    button.checked = value # button.setValue(value)
+  end
+
+  def setModelData(editor, model, index)
+    button = editor
+    value = button.isChecked # button.value
+    model.setData(index, value.to_v, Qt::EditRole)
+  end
+
+  def updateEditorGeometry(editor, option, index)
+    editor.setGeometry(option.rect)
+  end
+end

data/lib/tmis/interface/models/course_table_model.rb

@@ -0,0 +1,87 @@
+# encoding: UTF-8
+#~~~~~~~~~~~~~~~~~~~~~~~~~~
+require_relative '../../engine/models/course'
+#~~~~~~~~~~~~~~~~~~~~~~~~~~
+class CourseTableModel < Qt::AbstractTableModel
+
+  def initialize(courses, parent)
+    super()
+    @courses = courses
+    @view = parent
+  end
+
+  def refresh
+    @courses = Course.all
+    emit layoutChanged()
+  end
+
+  def rowCount(parent)
+    @courses.size
+  end
+
+  def columnCount(parent)
+    1
+  end
+
+  def data(index, role = Qt::DisplayRole)
+    invalid = Qt::Variant.new
+    return invalid unless role == Qt::DisplayRole or role == Qt::EditRole
+    course = @courses[index.row]
+    return invalid if course.nil?
+    v = case index.column
+        when 0
+          course.number
+        else
+          raise "invalid column #{index.column}"
+        end || ''
+    Qt::Variant.new(v)
+  end
+
+  def headerData(section, orientation, role = Qt::DisplayRole)
+    invalid = Qt::Variant.new
+    return invalid unless role == Qt::DisplayRole
+    v = case orientation
+        when Qt::Horizontal
+          %w(Номер)[section]
+        else
+          ''
+        end
+    Qt::Variant.new(v)
+  end
+
+  def flags(index)
+    Qt::ItemIsEditable | super(index)
+  end
+
+  def setData(index, variant, role = Qt::EditRole)
+    if index.valid? and role == Qt::EditRole
+      course = @courses[index.row]
+      case index.column
+      when 0
+        course.number = variant.toInt
+      else
+        raise "invalid column #{index.column}"
+      end
+      course.save
+      emit dataChanged(index, index)
+      true
+    else
+      false
+    end
+  end
+
+  def insert_new
+    @courses.prepend(Course.new)
+    emit layoutChanged()
+  end
+
+  def remove_current
+    if @view.currentIndex.valid?
+      @courses[@view.currentIndex.row].try(:destroy)
+      @courses.delete_at(@view.currentIndex.row)
+      emit layoutChanged()
+      @view.currentIndex = createIndex(-1, -1)
+    end
+  end
+
+end

data/lib/tmis/interface/models/group_table_model.rb

@@ -0,0 +1,190 @@
+# encoding: UTF-8
+#~~~~~~~~~~~~~~~~~~~~~~~~~~
+require_relative '../../engine/models/group'
+#~~~~~~~~~~~~~~~~~~~~~~~~~~
+class GroupTableModel < Qt::AbstractTableModel
+
+  def initialize(groups, parent)
+    super()
+    @groups = groups
+    @groups.size
+    @view = parent
+    @SpecialityComboBoxDelegate = SpecialityComboBoxDelegate.new(self)
+    @CourseComboBoxDelegate = CourseComboBoxDelegate.new(self)
+    @view.setItemDelegateForColumn(1, @SpecialityComboBoxDelegate)
+    @view.setItemDelegateForColumn(2, @CourseComboBoxDelegate)
+  end
+
+  def refresh
+    @groups = Group.all
+    @SpecialityComboBoxDelegate.setup
+    @CourseComboBoxDelegate.setup
+    emit layoutChanged()
+  end
+
+  def rowCount(parent)
+    @groups.size
+  end
+
+  def columnCount(parent)
+    4
+  end
+
+  def data(index, role = Qt::DisplayRole)
+    group = @groups[index.row]
+    default = Qt::Variant.new
+    case role
+    when Qt::DisplayRole
+      case index.column
+      when 0
+        group.title
+      when 1
+        group.speciality.try(:title)
+      when 2
+        group.course.try(:number)
+      when 3
+        group.emails.map(&:email).join(', ')
+      else
+        raise "invalid column #{index.column}"
+      end.try(:to_v) || default
+    when Qt::EditRole
+      case index.column
+      when 0
+        group.title
+      when 1
+        group.speciality_id
+      when 2
+        group.course_id
+      when 3
+        group.emails.map(&:email).join(', ')
+      else
+        raise "invalid column #{index.column}"
+      end.try(:to_v) || default
+    else
+      default
+    end
+  end
+
+  def headerData(section, orientation, role = Qt::DisplayRole)
+    invalid = Qt::Variant.new
+    return invalid unless role == Qt::DisplayRole
+    v = case orientation
+        when Qt::Horizontal
+          %w(Название Специальность Курс Email)[section]
+        else
+          ''
+        end
+    Qt::Variant.new(v)
+  end
+
+  def flags(index)
+    Qt::ItemIsEditable | super(index)
+  end
+
+  def setData(index, variant, role = Qt::EditRole)
+    if index.valid? and role == Qt::EditRole
+      group = @groups[index.row]
+      case index.column
+      when 0
+        group.title = variant.toString.force_encoding('UTF-8')
+      when 1
+        group.speciality_id = variant.toInt
+      when 2
+        group.course_id = variant.toInt
+      when 3
+        emails = variant.toString.force_encoding('UTF-8').split(/,\s*/)
+        group.emails.destroy_all
+        emails.each do |email|
+          group.emails.create(email: email)
+        end
+      else
+        raise "invalid column #{index.column}"
+      end
+      group.save
+      p group
+      emit dataChanged(index, index)
+      true
+    else
+      false
+    end
+  end
+
+  def insert_new
+    new_group = Group.new
+    @groups.prepend(new_group)
+    (1..2).map{ |i| Subgroup.create(group: new_group, number: i) }
+    emit layoutChanged()
+  end
+
+  def remove_current
+    if @view.currentIndex.valid?
+      @groups[@view.currentIndex.row].try(:destroy)
+      @groups.delete_at(@view.currentIndex.row)
+      emit layoutChanged()
+      @view.currentIndex = createIndex(-1, -1)
+    end
+  end
+
+end
+
+class SpecialityComboBoxDelegate < Qt::ItemDelegate
+  def initialize(parent)
+    super
+    setup
+  end
+
+  def setup
+    @specialities = Speciality.all.sort_by(&:title)
+  end
+
+  def createEditor(parent, option, index)
+    editor = Qt::ComboBox.new(parent)
+    @specialities.each{ |x| editor.addItem(x.title.to_s, x.id.to_v) }
+    editor
+  end
+
+  def setEditorData(editor, index)
+    value = index.data
+    editor.setCurrentIndex(editor.findData(value))
+  end
+
+  def setModelData(editor, model, index)
+    value = editor.itemData(editor.currentIndex)
+    model.setData(index, value, Qt::EditRole)
+  end
+
+  def updateEditorGeometry(editor, option, index)
+    editor.setGeometry(option.rect)
+  end
+end
+
+class CourseComboBoxDelegate < Qt::ItemDelegate
+  def initialize(parent)
+    super
+    setup
+  end
+
+  def setup
+    @courses = Course.all.sort_by(&:number)
+  end
+
+  def createEditor(parent, option, index)
+    editor = Qt::ComboBox.new(parent)
+    @courses.each{ |x| editor.addItem(x.number.to_s, x.id.to_v) }
+    editor
+  end
+
+  def setEditorData(editor, index)
+    value = index.data
+    editor.setCurrentIndex(editor.findData(value))
+  end
+
+  def setModelData(editor, model, index)
+    value = editor.itemData(editor.currentIndex)
+    model.setData(index, value, Qt::EditRole)
+  end
+
+  def updateEditorGeometry(editor, option, index)
+    editor.setGeometry(option.rect)
+  end
+end

data/lib/tmis/interface/models/lecturer_table_model.rb

@@ -0,0 +1,111 @@
+# encoding: UTF-8
+#~~~~~~~~~~~~~~~~~~~~~~~~~~
+require_relative '../../engine/models/lecturer'
+require_relative '../../engine/models/email'
+#~~~~~~~~~~~~~~~~~~~~~~~~~~
+class LecturerTableModel < Qt::AbstractTableModel
+
+  def initialize(lecturers, parent)
+    super()
+    @lecturers = lecturers
+    @view = parent
+  end
+
+  def refresh
+    @lecturers = Lecturer.all
+    emit layoutChanged()
+  end
+
+  def rowCount(parent)
+    @lecturers.size
+  end
+
+  def columnCount(parent)
+    5
+  end
+
+  def data(index, role = Qt::DisplayRole)
+    invalid = Qt::Variant.new
+    return invalid unless role == Qt::DisplayRole or role == Qt::EditRole
+    lecturer = @lecturers[index.row]
+    return invalid if lecturer.nil?
+    v = case index.column
+        when 0
+          lecturer.surname
+        when 1
+          lecturer.name
+        when 2
+          lecturer.patronymic
+        when 3
+          lecturer.emails.map(&:email).join(', ')
+        when 4
+          lecturer.preferred_days
+        else
+          raise "invalid column #{index.column}"
+        end || ''
+    Qt::Variant.new(v)
+  end
+
+  def headerData(section, orientation, role = Qt::DisplayRole)
+    invalid = Qt::Variant.new
+    return invalid unless role == Qt::DisplayRole
+    v = case orientation
+        when Qt::Horizontal
+          ['Фамилия', 'Имя', 'Отчество', 'Email', 'Предпочитаемые дни'][section]
+        else
+          ''
+        end
+    Qt::Variant.new(v)
+  end
+
+  def flags(index)
+    Qt::ItemIsEditable | super(index)
+  end
+
+  def setData(index, variant, role = Qt::EditRole)
+    if index.valid? and role == Qt::EditRole
+      lecturer = @lecturers[index.row]
+      case index.column
+      when 0
+        lecturer.surname = variant.toString.force_encoding('UTF-8')
+      when 1
+        lecturer.name = variant.toString.force_encoding('UTF-8')
+      when 2
+        lecturer.patronymic = variant.toString.force_encoding('UTF-8')
+      when 3
+        emails = variant.toString.force_encoding('UTF-8').split(/,\s*/)
+        lecturer.emails.destroy_all
+        emails.each do |email|
+          lecturer.emails.create(email: email)
+        end
+      when 4
+        lecturer.preferred_days = variant.toString.force_encoding('UTF-8').split(/,\s*/).select{|x| x[/^[1-7]$/]}.uniq.join(', ')
+      else
+        raise "invalid column #{index.column}"
+      end
+      lecturer.save
+      emit dataChanged(index, index)
+      true
+    else
+      false
+    end
+  end
+
+  def insert_new
+    @lecturers.prepend(Lecturer.new)
+    emit layoutChanged()
+  end
+
+  def remove_current
+    if @view.currentIndex.valid?
+      lecturer = @lecturers[@view.currentIndex.row]
+      unless lecturer.stub
+        lecturer.try(:destroy)
+        @lecturers.delete_at(@view.currentIndex.row)
+        emit layoutChanged()
+        @view.currentIndex = createIndex(-1, -1)
+      end
+    end
+  end
+
+end

data/lib/tmis/interface/models/semester_table_model.rb

@@ -0,0 +1,137 @@
+# encoding: UTF-8
+#~~~~~~~~~~~~~~~~~~~~~~~~~~
+require_relative '../../engine/models/semester'
+#~~~~~~~~~~~~~~~~~~~~~~~~~~
+class SemesterTableModel < Qt::AbstractTableModel
+
+  def initialize(semesters, parent)
+    super()
+    @semesters = semesters
+    @view = parent
+    @CourseComboBoxDelegate = CourseComboBoxDelegate.new(self)
+    @view.setItemDelegateForColumn(1, @CourseComboBoxDelegate)
+  end
+
+  def refresh
+    @semesters = Semester.all
+    @CourseComboBoxDelegate.setup
+    emit layoutChanged()
+  end
+
+  def rowCount(parent)
+    @semesters.size
+  end
+
+  def columnCount(parent)
+    2
+  end
+
+  def data(index, role = Qt::DisplayRole)
+    semester = @semesters[index.row]
+    default = Qt::Variant.new
+    case role
+    when Qt::DisplayRole
+      case index.column
+      when 0
+        semester.title
+      when 1
+        semester.course.try(:number)
+      else
+        raise "invalid column #{index.column}"
+      end.try(:to_v) || default
+    when Qt::EditRole
+      case index.column
+      when 0
+        semester.title
+      when 1
+        semester.course_id
+      else
+        raise "invalid column #{index.column}"
+      end.try(:to_v) || default
+    else
+      default
+    end
+  end
+
+  def headerData(section, orientation, role = Qt::DisplayRole)
+    invalid = Qt::Variant.new
+    return invalid unless role == Qt::DisplayRole
+    v = case orientation
+        when Qt::Horizontal
+          %w(Название Курс)[section]
+        else
+          ''
+        end
+    Qt::Variant.new(v)
+  end
+
+  def flags(index)
+    Qt::ItemIsEditable | super(index)
+  end
+
+  def setData(index, variant, role = Qt::EditRole)
+    if index.valid? and role == Qt::EditRole
+      semester = @semesters[index.row]
+      case index.column
+      when 0
+        semester.title = variant.toString.force_encoding('UTF-8')
+      when 1
+        semester.course_id = variant.toInt
+      else
+        raise "invalid column #{index.column}"
+      end
+      semester.save
+      p semester
+      emit dataChanged(index, index)
+      true
+    else
+      false
+    end
+  end
+
+  def insert_new
+    @semesters.prepend(Semester.new)
+    emit layoutChanged()
+  end
+
+  def remove_current
+    if @view.currentIndex.valid?
+      @semesters[@view.currentIndex.row].try(:destroy)
+      @semesters.delete_at(@view.currentIndex.row)
+      emit layoutChanged()
+      @view.currentIndex = createIndex(-1, -1)
+    end
+  end
+
+end
+
+class CourseComboBoxDelegate < Qt::ItemDelegate
+  def initialize(parent)
+    super
+    setup
+  end
+
+  def setup
+    @courses = Course.all.sort_by(&:number)
+  end
+
+  def createEditor(parent, option, index)
+    editor = Qt::ComboBox.new(parent)
+    @courses.each{ |x| editor.addItem(x.number.to_s, x.id.to_v) }
+    editor
+  end
+
+  def setEditorData(editor, index)
+    value = index.data
+    editor.setCurrentIndex(editor.findData(value))
+  end
+
+  def setModelData(editor, model, index)
+    value = editor.itemData(editor.currentIndex)
+    model.setData(index, value, Qt::EditRole)
+  end
+
+  def updateEditorGeometry(editor, option, index)
+    editor.setGeometry(option.rect)
+  end
+end