Salome HOME
Small fixes in sphinx documentation of python plugins:
[modules/smesh.git] / src / Tools / Verima / ajoutEnreg.py
index c1dc4bc34b466843a5b00cbcbf4335ce030521ca..de84fd7ae15fda6fe949efa342a032127efa01b5 100755 (executable)
@@ -1,11 +1,11 @@
-#!/usr/bin/env python
+#!/usr/bin/env python3
 
 import sys, os
 rep=os.path.dirname(os.path.abspath(__file__))
 installDir=os.path.join(rep,'..')
 sys.path.insert(0,installDir)
 
-from PyQt4 import QtGui,QtCore,QtSql
+from qtsalome import *
 from Base.dataBase import Base
 
 def completeDatabase(fichier,table,enregistrement):
@@ -13,13 +13,13 @@ def completeDatabase(fichier,table,enregistrement):
       maBase.initialise()
       nomTable="ma"+str(table)
       matable=getattr(maBase,nomTable)
-      model= QtSql.QSqlTableModel()
+      model= QSqlTableModel()
       model.setTable(matable.nom)
       nbCols=model.columnCount() -1
       if table == "TableGroupesRef" : nbCols==nbCols+1
       if len(enregistrement) != nbCols  :
-         print "mauvais nb de valeurs"
-         print "Attention, ne pas renter d'Id"
+         print("mauvais nb de valeurs")
+         print("Attention, ne pas renter d'Id")
       if table == "TableGroupesRef" : matable.insereLigne(enregistrement)
       else : matable.insereLigneAutoId(enregistrement)
       maBase.close()
@@ -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 : 
-         print  "table obligatoire"
+     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)