tmis 0.1.3

data/lib/tmis/engine/import/spreadsheet_roo.rb

@@ -0,0 +1,136 @@
+# coding: UTF-8
+#~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+require 'roo'
+require 'fileutils'
+require 'spreadsheet'
+#~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+#require '#Contracts'
+require_relative 'abstract_spreadsheet'
+#include #Contracts
+#~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+class SpreadsheetRoo < AbstractSpreadsheet
+
+  attr_reader :filepath
+
+  ##Contract String => Any
+  def initialize(filepath)
+    @filepath = filepath
+    @sheet = Roo::Spreadsheet.open(filepath)
+    @sheet.default_sheet = 0
+  end
+
+  #Contract None => Pos
+  def last_row
+    @sheet.last_row
+  end
+
+  #Contract None => Pos
+  def last_column
+    @sheet.last_column
+  end
+
+  #Contract Not[Neg] => Any
+  def sheet(number)
+    @sheet.default_sheet = @sheet.sheets[number-1]
+  end
+
+  #Contract Pos => Any
+  def row(n)
+    @sheet.row(n)
+  end
+
+  #Contract Pos => Any
+  def column(n)
+    @sheet.column(n)
+  end
+
+  #Contract Pos, Pos => Any
+  def [](r, c)
+    @sheet.cell(r, c)
+  end
+end
+
+class SpreadsheetSpreadsheet < AbstractSpreadsheet
+  include WritableSpreadsheet
+
+  attr_reader :filepath
+
+  #Contract String => Any
+  def initialize(filepath)
+    @filepath = filepath
+    Spreadsheet.client_encoding = 'UTF-8'
+    if File.file?(@filepath)
+      @book = Spreadsheet.open(@filepath)
+    else
+      @book = Spreadsheet::Workbook.new(@filepath)
+      @book.create_worksheet
+    end
+    @sheet = @book.worksheet(0)
+    fmt = Spreadsheet::Format.new text_wrap: true
+    fmt.horizontal_align = :center
+    fmt.vertical_align = :middle
+    fmt.font = Spreadsheet::Font.new('Times New Roman', :size => 12)
+    @sheet.default_format = fmt
+    @sheet
+  end
+
+  #Contract None => Pos
+  def last_row
+    @sheet.last_row_index
+  end
+
+  #Contract None => Pos
+  def last_column
+    @sheet.column_count
+  end
+
+  #Contract Not[Neg] => Any
+  def sheet(number)
+    @sheet = @book.worksheet(number - 1)
+  end
+
+  #Contract Pos => Any
+  def row(n)
+    @sheet.row(n - 1)
+  end
+
+  #Contract Pos => Any
+  def column(n)
+    @sheet.column(n - 1)
+  end
+
+  #Contract Pos, Pos => Any
+  def [](r, c)
+    @sheet[r - 1, c - 1]
+  end
+
+  #Contract Pos, Pos, Any => Any
+  def []=(r, c, obj)
+    @sheet[r - 1, c - 1] = obj
+  end
+
+  #Contract None => Any
+  def save
+    @book.write("#{@filepath}_temp")
+    FileUtils.mv("#{@filepath}_temp", @filepath) # Обход бага в библиотеке
+  end
+
+  def merge(start_row, start_col, end_row, end_col)
+    @sheet.merge_cells(start_row - 1, start_col - 1, end_row - 1, end_col - 1)
+  end
+
+  def format(r, c, fmt)
+    @sheet.row(r - 1).set_format(c - 1, fmt)
+  end
+end
+
+class SpreadsheetCreater
+  #Contract String => Or[IsA[AbstractSpreadsheet], IsA[WritableSpreadsheet]]
+  def self.create(filename)
+    if filename =~ /.*.csv/
+      SpreadsheetRoo.new(File.expand_path(filename))
+    else
+      SpreadsheetSpreadsheet.new(File.expand_path(filename))
+    end
+  end
+end

data/lib/tmis/engine/import/timetable_manager.rb

