db_stager 0.2.0 → 0.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 5596cc698905e2cb5c03fbcebee2a272403360f4
-  data.tar.gz: 2e27f20781268cc10f257c000cec3307d8b043cf
+  metadata.gz: 7144864a030786eac584c002945d972aebf2c2f9
+  data.tar.gz: 5322fb8b234927617f4696ae6d4d5dfbcd3c5fa1
 SHA512:
-  metadata.gz: ecd312cb600acc15779f8336443da10cd65997fd80a18a08744a5d3f0d7a88fde9aa99c253d972b128ab7779633b44b29654d2ca1830eefb44d69865cd9c3226
-  data.tar.gz: 9acec0d9ffaf4eaff6dcbd7b034b00828a93c0c31e2c2a1a8b53f3b24cd629dcf213b0e8af4efad56c93c0dc2ea65d015d52a932e07a46627bc6ae4601e668a6
+  metadata.gz: 3102262cec9d51155b1e83cfd56649b552e700c872ab2f5a23fe70f8cf641d14ec35b77ca5d1df351dc8d07f535cdb83e7fa027cf53654e179d1ab6289577bef
+  data.tar.gz: fe5e70470b372f04ed99a6cd2ba0d3a77f6fd285805dfa5c7619160e0ff7254ce71f89c9724a83129c95b375dc29c1b2334a037e89af2d89ab4394567c406511

data/Gemfile.lock

@@ -1,8 +1,10 @@
 PATH
   remote: .
   specs:
-    db_stager (0.2.0)
+    db_stager (0.2.1)
       dotenv (~> 2.2)
+      faker (~> 1.9)
+      hashie (~> 3.5)
       sequel (~> 4.41.0)
       thor (~> 0.19)
 
@@ -10,8 +12,14 @@ GEM
   remote: https://rubygems.org/
   specs:
     coderay (1.1.2)
+    concurrent-ruby (1.0.5)
     diff-lcs (1.3)
     dotenv (2.5.0)
+    faker (1.9.1)
+      i18n (>= 0.7)
+    hashie (3.5.7)
+    i18n (1.1.0)
+      concurrent-ruby (~> 1.0)
     method_source (0.9.0)
     pry (0.11.3)
       coderay (~> 1.1.0)

data/db_stager.gemspec

@@ -29,4 +29,6 @@ Gem::Specification.new do |spec|
   spec.add_runtime_dependency "dotenv", "~> 2.2"
   spec.add_runtime_dependency "thor", "~> 0.19"
   spec.add_runtime_dependency "sequel", "~> 4.41.0"
+  spec.add_runtime_dependency "hashie", "~> 3.5"
+  spec.add_runtime_dependency "faker", "~> 1.9"
 end

data/lib/db_stager/cli.rb

@@ -2,7 +2,8 @@ require 'thor'
 require 'sequel'
 require 'yaml'
 require 'erb'
-require 'erb'
+require 'hashie'
+require 'faker'
 require 'dotenv/load'
 
 module DbStager
@@ -18,18 +19,28 @@ module DbStager
     end
 
     method_option :environment, type: :string, aliases: '-e'
-    method_option :file, type: :string, aliases: '-f'
     desc "capture", "grab a new stager copy of your db"
     def capture(dir=".")
       Dir.chdir(dir)
 
-      sh 'rake db:structure:dump'
+      run 'rake db:structure:dump'
 
-      capture_file = options[:file] || File.join(Dir.pwd, "db", "#{environment}_capture.sql")
-      File.open(capture_file, "w") do |file|
+      capture_file = File.join(Dir.pwd, "db", "structure.sql")
+      File.open(capture_file, "a") do |file|
         db.tables.each do |table_name|
+          next if excluded_tables.include?(table_name)
           items = db[table_name]
           items.each do |item|
+            if fields[table_name]
+              fields[table_name].each do |field_name, task|
+                if task
+                  item[field_name.to_sym] = send(task)
+                else
+                  item[field_name.to_sym] = nil
+                end
+              end
+              item
+            end
             file.puts "#{items.insert_sql(item)};"
             file.puts
           end
@@ -38,8 +49,8 @@ module DbStager
     end
 
 
-    desc "upload", "upload your most recent capture to S3"
-    desc "download", "download your most recent capture from S3"
+    #desc "upload", "upload your most recent capture to S3"
+    #desc "download", "download your most recent capture from S3"
 
     method_option :environment, type: :string, aliases: '-e'
     method_option :structure_file, type: :string, aliases: '-s'
@@ -47,19 +58,48 @@ module DbStager
     desc "load", "overwrite the current db with the most recently downloaded stager db"
     def load(dir=".")
       Dir.chdir(dir)
-      structure_file = options[:structure_file] || File.join(Dir.pwd, "db", "structure.sql")
-      db.run File.read(struture_file)
-      capture_file = options[:file] || File.join(Dir.pwd, "db", "#{environment}_capture.sql")
-      db.run File.read(capture_file)
+      run 'rake db:structure:load'
     end
-  end
 
-  protected
-  def environment
-    @environment ||= options[:environment] || ENV['RAILS_ENV'] || ENV['RACK_ENV'] || "development"
-  end
+    protected
+    def environment
+      @environment ||= options[:environment] || ENV['RAILS_ENV'] || ENV['RACK_ENV'] || "development"
+    end
+
+    def db
+      @db ||= Sequel.connect(YAML.load(ERB.new(File.read(File.join(Dir.pwd, 'config', 'database.yml'))).result)[environment])
+    end
+
+    def config_file
+      return @config_file if @config_file
+      config_file_path = File.join(Dir.pwd, 'config', 'stage_hand.yml')
+      if File.exists?(config_file_path)
+        @config_file = Hashie::Mash.load(ERB.new(File.read(config_file_path)))
+      else
+        @config_file = {}
+      end
+    end
+
+    def excluded_tables
+      return @excluded_tables if @excluded_tables
+      @excluded_tables = config_file[:excluded_tables] || []
+      @excluded_tables.collect! { |t| t.to_sym }
+      # always skip schema_migrations, they get set by the structure file
+      @excluded_tables = @excluded_tables | [:schema_migrations]
+      return @excluded_tables
+    end
+
+    def fields
+      return @fields if @fields
+      # prototype is people: [{name: "Faker::Name.unique.name"}, {email: false}]
+      # key is table name
+      # fake can be any executable faker string
+      # if fake is not set, no edit is made
+      # skip defaults to false if not set
+      # you should set either fake or skip, not both... duh
+      @fields = config_file[:fields] || {}
+      return @fields
+    end
 
-  def db
-    @db ||= Sequel.connect(YAML.load(ERB.new(File.read(File.join(Dir.pwd, 'config', 'database.yml'))).result)[environment])
   end
 end

data/lib/db_stager/version.rb

@@ -1,3 +1,3 @@
 module DbStager
-  VERSION = "0.2.0"
+  VERSION = "0.2.1"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: db_stager
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.2.1
 platform: ruby
 authors:
 - Rob Kaufman
@@ -108,6 +108,34 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: 4.41.0
+- !ruby/object:Gem::Dependency
+  name: hashie
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.5'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.5'
+- !ruby/object:Gem::Dependency
+  name: faker
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.9'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.9'
 description: Perform backups and restores of databases, scrubbing records along the
   way. This allows taking limited snapshots of large databases and keeps customer
   info out of your staging site.