Support components by ensuring we remove any extra 'models' directori…

…es in the path and model names
ctran · Mar 22, 2022 · 96831c1 · 96831c1
1 parent c0aa6ca
commit 96831c1
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/lib/annotate/annotate_models.rb b/lib/annotate/annotate_models.rb
@@ -584,6 +584,10 @@ def get_model_class(file)
  begin
  get_loaded_model(model_path, file) || raise(BadModelFileError.new)
  rescue LoadError
+ if model_path.include?("models/")
+ model_path = model_path.gsub(/models\//, '')
+ retry
+ end
  # this is for non-rails projects, which don't get Rails auto-require magic
  file_path = File.expand_path(file)
  if File.file?(file_path) && Kernel.require(file_path)