Use absolute path in require and catch nil file_name in writer

plentz · Jan 25, 2014 · d6890fa · d6890fa
1 parent 9727359
commit d6890fa
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/lib/lol_dba/migration.rb b/lib/lol_dba/migration.rb
@@ -4,7 +4,7 @@ class Migration
 
     def initialize(migration_file)
       self.full_name = File.basename(migration_file, ".rb")
-      require Rails.root + "db/migrate/#{full_name}.rb"
+      require Rails.root.join(migration_file)
     end
 
     def number

diff --git a/lib/lol_dba/writer.rb b/lib/lol_dba/writer.rb
@@ -17,6 +17,7 @@ def path
       end
 
       def write(string)
+        return if file_name.nil?
         File.open(path, 'a') { |file| file << string; file << ";\n" }
       end
     end