Salome HOME
Porting to Python 3
[modules/homard.git] / doc / files / tutorial_util.py
index 19f3b53d846785182d0578898dd1ef2614a61e34..34fb19ae75ce891856a3f4109538268ee92a50c3 100755 (executable)
@@ -1,5 +1,5 @@
-# -*- coding: iso-8859-1 -*-
-# Copyright (C) 2011-2014  CEA/DEN, EDF R&D
+# -*- coding: utf-8 -*-
+# Copyright (C) 2011-2016  CEA/DEN, EDF R&D
 #
 # This library is free software; you can redistribute it and/or
 # modify it under the terms of the GNU Lesser General Public
 Python script for HOMARD
 Copyright EDF-R&D 2014
 """
-__revision__ = "V1.0"
+__revision__ = "V1.3"
 
 import os
 import sys
+
+PATH_HOMARD = os.getenv('HOMARD_ROOT_DIR')
+# Repertoire des scripts utilitaires
+REP_PYTHON = os.path.join(PATH_HOMARD, "bin", "salome", "test", "HOMARD")
+REP_PYTHON = os.path.normpath(REP_PYTHON)
+sys.path.append(REP_PYTHON)
+from test_util import remove_dir
+
 #========================================================================
 #========================================================================
 def gzip_gunzip(data_dir, num_tuto, option) :
@@ -38,14 +46,14 @@ Copyright EDF-R&D 2014
 #
   ficloc_basis = "tutorial_%d" % num_tuto
 #
-  ok = True
+  erreur = 0
   num = -1
 #
 # Uncompression
 #
   if ( option == -1 ) :
 #
-    while ok :
+    while not erreur :
       num += 1
       ficloc = ficloc_basis + ".%02d.med" % num
       nomfic = os.path.join(data_dir, ficloc)
@@ -55,7 +63,7 @@ Copyright EDF-R&D 2014
         if os.path.isfile(nomfic) :
           os.system("gunzip "+nomfic)
         else :
-          ok = False
+          erreur = 1
           break
 #
     ficloc = ficloc_basis + ".fr.med"
@@ -70,7 +78,7 @@ Copyright EDF-R&D 2014
 #
   elif ( option == 1 ) :
 #
-    while ok :
+    while not erreur :
       num += 1
       ficloc = ficloc_basis + ".%02d.med.gz" % num
       nomfic = os.path.join(data_dir, ficloc)
@@ -80,7 +88,7 @@ Copyright EDF-R&D 2014
         if os.path.isfile(nomfic) :
           os.system("gzip "+nomfic)
         else :
-          ok = False
+          erreur = 2
           break
 #
     ficloc = ficloc_basis + ".fr.med.gz"
@@ -95,3 +103,27 @@ Copyright EDF-R&D 2014
 #
 #========================================================================
 #========================================================================
+def creation_dircase(num_tuto) :
+  """
+Creation of a directory for the results of tutorial_x
+num_tuto: number of the tutorial
+Copyright EDF-R&D 2014
+  """
+#
+  if "LOGNAME" in os.environ :
+    user = os.environ ["LOGNAME"]
+  else :
+    user = "anonymous"
+  dircase = os.path.join( os.sep, "tmp", "HOMARD_"+user)
+  if not os.path.isdir(dircase) :
+    os.mkdir (dircase)
+  dirtuto_basis = "tutorial_%d" % num_tuto
+  dircase = os.path.join( dircase, dirtuto_basis )
+  if os.path.isdir(dircase) :
+    remove_dir(dircase)
+  os.mkdir (dircase)
+#
+  return dircase
+#========================================================================
+#========================================================================
+#