Porting Python 3

[modules/smesh.git] / src / Tools / Verima / ajoutEnreg.py

diff --git a/src/Tools/Verima/ajoutEnreg.py b/src/Tools/Verima/ajoutEnreg.py
index 57d18dcc60a1afe41d840a8613ad1a2301a39ca2..de84fd7ae15fda6fe949efa342a032127efa01b5 100755 (executable)
--- a/src/Tools/Verima/ajoutEnreg.py
+++ b/src/Tools/Verima/ajoutEnreg.py
@@ -1,4 +1,4 @@
-#!/usr/bin/env python
+#!/usr/bin/env python3
 
 import sys, os
 rep=os.path.dirname(os.path.abspath(__file__))
@@ -6,7 +6,7 @@ installDir=os.path.join(rep,'..')
 sys.path.insert(0,installDir)
 
 from qtsalome import *
-from .Base.dataBase import Base
+from Base.dataBase import Base
 
 def completeDatabase(fichier,table,enregistrement):
       maBase=Base(fichier)
@@ -27,17 +27,18 @@ def completeDatabase(fichier,table,enregistrement):
 
 
 if __name__ == "__main__":
-     from optparse import OptionParser
-     p=OptionParser()
-     p.add_option('-d',dest='database',default="myMesh.db",help='nom de la database')
-     p.add_option('-t',dest='table',help='nom de la table a completer')
-     options, args = p.parse_args()
-     if  options.table==None : 
+     from argparse import ArgumentParser
+     p=ArgumentParser()
+     p.add_argument('-d',dest='database',default="myMesh.db",help='nom de la database')
+     p.add_argument('table',help='nom de la table a completer')
+     p.add_argument('enregistrement', nargs="+")
+     args = p.parse_args()
+     if args.table is None : 
          print("table obligatoire")
          exit()
-     if  options.table not in ("TableMaillages","TableMailleurs","TableGroupesRef","TableVersions") : 
-         print("la table doit etre : TableMaillages ou TableMailleurs ou TableGroupesRef ou TableVersions") 
+     good_tables = ("TableMaillages","TableMailleurs","TableGroupesRef","TableVersions")
+     if args.table not in good_tables:
+         print("la table doit etre : %s" % ' ou '.join(good_tables))
          exit()
-     enregistrement=tuple(args)
-     completeDatabase(options.database,options.table,enregistrement)
+     completeDatabase(args.database,args.table,args.enregistrement)