X-Git-Url: https://projects.mako.cc/source/selectricity/blobdiff_plain/9abed97635edbac7fb1a687298fff5c5434cdff4..fd484b33a3920e0031007c5c2f1dec61913cbab5:/app/models/candidate.rb?ds=sidebyside

diff --git a/app/models/candidate.rb b/app/models/candidate.rb
index a0e492e..7b6e802 100644
--- a/app/models/candidate.rb
+++ b/app/models/candidate.rb
@@ -1,8 +1,35 @@
 class Candidate < ActiveRecord::Base
   belongs_to :election
+  validates_uniqueness_of :name
+  validates_presence_of :name
 
   def <=>(other)
-   self.name <=> other.name 
+    self.name <=> other.name 
+  end
+  
+  def to_s
+    name
+  end
+
+  def picture=(picture_field)
+    if picture_field
+      unless picture_field.content_type.match(/^image/)
+        return false
+      end
+      self.picture_filename = base_part_of(picture_field.original_filename)
+      self.picture_type =  picture_field.content_type.chomp
+      self.picture_data = picture_field.read
+    end
+  end
+
+  def base_part_of(filename)
+    name = File.basename(filename)
+    name.gsub(/[^\w._-]/, '')
+  end
+
+  def picture?
+    !self.picture_filename.nil?
   end
 
 end
+