ydd 0.0.3 → 0.0.4

data/VERSION

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

data/lib/ydd/application.rb

@@ -1,10 +1,13 @@
 require 'thor'
+require 'ruby-debug'
 
 module YDD
   class Application < Thor
     include Thor::Actions
     
-    method_options :env => :string, :force => :boolean
+    method_options :env => :string, :force => :boolean,
+                   :skip_schema => :boolean, :skip_data => :boolean,
+                   :tables => :string
     
     desc "dump DUMP_DIR [APP_DIR='.'] [OPTIONS]", "Dumps the database contents and schema."
     def dump(dump_dir, app_dir = ".")
@@ -27,7 +30,7 @@ module YDD
       @app_dir   = File.expand_path(app_dir)
       @db_config = File.join(@app_dir, "config", "database.yml")
       if !File.directory?(@app_dir)
-        die "The given application directory, #{@application}, does not exist."
+        die "The given application directory, #{@app_dir}, does not exist."
       elsif !File.exist?(@db_config)
         die "#{@db_config} does not exist."
       elsif YDD.configuration_from(@db_config).blank?
@@ -38,7 +41,10 @@ module YDD
     end
     
     def setup!(dump_dir, app_dir, extra = {})
-      validate_environment!
+      YDD.env         = options.env if options.env.present?
+      YDD.tables      = options.tables if options.tables.present?
+      YDD.skip_data   = options.skip_data?
+      YDD.skip_schema = options.skip_schema?
       setup_directories!    dump_dir, app_dir, extra
       connect_to_database!
     end
@@ -47,10 +53,6 @@ module YDD
       YDD.connect_from @db_config
     end
     
-    def validate_environment!
-      YDD.env = options.env if options.env.present?
-    end
-    
     def die(error, code = 1)
       say error
       exit code

data/lib/ydd/serialization_helper.rb

@@ -129,7 +129,9 @@ module YDD
       end
 
       def self.tables
-        YDD.connection.tables.reject { |table| ['schema_info', 'schema_migrations'].include?(table) }
+        base_tables = YDD.connection.tables
+        base_tables &= YDD.tables if YDD.tables.present?
+          
       end
 
       def self.dump_table(io, table)

data/lib/ydd.rb

@@ -12,6 +12,24 @@ module YDD
   autoload :SerializationHelper, 'ydd/serialization_helper'
   autoload :Application,         'ydd/application'
 
+  mattr_accessor :skip_schema, :skip_data
+  
+  def self.skip_schema?
+    !!skip_schema
+  end
+  
+  def self.skip_data?
+    !!skip_data
+  end
+
+  def self.tables=(value)
+    @@tables = value.blank? ? nil : Array(value).join(",").split(",")
+  end
+  
+  def self.tables
+    @@tables ||= nil
+  end
+
   def self.env=(value)
     if value.blank?
       @@env = nil
@@ -45,8 +63,8 @@ module YDD
   
   def self.dump(directory)
     FileUtils.mkdir_p directory
-    SchemaManager.dump File.join(directory, "schema.rb")
-    DataManager.dump   File.join(directory, "data.yml")
+    SchemaManager.dump File.join(directory, "schema.rb") unless skip_schema?
+    DataManager.dump   File.join(directory, "data.yml")  unless skip_data?
   end
   
   def self.load(directory)
@@ -54,8 +72,8 @@ module YDD
       raise Error, "Please provide a valid directory - #{directory} doesn't exist."
     end
     check_files! directory, "schema.rb", "data.yml"
-    SchemaManager.load File.join(directory, "schema.rb")
-    DataManager.load   File.join(directory, "data.yml")
+    SchemaManager.load File.join(directory, "schema.rb") unless skip_schema?
+    DataManager.load   File.join(directory, "data.yml")  unless skip_data?
   end
   
   def self.check_files!(dir, *files)

data/ydd.gemspec

@@ -5,11 +5,11 @@
 
 Gem::Specification.new do |s|
   s.name = %q{ydd}
-  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 = ["Adam Wiggins", "Orion Henry", "Darcy Laycock"]
-  s.date = %q{2010-09-19}
+  s.date = %q{2010-09-20}
   s.default_executable = %q{ydd}
   s.description = %q{
 YDD is a tool (a fork of yaml_db really) to make it generally easy

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: ydd
 version: !ruby/object:Gem::Version 
-  hash: 25
+  hash: 23
   prerelease: false
   segments: 
   - 0
   - 0
-  - 3
-  version: 0.0.3
+  - 4
+  version: 0.0.4
 platform: ruby
 authors: 
 - Adam Wiggins
@@ -17,7 +17,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-09-19 00:00:00 +08:00
+date: 2010-09-20 00:00:00 +08:00
 default_executable: ydd
 dependencies: 
 - !ruby/object:Gem::Dependency