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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
|
#! /usr/bin/env python
"""
Process the examples in the documentation for inclusion in the Gallery:
- create a "documentation" directory within "examples"
- add a README.txt file
- copy the examples from the documentation, bu remove the "doc_" from the
filename
- add the required docstring to the files for proper rendering
- copy the data files
"""
import os
import time
basedir = os.getcwd()
examples_dir = os.path.abspath(os.path.join(basedir, '..', 'examples'))
files = [fn for fn in os.listdir(examples_dir) if fn.startswith('doc_')]
examples_documentation_dir = os.path.join(examples_dir, 'documentation')
os.makedirs(examples_documentation_dir, exist_ok=True)
scripts_to_run = []
with open(os.path.join(examples_documentation_dir, 'README.txt'), 'w') as out:
out.write("Examples from the documentation\n")
out.write("===============================\n\n")
out.write("Below are all the examples that are part of the lmfit documentation.")
for fn in files:
inp_path = os.path.join(examples_dir, fn)
with open(inp_path) as inp:
script_text = inp.read()
gallery_file = os.path.join(examples_documentation_dir, fn[4:])
with open(gallery_file, 'w') as out:
msg = "" # add optional message f
out.write('"""\n{}\n{}\n\n{}\n"""\n'.format(fn, "="*len(fn), msg))
out.write('##\nimport warnings\nwarnings.filterwarnings("ignore")\n##\n')
out.write(script_text)
# make sure the saved Models and ModelResult are available
if 'save' in fn:
scripts_to_run.append(fn[4:])
time.sleep(1.0)
os.system('cp {}/*.dat {}'.format(examples_dir, examples_documentation_dir))
os.system('cp {}/*.csv {}'.format(examples_dir, examples_documentation_dir))
os.system('cp {}/*.sav {}'.format(examples_dir, examples_documentation_dir))
os.chdir(examples_documentation_dir)
for script in scripts_to_run:
os.system('python {}'.format(script))
os.chdir(basedir)
time.sleep(1.0)
# data files for the other Gallery examples
os.system('cp {}/*.dat .'.format(examples_documentation_dir))
os.system('cp {}/*.csv .'.format(examples_documentation_dir))
os.system('cp {}/*.sav .'.format(examples_documentation_dir))
|
