Home-brewed file upload in Ruby on Rails

seancribbs (Sean Cribbs) — Thu, 30 Mar 2006 02:33:29 GMT

Most of this comes from others' work, but I was able to tool it to my needs and fix some bugs. All of these lines go in the model, which for me has a :file and :content_type attributes. :file stores the complete path to the uploaded file. Be sure to change the string in path_to_file to the place where you want files stored, and that proper permissions are set on that path. Also, sanitize_filename doesn't HAVE to be a private method -- make it public if you want.



### Model ###

 def file=(uploaded_file)  

    @uploaded_file = uploaded_file

    @filename = sanitize_filename(@uploaded_file.original_filename)

    write_attribute("content_type", @uploaded_file.content_type)

  end

  

  def after_create

    if !File.exists?(File.dirname(path_to_file))

      Dir.mkdir(File.dirname(path_to_file))

    end

    if @uploaded_file.instance_of?(Tempfile)

      FileUtils.copy(@uploaded_file.local_path, path_to_file)

    else

      File.open(self.path_to_file, "wb") { |f| f.write(@uploaded_file.read) }

    end

    write_attribute("file", path_to_file)

  end



  def after_destroy

    if File.exists?(self.file)

      File.delete(self.file)

      Dir.rmdir(File.dirname(self.file))

    end

  end

  

  def path_to_file

    File.expand_path("#{RAILS_ROOT}/upload/#{self.id}/#{@filename}")

  end

  

  private

  def sanitize_filename(file_name)

    # get only the filename, not the whole path (from IE)

    just_filename = File.basename(file_name) 

    # replace all none alphanumeric, underscore or perioids with underscore

    just_filename.gsub(/[^\w\.\_]/,'_') 

  end

  

### View ###

...



...

DZone Snippets: rubyonrails code

Home-brewed file upload in Ruby on Rails