@@ -0,0 +1,110 @@
+# coding: UTF-8
+#~~~~~~~~~~~~~~~~~~~~~~~~~~
+require 'active_record'
+require 'date'
+require_relative '../models/group'
+require_relative '../models/subgroup'
+require_relative '../models/subject'
+require_relative '../models/lecturer'
+require_relative '../models/cabinet'
+require_relative '../models/study'
+require_relative '../models/course'
+require_relative '../models/semester'
+require_relative 'timetable_reader'
+#~~~~~~~~~~~~~~~~~~~~~~~~~~
+require 'contracts'
+require_relative 'abstract_spreadsheet'
+#include Contracts
+#~~~~~~~~~~~~~~~~~~~~~~~~~~
+class TimetableManager
+  #Contract IsA[TimetableReader], Any => Any
+  def initialize(timetable_reader, date)
+    @reader = timetable_reader
+    @days = {'понедельник' => date, 'вторник' => date + 1, 'среда' => date + 2, 'четверг' => date + 3, 'пятница' => date + 4, 'суббота' => date + 5 }
+  end
+
+  #Contract None => Any
+  def save_to_db
+    Database.instance.transaction do
+      create_subgroups
+      (1..4).each{ |n| add(Course, number: n) }
+      Course.all.zip(Course.all).flatten.each_with_index { |c, i| add(Semester, {title: i + 1, course_id: c}) }
+    end
+  end
+
+private
+  #Contract None => Any
+  def create_subgroups
+    @reader.groups.each do |data|
+      group = add(Group, title: data[:title] || "")
+      subgroups = (1..2).map{ |i| add(Subgroup, {group_id: group, number: i}) }
+      create_studies(data, group, subgroups)
+    end
+  end
+
+  #Contract Hash, Group, ArrayOf[Subgroup] => Any
+  def create_studies(data, group, subgroups)
+    data[:days].each do |day|
+      day[:studies].each_with_index do |study, number|
+        if study.size == 1
+          if (s = study.first[:info][:subgroup])
+            s = s.to_i
+            if s > 0 && s < 3
+              Study.create( get_study_options(study[0], day[:name], number.succ, subgroups[s.to_i - 1]) )
+            else
+              next
+            end
+          else
+            Study.create( get_study_options(study[0], day[:name], number.succ, group) )
+          end
+        else
+          study.each do |sepstudy|
+            Study.create( get_study_options(sepstudy, day[:name], number.succ,
+              sepstudy[:info][:subgroup] ? subgroups[sepstudy[:info][:subgroup].to_i - 1] : group) )
+          end
+        end
+      end
+    end
+  end
+
+  #Contract IsA[Class], Hash => IsA[ActiveRecord::Base]
+  def add(model, options)
+    model.where(options).first_or_create
+  end
+
+  #Contract Hash, String, Pos, Or[Group, Subgroup] => Hash
+  def get_study_options(study, day, study_number, groupable)
+    { subject: add(Subject, title: study[:info][:subject]),
+      cabinet: new_cabinet_or_stub(study),
+      lecturer: new_lecturer_or_stub(study),
+      date: @days[day.mb_chars.downcase.to_s.gsub(' ', '')],
+      number: study_number,
+      groupable: groupable }
+  end
+
+  def new_lecturer_or_stub(study)
+    case study[:info][:lecturer][:surname]
+    when nil
+      Lecturer.where(stub: true).first
+    when /#{Settings[:stubs, :lecturer]}/i
+      Lecturer.where(stub: true).first
+    else
+      add(Lecturer, { surname: study[:info][:lecturer][:surname],
+                      name: study[:info][:lecturer][:name],
+                      patronymic: study[:info][:lecturer][:patronymic] })
+    end
+  end
+
+  def new_cabinet_or_stub(study)
+    if study[:cabinet]
+      add(Cabinet, title: fix_cabinet(study[:cabinet]))
+    else
+      Cabinet.where(stub: true).first
+    end
+  end
+
+  #Contract Or[Num, String] => String
+  def fix_cabinet(title)
+    title.is_a?(Float) ? title.ceil.to_s : title.to_s
+  end
+end

data/lib/tmis/engine/import/timetable_reader.rb

