pg_dumper 0.0.2

data/.gitignore

@@ -0,0 +1,4 @@
+*.gem
+.bundle
+Gemfile.lock
+pkg/*

data/Gemfile

@@ -0,0 +1,4 @@
+source "http://rubygems.org"
+
+# Specify your gem's dependencies in pg_dumper.gemspec
+gemspec

data/README.md

@@ -0,0 +1,18 @@
+# PgDumper
+
+Abstraction layer between PostgreSQL utility pg_dump and Rails app.
+
+## Rake tasks
+You can specify few ENV variables:
+GZIP - to gzip sql dump with gzip utility
+FILE - dump output (else asks interactively)
+VERBOSE - verbose output
+Z - pg_dump compression level
+
+### db:dump
+Dumps database schema **with data**.
+
+### db:dump:schema
+Dumps database schema **without data**.
+
+

data/Rakefile

@@ -0,0 +1,2 @@
+require 'bundler'
+Bundler::GemHelper.install_tasks

data/lib/pg_dumper/railtie.rb

@@ -0,0 +1,12 @@
+require 'pg_dumper'
+require 'rails'
+
+class PgDumper
+  class Railtie < Rails::Railtie
+    #railtie_name :pg_dumper # Deprecated
+
+    rake_tasks do
+      load "tasks/db_dump.rake"
+    end
+  end
+end

data/lib/pg_dumper/version.rb

@@ -0,0 +1,3 @@
+class PgDumper
+  VERSION = "0.0.2"
+end

data/lib/pg_dumper.rb

@@ -0,0 +1,76 @@
+class PgDumper
+  require 'pg_dumper/railtie' if defined?(Rails)
+  
+  attr_reader :database
+  attr_reader :args
+  
+  def initialize database
+    @database = database
+    @args = []
+    @options = {}
+  end
+  
+  def run(mode = :silent)
+    @binary ||= find_executable
+    
+    raise "ERROR: pg_dump executable not found" unless @binary.present?
+
+    options = {}
+    
+    case mode
+    when :silent
+      options[:out] = "/dev/null" 
+    end
+    system @binary, *args, database, options
+  end
+  
+  def schema_only!
+    add_args "-s"
+  end
+  
+  def recreate!
+    add_args "-c"
+  end
+  
+  def data_only!
+    add_args "-a", '--disable-triggers'
+    add_args '-T', 'schema_migrations'
+  end
+  
+  def compress! level=9
+    add_args '-Z', level if level.present?
+  end
+  
+  def verbose!
+    add_args "-v"
+  end
+  
+  def pretty!
+    add_args '--column-inserts'
+  end
+  
+  def connection= opts
+    add_args '-p', opts['port'] if opts['port'].present?
+    add_args '-h', opts['host'] if opts['host'].present?
+    add_args '-U', opts['username'] if opts['host'].present?
+  end
+  
+  def output= filename
+    add_args "-f", filename
+  end
+  
+  private
+  def find_executable
+    [ENV['PG_DUMP'], %x{which pg_dump}.strip].each do |pg|
+      return pg if pg.present? && File.exists?(pg)
+    end
+    nil
+  end
+  
+  def add_args(*args)
+    @args.push *args.map!(&:to_s)
+    @args.uniq!
+  end
+  
+  
+end

data/lib/tasks/db_dump.rake

@@ -0,0 +1,63 @@
+namespace :db do
+  desc 'Dumps database (both data and schema) to FILE or asks for output'
+  task :dump => ["#{Rails.root}/config/database.yml", :environment] do
+    gzip = ENV['GZIP'].present?
+    db = ActiveRecord::Base.configurations[Rails.env]
+    backup_file = Time.now.strftime("backup_%Y%m%d%H%M%S.sql")
+    unless file = ENV['FILE']
+      print "Enter filename of dump [#{backup_file}#{'.gz' if gzip}]: "
+      input = STDIN.gets.chop
+      file  = (input.blank?)? backup_file : input
+    end
+    
+    dump = PgDumper.new db["database"]
+    dump.recreate!
+    dump.connection = db
+    dump.output = backup_file
+    dump.verbose! if ENV['VERBOSE'].present?
+    dump.compress! ENV['Z']
+    
+    print "Creating backup to '#{file}'\n"
+    
+    if dump.run
+      system 'gzip', '-v', file if gzip
+      print "Backup saved to #{file.inspect}\n"
+    else
+      print "Backup failed! Error status: #{$?}."
+    end
+    
+    @backup_file = file
+  end
+  
+  desc 'Dumps database (only data) to FILE or asks for output'
+  namespace :dump do
+    task :data =>  ["#{Rails.root}/config/database.yml", :environment] do
+      gzip = ENV['GZIP'].present?
+      db = ActiveRecord::Base.configurations[Rails.env]
+      backup_file = Time.now.strftime("backup_%Y%m%d%H%M%S.sql")
+      unless file = ENV['FILE']
+        print "Enter filename of dump [#{backup_file}#{'.gz' if gzip}]: "
+        input = STDIN.gets.chop
+        file  = (input.blank?)? backup_file : input
+      end
+
+      dump = PgDumper.new db["database"]
+      dump.data_only!
+      dump.connection = db
+      dump.output = backup_file
+      dump.verbose! if ENV['VERBOSE'].present?
+      dump.compress! ENV['Z']
+
+      print "Creating backup to '#{file}'\n"
+
+      if dump.run
+        system 'gzip', '-v', file if gzip
+        print "Backup saved to #{file.inspect}\n"
+      else
+        print "Backup failed! Error status: #{$?}."
+      end
+
+      @backup_file = file
+    end
+  end
+end

data/pg_dumper.gemspec

@@ -0,0 +1,21 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "pg_dumper/version"
+
+Gem::Specification.new do |s|
+  s.name        = "pg_dumper"
+  s.version     = PgDumper::VERSION
+  s.platform    = Gem::Platform::RUBY
+  s.authors     = ["mikz"]
+  s.email       = ["mikz@o2h.cz"]
+  s.homepage    = ""
+  s.summary     = %q{Adds Rake tasks to easily dump postgresql database with or without data}
+  s.description = %q{Provides abstraction layer between pg_dump utility and ruby. Also adds rake task to easily dump database with or without data.}
+
+  #s.rubyforge_project = "pg_dumper"
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+end

metadata

@@ -0,0 +1,73 @@
+--- !ruby/object:Gem::Specification 
+name: pg_dumper
+version: !ruby/object:Gem::Version 
+  prerelease: false
+  segments: 
+  - 0
+  - 0
+  - 2
+  version: 0.0.2
+platform: ruby
+authors: 
+- mikz
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-02-01 00:00:00 +01:00
+default_executable: 
+dependencies: []
+
+description: Provides abstraction layer between pg_dump utility and ruby. Also adds rake task to easily dump database with or without data.
+email: 
+- mikz@o2h.cz
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- .gitignore
+- Gemfile
+- README.md
+- Rakefile
+- lib/pg_dumper.rb
+- lib/pg_dumper/railtie.rb
+- lib/pg_dumper/version.rb
+- lib/tasks/db_dump.rake
+- pg_dumper.gemspec
+has_rdoc: true
+homepage: ""
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.7
+signing_key: 
+specification_version: 3
+summary: Adds Rake tasks to easily dump postgresql database with or without data
+test_files: []
+