Skip to content

Commit 7117da3

Browse files
authored
Merge branch 'master' into refactor
2 parents dd55cd0 + 2c9cb62 commit 7117da3

File tree

1 file changed

+3
-7
lines changed

1 file changed

+3
-7
lines changed

setup.py

Lines changed: 3 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
import os
2+
from pathlib import Path, PurePath
23
from setuptools import setup, find_packages
34

45
SUFFIX = '.jinja2'
@@ -9,13 +10,8 @@
910
def gen_data(data_root='static'):
1011
"""just for collect static files.
1112
"""
12-
data = []
13-
for fn in os.listdir(os.path.join(src_path, data_root)):
14-
if os.path.isdir(os.path.join(src_path, data_root, fn)):
15-
data.extend(gen_data(os.path.join(data_root, fn)))
16-
if fn.endswith(SUFFIX):
17-
data.append(os.path.join(data_root, fn))
18-
return data
13+
return [fpath for fpath in Path(
14+
PurePath(src_path) / data_root).glob(f'**/*{SUFFIX}')]
1915

2016

2117
package_data = gen_data()

0 commit comments

Comments
 (0)