Salome HOME
Copyright update 2021
[modules/smesh.git] / src / Tools / Verima / Base / importFromCSV.py
index a394ab16b897fd01162834a0acd1c8c36f7cb96b..34cc3b1a0b3517ad92dc8e9038860bf9c9ac0277 100755 (executable)
@@ -1,30 +1,42 @@
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
+#!/usr/bin/env python3
+# Copyright (C) 2013-2021  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
+# License as published by the Free Software Foundation; either
+# version 2.1 of the License, or (at your option) any later version.
+#
+# This library is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+# Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public
+# License along with this library; if not, write to the Free Software
+# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
+#
+# See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
+#
 
-import sys
 import os
 from dataBase import Base
 
 
-
 if __name__ == "__main__":
-    from optparse import OptionParser
-    p=OptionParser()
-    p.add_option('-p',dest='partiel',action="store_true", default=False,help='import de machine, groupe, ratio Maille et Perf uniquement')
-    p.add_option('-f',dest='force',action="store_true", default=False,help='ecrasement des valeurs dans la base par les valeurs dans les fichiers en cas de meme clef primaire')
-    p.add_option('-d',dest='database',default="myMesh.db",help='nom de la database')
-    options, args = p.parse_args()
-    if len(args) != 1 :
-        print "entrer SVP le nom de la directory ou sont rangees les fichiers a charger"
-        exit()
-    folder=args[0]
+    from argparse import ArgumentParser
+    p=ArgumentParser()
+    p.add_argument('-p',dest='partiel',action="store_true", default=False,help='import de machine, groupe, ratio Maille et Perf uniquement')
+    p.add_argument('-f',dest='force',action="store_true", default=False,help='ecrasement des valeurs dans la base par les valeurs dans les fichiers en cas de meme clef primaire')
+    p.add_argument('-d',dest='database',default="myMesh.db",help='nom de la database')
+    p.add_argument('folder', help='Repertoire ou sont ranges les fichiers a charger')
+    args = p.parse_args()
 
-    if not(os.path.isdir(folder)):
-        print folder , " n existe pas"
+    if not(os.path.isdir(args.folder)):
+        print(args.folder , " n existe pas")
         exit()
 
-    maBase=Base(options.database)
+    maBase=Base(args.database)
     maBase.create()
     maBase.initialise()
-    maBase.importFromCSV(folder,options.partiel,options.force)
+    maBase.importFromCSV(args.folder,args.partiel,args.force)
     maBase.close()