irm_yaml_db 0.0.3 → 0.0.4

data/VERSION

@@ -1 +1 @@
-0.0.3
+0.0.4

data/irm_yaml_db.gemspec

@@ -5,7 +5,7 @@
 
 Gem::Specification.new do |s|
   s.name = "irm_yaml_db"
-  s.version = "0.0.3"
+  s.version = "0.0.4"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["uudui"]

data/lib/serialization_helper.rb

@@ -172,11 +172,16 @@ module SerializationHelper
     end
 
     def self.base_tables
-      sql = "select table_name from information_schema.tables where table_schema='irm_dev' and table_type='base table';"
+
+      sql = "select table_name from information_schema.tables where table_schema='#{database}' and table_type='base table';"
       base_tables = ActiveRecord::Base.connection.execute(sql, 'SCHEMA').collect do |field|
         field.first
       end
-      base_tables.reject { |table| ['schema_info', 'schema_migrations','irm_data_accesses_top_org_t'].include?(table) }
+      base_tables.reject { |table| ['schema_info', 'schema_migrations'].include?(table) }
+    end
+
+    def self.database
+      Rails.configuration.database_configuration[Rails.env]["database"]
     end
 
     def self.tables

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: irm_yaml_db
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.4
   prerelease: 
 platform: ruby
 authors: