Skip to content

Commit bfa1993

Browse files
add script to generate a example folder for collecting many files
1 parent ccdb248 commit bfa1993

File tree

3 files changed

+31
-0
lines changed

3 files changed

+31
-0
lines changed
Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
foo_*
Lines changed: 28 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,28 @@
1+
import argparse
2+
import pathlib
3+
4+
5+
HERE = pathlib.Path(__file__).parent
6+
TEST_CONTENT = (HERE / "template_test.py").read_bytes()
7+
8+
parser = argparse.ArgumentParser()
9+
parser.add_argument("numbers", nargs="*", type=int)
10+
11+
12+
def generate_folders(root, elements, *more_numbers):
13+
fill_len = len(str(elements))
14+
if more_numbers:
15+
for i in range(elements):
16+
new_folder = root.joinpath(f"foo_{i:0>{fill_len}}")
17+
new_folder.mkdir()
18+
new_folder.joinpath("__init__.py").write_bytes(TEST_CONTENT)
19+
generate_folders(new_folder, *more_numbers)
20+
else:
21+
for i in range(elements):
22+
new_test = root.joinpath(f"test_{i:0<{fill_len}}.py")
23+
new_test.write_bytes(TEST_CONTENT)
24+
25+
26+
if __name__ == "__main__":
27+
args = parser.parse_args()
28+
generate_folders(HERE, *(args.numbers or (10, 100)))
Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,2 @@
1+
def test_x():
2+
pass

0 commit comments

Comments
 (0)