commands/jobs.py

   1 #!/usr/bin/env python
   2 #-*- coding:utf-8 -*-
   3 #  Copyright (C) 2010-2013  CEA/DEN
   4 #
   5 #  This library is free software; you can redistribute it and/or
   6 #  modify it under the terms of the GNU Lesser General Public
   7 #  License as published by the Free Software Foundation; either
   8 #  version 2.1 of the License.
   9 #
  10 #  This library is distributed in the hope that it will be useful,
  11 #  but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13 #  Lesser General Public License for more details.
  14 #
  15 #  You should have received a copy of the GNU Lesser General Public
  16 #  License along with this library; if not, write to the Free Software
  17 #  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  18
  19 import os
  20 import sys
  21 import tempfile
  22 import traceback
  23 import datetime
  24 import time
  25 import csv
  26 import shutil
  27 import itertools
  28 import re
  29 import paramiko
  30
  31 import src
  32 import src.ElementTree as etree
  33
  34 STYLESHEET_GLOBAL = "jobs_global_report.xsl"
  35 STYLESHEET_BOARD = "jobs_board_report.xsl"
  36
  37 DAYS_SEPARATOR = ","
  38 CSV_DELIMITER = ";"
  39
  40 parser = src.options.Options()
  41
  42 parser.add_option('n', 'name', 'list2', 'jobs_cfg',
  43                   _('Mandatory: The name of the config file that contains'
  44                   ' the jobs configuration. Can be a list.'))
  45 parser.add_option('o', 'only_jobs', 'list2', 'only_jobs',
  46                   _('Optional: the list of jobs to launch, by their name. '))
  47 parser.add_option('l', 'list', 'boolean', 'list',
  48                   _('Optional: list all available config files.'))
  49 parser.add_option('t', 'test_connection', 'boolean', 'test_connection',
  50                   _("Optional: try to connect to the machines. "
  51                     "Not executing the jobs."),
  52                   False)
  53 parser.add_option('p', 'publish', 'boolean', 'publish',
  54                   _("Optional: generate an xml file that can be read in a "
  55                     "browser to display the jobs status."),
  56                   False)
  57 parser.add_option('i', 'input_boards', 'string', 'input_boards', _("Optional: "
  58                                 "the path to csv file that contain "
  59                                 "the expected boards."),"")
  60 parser.add_option('', 'completion', 'boolean', 'no_label',
  61                   _("Optional (internal use): do not print labels, Works only "
  62                     "with --list."),
  63                   False)
  64
  65 class Machine(object):
  66     '''Class to manage a ssh connection on a machine
  67     '''
  68     def __init__(self,
  69                  name,
  70                  host,
  71                  user,
  72                  port=22,
  73                  passwd=None,
  74                  sat_path="salomeTools"):
  75         self.name = name
  76         self.host = host
  77         self.port = port
  78         self.distribution = None # Will be filled after copying SAT on the machine
  79         self.user = user
  80         self.password = passwd
  81         self.sat_path = sat_path
  82         self.ssh = paramiko.SSHClient()
  83         self._connection_successful = None
  84
  85     def connect(self, logger):
  86         '''Initiate the ssh connection to the remote machine
  87
  88         :param logger src.logger.Logger: The logger instance
  89         :return: Nothing
  90         :rtype: N\A
  91         '''
  92
  93         self._connection_successful = False
  94         self.ssh.load_system_host_keys()
  95         self.ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
  96         try:
  97             self.ssh.connect(self.host,
  98                              port=self.port,
  99                              username=self.user,
 100                              password = self.password)
 101         except paramiko.AuthenticationException:
 102             message = src.KO_STATUS + _("Authentication failed")
 103         except paramiko.BadHostKeyException:
 104             message = (src.KO_STATUS +
 105                        _("The server's host key could not be verified"))
 106         except paramiko.SSHException:
 107             message = ( _("SSHException error connecting or "
 108                           "establishing an SSH session"))
 109         except:
 110             message = ( _("Error connecting or establishing an SSH session"))
 111         else:
 112             self._connection_successful = True
 113             message = ""
 114         return message
 115
 116     def successfully_connected(self, logger):
 117         '''Verify if the connection to the remote machine has succeed
 118
 119         :param logger src.logger.Logger: The logger instance
 120         :return: True if the connection has succeed, False if not
 121         :rtype: bool
 122         '''
 123         if self._connection_successful == None:
 124             message = _("Warning : trying to ask if the connection to "
 125             "(name: %s host: %s, port: %s, user: %s) is OK whereas there were"
 126             " no connection request" %
 127                         (self.name, self.host, self.port, self.user))
 128             logger.write( src.printcolors.printcWarning(message))
 129         return self._connection_successful
 130
 131     def copy_sat(self, sat_local_path, job_file):
 132         '''Copy salomeTools to the remote machine in self.sat_path
 133         '''
 134         res = 0
 135         try:
 136             # open a sftp connection
 137             self.sftp = self.ssh.open_sftp()
 138             # Create the sat directory on remote machine if it is not existing
 139             self.mkdir(self.sat_path, ignore_existing=True)
 140             # Put sat
 141             self.put_dir(sat_local_path, self.sat_path, filters = ['.git'])
 142             # put the job configuration file in order to make it reachable
 143             # on the remote machine
 144             remote_job_file_name = ".%s" % os.path.basename(job_file)
 145             self.sftp.put(job_file, os.path.join(self.sat_path,
 146                                                  remote_job_file_name))
 147         except Exception as e:
 148             res = str(e)
 149             self._connection_successful = False
 150
 151         return res
 152
 153     def put_dir(self, source, target, filters = []):
 154         ''' Uploads the contents of the source directory to the target path. The
 155             target directory needs to exists. All sub-directories in source are
 156             created under target.
 157         '''
 158         for item in os.listdir(source):
 159             if item in filters:
 160                 continue
 161             source_path = os.path.join(source, item)
 162             destination_path = os.path.join(target, item)
 163             if os.path.islink(source_path):
 164                 linkto = os.readlink(source_path)
 165                 try:
 166                     self.sftp.symlink(linkto, destination_path)
 167                     self.sftp.chmod(destination_path,
 168                                     os.stat(source_path).st_mode)
 169                 except IOError:
 170                     pass
 171             else:
 172                 if os.path.isfile(source_path):
 173                     self.sftp.put(source_path, destination_path)
 174                     self.sftp.chmod(destination_path,
 175                                     os.stat(source_path).st_mode)
 176                 else:
 177                     self.mkdir(destination_path, ignore_existing=True)
 178                     self.put_dir(source_path, destination_path)
 179
 180     def mkdir(self, path, mode=511, ignore_existing=False):
 181         ''' Augments mkdir by adding an option to not fail
 182             if the folder exists
 183         '''
 184         try:
 185             self.sftp.mkdir(path, mode)
 186         except IOError:
 187             if ignore_existing:
 188                 pass
 189             else:
 190                 raise
 191
 192     def exec_command(self, command, logger):
 193         '''Execute the command on the remote machine
 194
 195         :param command str: The command to be run
 196         :param logger src.logger.Logger: The logger instance
 197         :return: the stdin, stdout, and stderr of the executing command,
 198                  as a 3-tuple
 199         :rtype: (paramiko.channel.ChannelFile, paramiko.channel.ChannelFile,
 200                 paramiko.channel.ChannelFile)
 201         '''
 202         try:
 203             # Does not wait the end of the command
 204             (stdin, stdout, stderr) = self.ssh.exec_command(command)
 205         except paramiko.SSHException:
 206             message = src.KO_STATUS + _(
 207                             ": the server failed to execute the command\n")
 208             logger.write( src.printcolors.printcError(message))
 209             return (None, None, None)
 210         except:
 211             logger.write( src.printcolors.printcError(src.KO_STATUS + '\n'))
 212             return (None, None, None)
 213         else:
 214             return (stdin, stdout, stderr)
 215
 216     def close(self):
 217         '''Close the ssh connection
 218
 219         :rtype: N\A
 220         '''
 221         self.ssh.close()
 222
 223     def write_info(self, logger):
 224         '''Prints the informations relative to the machine in the logger
 225            (terminal traces and log file)
 226
 227         :param logger src.logger.Logger: The logger instance
 228         :return: Nothing
 229         :rtype: N\A
 230         '''
 231         logger.write("host : " + self.host + "\n")
 232         logger.write("port : " + str(self.port) + "\n")
 233         logger.write("user : " + str(self.user) + "\n")
 234         if self.successfully_connected(logger):
 235             status = src.OK_STATUS
 236         else:
 237             status = src.KO_STATUS
 238         logger.write("Connection : " + status + "\n\n")
 239
 240
 241 class Job(object):
 242     '''Class to manage one job
 243     '''
 244     def __init__(self,
 245                  name,
 246                  machine,
 247                  application,
 248                  board,
 249                  commands,
 250                  timeout,
 251                  config,
 252                  job_file_path,
 253                  logger,
 254                  after=None,
 255                  prefix=None):
 256
 257         self.name = name
 258         self.machine = machine
 259         self.after = after
 260         self.timeout = timeout
 261         self.application = application
 262         self.board = board
 263         self.config = config
 264         self.logger = logger
 265         # The list of log files to download from the remote machine
 266         self.remote_log_files = []
 267
 268         # The remote command status
 269         # -1 means that it has not been launched,
 270         # 0 means success and 1 means fail
 271         self.res_job = "-1"
 272         self.cancelled = False
 273
 274         self._T0 = -1
 275         self._Tf = -1
 276         self._has_begun = False
 277         self._has_finished = False
 278         self._has_timouted = False
 279         self._stdin = None # Store the command inputs field
 280         self._stdout = None # Store the command outputs field
 281         self._stderr = None # Store the command errors field
 282
 283         self.out = ""
 284         self.err = ""
 285
 286         self.name_remote_jobs_pyconf = ".%s" % os.path.basename(job_file_path)
 287         self.commands = commands
 288         self.command = (os.path.join(self.machine.sat_path, "sat") +
 289                         " -l " +
 290                         os.path.join(self.machine.sat_path,
 291                                      "list_log_files.txt") +
 292                         " job --jobs_config " +
 293                         os.path.join(self.machine.sat_path,
 294                                      self.name_remote_jobs_pyconf) +
 295                         " --name " +
 296                         self.name)
 297         if prefix:
 298             self.command = prefix + ' "' + self.command +'"'
 299
 300     def get_pids(self):
 301         """ Get the pid(s) corresponding to the command that have been launched
 302             On the remote machine
 303
 304         :return: The list of integers corresponding to the found pids
 305         :rtype: List
 306         """
 307         pids = []
 308         cmd_pid = 'ps aux | grep "' + self.command + '" | awk \'{print $2}\''
 309         (_, out_pid, _) = self.machine.exec_command(cmd_pid, self.logger)
 310         pids_cmd = out_pid.readlines()
 311         pids_cmd = [str(src.only_numbers(pid)) for pid in pids_cmd]
 312         pids+=pids_cmd
 313         return pids
 314
 315     def kill_remote_process(self, wait=1):
 316         '''Kills the process on the remote machine.
 317
 318         :return: (the output of the kill, the error of the kill)
 319         :rtype: (str, str)
 320         '''
 321         try:
 322             pids = self.get_pids()
 323         except:
 324             return ("Unable to get the pid of the command.", "")
 325
 326         cmd_kill = " ; ".join([("kill -2 " + pid) for pid in pids])
 327         (_, out_kill, err_kill) = self.machine.exec_command(cmd_kill,
 328                                                             self.logger)
 329         time.sleep(wait)
 330         return (out_kill.read().decode(), err_kill.read().decode())
 331
 332     def has_begun(self):
 333         '''Returns True if the job has already begun
 334
 335         :return: True if the job has already begun
 336         :rtype: bool
 337         '''
 338         return self._has_begun
 339
 340     def has_finished(self):
 341         '''Returns True if the job has already finished
 342            (i.e. all the commands have been executed)
 343            If it is finished, the outputs are stored in the fields out and err.
 344
 345         :return: True if the job has already finished
 346         :rtype: bool
 347         '''
 348
 349         # If the method has already been called and returned True
 350         if self._has_finished:
 351             return True
 352
 353         # If the job has not begun yet
 354         if not self.has_begun():
 355             return False
 356
 357         if self._stdout.channel.closed:
 358             self._has_finished = True
 359             # Store the result outputs
 360             self.out += self._stdout.read().decode()
 361             self.err += self._stderr.read().decode()
 362             # Put end time
 363             self._Tf = time.time()
 364             # And get the remote command status and log files
 365             try:
 366                 self.get_log_files()
 367             except Exception as e:
 368                 self.err += _("Unable to get remote log files: %s" % e)
 369
 370         return self._has_finished
 371
 372     def get_log_files(self):
 373         """Get the log files produced by the command launched
 374            on the remote machine, and put it in the log directory of the user,
 375            so they can be accessible from
 376         """
 377         # Do not get the files if the command is not finished
 378         if not self.has_finished():
 379             msg = _("Trying to get log files whereas the job is not finished.")
 380             self.logger.write(src.printcolors.printcWarning(msg))
 381             return
 382
 383         # First get the file that contains the list of log files to get
 384         tmp_file_path = src.get_tmp_filename(self.config, "list_log_files.txt")
 385         remote_path = os.path.join(self.machine.sat_path, "list_log_files.txt")
 386         self.machine.sftp.get(
 387                     remote_path,
 388                     tmp_file_path)
 389
 390         # Read the file and get the result of the command and all the log files
 391         # to get
 392         fstream_tmp = open(tmp_file_path, "r")
 393         file_lines = fstream_tmp.readlines()
 394         file_lines = [line.replace("\n", "") for line in file_lines]
 395         fstream_tmp.close()
 396         os.remove(tmp_file_path)
 397
 398         try :
 399             # The first line is the result of the command (0 success or 1 fail)
 400             self.res_job = file_lines[0]
 401         except Exception as e:
 402             self.err += _("Unable to get status from remote file %s: %s" %
 403                                                     (remote_path, str(e)))
 404
 405         for i, job_path_remote in enumerate(file_lines[1:]):
 406             try:
 407                 # For each command, there is two files to get :
 408                 # 1- The xml file describing the command and giving the
 409                 # internal traces.
 410                 # 2- The txt file containing the system command traces (like
 411                 # traces produced by the "make" command)
 412                 # 3- In case of the test command, there is another file to get :
 413                 # the xml board that contain the test results
 414                 dirname = os.path.basename(os.path.dirname(job_path_remote))
 415                 if dirname != 'OUT' and dirname != 'TEST':
 416                     # Case 1-
 417                     local_path = os.path.join(os.path.dirname(
 418                                                         self.logger.logFilePath),
 419                                               os.path.basename(job_path_remote))
 420                     if i==0: # The first is the job command
 421                         self.logger.add_link(os.path.basename(job_path_remote),
 422                                              "job",
 423                                              self.res_job,
 424                                              self.command)
 425                 elif dirname == 'OUT':
 426                     # Case 2-
 427                     local_path = os.path.join(os.path.dirname(
 428                                                         self.logger.logFilePath),
 429                                               'OUT',
 430                                               os.path.basename(job_path_remote))
 431                 elif dirname == 'TEST':
 432                     # Case 3-
 433                     local_path = os.path.join(os.path.dirname(
 434                                                         self.logger.logFilePath),
 435                                               'TEST',
 436                                               os.path.basename(job_path_remote))
 437
 438                 # Get the file
 439                 if not os.path.exists(local_path):
 440                     self.machine.sftp.get(job_path_remote, local_path)
 441                 self.remote_log_files.append(local_path)
 442             except Exception as e:
 443                 self.err += _("Unable to get %s log file from remote: %s" %
 444                                                     (str(job_path_remote),
 445                                                      str(e)))
 446
 447     def has_failed(self):
 448         '''Returns True if the job has failed.
 449            A job is considered as failed if the machine could not be reached,
 450            if the remote command failed,
 451            or if the job finished with a time out.
 452
 453         :return: True if the job has failed
 454         :rtype: bool
 455         '''
 456         if not self.has_finished():
 457             return False
 458         if not self.machine.successfully_connected(self.logger):
 459             return True
 460         if self.is_timeout():
 461             return True
 462         if self.res_job == "1":
 463             return True
 464         return False
 465
 466     def cancel(self):
 467         """In case of a failing job, one has to cancel every job that depend
 468            on it. This method put the job as failed and will not be executed.
 469         """
 470         if self.cancelled:
 471             return
 472         self._has_begun = True
 473         self._has_finished = True
 474         self.cancelled = True
 475         self.out += _("This job was not launched because its father has failed.")
 476         self.err += _("This job was not launched because its father has failed.")
 477
 478     def is_running(self):
 479         '''Returns True if the job commands are running
 480
 481         :return: True if the job is running
 482         :rtype: bool
 483         '''
 484         return self.has_begun() and not self.has_finished()
 485
 486     def is_timeout(self):
 487         '''Returns True if the job commands has finished with timeout
 488
 489         :return: True if the job has finished with timeout
 490         :rtype: bool
 491         '''
 492         return self._has_timouted
 493
 494     def time_elapsed(self):
 495         """Get the time elapsed since the job launching
 496
 497         :return: The number of seconds
 498         :rtype: int
 499         """
 500         if not self.has_begun():
 501             return -1
 502         T_now = time.time()
 503         return T_now - self._T0
 504
 505     def check_time(self):
 506         """Verify that the job has not exceeded its timeout.
 507            If it has, kill the remote command and consider the job as finished.
 508         """
 509         if not self.has_begun():
 510             return
 511         if self.time_elapsed() > self.timeout:
 512             self._has_finished = True
 513             self._has_timouted = True
 514             self._Tf = time.time()
 515             (out_kill, _) = self.kill_remote_process()
 516             self.out += "TIMEOUT \n" + out_kill
 517             self.err += "TIMEOUT : %s seconds elapsed\n" % str(self.timeout)
 518             try:
 519                 self.get_log_files()
 520             except Exception as e:
 521                 # The 2 following lines must be suppressed after the bug is fixed
 522                 print("The error type: ")
 523                 print(type(e))
 524                 print("The error: ")
 525                 print(e)
 526                 print("Local scope:")
 527                 print(dir())
 528                 print("The str type: ")
 529                 print(type(str))
 530                 print("str: ")
 531                 print(str)
 532                 self.err += _("Unable to get remote log files!")
 533
 534     def total_duration(self):
 535         """Give the total duration of the job
 536
 537         :return: the total duration of the job in seconds
 538         :rtype: int
 539         """
 540         return self._Tf - self._T0
 541
 542     def run(self):
 543         """Launch the job by executing the remote command.
 544         """
 545
 546         # Prevent multiple run
 547         if self.has_begun():
 548             msg = _("Warning: A job can only be launched one time")
 549             msg2 = _("Trying to launch the job \"%s\" whereas it has "
 550                      "already been launched." % self.name)
 551             self.logger.write(src.printcolors.printcWarning("%s\n%s\n" % (msg,
 552                                                                         msg2)))
 553             return
 554
 555         # Do not execute the command if the machine could not be reached
 556         if not self.machine.successfully_connected(self.logger):
 557             self._has_finished = True
 558             self.out = "N\A"
 559             self.err += ("Connection to machine (name : %s, host: %s, port:"
 560                         " %s, user: %s) has failed\nUse the log command "
 561                         "to get more information."
 562                         % (self.machine.name,
 563                            self.machine.host,
 564                            self.machine.port,
 565                            self.machine.user))
 566         else:
 567             # Usual case : Launch the command on remote machine
 568             self._T0 = time.time()
 569             self._stdin, self._stdout, self._stderr = self.machine.exec_command(
 570                                                                   self.command,
 571                                                                   self.logger)
 572             # If the results are not initialized, finish the job
 573             if (self._stdin, self._stdout, self._stderr) == (None, None, None):
 574                 self._has_finished = True
 575                 self._Tf = time.time()
 576                 self.out += "N\A"
 577                 self.err += "The server failed to execute the command"
 578
 579         # Put the beginning flag to true.
 580         self._has_begun = True
 581
 582     def write_results(self):
 583         """Display on the terminal all the job's information
 584         """
 585         self.logger.write("name : " + self.name + "\n")
 586         if self.after:
 587             self.logger.write("after : %s\n" % self.after)
 588         self.logger.write("Time elapsed : %4imin %2is \n" %
 589                      (self.total_duration()//60 , self.total_duration()%60))
 590         if self._T0 != -1:
 591             self.logger.write("Begin time : %s\n" %
 592                          time.strftime('%Y-%m-%d %H:%M:%S',
 593                                        time.localtime(self._T0)) )
 594         if self._Tf != -1:
 595             self.logger.write("End time   : %s\n\n" %
 596                          time.strftime('%Y-%m-%d %H:%M:%S',
 597                                        time.localtime(self._Tf)) )
 598
 599         machine_head = "Informations about connection :\n"
 600         underline = (len(machine_head) - 2) * "-"
 601         self.logger.write(src.printcolors.printcInfo(
 602                                                 machine_head+underline+"\n"))
 603         self.machine.write_info(self.logger)
 604
 605         self.logger.write(src.printcolors.printcInfo("out : \n"))
 606         if self.out == "":
 607             self.logger.write("Unable to get output\n")
 608         else:
 609             self.logger.write(self.out + "\n")
 610         self.logger.write(src.printcolors.printcInfo("err : \n"))
 611         self.logger.write(self.err + "\n")
 612
 613     def get_status(self):
 614         """Get the status of the job (used by the Gui for xml display)
 615
 616         :return: The current status of the job
 617         :rtype: String
 618         """
 619         if not self.machine.successfully_connected(self.logger):
 620             return "SSH connection KO"
 621         if not self.has_begun():
 622             return "Not launched"
 623         if self.cancelled:
 624             return "Cancelled"
 625         if self.is_running():
 626             return "running since " + time.strftime('%Y-%m-%d %H:%M:%S',
 627                                                     time.localtime(self._T0))
 628         if self.has_finished():
 629             if self.is_timeout():
 630                 return "Timeout since " + time.strftime('%Y-%m-%d %H:%M:%S',
 631                                                     time.localtime(self._Tf))
 632             return "Finished since " + time.strftime('%Y-%m-%d %H:%M:%S',
 633                                                      time.localtime(self._Tf))
 634
 635 class Jobs(object):
 636     '''Class to manage the jobs to be run
 637     '''
 638     def __init__(self,
 639                  runner,
 640                  logger,
 641                  job_file_path,
 642                  config_jobs,
 643                  lenght_columns = 20):
 644         # The jobs configuration
 645         self.cfg_jobs = config_jobs
 646         self.job_file_path = job_file_path
 647         # The machine that will be used today
 648         self.lmachines = []
 649         # The list of machine (hosts, port) that will be used today
 650         # (a same host can have several machine instances since there
 651         # can be several ssh parameters)
 652         self.lhosts = []
 653         # The jobs to be launched today
 654         self.ljobs = []
 655         # The jobs that will not be launched today
 656         self.ljobs_not_today = []
 657         self.runner = runner
 658         self.logger = logger
 659         self.len_columns = lenght_columns
 660
 661         # the list of jobs that have not been run yet
 662         self._l_jobs_not_started = []
 663         # the list of jobs that have already ran
 664         self._l_jobs_finished = []
 665         # the list of jobs that are running
 666         self._l_jobs_running = []
 667
 668         self.determine_jobs_and_machines()
 669
 670     def define_job(self, job_def, machine):
 671         '''Takes a pyconf job definition and a machine (from class machine)
 672            and returns the job instance corresponding to the definition.
 673
 674         :param job_def src.config.Mapping: a job definition
 675         :param machine machine: the machine on which the job will run
 676         :return: The corresponding job in a job class instance
 677         :rtype: job
 678         '''
 679         name = job_def.name
 680         cmmnds = job_def.commands
 681         if not "timeout" in job_def:
 682             timeout = 4*60*60 # default timeout = 4h
 683         else:
 684             timeout = job_def.timeout
 685         after = None
 686         if 'after' in job_def:
 687             after = job_def.after
 688         application = None
 689         if 'application' in job_def:
 690             application = job_def.application
 691         board = None
 692         if 'board' in job_def:
 693             board = job_def.board
 694         prefix = None
 695         if "prefix" in job_def:
 696             prefix = job_def.prefix
 697
 698         return Job(name,
 699                    machine,
 700                    application,
 701                    board,
 702                    cmmnds,
 703                    timeout,
 704                    self.runner.cfg,
 705                    self.job_file_path,
 706                    self.logger,
 707                    after = after,
 708                    prefix = prefix)
 709
 710     def determine_jobs_and_machines(self):
 711         '''Function that reads the pyconf jobs definition and instantiates all
 712            the machines and jobs to be done today.
 713
 714         :return: Nothing
 715         :rtype: N\A
 716         '''
 717         today = datetime.date.weekday(datetime.date.today())
 718         host_list = []
 719
 720         for job_def in self.cfg_jobs.jobs :
 721
 722             if not "machine" in job_def:
 723                 msg = _('WARNING: The job "%s" do not have the key '
 724                        '"machine", this job is ignored.\n\n' % job_def.name)
 725                 self.logger.write(src.printcolors.printcWarning(msg))
 726                 continue
 727             name_machine = job_def.machine
 728
 729             a_machine = None
 730             for mach in self.lmachines:
 731                 if mach.name == name_machine:
 732                     a_machine = mach
 733                     break
 734
 735             if a_machine == None:
 736                 for machine_def in self.cfg_jobs.machines:
 737                     if machine_def.name == name_machine:
 738                         if 'host' not in machine_def:
 739                             host = self.runner.cfg.VARS.hostname
 740                         else:
 741                             host = machine_def.host
 742
 743                         if 'user' not in machine_def:
 744                             user = self.runner.cfg.VARS.user
 745                         else:
 746                             user = machine_def.user
 747
 748                         if 'port' not in machine_def:
 749                             port = 22
 750                         else:
 751                             port = machine_def.port
 752
 753                         if 'password' not in machine_def:
 754                             passwd = None
 755                         else:
 756                             passwd = machine_def.password
 757
 758                         if 'sat_path' not in machine_def:
 759                             sat_path = "salomeTools"
 760                         else:
 761                             sat_path = machine_def.sat_path
 762
 763                         a_machine = Machine(
 764                                             machine_def.name,
 765                                             host,
 766                                             user,
 767                                             port=port,
 768                                             passwd=passwd,
 769                                             sat_path=sat_path
 770                                             )
 771
 772                         self.lmachines.append(a_machine)
 773                         if (host, port) not in host_list:
 774                             host_list.append((host, port))
 775
 776                 if a_machine == None:
 777                     msg = _("WARNING: The job \"%(job_name)s\" requires the "
 778                             "machine \"%(machine_name)s\" but this machine "
 779                             "is not defined in the configuration file.\n"
 780                             "The job will not be launched\n")
 781                     self.logger.write(src.printcolors.printcWarning(
 782                                         msg % {"job_name" : job_def.name,
 783                                                "machine_name" : name_machine}))
 784                     continue
 785
 786             a_job = self.define_job(job_def, a_machine)
 787
 788             if today in job_def.when:
 789                 self.ljobs.append(a_job)
 790             else: # today in job_def.when
 791                 self.ljobs_not_today.append(a_job)
 792
 793         self.lhosts = host_list
 794
 795     def ssh_connection_all_machines(self, pad=50):
 796         '''Function that do the ssh connection to every machine
 797            to be used today.
 798
 799         :return: Nothing
 800         :rtype: N\A
 801         '''
 802         self.logger.write(src.printcolors.printcInfo((
 803                         "Establishing connection with all the machines :\n")))
 804         for machine in self.lmachines:
 805             # little algorithm in order to display traces
 806             begin_line = (_("Connection to %s: " % machine.name))
 807             if pad - len(begin_line) < 0:
 808                 endline = " "
 809             else:
 810                 endline = (pad - len(begin_line)) * "." + " "
 811
 812             step = "SSH connection"
 813             self.logger.write( begin_line + endline + step)
 814             self.logger.flush()
 815             # the call to the method that initiate the ssh connection
 816             msg = machine.connect(self.logger)
 817
 818             # Copy salomeTools to the remote machine
 819             if machine.successfully_connected(self.logger):
 820                 step = _("Remove SAT")
 821                 self.logger.write('\r%s%s%s' % (begin_line, endline, 20 * " "),3)
 822                 self.logger.write('\r%s%s%s' % (begin_line, endline, step), 3)
 823                 (__, out_dist, __) = machine.exec_command(
 824                                                 "rm -rf %s" % machine.sat_path,
 825                                                 self.logger)
 826                 out_dist.read()
 827
 828                 self.logger.flush()
 829                 step = _("Copy SAT")
 830                 self.logger.write('\r%s%s%s' % (begin_line, endline, 20 * " "),3)
 831                 self.logger.write('\r%s%s%s' % (begin_line, endline, step), 3)
 832                 self.logger.flush()
 833                 res_copy = machine.copy_sat(self.runner.cfg.VARS.salometoolsway,
 834                                             self.job_file_path)
 835                 # get the remote machine distribution using a sat command
 836                 (__, out_dist, __) = machine.exec_command(
 837                                 os.path.join(machine.sat_path,
 838                                     "sat config --value VARS.dist --no_label"),
 839                                 self.logger)
 840                 machine.distribution = out_dist.read().decode().replace("\n",
 841                                                                         "")
 842                 # Print the status of the copy
 843                 if res_copy == 0:
 844                     self.logger.write('\r%s' %
 845                                 ((len(begin_line)+len(endline)+20) * " "), 3)
 846                     self.logger.write('\r%s%s%s' %
 847                         (begin_line,
 848                          endline,
 849                          src.printcolors.printc(src.OK_STATUS)), 3)
 850                 else:
 851                     self.logger.write('\r%s' %
 852                             ((len(begin_line)+len(endline)+20) * " "), 3)
 853                     self.logger.write('\r%s%s%s %s' %
 854                         (begin_line,
 855                          endline,
 856                          src.printcolors.printc(src.KO_STATUS),
 857                          _("Copy of SAT failed: %s" % res_copy)), 3)
 858             else:
 859                 self.logger.write('\r%s' %
 860                                   ((len(begin_line)+len(endline)+20) * " "), 3)
 861                 self.logger.write('\r%s%s%s %s' %
 862                     (begin_line,
 863                      endline,
 864                      src.printcolors.printc(src.KO_STATUS),
 865                      msg), 3)
 866             self.logger.write("\n", 3)
 867
 868         self.logger.write("\n")
 869
 870
 871     def is_occupied(self, hostname):
 872         '''Function that returns True if a job is running on
 873            the machine defined by its host and its port.
 874
 875         :param hostname (str, int): the pair (host, port)
 876         :return: the job that is running on the host,
 877                 or false if there is no job running on the host.
 878         :rtype: job / bool
 879         '''
 880         host = hostname[0]
 881         port = hostname[1]
 882         for jb in self.ljobs:
 883             if jb.machine.host == host and jb.machine.port == port:
 884                 if jb.is_running():
 885                     return jb
 886         return False
 887
 888     def update_jobs_states_list(self):
 889         '''Function that updates the lists that store the currently
 890            running jobs and the jobs that have already finished.
 891
 892         :return: Nothing.
 893         :rtype: N\A
 894         '''
 895         jobs_finished_list = []
 896         jobs_running_list = []
 897         for jb in self.ljobs:
 898             if jb.is_running():
 899                 jobs_running_list.append(jb)
 900                 jb.check_time()
 901             if jb.has_finished():
 902                 jobs_finished_list.append(jb)
 903
 904         nb_job_finished_before = len(self._l_jobs_finished)
 905         self._l_jobs_finished = jobs_finished_list
 906         self._l_jobs_running = jobs_running_list
 907
 908         nb_job_finished_now = len(self._l_jobs_finished)
 909
 910         return nb_job_finished_now > nb_job_finished_before
 911
 912     def cancel_dependencies_of_failing_jobs(self):
 913         '''Function that cancels all the jobs that depend on a failing one.
 914
 915         :return: Nothing.
 916         :rtype: N\A
 917         '''
 918
 919         for job in self.ljobs:
 920             if job.after is None:
 921                 continue
 922             father_job = self.find_job_that_has_name(job.after)
 923             if father_job is not None and father_job.has_failed():
 924                 job.cancel()
 925
 926     def find_job_that_has_name(self, name):
 927         '''Returns the job by its name.
 928
 929         :param name str: a job name
 930         :return: the job that has the name.
 931         :rtype: job
 932         '''
 933         for jb in self.ljobs:
 934             if jb.name == name:
 935                 return jb
 936         # the following is executed only if the job was not found
 937         return None
 938
 939     def str_of_length(self, text, length):
 940         '''Takes a string text of any length and returns
 941            the most close string of length "length".
 942
 943         :param text str: any string
 944         :param length int: a length for the returned string
 945         :return: the most close string of length "length"
 946         :rtype: str
 947         '''
 948         if len(text) > length:
 949             text_out = text[:length-3] + '...'
 950         else:
 951             diff = length - len(text)
 952             before = " " * (diff//2)
 953             after = " " * (diff//2 + diff%2)
 954             text_out = before + text + after
 955
 956         return text_out
 957
 958     def display_status(self, len_col):
 959         '''Takes a lenght and construct the display of the current status
 960            of the jobs in an array that has a column for each host.
 961            It displays the job that is currently running on the host
 962            of the column.
 963
 964         :param len_col int: the size of the column
 965         :return: Nothing
 966         :rtype: N\A
 967         '''
 968
 969         display_line = ""
 970         for host_port in self.lhosts:
 971             jb = self.is_occupied(host_port)
 972             if not jb: # nothing running on the host
 973                 empty = self.str_of_length("empty", len_col)
 974                 display_line += "|" + empty
 975             else:
 976                 display_line += "|" + src.printcolors.printcInfo(
 977                                         self.str_of_length(jb.name, len_col))
 978
 979         self.logger.write("\r" + display_line + "|")
 980         self.logger.flush()
 981
 982
 983     def run_jobs(self):
 984         '''The main method. Runs all the jobs on every host.
 985            For each host, at a given time, only one job can be running.
 986            The jobs that have the field after (that contain the job that has
 987            to be run before it) are run after the previous job.
 988            This method stops when all the jobs are finished.
 989
 990         :return: Nothing
 991         :rtype: N\A
 992         '''
 993
 994         # Print header
 995         self.logger.write(src.printcolors.printcInfo(
 996                                                 _('Executing the jobs :\n')))
 997         text_line = ""
 998         for host_port in self.lhosts:
 999             host = host_port[0]
1000             port = host_port[1]
1001             if port == 22: # default value
1002                 text_line += "|" + self.str_of_length(host, self.len_columns)
1003             else:
1004                 text_line += "|" + self.str_of_length(
1005                                 "("+host+", "+str(port)+")", self.len_columns)
1006
1007         tiret_line = " " + "-"*(len(text_line)-1) + "\n"
1008         self.logger.write(tiret_line)
1009         self.logger.write(text_line + "|\n")
1010         self.logger.write(tiret_line)
1011         self.logger.flush()
1012
1013         # The infinite loop that runs the jobs
1014         l_jobs_not_started = src.deepcopy_list(self.ljobs)
1015         while len(self._l_jobs_finished) != len(self.ljobs):
1016             new_job_start = False
1017             for host_port in self.lhosts:
1018
1019                 if self.is_occupied(host_port):
1020                     continue
1021
1022                 for jb in l_jobs_not_started:
1023                     if (jb.machine.host, jb.machine.port) != host_port:
1024                         continue
1025                     if jb.after == None:
1026                         jb.run()
1027                         l_jobs_not_started.remove(jb)
1028                         new_job_start = True
1029                         break
1030                     else:
1031                         jb_before = self.find_job_that_has_name(jb.after)
1032                         if jb_before is None:
1033                             jb.cancel()
1034                             msg = _("This job was not launched because its "
1035                                     "father is not in the jobs list.")
1036                             jb.out = msg
1037                             jb.err = msg
1038                             break
1039                         if jb_before.has_finished():
1040                             jb.run()
1041                             l_jobs_not_started.remove(jb)
1042                             new_job_start = True
1043                             break
1044             self.cancel_dependencies_of_failing_jobs()
1045             new_job_finished = self.update_jobs_states_list()
1046
1047             if new_job_start or new_job_finished:
1048                 if self.gui:
1049                     self.gui.update_xml_files(self.ljobs)
1050                 # Display the current status
1051                 self.display_status(self.len_columns)
1052
1053             # Make sure that the proc is not entirely busy
1054             time.sleep(0.001)
1055
1056         self.logger.write("\n")
1057         self.logger.write(tiret_line)
1058         self.logger.write("\n\n")
1059
1060         if self.gui:
1061             self.gui.update_xml_files(self.ljobs)
1062             self.gui.last_update()
1063
1064     def write_all_results(self):
1065         '''Display all the jobs outputs.
1066
1067         :return: Nothing
1068         :rtype: N\A
1069         '''
1070
1071         for jb in self.ljobs:
1072             self.logger.write(src.printcolors.printcLabel(
1073                         "#------- Results for job %s -------#\n" % jb.name))
1074             jb.write_results()
1075             self.logger.write("\n\n")
1076
1077 class Gui(object):
1078     '''Class to manage the the xml data that can be displayed in a browser to
1079        see the jobs states
1080     '''
1081
1082     def __init__(self,
1083                  xml_dir_path,
1084                  l_jobs,
1085                  l_jobs_not_today,
1086                  prefix,
1087                  logger,
1088                  file_boards=""):
1089         '''Initialization
1090
1091         :param xml_dir_path str: The path to the directory where to put
1092                                  the xml resulting files
1093         :param l_jobs List: the list of jobs that run today
1094         :param l_jobs_not_today List: the list of jobs that do not run today
1095         :param file_boards str: the file path from which to read the
1096                                    expected boards
1097         '''
1098         # The logging instance
1099         self.logger = logger
1100
1101         # The prefix to add to the xml files : date_hour
1102         self.prefix = prefix
1103
1104         # The path of the csv files to read to fill the expected boards
1105         self.file_boards = file_boards
1106
1107         if file_boards != "":
1108             today = datetime.date.weekday(datetime.date.today())
1109             self.parse_csv_boards(today)
1110         else:
1111             self.d_input_boards = {}
1112
1113         # The path of the global xml file
1114         self.xml_dir_path = xml_dir_path
1115         # Initialize the xml files
1116         self.global_name = "global_report"
1117         xml_global_path = os.path.join(self.xml_dir_path,
1118                                        self.global_name + ".xml")
1119         self.xml_global_file = src.xmlManager.XmlLogFile(xml_global_path,
1120                                                          "JobsReport")
1121
1122         # Find history for each job
1123         self.history = {}
1124         self.find_history(l_jobs, l_jobs_not_today)
1125
1126         # The xml files that corresponds to the boards.
1127         # {name_board : xml_object}}
1128         self.d_xml_board_files = {}
1129
1130         # Create the lines and columns
1131         self.initialize_boards(l_jobs, l_jobs_not_today)
1132
1133         # Write the xml file
1134         self.update_xml_files(l_jobs)
1135
1136     def add_xml_board(self, name):
1137         '''Add a board to the board list
1138         :param name str: the board name
1139         '''
1140         xml_board_path = os.path.join(self.xml_dir_path, name + ".xml")
1141         self.d_xml_board_files[name] =  src.xmlManager.XmlLogFile(
1142                                                     xml_board_path,
1143                                                     "JobsReport")
1144         self.d_xml_board_files[name].add_simple_node("distributions")
1145         self.d_xml_board_files[name].add_simple_node("applications")
1146         self.d_xml_board_files[name].add_simple_node("board", text=name)
1147
1148     def initialize_boards(self, l_jobs, l_jobs_not_today):
1149         '''Get all the first information needed for each file and write the
1150            first version of the files
1151         :param l_jobs List: the list of jobs that run today
1152         :param l_jobs_not_today List: the list of jobs that do not run today
1153         '''
1154         # Get the boards to fill and put it in a dictionary
1155         # {board_name : xml instance corresponding to the board}
1156         for job in l_jobs + l_jobs_not_today:
1157             board = job.board
1158             if (board is not None and
1159                                 board not in self.d_xml_board_files.keys()):
1160                 self.add_xml_board(board)
1161
1162         # Verify that the boards given as input are done
1163         for board in list(self.d_input_boards.keys()):
1164             if board not in self.d_xml_board_files:
1165                 self.add_xml_board(board)
1166             root_node = self.d_xml_board_files[board].xmlroot
1167             src.xmlManager.append_node_attrib(root_node,
1168                                               {"input_file" : self.file_boards})
1169
1170         # Loop over all jobs in order to get the lines and columns for each
1171         # xml file
1172         d_dist = {}
1173         d_application = {}
1174         for board in self.d_xml_board_files:
1175             d_dist[board] = []
1176             d_application[board] = []
1177
1178         l_hosts_ports = []
1179
1180         for job in l_jobs + l_jobs_not_today:
1181
1182             if (job.machine.host, job.machine.port) not in l_hosts_ports:
1183                 l_hosts_ports.append((job.machine.host, job.machine.port))
1184
1185             distrib = job.machine.distribution
1186             application = job.application
1187
1188             board_job = job.board
1189             if board is None:
1190                 continue
1191             for board in self.d_xml_board_files:
1192                 if board_job == board:
1193                     if (distrib not in [None, ''] and
1194                                             distrib not in d_dist[board]):
1195                         d_dist[board].append(distrib)
1196                         src.xmlManager.add_simple_node(
1197                             self.d_xml_board_files[board].xmlroot.find(
1198                                                             'distributions'),
1199                                                    "dist",
1200                                                    attrib={"name" : distrib})
1201
1202                 if board_job == board:
1203                     if (application not in [None, ''] and
1204                                     application not in d_application[board]):
1205                         d_application[board].append(application)
1206                         src.xmlManager.add_simple_node(
1207                             self.d_xml_board_files[board].xmlroot.find(
1208                                                                 'applications'),
1209                                                    "application",
1210                                                    attrib={
1211                                                         "name" : application})
1212
1213         # Verify that there are no missing application or distribution in the
1214         # xml board files (regarding the input boards)
1215         for board in self.d_xml_board_files:
1216             l_dist = d_dist[board]
1217             if board not in self.d_input_boards.keys():
1218                 continue
1219             for dist in self.d_input_boards[board]["rows"]:
1220                 if dist not in l_dist:
1221                     src.xmlManager.add_simple_node(
1222                             self.d_xml_board_files[board].xmlroot.find(
1223                                                             'distributions'),
1224                                                    "dist",
1225                                                    attrib={"name" : dist})
1226             l_appli = d_application[board]
1227             for appli in self.d_input_boards[board]["columns"]:
1228                 if appli not in l_appli:
1229                     src.xmlManager.add_simple_node(
1230                             self.d_xml_board_files[board].xmlroot.find(
1231                                                                 'applications'),
1232                                                    "application",
1233                                                    attrib={"name" : appli})
1234
1235         # Initialize the hosts_ports node for the global file
1236         self.xmlhosts_ports = self.xml_global_file.add_simple_node(
1237                                                                 "hosts_ports")
1238         for host, port in l_hosts_ports:
1239             host_port = "%s:%i" % (host, port)
1240             src.xmlManager.add_simple_node(self.xmlhosts_ports,
1241                                            "host_port",
1242                                            attrib={"name" : host_port})
1243
1244         # Initialize the jobs node in all files
1245         for xml_file in [self.xml_global_file] + list(
1246                                             self.d_xml_board_files.values()):
1247             xml_jobs = xml_file.add_simple_node("jobs")
1248             # Get the jobs present in the config file but
1249             # that will not be launched today
1250             self.put_jobs_not_today(l_jobs_not_today, xml_jobs)
1251
1252             # add also the infos node
1253             xml_file.add_simple_node("infos",
1254                                      attrib={"name" : "last update",
1255                                              "JobsCommandStatus" : "running"})
1256
1257             # and put the history node
1258             history_node = xml_file.add_simple_node("history")
1259             name_board = os.path.basename(xml_file.logFile)[:-len(".xml")]
1260             # serach for board files
1261             expression = "^[0-9]{8}_+[0-9]{6}_" + name_board + ".xml$"
1262             oExpr = re.compile(expression)
1263             # Get the list of xml borad files that are in the log directory
1264             for file_name in os.listdir(self.xml_dir_path):
1265                 if oExpr.search(file_name):
1266                     date = os.path.basename(file_name).split("_")[0]
1267                     file_path = os.path.join(self.xml_dir_path, file_name)
1268                     src.xmlManager.add_simple_node(history_node,
1269                                                    "link",
1270                                                    text=file_path,
1271                                                    attrib={"date" : date})
1272
1273
1274         # Find in each board the squares that needs to be filled regarding the
1275         # input csv files but that are not covered by a today job
1276         for board in self.d_input_boards.keys():
1277             xml_root_board = self.d_xml_board_files[board].xmlroot
1278             # Find the missing jobs for today
1279             xml_missing = src.xmlManager.add_simple_node(xml_root_board,
1280                                                  "missing_jobs")
1281             for row, column in self.d_input_boards[board]["jobs"]:
1282                 found = False
1283                 for job in l_jobs:
1284                     if (job.application == column and
1285                         job.machine.distribution == row):
1286                         found = True
1287                         break
1288                 if not found:
1289                     src.xmlManager.add_simple_node(xml_missing,
1290                                             "job",
1291                                             attrib={"distribution" : row,
1292                                                     "application" : column })
1293             # Find the missing jobs not today
1294             xml_missing_not_today = src.xmlManager.add_simple_node(
1295                                                  xml_root_board,
1296                                                  "missing_jobs_not_today")
1297             for row, column in self.d_input_boards[board]["jobs_not_today"]:
1298                 found = False
1299                 for job in l_jobs_not_today:
1300                     if (job.application == column and
1301                         job.machine.distribution == row):
1302                         found = True
1303                         break
1304                 if not found:
1305                     src.xmlManager.add_simple_node(xml_missing_not_today,
1306                                             "job",
1307                                             attrib={"distribution" : row,
1308                                                     "application" : column })
1309
1310     def find_history(self, l_jobs, l_jobs_not_today):
1311         """find, for each job, in the existent xml boards the results for the
1312            job. Store the results in the dictionnary self.history = {name_job :
1313            list of (date, status, list links)}
1314
1315         :param l_jobs List: the list of jobs to run today
1316         :param l_jobs_not_today List: the list of jobs that do not run today
1317         """
1318         # load the all the history
1319         expression = "^[0-9]{8}_+[0-9]{6}_" + self.global_name + ".xml$"
1320         oExpr = re.compile(expression)
1321         # Get the list of global xml that are in the log directory
1322         l_globalxml = []
1323         for file_name in os.listdir(self.xml_dir_path):
1324             if oExpr.search(file_name):
1325                 file_path = os.path.join(self.xml_dir_path, file_name)
1326                 try:
1327                     global_xml = src.xmlManager.ReadXmlFile(file_path)
1328                     l_globalxml.append(global_xml)
1329                 except Exception as e:
1330                     msg = _("\nWARNING: the file %s can not be read, it will be "
1331                             "ignored\n%s" % (file_path, e))
1332                     self.logger.write("%s\n" % src.printcolors.printcWarning(
1333                                                                         msg), 5)
1334
1335         # Construct the dictionnary self.history
1336         for job in l_jobs + l_jobs_not_today:
1337             l_links = []
1338             for global_xml in l_globalxml:
1339                 date = os.path.basename(global_xml.filePath).split("_")[0]
1340                 global_root_node = global_xml.xmlroot.find("jobs")
1341                 job_node = src.xmlManager.find_node_by_attrib(
1342                                                               global_root_node,
1343                                                               "job",
1344                                                               "name",
1345                                                               job.name)
1346                 if job_node:
1347                     if job_node.find("remote_log_file_path") is not None:
1348                         link = job_node.find("remote_log_file_path").text
1349                         res_job = job_node.find("res").text
1350                         if link != "nothing":
1351                             l_links.append((date, res_job, link))
1352             l_links = sorted(l_links, reverse=True)
1353             self.history[job.name] = l_links
1354
1355     def put_jobs_not_today(self, l_jobs_not_today, xml_node_jobs):
1356         '''Get all the first information needed for each file and write the
1357            first version of the files
1358
1359         :param xml_node_jobs etree.Element: the node corresponding to a job
1360         :param l_jobs_not_today List: the list of jobs that do not run today
1361         '''
1362         for job in l_jobs_not_today:
1363             xmlj = src.xmlManager.add_simple_node(xml_node_jobs,
1364                                                  "job",
1365                                                  attrib={"name" : job.name})
1366             src.xmlManager.add_simple_node(xmlj, "application", job.application)
1367             src.xmlManager.add_simple_node(xmlj,
1368                                            "distribution",
1369                                            job.machine.distribution)
1370             src.xmlManager.add_simple_node(xmlj, "board", job.board)
1371             src.xmlManager.add_simple_node(xmlj,
1372                                        "commands", " ; ".join(job.commands))
1373             src.xmlManager.add_simple_node(xmlj, "state", "Not today")
1374             src.xmlManager.add_simple_node(xmlj, "machine", job.machine.name)
1375             src.xmlManager.add_simple_node(xmlj, "host", job.machine.host)
1376             src.xmlManager.add_simple_node(xmlj, "port", str(job.machine.port))
1377             src.xmlManager.add_simple_node(xmlj, "user", job.machine.user)
1378             src.xmlManager.add_simple_node(xmlj, "sat_path",
1379                                                         job.machine.sat_path)
1380             xml_history = src.xmlManager.add_simple_node(xmlj, "history")
1381             for i, (date, res_job, link) in enumerate(self.history[job.name]):
1382                 if i==0:
1383                     # tag the first one (the last one)
1384                     src.xmlManager.add_simple_node(xml_history,
1385                                                    "link",
1386                                                    text=link,
1387                                                    attrib={"date" : date,
1388                                                            "res" : res_job,
1389                                                            "last" : "yes"})
1390                 else:
1391                     src.xmlManager.add_simple_node(xml_history,
1392                                                    "link",
1393                                                    text=link,
1394                                                    attrib={"date" : date,
1395                                                            "res" : res_job,
1396                                                            "last" : "no"})
1397
1398     def parse_csv_boards(self, today):
1399         """ Parse the csv file that describes the boards to produce and fill
1400             the dict d_input_boards that contain the csv file contain
1401
1402         :param today int: the current day of the week
1403         """
1404         # open the csv file and read its content
1405         l_read = []
1406         with open(self.file_boards, 'r') as f:
1407             reader = csv.reader(f,delimiter=CSV_DELIMITER)
1408             for row in reader:
1409                 l_read.append(row)
1410         # get the delimiter for the boards (empty line)
1411         boards_delimiter = [''] * len(l_read[0])
1412         # Make the list of boards, by splitting with the delimiter
1413         l_boards = [list(y) for x, y in itertools.groupby(l_read,
1414                                     lambda z: z == boards_delimiter) if not x]
1415
1416         # loop over the csv lists of lines and get the rows, columns and jobs
1417         d_boards = {}
1418         for input_board in l_boards:
1419             # get board name
1420             board_name = input_board[0][0]
1421
1422             # Get columns list
1423             columns = input_board[0][1:]
1424
1425             rows = []
1426             jobs = []
1427             jobs_not_today = []
1428             for line in input_board[1:]:
1429                 row = line[0]
1430                 rows.append(row)
1431                 for i, square in enumerate(line[1:]):
1432                     if square=='':
1433                         continue
1434                     days = square.split(DAYS_SEPARATOR)
1435                     days = [int(day) for day in days]
1436                     job = (row, columns[i])
1437                     if today in days:
1438                         jobs.append(job)
1439                     else:
1440                         jobs_not_today.append(job)
1441
1442             d_boards[board_name] = {"rows" : rows,
1443                                     "columns" : columns,
1444                                     "jobs" : jobs,
1445                                     "jobs_not_today" : jobs_not_today}
1446
1447         self.d_input_boards = d_boards
1448
1449     def update_xml_files(self, l_jobs):
1450         '''Write all the xml files with updated information about the jobs
1451
1452         :param l_jobs List: the list of jobs that run today
1453         '''
1454         for xml_file in [self.xml_global_file] + list(
1455                                             self.d_xml_board_files.values()):
1456             self.update_xml_file(l_jobs, xml_file)
1457
1458         # Write the file
1459         self.write_xml_files()
1460
1461     def update_xml_file(self, l_jobs, xml_file):
1462         '''update information about the jobs for the file xml_file
1463
1464         :param l_jobs List: the list of jobs that run today
1465         :param xml_file xmlManager.XmlLogFile: the xml instance to update
1466         '''
1467
1468         xml_node_jobs = xml_file.xmlroot.find('jobs')
1469         # Update the job names and status node
1470         for job in l_jobs:
1471             # Find the node corresponding to the job and delete it
1472             # in order to recreate it
1473             for xmljob in xml_node_jobs.findall('job'):
1474                 if xmljob.attrib['name'] == job.name:
1475                     xml_node_jobs.remove(xmljob)
1476
1477             T0 = str(job._T0)
1478             if T0 != "-1":
1479                 T0 = time.strftime('%Y-%m-%d %H:%M:%S',
1480                                        time.localtime(job._T0))
1481             Tf = str(job._Tf)
1482             if Tf != "-1":
1483                 Tf = time.strftime('%Y-%m-%d %H:%M:%S',
1484                                        time.localtime(job._Tf))
1485
1486             # recreate the job node
1487             xmlj = src.xmlManager.add_simple_node(xml_node_jobs,
1488                                                   "job",
1489                                                   attrib={"name" : job.name})
1490             src.xmlManager.add_simple_node(xmlj, "machine", job.machine.name)
1491             src.xmlManager.add_simple_node(xmlj, "host", job.machine.host)
1492             src.xmlManager.add_simple_node(xmlj, "port", str(job.machine.port))
1493             src.xmlManager.add_simple_node(xmlj, "user", job.machine.user)
1494             xml_history = src.xmlManager.add_simple_node(xmlj, "history")
1495             for date, res_job, link in self.history[job.name]:
1496                 src.xmlManager.add_simple_node(xml_history,
1497                                                "link",
1498                                                text=link,
1499                                                attrib={"date" : date,
1500                                                        "res" : res_job})
1501
1502             src.xmlManager.add_simple_node(xmlj, "sat_path",
1503                                            job.machine.sat_path)
1504             src.xmlManager.add_simple_node(xmlj, "application", job.application)
1505             src.xmlManager.add_simple_node(xmlj, "distribution",
1506                                            job.machine.distribution)
1507             src.xmlManager.add_simple_node(xmlj, "board", job.board)
1508             src.xmlManager.add_simple_node(xmlj, "timeout", str(job.timeout))
1509             src.xmlManager.add_simple_node(xmlj, "commands",
1510                                            " ; ".join(job.commands))
1511             src.xmlManager.add_simple_node(xmlj, "state", job.get_status())
1512             src.xmlManager.add_simple_node(xmlj, "begin", T0)
1513             src.xmlManager.add_simple_node(xmlj, "end", Tf)
1514             src.xmlManager.add_simple_node(xmlj, "out",
1515                                            src.printcolors.cleancolor(job.out))
1516             src.xmlManager.add_simple_node(xmlj, "err",
1517                                            src.printcolors.cleancolor(job.err))
1518             src.xmlManager.add_simple_node(xmlj, "res", str(job.res_job))
1519             if len(job.remote_log_files) > 0:
1520                 src.xmlManager.add_simple_node(xmlj,
1521                                                "remote_log_file_path",
1522                                                job.remote_log_files[0])
1523             else:
1524                 src.xmlManager.add_simple_node(xmlj,
1525                                                "remote_log_file_path",
1526                                                "nothing")
1527             # Search for the test log if there is any
1528             l_test_log_files = self.find_test_log(job.remote_log_files)
1529             xml_test = src.xmlManager.add_simple_node(xmlj,
1530                                                       "test_log_file_path")
1531             for test_log_path, res_test, nb_fails in l_test_log_files:
1532                 test_path_node = src.xmlManager.add_simple_node(xml_test,
1533                                                "path",
1534                                                test_log_path)
1535                 test_path_node.attrib["res"] = res_test
1536                 test_path_node.attrib["nb_fails"] = nb_fails
1537
1538             xmlafter = src.xmlManager.add_simple_node(xmlj, "after", job.after)
1539             # get the job father
1540             if job.after is not None:
1541                 job_father = None
1542                 for jb in l_jobs:
1543                     if jb.name == job.after:
1544                         job_father = jb
1545
1546                 if (job_father is not None and
1547                         len(job_father.remote_log_files) > 0):
1548                     link = job_father.remote_log_files[0]
1549                 else:
1550                     link = "nothing"
1551                 src.xmlManager.append_node_attrib(xmlafter, {"link" : link})
1552
1553             # Verify that the job is to be done today regarding the input csv
1554             # files
1555             if job.board and job.board in self.d_input_boards.keys():
1556                 found = False
1557                 for dist, appli in self.d_input_boards[job.board]["jobs"]:
1558                     if (job.machine.distribution == dist
1559                         and job.application == appli):
1560                         found = True
1561                         src.xmlManager.add_simple_node(xmlj,
1562                                                "extra_job",
1563                                                "no")
1564                         break
1565                 if not found:
1566                     src.xmlManager.add_simple_node(xmlj,
1567                                                "extra_job",
1568                                                "yes")
1569
1570
1571         # Update the date
1572         xml_node_infos = xml_file.xmlroot.find('infos')
1573         src.xmlManager.append_node_attrib(xml_node_infos,
1574                     attrib={"value" :
1575                     datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")})
1576
1577
1578     def find_test_log(self, l_remote_log_files):
1579         '''Find if there is a test log (board) in the remote log files and
1580            the path to it. There can be several test command, so the result is
1581            a list.
1582
1583         :param l_remote_log_files List: the list of all remote log files
1584         :return: the list of (test log files path, res of the command)
1585         :rtype: List
1586         '''
1587         res = []
1588         for file_path in l_remote_log_files:
1589             dirname = os.path.basename(os.path.dirname(file_path))
1590             file_name = os.path.basename(file_path)
1591             regex = src.logger.log_all_command_file_expression
1592             oExpr = re.compile(regex)
1593             if dirname == "TEST" and oExpr.search(file_name):
1594                 # find the res of the command
1595                 prod_node = etree.parse(file_path).getroot().find("product")
1596                 res_test = prod_node.attrib["global_res"]
1597                 # find the number of fails
1598                 testbase_node = prod_node.find("tests").find("testbase")
1599                 nb_fails = int(testbase_node.attrib["failed"])
1600                 # put the file path, the res of the test command and the number
1601                 # of fails in the output
1602                 res.append((file_path, res_test, nb_fails))
1603
1604         return res
1605
1606     def last_update(self, finish_status = "finished"):
1607         '''update information about the jobs for the file xml_file
1608
1609         :param l_jobs List: the list of jobs that run today
1610         :param xml_file xmlManager.XmlLogFile: the xml instance to update
1611         '''
1612         for xml_file in [self.xml_global_file] + list(self.d_xml_board_files.values()):
1613             xml_node_infos = xml_file.xmlroot.find('infos')
1614             src.xmlManager.append_node_attrib(xml_node_infos,
1615                         attrib={"JobsCommandStatus" : finish_status})
1616         # Write the file
1617         self.write_xml_files()
1618
1619     def write_xml_file(self, xml_file, stylesheet):
1620         ''' Write one xml file and the same file with prefix
1621         '''
1622         xml_file.write_tree(stylesheet)
1623         file_path = xml_file.logFile
1624         file_dir = os.path.dirname(file_path)
1625         file_name = os.path.basename(file_path)
1626         file_name_with_prefix = self.prefix + "_" + file_name
1627         xml_file.write_tree(stylesheet, os.path.join(file_dir,
1628                                                      file_name_with_prefix))
1629
1630     def write_xml_files(self):
1631         ''' Write the xml files
1632         '''
1633         self.write_xml_file(self.xml_global_file, STYLESHEET_GLOBAL)
1634         for xml_file in self.d_xml_board_files.values():
1635             self.write_xml_file(xml_file, STYLESHEET_BOARD)
1636
1637 def get_config_file_path(job_config_name, l_cfg_dir):
1638     found = False
1639     file_jobs_cfg = None
1640     if os.path.exists(job_config_name) and job_config_name.endswith(".pyconf"):
1641         found = True
1642         file_jobs_cfg = job_config_name
1643     else:
1644         for cfg_dir in l_cfg_dir:
1645             file_jobs_cfg = os.path.join(cfg_dir, job_config_name)
1646             if not file_jobs_cfg.endswith('.pyconf'):
1647                 file_jobs_cfg += '.pyconf'
1648
1649             if not os.path.exists(file_jobs_cfg):
1650                 continue
1651             else:
1652                 found = True
1653                 break
1654     return found, file_jobs_cfg
1655
1656 ##
1657 # Describes the command
1658 def description():
1659     return _("The jobs command launches maintenances that are described"
1660              " in the dedicated jobs configuration file.\n\nexample:\nsat "
1661              "jobs --name my_jobs --publish")
1662
1663 ##
1664 # Runs the command.
1665 def run(args, runner, logger):
1666
1667     (options, args) = parser.parse_args(args)
1668
1669     l_cfg_dir = runner.cfg.PATHS.JOBPATH
1670
1671     # list option : display all the available config files
1672     if options.list:
1673         for cfg_dir in l_cfg_dir:
1674             if not options.no_label:
1675                 logger.write("------ %s\n" %
1676                                  src.printcolors.printcHeader(cfg_dir))
1677             if not os.path.exists(cfg_dir):
1678                 continue
1679             for f in sorted(os.listdir(cfg_dir)):
1680                 if not f.endswith('.pyconf'):
1681                     continue
1682                 cfilename = f[:-7]
1683                 logger.write("%s\n" % cfilename)
1684         return 0
1685
1686     # Make sure the jobs_config option has been called
1687     if not options.jobs_cfg:
1688         message = _("The option --jobs_config is required\n")
1689         src.printcolors.printcError(message)
1690         return 1
1691
1692     # Find the file in the directories, unless it is a full path
1693     # merge all in a config
1694     merger = src.pyconf.ConfigMerger()
1695     config_jobs = src.pyconf.Config()
1696     l_conf_files_path = []
1697     for config_file in options.jobs_cfg:
1698         found, file_jobs_cfg = get_config_file_path(config_file, l_cfg_dir)
1699         if not found:
1700             msg = _("The file configuration %s was not found."
1701                     "\nUse the --list option to get the "
1702                     "possible files." % config_file)
1703             logger.write("%s\n" % src.printcolors.printcError(msg), 1)
1704             return 1
1705         l_conf_files_path.append(file_jobs_cfg)
1706         # Read the config that is in the file
1707         one_config_jobs = src.read_config_from_a_file(file_jobs_cfg)
1708         merger.merge(config_jobs, one_config_jobs)
1709
1710     info = [
1711         (_("Platform"), runner.cfg.VARS.dist),
1712         (_("Files containing the jobs configuration"), l_conf_files_path)
1713     ]
1714     src.print_info(logger, info)
1715
1716     if options.only_jobs:
1717         l_jb = src.pyconf.Sequence()
1718         for jb in config_jobs.jobs:
1719             if jb.name in options.only_jobs:
1720                 l_jb.append(jb,
1721                 "Job that was given in only_jobs option parameters\n")
1722         config_jobs.jobs = l_jb
1723
1724     # Make a unique file that contain all the jobs in order to use it
1725     # on every machine
1726     name_pyconf = "_".join([os.path.basename(path)[:-len('.pyconf')]
1727                             for path in l_conf_files_path]) + ".pyconf"
1728     path_pyconf = src.get_tmp_filename(runner.cfg, name_pyconf)
1729     #Save config
1730     f = file( path_pyconf , 'w')
1731     config_jobs.__save__(f)
1732
1733     # log the paramiko problems
1734     paramiko_log_dir_path = os.path.join(runner.cfg.USER.log_dir, "JOBS")
1735     src.ensure_path_exists(paramiko_log_dir_path)
1736     paramiko.util.log_to_file(os.path.join(paramiko_log_dir_path,
1737                                            logger.txtFileName))
1738
1739     # Initialization
1740     today_jobs = Jobs(runner,
1741                       logger,
1742                       path_pyconf,
1743                       config_jobs)
1744     # SSH connection to all machines
1745     today_jobs.ssh_connection_all_machines()
1746     if options.test_connection:
1747         return 0
1748
1749     gui = None
1750     if options.publish:
1751         logger.write(src.printcolors.printcInfo(
1752                                         _("Initialize the xml boards : ")), 5)
1753         logger.flush()
1754
1755         # Copy the stylesheets in the log directory
1756         log_dir = runner.cfg.USER.log_dir
1757         xsl_dir = os.path.join(runner.cfg.VARS.srcDir, 'xsl')
1758         files_to_copy = []
1759         files_to_copy.append(os.path.join(xsl_dir, STYLESHEET_GLOBAL))
1760         files_to_copy.append(os.path.join(xsl_dir, STYLESHEET_BOARD))
1761         files_to_copy.append(os.path.join(xsl_dir, "running.gif"))
1762         for file_path in files_to_copy:
1763             shutil.copy2(file_path, log_dir)
1764
1765         # Instanciate the Gui in order to produce the xml files that contain all
1766         # the boards
1767         gui = Gui(runner.cfg.USER.log_dir,
1768                   today_jobs.ljobs,
1769                   today_jobs.ljobs_not_today,
1770                   runner.cfg.VARS.datehour,
1771                   logger,
1772                   file_boards = options.input_boards)
1773
1774         logger.write(src.printcolors.printcSuccess("OK"), 5)
1775         logger.write("\n\n", 5)
1776         logger.flush()
1777
1778         # Display the list of the xml files
1779         logger.write(src.printcolors.printcInfo(("Here is the list of published"
1780                                                  " files :\n")), 4)
1781         logger.write("%s\n" % gui.xml_global_file.logFile, 4)
1782         for board in gui.d_xml_board_files.keys():
1783             file_path = gui.d_xml_board_files[board].logFile
1784             file_name = os.path.basename(file_path)
1785             logger.write("%s\n" % file_path, 4)
1786             logger.add_link(file_name, "board", 0, board)
1787
1788         logger.write("\n", 4)
1789
1790     today_jobs.gui = gui
1791
1792     interruped = False
1793     try:
1794         # Run all the jobs contained in config_jobs
1795         today_jobs.run_jobs()
1796     except KeyboardInterrupt:
1797         interruped = True
1798         logger.write("\n\n%s\n\n" %
1799                 (src.printcolors.printcWarning(_("Forced interruption"))), 1)
1800     except Exception as e:
1801         msg = _("CRITICAL ERROR: The jobs loop has been interrupted\n")
1802         logger.write("\n\n%s\n" % src.printcolors.printcError(msg) )
1803         logger.write("%s\n" % str(e))
1804         # get stack
1805         __, __, exc_traceback = sys.exc_info()
1806         fp = tempfile.TemporaryFile()
1807         traceback.print_tb(exc_traceback, file=fp)
1808         fp.seek(0)
1809         stack = fp.read()
1810         logger.write("\nTRACEBACK: %s\n" % stack.replace('"',"'"), 1)
1811
1812     finally:
1813         res = 0
1814         if interruped:
1815             res = 1
1816             msg = _("Killing the running jobs and trying"
1817                     " to get the corresponding logs\n")
1818             logger.write(src.printcolors.printcWarning(msg))
1819
1820         # find the potential not finished jobs and kill them
1821         for jb in today_jobs.ljobs:
1822             if not jb.has_finished():
1823                 res = 1
1824                 try:
1825                     jb.kill_remote_process()
1826                 except Exception as e:
1827                     msg = _("Failed to kill job %s: %s\n" % (jb.name, e))
1828                     logger.write(src.printcolors.printcWarning(msg))
1829             if jb.res_job != "0":
1830                 res = 1
1831         if interruped:
1832             if today_jobs.gui:
1833                 today_jobs.gui.last_update(_("Forced interruption"))
1834         else:
1835             if today_jobs.gui:
1836                 today_jobs.gui.last_update()
1837         # Output the results
1838         today_jobs.write_all_results()
1839         # Remove the temporary pyconf file
1840         if os.path.exists(path_pyconf):
1841             os.remove(path_pyconf)
1842         return res