src/Tools/Verima/ajoutEnreg.py

   1 #!/usr/bin/env python3
   2
   3 import sys, os
   4 rep=os.path.dirname(os.path.abspath(__file__))
   5 installDir=os.path.join(rep,'..')
   6 sys.path.insert(0,installDir)
   7
   8 from qtsalome import *
   9 from Base.dataBase import Base
  10
  11 def completeDatabase(fichier,table,enregistrement):
  12       maBase=Base(fichier)
  13       maBase.initialise()
  14       nomTable="ma"+str(table)
  15       matable=getattr(maBase,nomTable)
  16       model= QSqlTableModel()
  17       model.setTable(matable.nom)
  18       nbCols=model.columnCount() -1
  19       if table == "TableGroupesRef" : nbCols==nbCols+1
  20       if len(enregistrement) != nbCols  :
  21          print("mauvais nb de valeurs")
  22          print("Attention, ne pas renter d'Id")
  23       if table == "TableGroupesRef" : matable.insereLigne(enregistrement)
  24       else : matable.insereLigneAutoId(enregistrement)
  25       maBase.close()
  26
  27
  28
  29 if __name__ == "__main__":
  30      from argparse import ArgumentParser
  31      p=ArgumentParser()
  32      p.add_argument('-d',dest='database',default="myMesh.db",help='nom de la database')
  33      p.add_argument('table',help='nom de la table a completer')
  34      p.add_argument('enregistrement', nargs="+")
  35      args = p.parse_args()
  36      if args.table is None :
  37          print("table obligatoire")
  38          exit()
  39      good_tables = ("TableMaillages","TableMailleurs","TableGroupesRef","TableVersions")
  40      if args.table not in good_tables:
  41          print("la table doit etre : %s" % ' ou '.join(good_tables))
  42          exit()
  43      completeDatabase(args.database,args.table,args.enregistrement)
  44