@@ -0,0 +1,79 @@
+# coding: UTF-8
+#~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+require 'contracts'
+require_relative 'abstract_spreadsheet'
+require_relative '../../interface/forms/settings'
+#include Contracts
+#~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+class TimetableReader
+  #Contract IsA[AbstractSpreadsheet], Pos => Any
+  def initialize(spreadsheet, sheet_number=1)
+    @table = spreadsheet
+    week(sheet_number)
+  end
+
+  #Contract Pos => TimetableReader
+  def week(num)
+    @table.sheet(num); self
+  end
+
+  #Contract None => Array
+  def groups
+    (3..@table.last_column).each_slice(2).map do |cols|
+      { title: @table[7, cols.first], days: get_days(cols) }
+    end
+  end
+
+private
+  #Contract [Pos, Pos] => Array
+  def get_days(cols)
+    (7..84).each_slice(13).map{ |i| [i[1], i.last] }.map do |rows|
+      p @table[rows.first, 1]
+      { name: @table[rows.first, 1], studies: get_studies(rows, cols) }
+    end
+  end
+
+  #Contract [Pos, Pos], [Pos, Pos] => Array
+  def get_studies(rows, cols)
+    (rows.first..rows.last).each_slice(2).map do |study_rows|
+      study_rows.each_with_index.map do |row, index|
+        cabinet = @table[row, cols.last]
+        { info: parse_info(@table[row, cols.first]),
+          cabinet: if cabinet.nil? || if cabinet.kind_of?(String); cabinet.empty?; else; false end;
+                     @table[row - index, cols.last]
+                   else
+                     cabinet
+                   end }
+      end.reject{ |s| s[:info].nil? }
+    end
+  end
+
+  #Contract Maybe[String] => Maybe[Hash]
+  def parse_info(info)
+    unless info.nil? || info.empty?
+      reversed_info = info.split(/[\s\n\.\(\)\-_,]/).reverse.join(' ')
+      # TODO Переписать регулярку с использованием условий в движке Onigmo (Ruby 2.0)
+      r = /
+          (
+            ([Пп]|[Пп][Гг]|[Пп][Оо][Дд][Гг][Рр]|[Пп][Оо][Дд][Гг][Рр][Уу][Пп][Пп][Аа])*\s*
+            (?<subgroup>[[:digit:]])
+            \s*([Пп]|[Пп][Гг]|[Пп][Оо][Дд][Гг][Рр]|[Пп][Оо][Дд][Гг][Рр][Уу][Пп][Пп][Аа])*\s*
+          )?
+            (
+              (?<patronymic>[[:upper:]])\s
+              (?<name>[[:upper:]])\s*
+            )?
+          (
+            (?<surname>
+              ([Вв][Аа][Кк][Аа][Нн][Сс][Ии][Яя]|[[:upper:]][[:lower:]]+)
+            )
+          )?
+          (?<title>.*)/mx
+      data = r.match(reversed_info.strip)
+      res = { subject: data[:title].split.reverse.join(' ').strip,
+        lecturer: { surname: data[:surname], name: data[:name], patronymic: data[:patronymic] },
+        subgroup: data[:subgroup] }
+      res
+    end
+  end
+end

data/lib/tmis/engine/mailer/mailer.rb

@@ -0,0 +1,51 @@
+# coding: UTF-8
+#~~~~~~~~~~~~~~~~~~~~~~~~~~
+require 'mail'
+#~~~~~~~~~~~~~~~~~~~~~~~~~~
+require 'contracts'
+#include Contracts
+#~~~~~~~~~~~~~~~~~~~~~~~~~~
+class Mailer
+  #Contract String, String, Proc => Any
+  def initialize(email, password, &block)
+    @email = Mailer.email_valid?(email) ? email : (raise ArgumentError, "Incorrect email: #{email}")
+    @password = password
+    set_defaults
+    @mail = Mail.new(&block)
+    @mail.charset = 'UTF-8'
+  end
+
+  #Contract String => Bool
+  def self.email_valid?(email)
+    true & (/[\w\d\._\-]+@[\w\d\.\-]+[\.][\w\d\.\-]+/i =~ email)
+  end
+
+  #Contract None => Any
+  def send!
+    @mail.deliver!
+  end
+
+private
+  #Contract None => ({ local: String, domain: String })
+  def email_parts
+    @email[/([\w\d._-]+)@([\w\d.-]+)/i]
+    { local: $1, domain: $2 }
+  end
+
+  #Contract None => Any
+  def set_defaults
+    parts = email_parts
+    p email_parts
+    pass = @password
+    Mail.defaults do
+      delivery_method :smtp, {
+        address: "smtp.#{parts[:domain]}",
+        port: '587',
+        user_name: parts[:local],
+        password: pass,
+        authentication: :plain,
+        enable_starttls_auto: true
+      }
+    end
+  end
+end

data/lib/tmis/engine/migrations/10_create_speciality_subjects.rb

