capistrano3-postgres 0.0.3 → 0.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: f7a08e03347397dea2b027afa75fac9929f82184
-  data.tar.gz: 7c3b1d8765ea24754701b0e31951d4f30a7531cd
+  metadata.gz: 087895a85fd5bcae1fd8aa7ce96a921fc7f738e5
+  data.tar.gz: 9a919e94bea2d1b286949a0b435742b05c686041
 SHA512:
-  metadata.gz: 8527c0f1c0a9a97195909d7c095a267c0508a43bca5615098d1c3aa238e3c518feb9692e7771d6d103bf522b9e80260fe91f1bac26b3ed16b26f58e1c7611e66
-  data.tar.gz: 703183c9f4a37a4632aa9e960988cd110a13a515b4dd8929f9a837080a7e24bd09a3b806620ccfac47f67de60db14b7b49a72bd67ca053feacd5729e62db6de0
+  metadata.gz: f4b852cbfd8e8eadbd784fb1eaab73a9908ea3fcf7cf94689e490f8ae85dfd5af229388fe25c2d4717fe16ea00a95be2f7ee083abd30f58ec2e0370fc82f33a4
+  data.tar.gz: b932647d0ed037a1b21c2bffeeb4720d05fde1e596acb908ce006008126f58675db5efd716e12c473bd61f897db862eb519e07ce04b0eb295dc2ef7b30fb98fd

data/README.md

@@ -4,11 +4,11 @@
 
 Add this line to your application's Gemfile:
 
-    gem 'capistrano3-postgres'
+    gem 'capistrano3-postgres', require: false
 
 or:
 
-    gem 'capistrano3-postgres' , group: :development
+    gem 'capistrano3-postgres', require: false, group: :development
 
 And then execute:
 
@@ -43,6 +43,10 @@ Sometimes it's a good idea to create dump before each deploy.
 before 'deploy:starting', 'postgres:backup:create'
 ```
 
+All downloaded dump files will be deleted after importing. If you want to keep them, you can set:
+```
+set :postgres_keep_local_dumps, 5 # Will keep 5 last dump files.
+```
 
 ## Contributing
 
@@ -51,9 +55,3 @@ before 'deploy:starting', 'postgres:backup:create'
 3. Commit your changes (`git commit -am 'Add some feature'`)
 4. Push to the branch (`git push origin my-new-feature`)
 5. Create new Pull Request
-
-
-
-
-
-

data/lib/capistrano3/postgres/version.rb

@@ -1,5 +1,5 @@
 module Capistrano3
   module Postgres
-    VERSION = "0.0.3"
+    VERSION = "0.1.0"
   end
 end

data/lib/capistrano3/tasks/postgres.rb

@@ -3,6 +3,7 @@ namespace :load do
     set :postgres_backup_dir, -> { 'postgres_backup' }
     set :postgres_role, :db
     set :postgres_env, -> { fetch(:rack_env, fetch(:rails_env, fetch(:stage))) }
+    set :postgres_keep_local_dumps, 0
     set :postgres_remote_sqlc_file_path, -> { nil }
     set :postgres_local_database_config, -> { nil }
     set :postgres_remote_database_config, -> { nil }
@@ -24,7 +25,7 @@ namespace :postgres do
           set :postgres_remote_sqlc_file_path, "#{shared_path}/#{fetch(:postgres_backup_dir)}/#{file_name}"
         end
 
-        execute "PGPASSWORD=#{config['password']} pg_dump -U #{config['user'] || config['username']} -h #{config['host']} -Fc --file=#{fetch(:postgres_remote_sqlc_file_path)} #{config['database']}"
+        execute "PGPASSWORD=#{config['password']} pg_dump -U #{config['username'] || config['user']} -h #{config['host']} -Fc --file=#{fetch(:postgres_remote_sqlc_file_path)} #{config['database']}"
       end
     end
 
@@ -37,6 +38,13 @@ namespace :postgres do
         end
 
         download!(fetch(:postgres_remote_sqlc_file_path), "tmp/#{fetch :postgres_backup_dir}/#{Pathname.new(fetch(:postgres_remote_sqlc_file_path)).basename}")
+        begin
+          remote_file = fetch(:postgres_remote_sqlc_file_path)
+        rescue SSHKit::Command::Failed => e
+          warn e.inspect
+        ensure
+          execute "rm #{remote_file}"
+        end
       end
     end
 
@@ -55,13 +63,14 @@ namespace :postgres do
           file_path = "tmp/#{fetch :postgres_backup_dir}/#{file_name}"
           begin
             pgpass_path = File.join(Dir.pwd, '.pgpass')
-            File.open(pgpass_path, 'w+', 0600) { |file| file.write("*:*:*:#{config['user'] || config['username']}:#{config['password']}") }
-            execute "PGPASSFILE=#{pgpass_path} pg_restore -c -U #{config['user'] || config['username']} --no-owner -h #{config['host']} -d #{fetch(:database_name)} #{file_path}"
+            File.open(pgpass_path, 'w+', 0600) { |file| file.write("*:*:*:#{config['username'] || config['user']}:#{config['password']}") }
+            execute "PGPASSFILE=#{pgpass_path} pg_restore -c -U #{config['username'] || config['user']} --no-owner -h #{config['host']} -p #{config['port'] || 5432 } -d #{fetch(:database_name)} #{file_path}"
           rescue SSHKit::Command::Failed => e
             warn e.inspect
             info 'Import performed successfully!'
           ensure
             File.delete(pgpass_path) if File.exist?(pgpass_path)
+            File.delete(file_path) if (fetch(:postgres_keep_local_dumps) == 0) && File.exist?(file_path)
           end
         end
       end
@@ -82,6 +91,15 @@ namespace :postgres do
         end
       end
     end
+
+    desc "Cleanup old local dumps"
+    task :cleanup do
+      run_locally do
+        dir = "tmp/#{fetch :postgres_backup_dir}"
+        file_names = capture("ls -v #{dir}").split(/\n/).sort
+        file_names[0...-fetch(:postgres_keep_local_dumps)].each {|file_name| File.delete("#{dir}/#{file_name}") }
+      end
+    end
   end
 
   desc 'Replecate database locally'
@@ -92,6 +110,7 @@ namespace :postgres do
     invoke "postgres:backup:create"
     invoke "postgres:backup:download"
     invoke "postgres:backup:import"
+    invoke("postgres:backup:cleanup") if fetch(:postgres_keep_local_dumps) > 0
   end
 
   # Grabs local database config before importing dump

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: capistrano3-postgres
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.1.0
 platform: ruby
 authors:
 - Alex Krasynskyi
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-06-03 00:00:00.000000000 Z
+date: 2015-09-03 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: capistrano