File size: 465 Bytes
52007f8
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
#!/usr/bin/python
# -*- coding:utf-8 -*-
import os
from os.path import basename, splitext


def get_filename(path):
    return basename(splitext(path)[0])


def cnt_num_files(directory, recursive=False):
    cnt = 0
    for sub in os.listdir(directory):
        sub = os.path.join(directory, sub)
        if os.path.isfile(sub):
            cnt += 1
        elif os.path.isdir(sub) and recursive:
            cnt += cnt_num_files(sub, recursive=True)
    return cnt