@@ -0,0 +1,17 @@
+class CreateSpecialitySubjects < ActiveRecord::Migration
+  def change
+    create_table :speciality_subjects do |t|
+      t.integer :lecturer_id
+      t.integer :subject_id
+      t.integer :semester_id
+      t.integer :speciality_id
+      t.integer :lecture_hours
+      t.integer :practical_hours
+      t.integer :consultations_hours
+      t.string  :preffered_days
+      t.boolean :facultative
+
+      t.timestamps
+    end
+  end
+end

data/lib/tmis/engine/migrations/11_create_emails.rb

@@ -0,0 +1,10 @@
+class CreateEmails < ActiveRecord::Migration
+  def change
+    create_table :emails do |t|
+      t.references :emailable, :polymorphic => true
+      t.string :email
+
+      t.timestamps
+    end
+  end
+end

data/lib/tmis/engine/migrations/12_add_indexes.rb

@@ -0,0 +1,32 @@
+class AddIndexes < ActiveRecord::Migration
+  def change
+    add_index(:groups, :title)
+
+    add_index(:subgroups, :group_id)
+    add_index(:subgroups, :number)
+
+    add_index(:subjects, :title)
+
+    add_index(:cabinets, :title)
+
+    add_index(:lecturers, :surname)
+
+    add_index(:studies, :subject_id)
+    add_index(:studies, :lecturer_id)
+    add_index(:studies, :cabinet_id)
+    add_index(:studies, [:groupable_id, :groupable_type])
+
+    add_index(:courses, :number)
+
+    add_index(:specialities, :title)
+
+    add_index(:semesters, :course_id)
+    add_index(:semesters, :title)
+
+    add_index(:speciality_subjects, :subject_id)
+    add_index(:speciality_subjects, :lecturer_id)
+    add_index(:speciality_subjects, [:speciality_id, :semester_id])
+
+    add_index(:emails, [:emailable_id, :emailable_type])
+  end
+end

data/lib/tmis/engine/migrations/1_create_groups.rb

@@ -0,0 +1,11 @@
+class CreateGroups < ActiveRecord::Migration
+  def change
+    create_table :groups do |t|
+      t.string :title
+      t.integer :speciality_id
+      t.integer :course_id
+
+      t.timestamps
+    end
+  end
+end

data/lib/tmis/engine/migrations/2_create_subgroups.rb

@@ -0,0 +1,10 @@
+class CreateSubgroups < ActiveRecord::Migration
+  def change
+    create_table :subgroups do |t|
+      t.integer :number
+      t.integer :group_id
+
+      t.timestamps
+    end
+  end
+end

data/lib/tmis/engine/migrations/3_create_subjects.rb

@@ -0,0 +1,11 @@
+# coding: UTF-8
+class CreateSubjects < ActiveRecord::Migration
+  def change
+    create_table :subjects do |t|
+      t.string :title
+      t.boolean :stub, default: false
+
+      t.timestamps
+    end
+  end
+end

data/lib/tmis/engine/migrations/4_create_cabinets.rb

@@ -0,0 +1,12 @@
+# coding: UTF-8
+class CreateCabinets < ActiveRecord::Migration
+  def change
+    create_table :cabinets do |t|
+      t.string :title
+      t.boolean :stub, default: false
+      t.boolean :with_computers, default: false
+
+      t.timestamps
+    end
+  end
+end

data/lib/tmis/engine/migrations/5_create_lecturers.rb

@@ -0,0 +1,14 @@
+# coding: UTF-8
+class CreateLecturers < ActiveRecord::Migration
+  def change
+    create_table :lecturers do |t|
+      t.string :surname
+      t.string :name
+      t.string :patronymic
+      t.string :preferred_days
+      t.boolean :stub, default: false
+
+      t.timestamps
+    end
+  end
+end

data/lib/tmis/engine/migrations/6_create_studies.rb

@@ -0,0 +1,15 @@
+class CreateStudies < ActiveRecord::Migration
+  def change
+    create_table :studies do |t|
+      t.references :groupable, :polymorphic => true
+      t.integer :subject_id
+      t.integer :lecturer_id
+      t.integer :cabinet_id
+      t.integer :number
+      t.date    :date
+      t.string  :color
+
+      t.timestamps
+    end
+  end
+end

data/lib/tmis/engine/migrations/7_create_courses.rb

@@ -0,0 +1,9 @@
+class CreateCourses < ActiveRecord::Migration
+  def change
+    create_table :courses do |t|
+      t.integer :number
+
+      t.timestamps
+    end
+  end
+end

