pg_gnostic 0.0.1 → 0.0.2

data/README.rdoc

@@ -24,7 +24,9 @@ To install pg_gnostic rake tasks u need add this lines into  RAILS_ROOT/Rakefile
 * rake pg: pg:drop_views, pg:functions, pg:views and all in one pg:update
 
 
-== CHANGE LIST:
+== CHANGE LOG:
+
+0.0.2  - backup/restore rakes
 
 == TODO 
 

data/Rakefile

@@ -28,7 +28,7 @@ PKG_FILES = FileList[ '[a-zA-Z]*', 'generators/**/*', 'lib/**/*', 'rails/**/*',
 
 spec = Gem::Specification.new do |s|
   s.name = "pg_gnostic"
-  s.version = "0.0.1"
+  s.version = "0.0.2"
   s.author = "niquola"
   s.email = "niquola@gmail.com"
   s.homepage = "http://github.com/niquola/pg_gnostic"

data/lib/pg_gnostic/config.rb

@@ -1,5 +1,12 @@
 module PgGnostic
   class Config < KungFigure::Base
+
+    class Backup < KungFigure::Base
+      define_prop :path, 'db/backup'
+      define_prop :archive_command, 'gzip'
+      define_prop :unarchive_command, 'gunzip -c'
+    end
+
     class ViewModel < KungFigure::Base
       define_prop :nest_in_module,'Views'
       define_prop :prefix_view_name,'view_'
@@ -7,5 +14,6 @@ module PgGnostic
         nest_in_module ? "app/model/#{nest_in_module.underscore}" : "app/model"
       end
     end
+
   end
 end

data/tasks/pg_gnostic_tasks.rake

@@ -1,17 +1,92 @@
-puts "loading task"
+require 'fileutils'
+require 'date'
+
 namespace :pg do
+  def get_backup_dir()
+    dir = PgGnostic.config.backup.path
+    dir = File.join(Rails.root,dir) unless dir =~ /^\//
+    dir
+  end
+
+  desc "Backup database"
+  task :backup=>:environment do
+    config    = ActiveRecord::Base.configurations[RAILS_ENV || 'production']
+    backupdir=get_backup_dir
+    FileUtils.mkdir_p backupdir
+    date = DateTime.now.to_s.gsub(/[-:T+]/,'_')
+    backupfile = File.join(backupdir,"#{config["database"]}_#{date}.sql")
+
+    params={
+      'host'=>config["host"],
+      'port'=>config["port"],
+      'username'=>config["username"]
+    }.map{|k,v| "--#{k}=#{v}" if v}.join(' ')
+    archiver = PgGnostic.config.backup.archive_command
+    puts "Backup #{config["database"]} to #{backupfile}"
+    command =  "export PGPASSWORD=#{config["password"]} && pg_dump #{params} -w #{config["database"]} | #{archiver} > #{backupfile}"
+    puts "Execute: #{command}"
+    system command
+    system "ls -lh #{backupfile}"
+  end
+
+  desc "Restore database"
+  task :restore=>:environment do |args|
+
+    config    = ActiveRecord::Base.configurations[RAILS_ENV || 'production']
+    @encoding = config[:encoding] || ENV['CHARSET'] || 'utf8'
+    begin
+      ActiveRecord::Base.establish_connection(config.merge('database' => 'postgres', 'schema_search_path' => 'public'))
+      ActiveRecord::Base.connection.create_database(config['database'], config.merge('encoding' => @encoding))
+      ActiveRecord::Base.establish_connection(config)
+
+      file = ENV["PGBACKUP"]
+      config    = ActiveRecord::Base.configurations[RAILS_ENV || 'development']
+      backupdir = get_backup_dir 
+      if file && File.exists?(file)
+        file = file
+      elsif file && File.exists?(File.join(backupdir,file))
+        file = File.join(backupdir,file)
+      else
+        STDOUT.puts "Select which backup file to restore:"
+        files = Dir["#{backupdir}/*"].to_a
+        files.each_with_index{|f,i| STDOUT.puts "#{i}: #{f}"}
+        print "> "
+        num = STDIN.gets
+        file = files[num.to_i]
+      end
+      exit unless file && File.exists?(file)
+      puts "Restoring database #{config["database"]} from file #{file}"
+      params={
+        'host'=>config["host"],
+        'port'=>config["port"],
+        'username'=>config["username"]
+      }.map{|k,v| "--#{k}=#{v}" if v}.join(' ')
+      unarchiver = PgGnostic.config.backup.unarchive_command 
+      command =  "export PGPASSWORD=#{config["password"]} && #{unarchiver} #{file} | psql #{params} -w #{config["database"]}"
+      puts "Execute: #{command}"
+      system command
+      puts "Restore complete!"
+
+    rescue
+      $stderr.puts "It seems database #{config["database"]} already exists. Please drop it by hands (rake db:drop) before restore backup."
+      $stderr.puts $!, *($!.backtrace)
+    end
+  end
+
   desc "drop views"
   task :drop_views=>:environment do
     #ActiveRecord::Base.logger=Logger.new STDOUT
     PgGnostic::ViewDefinition.load_declarations(File.join(RAILS_ROOT,'db','views'))
     PgGnostic::ViewDefinition.delete_all
   end
+
   desc "update views"
   task :views=>:environment do
     #ActiveRecord::Base.logger=Logger.new STDOUT
     PgGnostic::ViewDefinition.load_declarations File.join(RAILS_ROOT,'db','views')
     PgGnostic::ViewDefinition.update
   end
+
   desc "updates functions"
   task :functions => [:environment] do
     Dir["#{File.join(RAILS_ROOT,'db','functions')}/*.sql"].sort.each do |f|
@@ -24,6 +99,7 @@ namespace :pg do
       end
     end
   end
+
   desc "updates functions and views"
   task :update=>[:functions,:views] do
   end

data/test/config_test.rb

@@ -3,15 +3,29 @@ require File.dirname(__FILE__) + '/test_helper.rb'
 class PgGnosticViewUtilsTest < ActiveSupport::TestCase
   def test_default_config
     assert_equal('Views',PgGnostic.config.view_model.nest_in_module)
+    assert_equal('db/backup',PgGnostic.config.backup.path)
+    assert_equal('gzip',PgGnostic.config.backup.archive_command)
+    assert_equal('gunzip -c',PgGnostic.config.backup.unarchive_command)
 
     PgGnostic.configure do
+
       view_model do
         nest_in_module 'OtherModule'
       end
+
+      backup do
+        path '/tmp/backups'
+        archive_command 'arch'
+        unarchive_command 'unarch'
+      end
+
     end
     assert_equal('OtherModule',PgGnostic.config.view_model.nest_in_module)
     assert_equal('view_',PgGnostic.config.view_model.prefix_view_name)
-    
+    assert_equal('/tmp/backups',PgGnostic.config.backup.path)
+    assert_equal('arch',PgGnostic.config.backup.archive_command)
+    assert_equal('unarch',PgGnostic.config.backup.unarchive_command)
+
     PgGnostic.clear_config!
   end
 end

metadata

@@ -1,7 +1,12 @@
 --- !ruby/object:Gem::Specification 
 name: pg_gnostic
 version: !ruby/object:Gem::Version 
-  version: 0.0.1
+  prerelease: false
+  segments: 
+  - 0
+  - 0
+  - 2
+  version: 0.0.2
 platform: ruby
 authors: 
 - niquola
@@ -9,19 +14,23 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-03-10 00:00:00 +03:00
+date: 2010-03-31 00:00:00 +04:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
   name: kung_figure
-  type: :runtime
-  version_requirement: 
-  version_requirements: !ruby/object:Gem::Requirement 
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
     requirements: 
     - - "="
       - !ruby/object:Gem::Version 
+        segments: 
+        - 0
+        - 0
+        - 2
         version: 0.0.2
-    version: 
+  type: :runtime
+  version_requirements: *id001
 description: 
 email: niquola@gmail.com
 executables: []
@@ -70,18 +79,20 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements: 
   - - ">="
     - !ruby/object:Gem::Version 
+      segments: 
+      - 0
       version: "0"
-  version: 
 required_rubygems_version: !ruby/object:Gem::Requirement 
   requirements: 
   - - ">="
     - !ruby/object:Gem::Version 
+      segments: 
+      - 0
       version: "0"
-  version: 
 requirements: []
 
 rubyforge_project: 
-rubygems_version: 1.3.5
+rubygems_version: 1.3.6
 signing_key: 
 specification_version: 3
 summary: Rails plugin for postgres