From 9e2bec53a6353268a14a13df05f5f0328e5dbc95 Mon Sep 17 00:00:00 2001
From: liyijie <liyijie825@gmail.com>
Date: Wed, 14 Apr 2021 15:41:32 +0800
Subject: [PATCH] import_headers fix with import_headers_result

---
 lib/tallty_import_export/import.rb | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/lib/tallty_import_export/import.rb b/lib/tallty_import_export/import.rb
index 708169d..19c8f9f 100644
--- a/lib/tallty_import_export/import.rb
+++ b/lib/tallty_import_export/import.rb
@@ -97,7 +97,11 @@ module TalltyImportExport
     end
 
     def import_headers_result
-      klass.try(:headers) || klass.try(:import_headers) || klass.try(:model_headers) || (raise ArgumentError.new('missing import_headers_result'))
+      @headers || import_headers
+    end
+
+    def import_headers
+      @headers || klass.try(:import_headers) || klass.try(:model_headers) || (raise ArgumentError.new('missing import_headers'))
     end
 
     # 只保留 key, name, json, 合并到 import_header
--
libgit2 0.27.1