2 <!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN"
3 "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">
6 <html xmlns="http://www.w3.org/1999/xhtml">
8 <meta http-equiv="Content-Type" content="text/html; charset=utf-8" />
10 <title>src.fork — salomeTools 5.0.0dev documentation</title>
12 <link rel="stylesheet" href="../../_static/alabaster.css" type="text/css" />
13 <link rel="stylesheet" href="../../_static/pygments.css" type="text/css" />
14 <link rel="stylesheet" href="/volatile/wambeke/SAT5/SAT5_S840_MATIX24/SAT/doc/src/custom.css" type="text/css" />
16 <script type="text/javascript">
17 var DOCUMENTATION_OPTIONS = {
20 COLLAPSE_INDEX: false,
25 <script type="text/javascript" src="../../_static/jquery.js"></script>
26 <script type="text/javascript" src="../../_static/underscore.js"></script>
27 <script type="text/javascript" src="../../_static/doctools.js"></script>
28 <script type="text/javascript" src="http://cdn.mathjax.org/mathjax/latest/MathJax.js?config=TeX-AMS-MML_HTMLorMML"></script>
29 <link rel="top" title="salomeTools 5.0.0dev documentation" href="../../index.html" />
30 <link rel="up" title="Module code" href="../index.html" />
32 <link rel="stylesheet" href="../../_static/custom.css" type="text/css" />
35 <meta name="viewport" content="width=device-width, initial-scale=0.9, maximum-scale=0.9" />
41 <div class="document">
42 <div class="documentwrapper">
43 <div class="bodywrapper">
44 <div class="body" role="main">
46 <h1>Source code for src.fork</h1><div class="highlight"><pre>
47 <span class="c">#!/usr/bin/env python</span>
48 <span class="c">#-*- coding:utf-8 -*-</span>
49 <span class="c"># Copyright (C) 2010-2013 CEA/DEN</span>
50 <span class="c">#</span>
51 <span class="c"># This library is free software; you can redistribute it and/or</span>
52 <span class="c"># modify it under the terms of the GNU Lesser General Public</span>
53 <span class="c"># License as published by the Free Software Foundation; either</span>
54 <span class="c"># version 2.1 of the License.</span>
55 <span class="c">#</span>
56 <span class="c"># This library is distributed in the hope that it will be useful,</span>
57 <span class="c"># but WITHOUT ANY WARRANTY; without even the implied warranty of</span>
58 <span class="c"># MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU</span>
59 <span class="c"># Lesser General Public License for more details.</span>
60 <span class="c">#</span>
61 <span class="c"># You should have received a copy of the GNU Lesser General Public</span>
62 <span class="c"># License along with this library; if not, write to the Free Software</span>
63 <span class="c"># Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA</span>
65 <span class="kn">import</span> <span class="nn">os</span>
66 <span class="kn">import</span> <span class="nn">sys</span>
67 <span class="kn">import</span> <span class="nn">time</span>
68 <span class="kn">import</span> <span class="nn">pickle</span>
69 <span class="kn">import</span> <span class="nn">subprocess</span>
71 <div class="viewcode-block" id="show_progress"><a class="viewcode-back" href="../../commands/apidoc/src.html#src.fork.show_progress">[docs]</a><span class="k">def</span> <span class="nf">show_progress</span><span class="p">(</span><span class="n">logger</span><span class="p">,</span> <span class="n">top</span><span class="p">,</span> <span class="n">delai</span><span class="p">,</span> <span class="n">ss</span><span class="o">=</span><span class="s">""</span><span class="p">):</span>
72 <span class="sd">"""shortcut function to display the progression</span>
73 <span class="sd"> </span>
74 <span class="sd"> :param logger Logger: The logging instance</span>
75 <span class="sd"> :param top int: the number to display</span>
76 <span class="sd"> :param delai int: the number max</span>
77 <span class="sd"> :param ss str: the string to display</span>
78 <span class="sd"> """</span>
79 <span class="n">logger</span><span class="o">.</span><span class="n">write</span><span class="p">(</span><span class="s">"</span><span class="se">\r</span><span class="si">%s</span><span class="se">\r</span><span class="si">%s</span><span class="s"> </span><span class="si">%s</span><span class="s"> / </span><span class="si">%s</span><span class="s"> "</span> <span class="o">%</span> <span class="p">((</span><span class="s">" "</span> <span class="o">*</span> <span class="mi">30</span><span class="p">),</span> <span class="n">ss</span><span class="p">,</span> <span class="n">top</span><span class="p">,</span> <span class="p">(</span><span class="n">delai</span> <span class="o">-</span> <span class="n">top</span><span class="p">)),</span> <span class="mi">4</span><span class="p">,</span>
80 <span class="bp">False</span><span class="p">)</span>
81 <span class="n">logger</span><span class="o">.</span><span class="n">flush</span><span class="p">()</span>
83 <div class="viewcode-block" id="write_back"><a class="viewcode-back" href="../../commands/apidoc/src.html#src.fork.write_back">[docs]</a><span class="k">def</span> <span class="nf">write_back</span><span class="p">(</span><span class="n">logger</span><span class="p">,</span> <span class="n">message</span><span class="p">,</span> <span class="n">level</span><span class="p">):</span>
84 <span class="sd">"""shortcut function to write at the begin of the line</span>
85 <span class="sd"> </span>
86 <span class="sd"> :param logger Logger: The logging instance</span>
87 <span class="sd"> :param message str: the text to display</span>
88 <span class="sd"> :param level int: the level of verbosity</span>
89 <span class="sd"> """</span>
90 <span class="n">logger</span><span class="o">.</span><span class="n">write</span><span class="p">(</span><span class="s">"</span><span class="se">\r</span><span class="si">%s</span><span class="se">\r</span><span class="si">%s</span><span class="s">"</span> <span class="o">%</span> <span class="p">((</span><span class="s">" "</span> <span class="o">*</span> <span class="mi">40</span><span class="p">),</span> <span class="n">message</span><span class="p">),</span> <span class="n">level</span><span class="p">)</span>
92 <span class="c"># Launch command</span>
93 <span class="c"># --------------</span></div>
94 <div class="viewcode-block" id="launch_command"><a class="viewcode-back" href="../../commands/apidoc/src.html#src.fork.launch_command">[docs]</a><span class="k">def</span> <span class="nf">launch_command</span><span class="p">(</span><span class="n">cmd</span><span class="p">,</span> <span class="n">logger</span><span class="p">,</span> <span class="n">cwd</span><span class="p">,</span> <span class="n">args</span><span class="o">=</span><span class="p">[],</span> <span class="n">log</span><span class="o">=</span><span class="bp">None</span><span class="p">):</span>
95 <span class="k">if</span> <span class="n">log</span><span class="p">:</span>
96 <span class="n">log</span> <span class="o">=</span> <span class="nb">file</span><span class="p">(</span><span class="n">log</span><span class="p">,</span> <span class="s">"a"</span><span class="p">)</span>
97 <span class="n">logger</span><span class="o">.</span><span class="n">write</span><span class="p">(</span><span class="s">"launch: </span><span class="si">%s</span><span class="se">\n</span><span class="s">"</span> <span class="o">%</span> <span class="n">cmd</span><span class="p">,</span> <span class="mi">5</span><span class="p">,</span> <span class="n">screenOnly</span><span class="o">=</span><span class="bp">True</span><span class="p">)</span>
98 <span class="k">for</span> <span class="n">arg</span> <span class="ow">in</span> <span class="n">args</span><span class="p">:</span>
99 <span class="n">cmd</span> <span class="o">+=</span> <span class="s">" "</span> <span class="o">+</span> <span class="n">arg</span>
100 <span class="n">prs</span> <span class="o">=</span> <span class="n">subprocess</span><span class="o">.</span><span class="n">Popen</span><span class="p">(</span><span class="n">cmd</span><span class="p">,</span>
101 <span class="n">shell</span><span class="o">=</span><span class="bp">True</span><span class="p">,</span>
102 <span class="n">stdout</span><span class="o">=</span><span class="n">log</span><span class="p">,</span>
103 <span class="n">stderr</span><span class="o">=</span><span class="n">subprocess</span><span class="o">.</span><span class="n">STDOUT</span><span class="p">,</span>
104 <span class="n">cwd</span><span class="o">=</span><span class="n">cwd</span><span class="p">,</span>
105 <span class="n">executable</span><span class="o">=</span><span class="s">'/bin/bash'</span><span class="p">)</span>
106 <span class="k">return</span> <span class="n">prs</span>
108 <span class="c"># Launch a batch</span>
109 <span class="c"># --------------</span></div>
110 <div class="viewcode-block" id="batch"><a class="viewcode-back" href="../../commands/apidoc/src.html#src.fork.batch">[docs]</a><span class="k">def</span> <span class="nf">batch</span><span class="p">(</span><span class="n">cmd</span><span class="p">,</span> <span class="n">logger</span><span class="p">,</span> <span class="n">cwd</span><span class="p">,</span> <span class="n">args</span><span class="o">=</span><span class="p">[],</span> <span class="n">log</span><span class="o">=</span><span class="bp">None</span><span class="p">,</span> <span class="n">delai</span><span class="o">=</span><span class="mi">20</span><span class="p">,</span> <span class="n">sommeil</span><span class="o">=</span><span class="mi">1</span><span class="p">):</span>
111 <span class="n">proc</span> <span class="o">=</span> <span class="n">launch_command</span><span class="p">(</span><span class="n">cmd</span><span class="p">,</span> <span class="n">logger</span><span class="p">,</span> <span class="n">cwd</span><span class="p">,</span> <span class="n">args</span><span class="p">,</span> <span class="n">log</span><span class="p">)</span>
112 <span class="n">top</span> <span class="o">=</span> <span class="mi">0</span>
113 <span class="n">sys</span><span class="o">.</span><span class="n">stdout</span><span class="o">.</span><span class="n">softspace</span> <span class="o">=</span> <span class="bp">True</span>
114 <span class="n">begin</span> <span class="o">=</span> <span class="n">time</span><span class="o">.</span><span class="n">time</span><span class="p">()</span>
115 <span class="k">while</span> <span class="n">proc</span><span class="o">.</span><span class="n">poll</span><span class="p">()</span> <span class="ow">is</span> <span class="bp">None</span><span class="p">:</span>
116 <span class="k">if</span> <span class="n">time</span><span class="o">.</span><span class="n">time</span><span class="p">()</span> <span class="o">-</span> <span class="n">begin</span> <span class="o">>=</span> <span class="mi">1</span><span class="p">:</span>
117 <span class="n">show_progress</span><span class="p">(</span><span class="n">logger</span><span class="p">,</span> <span class="n">top</span><span class="p">,</span> <span class="n">delai</span><span class="p">,</span> <span class="s">"batch:"</span><span class="p">)</span>
118 <span class="k">if</span> <span class="n">top</span> <span class="o">==</span> <span class="n">delai</span><span class="p">:</span>
119 <span class="n">logger</span><span class="o">.</span><span class="n">write</span><span class="p">(</span><span class="s">"batch: time out KILL</span><span class="se">\n</span><span class="s">"</span><span class="p">,</span> <span class="mi">3</span><span class="p">)</span>
120 <span class="kn">import</span> <span class="nn">signal</span>
121 <span class="n">os</span><span class="o">.</span><span class="n">kill</span><span class="p">(</span><span class="n">proc</span><span class="o">.</span><span class="n">pid</span><span class="p">,</span> <span class="n">signal</span><span class="o">.</span><span class="n">SIGTERM</span><span class="p">)</span>
122 <span class="k">break</span>
123 <span class="k">else</span><span class="p">:</span>
124 <span class="n">begin</span> <span class="o">=</span> <span class="n">time</span><span class="o">.</span><span class="n">time</span><span class="p">()</span>
125 <span class="n">time</span><span class="o">.</span><span class="n">sleep</span><span class="p">(</span><span class="n">sommeil</span><span class="p">)</span>
126 <span class="n">top</span> <span class="o">+=</span> <span class="mi">1</span>
127 <span class="n">sys</span><span class="o">.</span><span class="n">stdout</span><span class="o">.</span><span class="n">flush</span><span class="p">()</span>
128 <span class="k">else</span><span class="p">:</span>
129 <span class="n">write_back</span><span class="p">(</span><span class="n">logger</span><span class="p">,</span> <span class="s">"batch: exit (</span><span class="si">%s</span><span class="s">)</span><span class="se">\n</span><span class="s">"</span> <span class="o">%</span> <span class="nb">str</span><span class="p">(</span><span class="n">proc</span><span class="o">.</span><span class="n">returncode</span><span class="p">),</span> <span class="mi">5</span><span class="p">)</span>
130 <span class="k">return</span> <span class="p">(</span><span class="n">proc</span><span class="o">.</span><span class="n">returncode</span> <span class="o">==</span> <span class="mi">0</span><span class="p">),</span> <span class="n">top</span>
132 <span class="c"># Launch a salome process</span>
133 <span class="c"># -----------------------</span></div>
134 <div class="viewcode-block" id="batch_salome"><a class="viewcode-back" href="../../commands/apidoc/src.html#src.fork.batch_salome">[docs]</a><span class="k">def</span> <span class="nf">batch_salome</span><span class="p">(</span><span class="n">cmd</span><span class="p">,</span> <span class="n">logger</span><span class="p">,</span> <span class="n">cwd</span><span class="p">,</span> <span class="n">args</span><span class="p">,</span> <span class="n">getTmpDir</span><span class="p">,</span>
135 <span class="n">pendant</span><span class="o">=</span><span class="s">"SALOME_Session_Server"</span><span class="p">,</span> <span class="n">fin</span><span class="o">=</span><span class="s">"killSalome.py"</span><span class="p">,</span>
136 <span class="n">log</span><span class="o">=</span><span class="bp">None</span><span class="p">,</span> <span class="n">delai</span><span class="o">=</span><span class="mi">20</span><span class="p">,</span> <span class="n">sommeil</span><span class="o">=</span><span class="mi">1</span><span class="p">,</span> <span class="n">delaiapp</span><span class="o">=</span><span class="mi">0</span><span class="p">):</span>
138 <span class="n">beginTime</span> <span class="o">=</span> <span class="n">time</span><span class="o">.</span><span class="n">time</span><span class="p">()</span>
139 <span class="n">launch_command</span><span class="p">(</span><span class="n">cmd</span><span class="p">,</span> <span class="n">logger</span><span class="p">,</span> <span class="n">cwd</span><span class="p">,</span> <span class="n">args</span><span class="p">,</span> <span class="n">log</span><span class="p">)</span>
141 <span class="k">if</span> <span class="n">delaiapp</span> <span class="o">==</span> <span class="mi">0</span><span class="p">:</span>
142 <span class="n">delaiapp</span> <span class="o">=</span> <span class="n">delai</span>
144 <span class="c"># first launch salome (looking for .pidict file)</span>
145 <span class="n">top</span> <span class="o">=</span> <span class="mi">0</span>
146 <span class="n">found</span> <span class="o">=</span> <span class="bp">False</span>
147 <span class="n">tmp_dir</span> <span class="o">=</span> <span class="n">getTmpDir</span><span class="p">()</span>
148 <span class="k">while</span> <span class="p">(</span><span class="ow">not</span> <span class="n">found</span> <span class="ow">and</span> <span class="n">top</span> <span class="o"><</span> <span class="n">delaiapp</span><span class="p">):</span>
149 <span class="k">if</span> <span class="n">os</span><span class="o">.</span><span class="n">path</span><span class="o">.</span><span class="n">exists</span><span class="p">(</span><span class="n">tmp_dir</span><span class="p">):</span>
150 <span class="n">listFile</span> <span class="o">=</span> <span class="n">os</span><span class="o">.</span><span class="n">listdir</span><span class="p">(</span><span class="n">tmp_dir</span><span class="p">)</span>
151 <span class="k">else</span><span class="p">:</span>
152 <span class="n">listFile</span> <span class="o">=</span> <span class="p">[]</span>
154 <span class="k">for</span> <span class="n">file_name</span> <span class="ow">in</span> <span class="n">listFile</span><span class="p">:</span>
155 <span class="k">if</span> <span class="n">file_name</span><span class="o">.</span><span class="n">endswith</span><span class="p">(</span><span class="s">"pidict"</span><span class="p">):</span>
156 <span class="c"># sometime we get a old file that will be removed by runSalome.</span>
157 <span class="c"># So we test that we can read it.</span>
158 <span class="n">currentTime</span> <span class="o">=</span> <span class="bp">None</span>
159 <span class="k">try</span><span class="p">:</span>
160 <span class="n">statinfo</span> <span class="o">=</span> <span class="n">os</span><span class="o">.</span><span class="n">stat</span><span class="p">(</span><span class="n">os</span><span class="o">.</span><span class="n">path</span><span class="o">.</span><span class="n">join</span><span class="p">(</span><span class="n">tmp_dir</span><span class="p">,</span> <span class="n">file_name</span><span class="p">))</span>
161 <span class="n">currentTime</span> <span class="o">=</span> <span class="n">statinfo</span><span class="o">.</span><span class="n">st_mtime</span>
162 <span class="k">except</span><span class="p">:</span> <span class="k">pass</span>
164 <span class="k">if</span> <span class="n">currentTime</span> <span class="ow">and</span> <span class="n">currentTime</span> <span class="o">></span> <span class="n">beginTime</span><span class="p">:</span>
165 <span class="k">try</span><span class="p">:</span>
166 <span class="n">file_</span> <span class="o">=</span> <span class="nb">open</span><span class="p">(</span><span class="n">os</span><span class="o">.</span><span class="n">path</span><span class="o">.</span><span class="n">join</span><span class="p">(</span><span class="n">tmp_dir</span><span class="p">,</span> <span class="n">file_name</span><span class="p">),</span> <span class="s">"r"</span><span class="p">)</span>
167 <span class="n">process_ids</span> <span class="o">=</span> <span class="n">pickle</span><span class="o">.</span><span class="n">load</span><span class="p">(</span><span class="n">file_</span><span class="p">)</span>
168 <span class="n">file_</span><span class="o">.</span><span class="n">close</span><span class="p">()</span>
169 <span class="k">for</span> <span class="n">process_id</span> <span class="ow">in</span> <span class="n">process_ids</span><span class="p">:</span>
170 <span class="k">for</span> <span class="n">__</span><span class="p">,</span> <span class="n">cmd</span> <span class="ow">in</span> <span class="n">process_id</span><span class="o">.</span><span class="n">items</span><span class="p">():</span>
171 <span class="k">if</span> <span class="n">cmd</span> <span class="o">==</span> <span class="p">[</span><span class="n">pendant</span><span class="p">]:</span>
172 <span class="n">found</span> <span class="o">=</span> <span class="bp">True</span>
173 <span class="n">pidictFile</span> <span class="o">=</span> <span class="n">file_name</span>
174 <span class="k">except</span><span class="p">:</span>
175 <span class="n">file_</span><span class="o">.</span><span class="n">close</span><span class="p">()</span>
177 <span class="n">time</span><span class="o">.</span><span class="n">sleep</span><span class="p">(</span><span class="n">sommeil</span><span class="p">)</span>
178 <span class="n">top</span> <span class="o">+=</span> <span class="mi">1</span>
179 <span class="n">show_progress</span><span class="p">(</span><span class="n">logger</span><span class="p">,</span> <span class="n">top</span><span class="p">,</span> <span class="n">delaiapp</span><span class="p">,</span> <span class="s">"launching salome or appli:"</span><span class="p">)</span>
181 <span class="c"># continue or not</span>
182 <span class="k">if</span> <span class="n">found</span><span class="p">:</span>
183 <span class="n">write_back</span><span class="p">(</span><span class="n">logger</span><span class="p">,</span> <span class="s">"batch_salome: started</span><span class="se">\n</span><span class="s">"</span><span class="p">,</span> <span class="mi">5</span><span class="p">)</span>
184 <span class="k">else</span><span class="p">:</span>
185 <span class="n">logger</span><span class="o">.</span><span class="n">write</span><span class="p">(</span><span class="s">"batch_salome: FAILED to launch salome or appli</span><span class="se">\n</span><span class="s">"</span><span class="p">,</span> <span class="mi">3</span><span class="p">)</span>
186 <span class="k">return</span> <span class="bp">False</span><span class="p">,</span> <span class="o">-</span><span class="mi">1</span>
188 <span class="c"># salome launched run the script</span>
189 <span class="n">top</span> <span class="o">=</span> <span class="mi">0</span>
190 <span class="n">code</span> <span class="o">=</span> <span class="bp">None</span>
191 <span class="k">while</span> <span class="n">code</span> <span class="ow">is</span> <span class="bp">None</span><span class="p">:</span>
192 <span class="n">show_progress</span><span class="p">(</span><span class="n">logger</span><span class="p">,</span> <span class="n">top</span><span class="p">,</span> <span class="n">delai</span><span class="p">,</span> <span class="s">"running salome or appli:"</span><span class="p">)</span>
194 <span class="k">if</span> <span class="ow">not</span> <span class="n">os</span><span class="o">.</span><span class="n">access</span><span class="p">(</span><span class="n">os</span><span class="o">.</span><span class="n">path</span><span class="o">.</span><span class="n">join</span><span class="p">(</span><span class="n">tmp_dir</span><span class="p">,</span> <span class="n">pidictFile</span><span class="p">),</span> <span class="n">os</span><span class="o">.</span><span class="n">F_OK</span><span class="p">):</span>
195 <span class="n">write_back</span><span class="p">(</span><span class="n">logger</span><span class="p">,</span> <span class="s">"batch_salome: exit</span><span class="se">\n</span><span class="s">"</span><span class="p">,</span> <span class="mi">5</span><span class="p">)</span>
196 <span class="n">code</span> <span class="o">=</span> <span class="bp">True</span>
197 <span class="k">elif</span> <span class="n">top</span> <span class="o">>=</span> <span class="n">delai</span><span class="p">:</span>
198 <span class="c"># timeout kill the test</span>
199 <span class="n">os</span><span class="o">.</span><span class="n">system</span><span class="p">(</span><span class="n">fin</span><span class="p">)</span>
200 <span class="n">logger</span><span class="o">.</span><span class="n">write</span><span class="p">(</span><span class="s">"batch_salome: time out KILL</span><span class="se">\n</span><span class="s">"</span><span class="p">,</span> <span class="mi">3</span><span class="p">)</span>
201 <span class="n">code</span> <span class="o">=</span> <span class="bp">False</span>
202 <span class="k">else</span><span class="p">:</span>
203 <span class="c"># still waiting</span>
204 <span class="n">time</span><span class="o">.</span><span class="n">sleep</span><span class="p">(</span><span class="n">sommeil</span><span class="p">)</span>
205 <span class="n">top</span> <span class="o">=</span> <span class="n">top</span> <span class="o">+</span> <span class="mi">1</span>
207 <span class="k">return</span> <span class="n">code</span><span class="p">,</span> <span class="n">top</span></div>
213 <div class="sphinxsidebar">
214 <div class="sphinxsidebarwrapper">
215 <p class="logo"><a href="../../index.html">
216 <img class="logo" src="../../_static/sat_v5.0.png" alt="Logo"/>
217 </a></p><div class="relations">
218 <h3>Related Topics</h3>
220 <li><a href="../../index.html">Documentation overview</a><ul>
221 <li><a href="../index.html">Module code</a><ul>
226 <div id="searchbox" style="display: none">
227 <h3>Quick search</h3>
228 <form class="search" action="../../search.html" method="get">
229 <input type="text" name="q" />
230 <input type="submit" value="Go" />
231 <input type="hidden" name="check_keywords" value="yes" />
232 <input type="hidden" name="area" value="default" />
234 <p class="searchtip" style="font-size: 90%">
235 Enter search terms or a module, class or function name.
238 <script type="text/javascript">$('#searchbox').show(0);</script>
241 <div class="clearer"></div>
247 Powered by <a href="http://sphinx-doc.org/">Sphinx 1.1.3</a>
248 & <a href="https://github.com/bitprophet/alabaster">Alabaster </a>