Merge pull request #98 from TheoTT/feature/rewrite-gen_data-by-pathlib

rewrite gen_data function by pathlib

Author: TTWShell

setup.py

@@ -1,4 +1,5 @@
 import os
+from pathlib import Path, PurePath
 from setuptools import setup, find_packages
 
 SUFFIX = '.jinja2'
@@ -9,13 +10,8 @@
 def gen_data(data_root='static'):
     """just for collect static files.
     """
-    data = []
-    for fn in os.listdir(os.path.join(src_path, data_root)):
-        if os.path.isdir(os.path.join(src_path, data_root, fn)):
-            data.extend(gen_data(os.path.join(data_root, fn)))
-        if fn.endswith(SUFFIX):
-            data.append(os.path.join(data_root, fn))
-    return data
+    return [fpath for fpath in Path(
+        PurePath(src_path) / data_root).glob(f'**/*{SUFFIX}')]
 
 
 package_data = gen_data()