tmis 0.1.3

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

@@ -0,0 +1,136 @@
+# encoding: UTF-8
+#~~~~~~~~~~~~~~~~~~~~~~~~~~
+require_relative '../../engine/models/subgroup'
+#~~~~~~~~~~~~~~~~~~~~~~~~~~
+class SubgroupTableModel < Qt::AbstractTableModel
+
+  def initialize(subgroups, parent)
+    super()
+    @subgroups = subgroups
+    @view = parent
+    @GroupComboBoxDelegate = GroupComboBoxDelegate.new(self)
+    @view.setItemDelegateForColumn(1, @GroupComboBoxDelegate)
+  end
+
+  def refresh
+    @subgroups = Subgroup.all
+    @GroupComboBoxDelegate.setup
+    emit layoutChanged()
+  end
+
+  def rowCount(parent)
+    @subgroups.size
+  end
+
+  def columnCount(parent)
+    2
+  end
+
+  def data(index, role = Qt::DisplayRole)
+    subgroup = @subgroups[index.row]
+    default = Qt::Variant.new
+    case role
+    when Qt::DisplayRole
+      case index.column
+      when 0
+        subgroup.number
+      when 1
+        subgroup.group.try(:title)
+      else
+        raise "invalid column #{index.column}"
+      end.try(:to_v) || default
+    when Qt::EditRole
+      case index.column
+      when 0
+        subgroup.number
+      when 1
+        subgroup.group_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
+      subgroup = @subgroups[index.row]
+      case index.column
+      when 0
+        subgroup.number = variant.toInt
+      when 1
+        subgroup.group_id = variant.toInt
+      else
+        raise "invalid column #{index.column}"
+      end
+      subgroup.save
+      emit dataChanged(index, index)
+      true
+    else
+      false
+    end
+  end
+
+  def insert_new
+    @subgroups.prepend(Subgroup.new)
+    emit layoutChanged()
+  end
+
+  def remove_current
+    if @view.currentIndex.valid?
+      @subgroups[@view.currentIndex.row].try(:destroy)
+      @subgroups.delete_at(@view.currentIndex.row)
+      emit layoutChanged()
+      @view.currentIndex = createIndex(-1, -1)
+    end
+  end
+
+end
+
+class GroupComboBoxDelegate < Qt::ItemDelegate
+  def initialize(parent)
+    super
+    setup
+  end
+
+  def setup
+    @groups = Group.all.sort_by(&:title_for_sort)
+  end
+
+  def createEditor(parent, option, index)
+    editor = Qt::ComboBox.new(parent)
+    @groups.each{ |x| editor.addItem(x.title.to_s, x.id.to_v) }
+    editor
+  end
+
+  def setEditorData(editor, index)
+    value = index.model.data(index, Qt::EditRole)
+    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/subject_table_model.rb

@@ -0,0 +1,90 @@
+# encoding: UTF-8
+#~~~~~~~~~~~~~~~~~~~~~~~~~~
+require_relative '../../engine/models/subject'
+#~~~~~~~~~~~~~~~~~~~~~~~~~~
+class SubjectTableModel < Qt::AbstractTableModel
+
+  def initialize(subjects, parent)
+    super()
+    @subjects = subjects
+    @view = parent
+  end
+
+  def refresh
+    @subjects = Subject.all
+    emit layoutChanged()
+  end
+
+  def rowCount(parent)
+    @subjects.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
+    subject = @subjects[index.row]
+    return invalid if subject.nil?
+    v = case index.column
+        when 0
+          subject.title
+        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
+      subject = @subjects[index.row]
+      case index.column
+      when 0
+        subject.title = variant.toString.force_encoding('UTF-8')
+      else
+        raise "invalid column #{index.column}"
+      end
+      subject.save
+      emit dataChanged(index, index)
+      true
+    else
+      false
+    end
+  end
+
+  def insert_new
+    @subjects.prepend(Subject.new)
+    emit layoutChanged()
+  end
+
+  def remove_current
+    if @view.currentIndex.valid?
+      subject = @subjects[@view.currentIndex.row]
+      unless subject.stub
+        subject.try(:destroy)
+        @subjects.delete_at(@view.currentIndex.row)
+        emit layoutChanged()
+        @view.currentIndex = createIndex(-1, -1)
+      end
+    end
+  end
+
+end