data/lib/tmis/engine/migrations/8_create_specialities.rb

@@ -0,0 +1,9 @@
+class CreateSpecialities < ActiveRecord::Migration
+  def change
+    create_table :specialities do |t|
+      t.string :title
+
+      t.timestamps
+    end
+  end
+end

data/lib/tmis/engine/migrations/9_create_semesters.rb

@@ -0,0 +1,10 @@
+class CreateSemesters < ActiveRecord::Migration
+  def change
+    create_table :semesters do |t|
+      t.string :title
+      t.integer :course_id
+
+      t.timestamps
+    end
+  end
+end

data/lib/tmis/engine/models/cabinet.rb

@@ -0,0 +1,18 @@
+class Cabinet < ActiveRecord::Base
+  has_many :studies
+
+  before_destroy :set_stubs_for_studies
+
+  def to_s
+    title
+  end
+
+  def set_stubs_for_studies
+    raise "Stub can't be destroyed!" if self.stub
+    stub = Cabinet.where(stub: true).first
+    studies.each do |s|
+      s.cabinet = stub
+      s.save
+    end
+  end
+end

data/lib/tmis/engine/models/course.rb

@@ -0,0 +1,11 @@
+class Course < ActiveRecord::Base
+  has_many :semesters,  :dependent => :destroy
+
+  def current_semester
+    if Date.today.yday > 182
+      semesters.first
+    else
+      semesters.last
+    end
+  end
+end

data/lib/tmis/engine/models/email.rb

@@ -0,0 +1,19 @@
+# coding: UTF-8
+class Email < ActiveRecord::Base
+  belongs_to :emailable, :polymorphic => true
+
+  def validate
+  end
+
+  def to_group?
+    emailable_type == 'Group'
+  end
+
+  def to_lecturer?
+    emailable_type == 'Lecturer'
+  end
+
+  def email_valid?
+    true & (/[\w\d._-]+@[\w\d.-]+[.][\w\d.-]+/i =~ email)
+  end
+end

data/lib/tmis/engine/models/group.rb

@@ -0,0 +1,31 @@
+class Group < ActiveRecord::Base
+  belongs_to :speciality
+  belongs_to :course
+  has_many :subgroups, :dependent => :destroy
+  has_many :studies, :as => :groupable, :dependent => :destroy
+  has_many :emails, :as => :emailable,  :dependent => :destroy
+
+  def group?
+    true
+  end
+
+  def subgroup?
+    false
+  end
+
+  def get_group
+    self
+  end
+
+  def number
+    0
+  end
+
+  def title_for_sort
+    self.title[/(.*)-(.*)/i]; "#{$2}-#{$1}"
+  end
+
+  def to_s
+    title
+  end
+end

data/lib/tmis/engine/models/lecturer.rb

@@ -0,0 +1,45 @@
+# coding: UTF-8
+#~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+require 'contracts'
+#include Contracts
+#~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+class Lecturer < ActiveRecord::Base
+  has_many :studies
+  has_many :speciality_subjects
+  has_many :emails, :as => :emailable, :dependent => :destroy
+
+  before_destroy :set_stubs_for_studies
+
+  #Contract None => String
+  def to_s
+    first = surname
+    if name.nil?
+      second = ""
+    else
+      if name.empty?
+        second = name
+      else
+        second = name[0].mb_chars.capitalize.to_s
+      end
+    end
+    if patronymic.nil?
+      third = ""
+    else
+      if patronymic.empty?
+        third = name
+      else
+        third = patronymic[0].mb_chars.capitalize.to_s
+      end
+    end
+    "#{surname} #{second}.#{third}."
+  end
+
+  def set_stubs_for_studies
+    raise "Stub can't be destroyed!" if self.stub
+    stub = Lecturer.where(stub: true).first
+    studies.each do |s|
+      s.lecturer = stub
+      s.save
+    end
+  end
+end

data/lib/tmis/engine/models/semester.rb

@@ -0,0 +1,4 @@
+class Semester < ActiveRecord::Base
+  belongs_to :course
+  has_many :speciality_subjects,  :dependent => :destroy
+end

data/lib/tmis/engine/models/speciality.rb

@@ -0,0 +1,3 @@
+class Speciality < ActiveRecord::Base
+  has_many :speciality_subjects,  :dependent => :destroy
+end