notebook/generate_script.py at master · namnd95/notebook · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
import os

# ttt is local server folder
NON_DIRECTORY = ['.git', 'content', 'ttt']

directories = []
files = []

def get_list_file(directory):
    return os.walk(directory).next()[2]

def get_list_dir(directory):
    return os.walk(directory).next()[1]

def travel(directory, root=True):
    list_dir = get_list_dir(directory)
    list_file = get_list_file(directory)

    directories.append(directory)
    for dir in list_dir:
        if dir[0] != '.' and dir not in NON_DIRECTORY:
            travel(directory+'/'+dir, False)

    if not root:
        for file in list_file:
            files.append(directory+'/'+file)

def name_part(file_name):
    return file_name[:file_name.rfind('.')]

def generate_script():
    with open('script.sh','w') as f:
        #gen directory
        for dir in directories:
            if dir=='.':
                dir = ''
            f.write('mkdir content/%s\n' % dir)

        #gen nbconvert
        for file in files:
            extra = ''
            for i in xrange(file.count('/')-1):
                extra += '../'
            f.write('jupyter nbconvert --to html --template basic %s --output %scontent/%s\n' % (file, extra, name_part(file)))

if __name__ == '__main__':
    travel('.')
    generate_script()
    print directories
    print files