copy_db_from_prod 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: b9e19bb98ebc1cfd55e787e300b1435108fa2335
-  data.tar.gz: b578debcc2bbd4d3b617285a1333795a71998103
+  metadata.gz: 3304f3a840aae3a772dcd7ee6a0c15978d0a9d6e
+  data.tar.gz: eef944dcfc6e25a7434082f670fba43548b080ff
 SHA512:
-  metadata.gz: db3bb93ac39b0dfb0929915b3f8db13a85dd99759cec9b8e825698db837cdf4dfa4dbcde55b84bc53713c54cb547ba0aa808ae850fbbe7a75dad2d00b69a7e81
-  data.tar.gz: d7456991274c5b99bb1d658f6fce837ed976bd9532074cd919c412860f6b6fa370e7b1a68004b37ac4882abca1064eb7462dda9a94e1e05aa4edb7507894bd04
+  metadata.gz: 2dbafac3077cae23f82ea84827eff4a1c5de20c202ae33f98363638742f96c53ca745c2de7d9d13f31e374c4c4a4be9754f78a1f3c3dde12f3dbc3fdd10a6550
+  data.tar.gz: cf2ca1887b2d161d2540a5323d1c1fb79ca72a4da550d3294ecc7ee5cc9aa1936304906f44f5b569decc6b0da4a0a05a7d4788ccae1ee9eb1805cc24ad6b854f

data/bin/copy_db_from_prod

@@ -7,7 +7,7 @@ def echo_wrong_args
 end
 
 def echo_help
-  puts("copy_db_from_prod -h <host> -u <user> -p <path>
+  puts("copy_db_from_prod -h <host> -u <user> -p <path> --schema-only
     <user> - default 'deploy'
     <path> - default '/projects/insales3', without / at the end")
 end

data/lib/copy_db_from_prod.rb

@@ -3,7 +3,7 @@ require 'yaml'
 require 'pg'
 
 class CopyDbFromProd
-  attr_accessor :cf
+  attr_accessor :cf, :conn
 
   def self.hi
     puts "Hello world!"
@@ -21,12 +21,21 @@ class CopyDbFromProd
 
   def download_schema
     Net::SSH.start(cf[:host], cf[:user], key_data: cf[:keys], keys_only: TRUE) do |ssh|
-      @prod_conf = YAML.load(ssh.exec!("cat #{cf[:deploy_to]}/config/database.yml"))['production']
+      yml_text = ssh.exec!("cat #{cf[:deploy_to]}/config/database.yml")
+      begin
+        @prod_conf ||= YAML.load(yml_text)['production']
+      rescue => e
+        raise "Can't access to database.yml, check -p params\n #{e}"
+      end
       prod_data = ssh.exec!(dump_schema)
       if prod_data.include?('pg_dump') || prod_data.include?('warn')
-        raise "error: pg_dump dosen't work correct \n #{prod_data}"
+        raise "error: pg_dump dosen't work correct\n #{prod_data}"
       else
-        return Zlib::GzipReader.new(StringIO.new(ssh.exec!(prod_data))).read
+        begin
+          return Zlib::GzipReader.new(StringIO.new(prod_data)).read
+        rescue => e
+          raise "#{e}\n #{prod_data}"
+        end
       end
     end
   end
@@ -40,18 +49,39 @@ class CopyDbFromProd
     "#{cmd.join(' ')} --schema-only --no-owner --no-privileges -Z9"
   end
 
+  def drop_restore_table
+    conn.exec("SELECT table_name
+                FROM information_schema.tables
+                WHERE table_schema = 'public'
+                AND table_name LIKE '%_201%'
+                ORDER BY table_name"
+    ) do |result|
+      conn.exec(result.map { |row| "DROP TABLE #{row['table_name']}" }.join(';'))
+    end
+  end
+
   def load_schema
     @dev_conf = YAML.load(File.open('config/database.yml', 'r').read)['development']
-    conn = PGconn.open(host: @dev_conf['host'],
+    @conn = PGconn.open(host: @dev_conf['host'],
                      dbname: 'postgres',
                        user: @dev_conf['username'],
                    password: @dev_conf['password'])
     conn.exec("DROP DATABASE IF EXISTS #{@dev_conf['database']}")
     conn.exec("CREATE DATABASE #{@dev_conf['database']}")
-    conn = PGconn.open(host: @dev_conf['host'],
+    @conn = PGconn.open(host: @dev_conf['host'],
                      dbname: @dev_conf['database'],
                        user: @dev_conf['username'],
                    password: @dev_conf['password'])
-    conn.exec(download_schema)
+    schema = download_schema
+    begin
+      conn.exec(schema)
+    rescue => e
+      raise "Database #{@dev_conf['database']}\n#{e}\n #{schema}"
+    end
+    drop_restore_table if 'insales_dev' == @dev_conf['database']
+  end
+
+  def copy_env_table
+
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: copy_db_from_prod
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - vgulaev