优化

5 years ago · 0aafde21e4
parent 427eac4313
commit 0aafde21e4
1 changed files with 6 additions and 2 deletions
--- a/app/src/main/java/io/legado/app/ui/importbook/ImportBookActivity.kt
+++ b/app/src/main/java/io/legado/app/ui/importbook/ImportBookActivity.kt
@ -87,9 +87,13 @@ class ImportBookActivity : VMBaseActivity<ImportBookViewModel>(R.layout.activity
                    tv_path.text = tv_path.text.toString() + it.name + File.separator
                }
            }
-            doc?.listFiles()?.let {
-                importBookAdapter.setItems(it.toList())
+            val docList = arrayListOf<DocumentFile>()
+            doc?.listFiles()?.forEach {
+                if (it.isDirectory || it.name?.endsWith(".txt", true) == true) {
+                    docList.add(it)
+                }
            }
+            importBookAdapter.setItems(docList)
        }
    }