MultiSystems.from_dir support wildcard

Author: felix5572

README.md

@@ -65,7 +65,10 @@ The following commands relating to `Class dpdata.MultiSystems` may be useful.
 # load data
 
 xyz_multi_systems = dpdata.MultiSystems.from_file(file_name='tests/xyz/xyz_unittest.xyz',fmt='quip/gap/xyz')
-vasp_multi_systems =  dpdata.MultiSystems.from_dir(dir_name='./mgal_outcar', file_name='OUTCAR', fmt='vasp/outcar')
+vasp_multi_systems = dpdata.MultiSystems.from_dir(dir_name='./mgal_outcar', file_name='OUTCAR', fmt='vasp/outcar')
+
+# use wildcard
+vasp_multi_systems = dpdata.MultiSystems.from_dir(dir_name='./mgal_outcar', file_name='*OUTCAR', fmt='vasp/outcar')
 
 # print the multi_system infomation
 print(xyz_multi_systems)

dpdata/system.py

@@ -1,5 +1,6 @@
 #%%
 import os
+import glob
 import numpy as np
 import dpdata.lammps.lmp
 import dpdata.lammps.dump
@@ -954,11 +955,7 @@ def from_file(cls,file_name,fmt):
     @classmethod
     def from_dir(cls,dir_name, file_name, fmt='auto'):
         multi_systems = cls()
-        target_file_list = []
-        for ii in os.walk(dir_name):
-            if file_name in ii[2]:
-                target_file = os.path.join(ii[0], file_name)
-                target_file_list.append(target_file)
+        target_file_list = glob.glob('./{}/**/{}'.format(dir_name, file_name), recursive=True)
         for target_file in target_file_list:
             multi_systems.append(LabeledSystem(file_name=target_file, fmt=fmt))
         return multi_systems