lib/models/assignment.rb in bobross-0.1.7 vs lib/models/assignment.rb in bobross-0.1.8

@@ -1,9 +1,17 @@
 class Assignment < Forgery
-  attr_reader :name, :description, :due_at, :lock_at, :course_id, :assignment_group
-  attr_writer :due_at, :lock_at
+  attr_reader :name, :description, :course_uid, :assignment_group
+  attr_accessor :due_at, :lock_at
 
+  def to_s
+    string = "#{name}, #{description}, #{due_at}, #{lock_at}, #{course_uid}, #{assignment_group}"
+  end
+
+  def to_csv
+    row = [name, description, due_at, lock_at, course_uid, assignment_group]
+  end
+
   def initialize(opts = {})
     @name = opts[:name] if opts[:name]
     @description = opts[:description] if opts[:description]
     @due_at = opts[:due_at] if opts[:due_at]
     @lock_at = opts[:lock_at] if opts[:lock_at]
@@ -22,6 +30,27 @@
         course_id: course,
         assignment_group: group
         }
       )
   end
+
+  def self.gen_file(opts = {})
+    course, group = 1
+    rows = 0
+    course = opts[:course] if opts[:course]
+    group = opts[:group] if opts[:group]
+    rows = opts[:rows] if opts[:rows]
+    assignments = []
+    if(opts[:rows])
+      rows.times do |x|
+        assignments.push(Assignment.random(course, group))
+      end
+    end
+    header = ["name", "description", "due_at", "lock_at", "course_id", "assignment_group"]
+    CSV.open("./assignments.csv", "wb", write_headers: true, headers: header) do |csv|
+      assignments.each do |acc|
+        csv << acc.to_csv
+      end
+    end
+  end
+
 end