Exemple #1
0
    def __init__(self, test_run=False):
        self.master_db = SPGMasterDB(
            EnsembleConstructor=ParameterEnsembleThreaded)
        self.test_run = test_run

        self.lock = threading.Lock()
        self.db_locks = {}
        ### :::~ The number of processes that are active in each spg file
        self.active_processes = defaultdict(lambda: 0)
Exemple #2
0
 def __init__(self):
     cmd.Cmd.__init__(self)
     self.prompt = "| spg-queue :::~ "
     self.current_queue = None
     self.master_db = SPGMasterDB()
     # self.possible_keys = ['max_jobs', 'status']
     self.__update_queue_list()
Exemple #3
0
    def __init__(self, test_run = False):
        self.master_db = SPGMasterDB( EnsembleConstructor = ParameterEnsembleThreaded )
        self.test_run = test_run

        self.lock = threading.Lock()
        self.db_locks = {}
        ### :::~ The number of processes that are active in each spg file
        self.active_processes = defaultdict(lambda : 0)
Exemple #4
0
class SPGRunningPool():
    def __init__(self):
        self.master_db = SPGMasterDB( EnsembleConstructor = ParameterEnsembleThreaded )
        self.lock = threading.Lock()
        self.db_locks = {}

    def get_lock(self, i_db):
        if not self.db_locks.has_key( i_db.full_name ):
            self.db_locks[ i_db.full_name  ] = threading.Lock()
        return self.db_locks[ i_db.full_name  ]

    def launch_workers(self):
        target_jobs, = self.master_db.query_master_fetchone('SELECT max_jobs FROM queues WHERE name = "default"')

        self.master_db.update_list_ensemble_dbs()
        if len(self.master_db.active_dbs) == 0:
            utils.inline_msg("MSG", "No active dbs... sleeping ")
            return

        current_count = self.active_threads()
        to_launch = target_jobs - current_count
        if to_launch >= 0:
             utils.newline_msg("STATUS", "[n_jobs=%d] run=%d ::: new=%d" % (target_jobs,current_count,to_launch ) )
        else:
             utils.newline_msg("STATUS", "[n_jobs=%d] run=%d :!: exceed" % (target_jobs,current_count))


        for i_t in range(to_launch):
            self.lock.acquire()
            pick = self.master_db.pick_ensemble()
            status = pick.get_updated_status()
            if status['process_not_run'] == 0:
                print "+D+ ----- %s " % (pick.full_name)
                self.master_db.query_master_db('UPDATE dbs SET status= ? WHERE full_name = ?', "D", pick.full_name)
                return

            self.lock.release()

            nt = SPGRunningAtom(pick, self.lock)
            # nt = SPGRunningAtom(pick, lock=self.get_lock( pick ) )

            nt.start()

    def active_threads(self):
        return threading.active_count() - 1











#rp = SPGRunningPool(50, 2)
#rp.run()
Exemple #5
0
 def __init__(self, EnsembleConstructor = ParameterEnsemble):
     BaseSPGCommandLine.__init__(self, EnsembleConstructor)
     self.prompt = "| spg-db :::~ "
     # self.current_param_db = None
     self.master_db =  SPGMasterDB(EnsembleConstructor = EnsembleConstructor)
Exemple #6
0
class SPGDBCommandLine(BaseSPGCommandLine):
    """A command handler which interfaces a master DB"""
 
    def __init__(self, EnsembleConstructor = ParameterEnsemble):
        BaseSPGCommandLine.__init__(self, EnsembleConstructor)
        self.prompt = "| spg-db :::~ "
        # self.current_param_db = None
        self.master_db =  SPGMasterDB(EnsembleConstructor = EnsembleConstructor)


    def filter_db_list(self, ls = None, filter = None):
        if ls == None:
            ls = self.master_db.result_dbs.keys()
        if  re.match("^\d+?$", filter): #:::~ Is filter an integer???
            id = int(filter)
            rdb = self.master_db.result_dbs
            filtered = [ x for x in ls if rdb.has_key(x) and rdb[x] is not None and rdb[x]['id'] == id  ]
            return filtered

        if filter:
                ret = fnmatch.filter(ls, filter)
        else:
                ret = ls
        return sorted( ret )

    # def get_db(self, c):
    #     """it returns the db name (or None) of a database identified either from its id or """
    #     db_name = c.strip()
    #     if db_name.isdigit():
    #         id = int( db_name )
    #         rdb = self.master_db.result_dbs
    #         filtered = [x for x in rdb if rdb[x] is not None and rdb[x]['id'] == id]
    #         if filtered:
    #             db_name = filtered[0]
    #         else:
    #             utils.newline_msg("ERR", "database with id '%s' doesn't exist." % c)
    #             return None
    #
    #     full_name, path, base_name, extension = utils.translate_name(db_name)
    #     db_name = "%s/%s.spgql" % (path, base_name)
    #     sim_name = "%s/%s.spg" % (path, base_name)
    #     if not os.path.exists(db_name) and not os.path.exists(sim_name):
    #         utils.newline_msg("ERR", "database with name '%s' doesn't exist." % c)
    #         return None
    #     return self.EnsembleConstructor(db_name, init_db=True)
    #     #
    #     #     # if self.master_db.result_dbs.has_key(db_name):
    #     #     #     return self.master_db.result_dbs[db_name]
    #     #     # else:
    #     #     #     utils.newline_msg("WRN", "database '%s' is not registered, loading it anyhow" %  db_name )
    #     # return None
    #

    def get_flags_and_db(self, c,init_db = True):

        flags, args =  self.parse_command_line(c)
        # if len(args)== 1:
        #     utils.newline_msg("ERR", "a single file was expected or could not parse flags")
        #     return flags, args, None

        db_name = args[-1]
        args = args[:-1]


        if db_name.isdigit():
            id = int(db_name)
            rdb = self.master_db.result_dbs
            filtered = [x for x in rdb if rdb[x]['id'] == id]
            if filtered:
                db_name = filtered[0]
            else:
                utils.newline_msg("ERR", "database with id '%s' doesn't exist." % db_name)
                return flags, args.append(db_name), None
        else:
                full_name, path, base_name, extension = utils.translate_name(db_name)
                # print "do_init::: ",self.translate_name(i_arg)
                db_name = "%s/%s.spgql" % (path, base_name)
                # sim_name = "%s/%s.spg" % (path, base_name)
                if not os.path.exists(db_name):
                    utils.newline_msg("ERR", "database with name '%s' doesn't exist." % utils.shorten_name(db_name))
                    return flags, args.append(db_name), None

        return flags, args, self.EnsembleConstructor(db_name, init_db)


    def do_ls(self, c):
        """ls REGEXP|DB_ID
        lists the databases already registered in the master database and the possible ones found in the current directory"""

        ls_res_db = self.filter_db_list( filter = c )

        if ls_res_db: 
            print " --- registered dbs" 
            for i in sorted( ls_res_db  ):
                # :::~FIXME workaround for non-existing dbs
                curr_db = self.master_db.result_dbs[i]
                short_name = utils.shorten_name(i)

                try:
                    print "%5d: %s (%5.5f)"%(curr_db['id'], short_name  , curr_db['weight'] )
                except:
                    print "%5d: %s "%(curr_db['id'],  short_name )

        BaseSPGCommandLine.do_ls(self, c )

    # def do_load(self,c):
    #     """load DB_NAME|DB_ID
    #     loads one of the registered databases from the master"""
    #     c = c.split()
    #     if len(c) >1:
    #         utils.newline_msg("ERR", "only one db can be loaded at a time", 2)
    #         return
    #     ret = self.get_db_from_cmdline(c[0])
    #     if ret:
    #         self.current_param_db = ret
    #         print " --- loaded db '%s'"% utils.shorten_name(self.current_param_db.full_name)
    #     # else:
    #     #     utils.newline_msg("ERR", "db does not exist", 2)

    def do_init(self, c):
        """init [--flag ...] PARAMETERS_NAME|DB_NAME [VAR1=VALUE1[:VAR2=VALUE2]]
        Generates a new database out of a simulation.dat
        FLAGS::: --purge:         deletes the spgql database, if it already exists
                 --repeat=REPEAT  repeats the parameter generation REPEAT times
        """

        flags, db_arg = self.parse_command_line(c)
        if len(db_arg) != 1:
            utils.newline_msg("WRN", "init must be called with a database", 2)
            return
        db_arg = db_arg[0]
        # i_arg = c[0]

        full_name, path, base_name, extension = utils.translate_name(db_arg)
        # print "do_init::: ",self.translate_name(i_arg)
        full_db_name = "%s/%s.spgql" % (path, base_name)
        sim_name = "%s/%s.spg" % (path, base_name)
        if os.path.exists(full_db_name) and "purge" not in flags:
            utils.newline_msg("ERR", "database '%s' already exists. Cannot init it twice" % utils.shorten_name(full_db_name), 2)
            return
        if not os.path.exists(sim_name):
            utils.newline_msg("ERR", "configuration '%s' doesn't exist. Cannot init it" % utils.shorten_name(sim_name), 2)
            return

        if "purge" in flags:
            try:
                self.do_deregister(db_arg)
                os.remove(full_db_name)
            except:
                utils.newline_msg("WRN", "database '%s' could not be removed... skipping" % full_db_name)

        if 'repeat' in flags:
            repeat = int(flags['repeat'])
        else:
            repeat = 1

        parser = MultIteratorDBBuilder(db_name=full_db_name)
        parser.init_db()
        parser.fill_status(repeat=repeat)

        current_param_db = ParameterEnsemble(full_db_name, init_db=True)
        current_param_db.repeat = repeat

        # if len(c) > 1: self.do_set(":".join(c[1:]))
        self.master_db.write_ensemble_to_master(current_param_db)

        self.master_db.update_list_ensemble_dbs()

        print " **-- init       - %d: '%s'   " % (current_param_db.id, utils.shorten_name(current_param_db.full_name))

    def complete_init(self, text, line, begidx, endidx):

        completions = fnmatch.filter(os.listdir("."), ".spgql")
        completions.extend(fnmatch.filter(os.listdir("."), "*.spg"))
        if text:
            completions = [f
                           for f in completions
                           if f.startswith(text)
                           ]
        return completions

    def do_register(self, c):
        """registers a given results database into the master database"""
        flags, cmds, ensemble = self.get_flags_and_db(c)
        if ensemble  is None:
            # utils.newline_msg("ERR", "no database supplied ... skipping")
            return

        if self.master_db.result_dbs.has_key(ensemble.full_name):
            utils.newline_msg("WRN", "skipping... database '%s' is already registered" % utils.shorten_name(ensemble.full_name), 2)
            return

#        current_param_db = ParameterEnsemble(db_name, init_db=True)

        self.master_db.write_ensemble_to_master(ensemble )
        self.master_db.update_list_ensemble_dbs()
        print " *--- registered - %d: '%s'   " % (ensemble.id, utils.shorten_name(ensemble.full_name) )

    def complete_register(self, text, line, begidx, endidx):

        return self.complete_init(text, line, begidx, endidx)

    def do_clean(self, c):
        """clean [-flag ...] PARAMETERS_NAME|DB_NAME [VAR1=VALUE1[:VAR2=VALUE2]]
           if not arguments are given  sets all the rows in run_status with status R, E to N
           FLAGS::: --all: sets all the rows in run_status to N  """
        #:::~ OK, as of 13.10.11

        flags, cmds, ensemble = self.get_flags_and_db(c)
        # ensemble = self.get_db(db_name)
        if ensemble is None:
            # utils.newline_msg("ERR", "database not found... aborting")
            return


        if "all" in flags:
            ensemble.execute_query('UPDATE run_status SET status = "N"  ')
        else:
            ensemble.execute_query('UPDATE run_status SET status = "N" WHERE status ="R" OR status ="E" ')

    def complete_clean(self, text, line, begidx, endidx):

        return self.complete_init(text, line, begidx, endidx)


    def do_deregister(self, c):
        """remove current_db|FILENAME|_ID_
           deregisters a simulation file simulations. Does not remove them from disk except --purge is used
           FLAGS::: --purge:         deletes the spgql database, if it already exists"""

        flags, cmds, ensemble = self.get_flags_and_db(c,init_db=False)
        if ensemble is None:
            # utils.newline_msg("ERR", "no database supplied nor currently set... skipping")
            return

        # ensemble = self.get_db_from_cmdline(db_name)

        # if not self.current_param_db is None and self.current_param_db.full_name == db_name:
        #     self.current_param_db = None

        if "purge" in flags and os.path.exists(ensemble.full_name):
            os.remove(ensemble.full_name)

        self.master_db.query_master_db("DELETE FROM dbs WHERE full_name = ?", ensemble.full_name)
        if self.master_db.result_dbs.has_key(ensemble.full_name):
            del self.master_db.result_dbs[ensemble.full_name]
        # :::~ FIXME
        self.master_db.synchronise_master_db()

    def complete_deregister(self, text, line, begidx, endidx):

        return self.complete_init(text, line, begidx, endidx)

    def do_set_weight(self, c):
        flags, args, ensemble = self.get_flags_and_db(c)
     #   print flags, args, ensemble

        if ensemble == None:
            return

        try:
            new_weight = float(args[0])
        except:
            utils.newline_msg("ERR", "cannot parse weight")
            return
       # print "UPDATE dbs SET weight=%f WHERE full_name = '%s' " %  ( new_weight, ensemble.full_name )
        try:
            self.master_db.query_master_db("UPDATE dbs SET weight=%f WHERE full_name = '%s' " %  ( new_weight, ensemble.full_name ) )
        except:
            utils.newline_msg("ERR", "cannot parse command")
            return
        self.master_db.update_list_ensemble_dbs()



    # def do_set(self, c):
    #     """set  VAR1=VALUE1 VAR2=VALUE2
    #     sets some values in the currently loaded database
    #     FLAGS::: --help, the possible keys are printed """
    #
    #     # print c
    #     flags, c = self.parse_command_line(c)
    #
    #     if "help" in flags:
    #         print utils.newline_msg("HELP", " possible_keys = %s" % self.possible_keys)
    #         return
    #
    #     if not self.current_param_db:
    #         utils.newline_msg("WRN", "not database loaded... skipping")
    #         return
    #
    #     for iarg in c:
    #         ret = utils.parse_to_dict(iarg, allowed_keys=self.possible_keys)
    #         if not ret:
    #             utils.newline_msg("ERR", "'%s' not understood" % iarg)
    #             return
    #
    #         # if k == "repeat": continue # repeat is not in the master db (should it be added)
    #         for k in ret:
    #             self.current_param_db.__dict__[k] = ret[k]
    #             self.master_db.query_master_db('UPDATE dbs SET %s= ? WHERE id = ?' % k, ret[k],
    #                                            self.current_param_db.id)

    def __set_status(self, c, st):
        # if not c:
        #     ls_res_db = [ self.current_param_db.full_name ]
        # else:
        #     ls_res_db = self.filter_db_list( filter = c )
        # if not ls_res_db: return
        #
        # for i in ls_res_db:

        flags, cmds, ensemble = self.get_flags_and_db(c)

        # ensemble = self.get_db(db_name)
        if ensemble is None:
            # utils.newline_msg("ERR", "database not found... aborting")
            return

        ensemble.status = st

        print " +---  '%s' [status : %s ]" % (utils.shorten_name(ensemble.full_name), st)
        self.master_db.query_master_db('UPDATE dbs SET status= ? WHERE full_name = ?', st, ensemble.full_name)

    #
    # def do_stop(self, c):
    #     """stops the currently loaded registered database"""
    #     self.__set_status(c, 'S')
    #
    def do_start(self, c):
        """starts the currently loaded registered database"""
        self.__set_status(c, 'R')

    def do_pause(self, c):
        """pauses the currently loaded registered database"""
        self.__set_status(c, 'P')

    def do_set_jobs(self, c):
        """sets the maximum number of jobs running concurrently
           usage: N_JOBS"""
        c = c.split()
        if len(c) == 1:
            max_jobs = int(c[0])
            self.master_db.query_master_db('UPDATE queues SET max_jobs= ? WHERE name = "default"', max_jobs)

    def do_get_jobs(self, c):
        """returns the number of jobs that would concurrently run in a multi-threaded run"""
        nj, = self.master_db.query_master_fetchone('SELECT max_jobs FROM queues WHERE name = "default"')
        print " +--- no_jobs = %d " % nj

    def do_info(self, c):
        """info REGEXP
           prints the information of the results databases, filtered by a regular expression, or its id """

        flags, cmds, ensemble = self.get_flags_and_db(c)
        # ensemble = self.get_db(db_name)
        if ensemble is None:
            # utils.newline_msg("ERR", "database not found... aborting")
            return

        db_status = ensemble.get_updated_status()

        param_db_id = self.master_db.query_master_fetchone("SELECT id, status, weight FROM dbs WHERE full_name = ?", ensemble.full_name)
        if param_db_id is None:
            param_db_id = "X"
        else:
            [param_db_id, status, weight] = param_db_id

        print " ---%5s: %s" % (param_db_id, utils.shorten_name(ensemble.full_name))
        frac_done = float(db_status['process_done']) / float(db_status['value_set'])

        n_repet = db_status['value_set_with_rep'] / db_status['value_set']

        print "   -+ status = %s /  weight: %5.5f " % (status, weight)
        print "   -+ total  = %d*%d / done: %d (%.5f) - running: %d - error: %d " % (
            db_status['value_set'], n_repet, db_status['process_done'], frac_done,
            db_status['process_running'], db_status['process_error'])
        try:
            print "   -+ time   = %f / mean: %f - min: %f - max: %f" % (
            db_status['total_run_time'], db_status['avg_run_time'], db_status['min_run_time'],
            db_status['max_run_time'])
        except:
            pass
Exemple #7
0
 def __init__(self, EnsembleConstructor=ParameterEnsemble):
     BaseSPGCommandLine.__init__(self, EnsembleConstructor)
     self.prompt = "| spg-db :::~ "
     # self.current_param_db = None
     self.master_db = SPGMasterDB(EnsembleConstructor=EnsembleConstructor)
Exemple #8
0
class SPGDBCommandLine(BaseSPGCommandLine):
    """A command handler which interfaces a master DB"""
    def __init__(self, EnsembleConstructor=ParameterEnsemble):
        BaseSPGCommandLine.__init__(self, EnsembleConstructor)
        self.prompt = "| spg-db :::~ "
        # self.current_param_db = None
        self.master_db = SPGMasterDB(EnsembleConstructor=EnsembleConstructor)

    def filter_db_list(self, ls=None, filter=None):
        if ls == None:
            ls = list(self.master_db.result_dbs.keys())
        if re.match("^\d+?$", filter):  #:::~ Is filter an integer???
            id = int(filter)
            rdb = self.master_db.result_dbs
            filtered = [
                x for x in ls
                if x in rdb and rdb[x] is not None and rdb[x]['id'] == id
            ]
            return filtered

        if filter:
            ret = fnmatch.filter(ls, filter)
        else:
            ret = ls
        return sorted(ret)

    # def get_db(self, c):
    #     """it returns the db name (or None) of a database identified either from its id or """
    #     db_name = c.strip()
    #     if db_name.isdigit():
    #         id = int( db_name )
    #         rdb = self.master_db.result_dbs
    #         filtered = [x for x in rdb if rdb[x] is not None and rdb[x]['id'] == id]
    #         if filtered:
    #             db_name = filtered[0]
    #         else:
    #             utils.newline_msg("ERR", "database with id '%s' doesn't exist." % c)
    #             return None
    #
    #     full_name, path, base_name, extension = utils.translate_name(db_name)
    #     db_name = "%s/%s.spgql" % (path, base_name)
    #     sim_name = "%s/%s.spg" % (path, base_name)
    #     if not os.path.exists(db_name) and not os.path.exists(sim_name):
    #         utils.newline_msg("ERR", "database with name '%s' doesn't exist." % c)
    #         return None
    #     return self.EnsembleConstructor(db_name, init_db=True)
    #     #
    #     #     # if self.master_db.result_dbs.has_key(db_name):
    #     #     #     return self.master_db.result_dbs[db_name]
    #     #     # else:
    #     #     #     utils.newline_msg("WRN", "database '%s' is not registered, loading it anyhow" %  db_name )
    #     # return None
    #

    def get_flags_and_db(self, c, init_db=True):

        flags, args = self.parse_command_line(c)
        # if len(args)== 1:
        #     utils.newline_msg("ERR", "a single file was expected or could not parse flags")
        #     return flags, args, None

        db_name = args[-1]
        args = args[:-1]

        if db_name.isdigit():
            id = int(db_name)
            rdb = self.master_db.result_dbs
            filtered = [x for x in rdb if rdb[x]['id'] == id]
            if filtered:
                db_name = filtered[0]
            else:
                utils.newline_msg(
                    "ERR", "database with id '%s' doesn't exist." % db_name)
                return flags, args.append(db_name), None
        else:
            full_name, path, base_name, extension = utils.translate_name(
                db_name)
            # print "do_init::: ",self.translate_name(i_arg)
            db_name = "%s/%s.spgql" % (path, base_name)
            # sim_name = "%s/%s.spg" % (path, base_name)
            if not os.path.exists(db_name):
                utils.newline_msg(
                    "ERR", "database with name '%s' doesn't exist." %
                    utils.shorten_name(db_name))
                return flags, args.append(db_name), None

        return flags, args, self.EnsembleConstructor(db_name, init_db)

    def do_ls(self, c):
        """ls REGEXP|DB_ID
        lists the databases already registered in the master database and the possible ones found in the current directory"""

        ls_res_db = self.filter_db_list(filter=c)

        if ls_res_db:
            print(" --- registered dbs")
            for i in sorted(ls_res_db):
                # :::~FIXME workaround for non-existing dbs
                curr_db = self.master_db.result_dbs[i]
                short_name = utils.shorten_name(i)

                try:
                    print("%5d: %s (%5.5f)" %
                          (curr_db['id'], short_name, curr_db['weight']))
                except:
                    print("%5d: %s " % (curr_db['id'], short_name))

        BaseSPGCommandLine.do_ls(self, c)

    # def do_load(self,c):
    #     """load DB_NAME|DB_ID
    #     loads one of the registered databases from the master"""
    #     c = c.split()
    #     if len(c) >1:
    #         utils.newline_msg("ERR", "only one db can be loaded at a time", 2)
    #         return
    #     ret = self.get_db_from_cmdline(c[0])
    #     if ret:
    #         self.current_param_db = ret
    #         print " --- loaded db '%s'"% utils.shorten_name(self.current_param_db.full_name)
    #     # else:
    #     #     utils.newline_msg("ERR", "db does not exist", 2)

    def do_init(self, c):
        """init [--flag ...] PARAMETERS_NAME|DB_NAME [VAR1=VALUE1[:VAR2=VALUE2]]
        Generates a new database out of a simulation.dat
        FLAGS::: --purge:         deletes the spgql database, if it already exists
                 --repeat=REPEAT  repeats the parameter generation REPEAT times
        """

        flags, db_arg = self.parse_command_line(c)
        if len(db_arg) != 1:
            utils.newline_msg("WRN", "init must be called with a database", 2)
            return
        db_arg = db_arg[0]
        # i_arg = c[0]

        full_name, path, base_name, extension = utils.translate_name(db_arg)
        # print "do_init::: ",self.translate_name(i_arg)
        full_db_name = "%s/%s.spgql" % (path, base_name)
        sim_name = "%s/%s.spg" % (path, base_name)
        if os.path.exists(full_db_name) and "purge" not in flags:
            utils.newline_msg(
                "ERR", "database '%s' already exists. Cannot init it twice" %
                utils.shorten_name(full_db_name), 2)
            return
        if not os.path.exists(sim_name):
            utils.newline_msg(
                "ERR", "configuration '%s' doesn't exist. Cannot init it" %
                utils.shorten_name(sim_name), 2)
            return

        if "purge" in flags:
            try:
                self.do_deregister(db_arg)
                os.remove(full_db_name)
            except:
                utils.newline_msg(
                    "WRN", "database '%s' could not be removed... skipping" %
                    full_db_name)

        if 'repeat' in flags:
            repeat = int(flags['repeat'])
        else:
            repeat = 1

        parser = MultIteratorDBBuilder(db_name=full_db_name)
        parser.init_db()
        parser.fill_status(repeat=repeat)

        current_param_db = ParameterEnsemble(full_db_name, init_db=True)
        current_param_db.repeat = repeat

        # if len(c) > 1: self.do_set(":".join(c[1:]))
        self.master_db.write_ensemble_to_master(current_param_db)

        self.master_db.update_list_ensemble_dbs()

        print(" **-- init       - %d: '%s'   " %
              (current_param_db.id,
               utils.shorten_name(current_param_db.full_name)))

    def complete_init(self, text, line, begidx, endidx):

        completions = fnmatch.filter(os.listdir("."), ".spgql")
        completions.extend(fnmatch.filter(os.listdir("."), "*.spg"))
        if text:
            completions = [f for f in completions if f.startswith(text)]
        return completions

    def do_register(self, c):
        """registers a given results database into the master database"""
        flags, cmds, ensemble = self.get_flags_and_db(c)
        if ensemble is None:
            # utils.newline_msg("ERR", "no database supplied ... skipping")
            return

        if ensemble.full_name in self.master_db.result_dbs:
            utils.newline_msg(
                "WRN", "skipping... database '%s' is already registered" %
                utils.shorten_name(ensemble.full_name), 2)
            return

#        current_param_db = ParameterEnsemble(db_name, init_db=True)

        self.master_db.write_ensemble_to_master(ensemble)
        self.master_db.update_list_ensemble_dbs()
        print(" *--- registered - %d: '%s'   " %
              (ensemble.id, utils.shorten_name(ensemble.full_name)))

    def complete_register(self, text, line, begidx, endidx):

        return self.complete_init(text, line, begidx, endidx)

    def do_clean(self, c):
        """clean [-flag ...] PARAMETERS_NAME|DB_NAME [VAR1=VALUE1[:VAR2=VALUE2]]
           if not arguments are given  sets all the rows in run_status with status R, E to N
           FLAGS::: --all: sets all the rows in run_status to N  """
        #:::~ OK, as of 13.10.11

        flags, cmds, ensemble = self.get_flags_and_db(c)
        # ensemble = self.get_db(db_name)
        if ensemble is None:
            # utils.newline_msg("ERR", "database not found... aborting")
            return

        if "all" in flags:
            ensemble.execute_query('UPDATE run_status SET status = "N"  ')
        else:
            ensemble.execute_query(
                'UPDATE run_status SET status = "N" WHERE status ="R" OR status ="E" '
            )

    def complete_clean(self, text, line, begidx, endidx):

        return self.complete_init(text, line, begidx, endidx)

    def do_deregister(self, c):
        """remove current_db|FILENAME|_ID_
           deregisters a simulation file simulations. Does not remove them from disk except --purge is used
           FLAGS::: --purge:         deletes the spgql database, if it already exists"""

        flags, cmds, ensemble = self.get_flags_and_db(c, init_db=False)
        if ensemble is None:
            # utils.newline_msg("ERR", "no database supplied nor currently set... skipping")
            return

        # ensemble = self.get_db_from_cmdline(db_name)

        # if not self.current_param_db is None and self.current_param_db.full_name == db_name:
        #     self.current_param_db = None

        if "purge" in flags and os.path.exists(ensemble.full_name):
            os.remove(ensemble.full_name)

        self.master_db.query_master_db("DELETE FROM dbs WHERE full_name = ?",
                                       ensemble.full_name)
        if ensemble.full_name in self.master_db.result_dbs:
            del self.master_db.result_dbs[ensemble.full_name]
        # :::~ FIXME
        self.master_db.synchronise_master_db()

    def complete_deregister(self, text, line, begidx, endidx):

        return self.complete_init(text, line, begidx, endidx)

    def do_set_weight(self, c):
        flags, args, ensemble = self.get_flags_and_db(c)
        #   print flags, args, ensemble

        if ensemble == None:
            return

        try:
            new_weight = float(args[0])
        except:
            utils.newline_msg("ERR", "cannot parse weight")
            return
    # print "UPDATE dbs SET weight=%f WHERE full_name = '%s' " %  ( new_weight, ensemble.full_name )
        try:
            self.master_db.query_master_db(
                "UPDATE dbs SET weight=%f WHERE full_name = '%s' " %
                (new_weight, ensemble.full_name))
        except:
            utils.newline_msg("ERR", "cannot parse command")
            return
        self.master_db.update_list_ensemble_dbs()

    # def do_set(self, c):
    #     """set  VAR1=VALUE1 VAR2=VALUE2
    #     sets some values in the currently loaded database
    #     FLAGS::: --help, the possible keys are printed """
    #
    #     # print c
    #     flags, c = self.parse_command_line(c)
    #
    #     if "help" in flags:
    #         print utils.newline_msg("HELP", " possible_keys = %s" % self.possible_keys)
    #         return
    #
    #     if not self.current_param_db:
    #         utils.newline_msg("WRN", "not database loaded... skipping")
    #         return
    #
    #     for iarg in c:
    #         ret = utils.parse_to_dict(iarg, allowed_keys=self.possible_keys)
    #         if not ret:
    #             utils.newline_msg("ERR", "'%s' not understood" % iarg)
    #             return
    #
    #         # if k == "repeat": continue # repeat is not in the master db (should it be added)
    #         for k in ret:
    #             self.current_param_db.__dict__[k] = ret[k]
    #             self.master_db.query_master_db('UPDATE dbs SET %s= ? WHERE id = ?' % k, ret[k],
    #                                            self.current_param_db.id)

    def __set_status(self, c, st):
        # if not c:
        #     ls_res_db = [ self.current_param_db.full_name ]
        # else:
        #     ls_res_db = self.filter_db_list( filter = c )
        # if not ls_res_db: return
        #
        # for i in ls_res_db:

        flags, cmds, ensemble = self.get_flags_and_db(c)

        # ensemble = self.get_db(db_name)
        if ensemble is None:
            # utils.newline_msg("ERR", "database not found... aborting")
            return

        ensemble.status = st

        print(" +---  '%s' [status : %s ]" %
              (utils.shorten_name(ensemble.full_name), st))
        self.master_db.query_master_db(
            'UPDATE dbs SET status= ? WHERE full_name = ?', st,
            ensemble.full_name)

    #
    # def do_stop(self, c):
    #     """stops the currently loaded registered database"""
    #     self.__set_status(c, 'S')
    #
    def do_start(self, c):
        """starts the currently loaded registered database"""
        self.__set_status(c, 'R')

    def do_pause(self, c):
        """pauses the currently loaded registered database"""
        self.__set_status(c, 'P')

    def do_set_jobs(self, c):
        """sets the maximum number of jobs running concurrently
           usage: N_JOBS"""
        c = c.split()
        if len(c) == 1:
            max_jobs = int(c[0])
            self.master_db.query_master_db(
                'UPDATE queues SET max_jobs= ? WHERE name = "default"',
                max_jobs)

    def do_get_jobs(self, c):
        """returns the number of jobs that would concurrently run in a multi-threaded run"""
        nj, = self.master_db.query_master_fetchone(
            'SELECT max_jobs FROM queues WHERE name = "default"')
        print(" +--- no_jobs = %d " % nj)

    def do_info(self, c):
        """info REGEXP
           prints the information of the results databases, filtered by a regular expression, or its id """

        flags, cmds, ensemble = self.get_flags_and_db(c)
        # ensemble = self.get_db(db_name)
        if ensemble is None:
            # utils.newline_msg("ERR", "database not found... aborting")
            return

        db_status = ensemble.get_updated_status()

        param_db_id = self.master_db.query_master_fetchone(
            "SELECT id, status, weight FROM dbs WHERE full_name = ?",
            ensemble.full_name)
        if param_db_id is None:
            param_db_id = "X"
        else:
            [param_db_id, status, weight] = param_db_id

        print(" ---%5s: %s" %
              (param_db_id, utils.shorten_name(ensemble.full_name)))
        frac_done = float(db_status['process_done']) / float(
            db_status['value_set'])

        n_repet = db_status['value_set_with_rep'] / db_status['value_set']

        print("   -+ status = %s /  weight: %5.5f " % (status, weight))
        print(
            "   -+ total  = %d*%d / done: %d (%.5f) - running: %d - error: %d "
            % (db_status['value_set'], n_repet, db_status['process_done'],
               frac_done, db_status['process_running'],
               db_status['process_error']))
        try:
            print("   -+ time   = %f / mean: %f - min: %f - max: %f" %
                  (db_status['total_run_time'], db_status['avg_run_time'],
                   db_status['min_run_time'], db_status['max_run_time']))
        except:
            pass
Exemple #9
0
class QueueCommandParser(cmd.Cmd):
    """command processor for the queues."""

    def __init__(self):
        cmd.Cmd.__init__(self)
        self.prompt = "| spg-queue :::~ "
        self.current_queue = None
        self.master_db = SPGMasterDB()
        # self.possible_keys = ['max_jobs', 'status']
        self.__update_queue_list()

    def __update_queue_list(self):
        self.queues = self.master_db.query_master_db("SELECT * FROM queues ORDER BY id")

    def do_load(self, c):
        """loads a queue as active one"""
        self.__update_queue_list()
        if len(filter(lambda x: x[1] == c, self.queues)):
            self.current_queue = c
        else:
            newline_msg("ERR", "queue '%s' not found" % c)
        print "loaded '%s'" % self.current_queue

    def completedefault(self, text, line, begidx, endidx):
        self.__update_queue_list()
        completions = [name for (id, name, max_jobs, status) in self.queues]

        if text:
            completions = [f for f in completions if f.startswith(text)]
        return completions

    def do_ls(self, c):
        """lists the queues in the master database """
        self.__update_queue_list()
        for (id, name, max_jobs, status) in self.queues:
            print "   %16s  - J: %2d - S: '%s'" % (name, max_jobs, status)

    ###  CREATE TABLE IF NOT EXISTS queues
    ###           ( id INTEGER PRIMARY KEY, name CHAR(64), max_jobs INTEGER,
    ###           status CHAR(1) )

    def do_add(self, c):
        """adds queue queue"""
        self.__update_queue_list()

        if len(filter(lambda x: x[1] == c, self.queues)):
            newline_msg("ERR", "queue '%s' already exists" % queue, 2)
            return
        self.master_db.query_master_db("INSERT INTO queues (name, max_jobs, status) VALUES (?,?,?)", c, 1, "S")
        os.makedirs("%s/queue/%s" % (VAR_PATH, c))
        self.current_queue = c

    def do_set_max_jobs(self, c):
        """sets the maximum number of jobs in the given queue 
           usage: [regexp] N_JOBS"""
        c = c.split()
        if len(c) == 1 and self.current_queue:
            max_jobs = int(c[0])
            self.master_db.query_master_db("UPDATE queues SET max_jobs= ? WHERE name = ?", max_jobs, self.current_queue)
        elif len(c) == 2:
            re = c[0]
            max_jobs = int(c[1])
            #            print re, max_jobs
            lsq = [n for (id, n, mj, s) in self.queues]
            lsq = fnmatch.filter(lsq, re)
            #        print lsq
            for q in lsq:
                # print status, q
                self.master_db.query_master_db("UPDATE queues SET max_jobs= ? WHERE name = ?", max_jobs, q)

    def do_clean_pool(self, cmd):
        """cleans all the hanging pickled input/output files"""

        proc = Popen("rm -f %s/run/*" % VAR_PATH, shell=True, stdin=PIPE, stdout=PIPE, stderr=PIPE)
        proc.wait()
        self.__update_queue_list()

        for (id, name, max_jobs, status) in self.queues:
            proc = Popen("rm -f %s/queue/%s/*" % (VAR_PATH, name), shell=True, stdin=PIPE, stdout=PIPE, stderr=PIPE)
            proc.wait()

    def do_remove(self, c):
        """sets the maximum number of jobs in the given queue 
           usage: [regexp] N_JOBS"""
        c = c.split()
        if len(c) == 0 and self.current_queue:
            self.master_db.query_master_db("DELETE FROM queues WHERE name = ?", self.current_queue)
        elif len(c) == 1:
            re = c[0]
            #            print re, max_jobs
            lsq = [n for (id, n, mj, s) in self.queues]
            lsq = fnmatch.filter(lsq, re)
            #        print lsq
            for q in lsq:
                # print status, q
                self.master_db.query_master_db("DELETE FROM queues WHERE name = ?", q)

    #        ret = utils.parse_to_dict(c, allowed_keys=self.possible_keys)
    #        for k in ret:
    #            self.values[k] = ret[k]
    #            print "%s = %s" %(k,ret[k])
    #        if self.current_queue:
    #            if ret.has_key('max_jobs'):
    #            if ret.has_key('status'):
    #                self.master_db.execute_query( 'UPDATE queues SET status= ? WHERE name = ?', self.current_queue  )
    #                self.master_db.execute( 'UPDATE dbs SET status="%s" WHERE id = %s'% ( self.current_param_db.status, self.current_param_db.id ) )

    #        self.doc_header = "default values: %s"%(self.values )

    def __set_status(self, status, name=None):
        # print name, status
        if not name and self.current_queue:
            #            print  'UPDATE queues SET status= ? WHERE name = ?'
            self.master_db.query_master_db("UPDATE queues SET status= ? WHERE name = ?", status, self.current_queue)
            return
        lsq = [n for (id, n, max_jobs, s) in self.queues]
        if name:
            lsq = fnmatch.filter(lsq, name)
        #        print lsq
        for q in lsq:
            #            print status, q
            self.master_db.query_master_db("UPDATE queues SET status= ? WHERE name = ?", status, q)

    def do_stop(self, c):
        """stops the currently loaded registered database"""
        self.__set_status("S", c)

    def do_start(self, c):
        """starts the currently loaded registered database"""
        self.__set_status("R", c)

    def do_pause(self, c):
        """pauses the currently loaded registered database"""
        self.__set_status("P", c)

    def do_EOF(self, line):
        return True
Exemple #10
0
 def __init__(self):
     self.master_db = SPGMasterDB( EnsembleConstructor = ParameterEnsembleThreaded )
     self.lock = threading.Lock()
     self.db_locks = {}
Exemple #11
0
class SPGRunningPool():
    def __init__(self, test_run=False):
        self.master_db = SPGMasterDB(
            EnsembleConstructor=ParameterEnsembleThreaded)
        self.test_run = test_run

        self.lock = threading.Lock()
        self.db_locks = {}
        ### :::~ The number of processes that are active in each spg file
        self.active_processes = defaultdict(lambda: 0)

    def get_lock(self, i_db):
        if i_db.full_name not in self.db_locks:
            self.db_locks[i_db.full_name] = threading.Lock()
        return self.db_locks[i_db.full_name]

    def launch_workers(self):
        target_jobs, = self.master_db.query_master_fetchone(
            'SELECT max_jobs FROM queues WHERE name = "default"')

        self.master_db.update_list_ensemble_dbs()
        if len(self.master_db.active_dbs) == 0:
            utils.inline_msg("MSG", "No active dbs... sleeping ")
            return

        current_count = self.active_threads()
        #        print "+++++++++++", to_launch
        vec_to_launch = []

        launch = defaultdict(lambda: 0)
        running = {}

        for ae in self.master_db.active_dbs:
            ens = self.master_db.result_dbs[ae]
            running[ens['id']] = self.active_processes[ae]
            qty_to_launch = int(
                m.floor(0.5 + target_jobs * ens['weight'] /
                        self.master_db.normalising) -
                self.active_processes[ae])

            vec_to_launch += qty_to_launch * [ae]
            launch[ens['id']] += qty_to_launch
#       for id in launch:
#           print "+++ (%d) %d + %d = //%d//, "%( id, launch[id], running[id],launch[id]+running[id] )
#       print

        to_launch = len(vec_to_launch)
        if to_launch >= 0:
            utils.newline_msg(
                "STATUS",
                utils.str_color(
                    "@green[n_jobs=%d] run=%d %s ::: new=%d" %
                    (target_jobs, current_count, dict(running), to_launch)))
        else:
            utils.newline_msg(
                "STATUS",
                utils.str_color(
                    "@yellow[n_jobs=%d] run=%d :!: exceeded number" %
                    (target_jobs, current_count)))

#       print to_launch, len( vec_to_launch ), launch

#        for i_t in range(to_launch):
        for ae in vec_to_launch:
            pick = self.master_db.EnsembleConstructor(ae, init_db=True)
            self.lock.acquire()
            #            pick = self.master_db.pick_ensemble()

            pick.test_run = self.test_run

            status = pick.get_updated_status()
            if status['process_not_run'] == 0:
                print("+D+ ----- %s " % (pick.full_name))
                self.master_db.query_master_db(
                    'UPDATE dbs SET status= ? WHERE full_name = ?', "D",
                    pick.full_name)
                return

            self.lock.release()

            nt = SPGRunningAtom(pick, self.lock, self.active_processes)
            #            nt.test_run = self.test_run
            # nt = SPGRunningAtom(pick, lock=self.get_lock( pick ) )

            nt.start()

    def active_threads(self):
        return threading.active_count() - 1


#rp = SPGRunningPool(50, 2)
#rp.run()
Exemple #12
0
class SPGRunningPool():
    def __init__(self, test_run = False):
        self.master_db = SPGMasterDB( EnsembleConstructor = ParameterEnsembleThreaded )
        self.test_run = test_run

        self.lock = threading.Lock()
        self.db_locks = {}
        ### :::~ The number of processes that are active in each spg file
        self.active_processes = defaultdict(lambda : 0)

    def get_lock(self, i_db):
        if not self.db_locks.has_key( i_db.full_name ):
            self.db_locks[ i_db.full_name ] = threading.Lock()
        return self.db_locks[ i_db.full_name ]

    def launch_workers(self):
        target_jobs, = self.master_db.query_master_fetchone('SELECT max_jobs FROM queues WHERE name = "default"')

        self.master_db.update_list_ensemble_dbs()
        if len(self.master_db.active_dbs) == 0:
            utils.inline_msg("MSG", "No active dbs... sleeping ")
            return

        current_count = self.active_threads()
#        print "+++++++++++", to_launch
        vec_to_launch = []

        launch = defaultdict(lambda: 0)
        running = {}

        for ae in self.master_db.active_dbs:
            ens = self.master_db.result_dbs[ae  ]
            running[ ens['id'] ] = self.active_processes[ ae ]
            qty_to_launch = int( m.floor(0.5 + target_jobs*ens['weight']/self.master_db.normalising) - self.active_processes[ ae ] )

            vec_to_launch += qty_to_launch * [ae]
            launch[ ens['id'] ] += qty_to_launch
 #       for id in launch:
 #           print "+++ (%d) %d + %d = //%d//, "%( id, launch[id], running[id],launch[id]+running[id] )
 #       print

        to_launch = len(vec_to_launch)
        if to_launch >= 0:
             utils.newline_msg("STATUS", utils.str_color( "@green[n_jobs=%d] run=%d %s ::: new=%d" % (target_jobs,current_count, dict(running),to_launch) ) )
        else:
             utils.newline_msg("STATUS", utils.str_color( "@yellow[n_jobs=%d] run=%d :!: exceeded number" % (target_jobs,current_count)) )

 #       print to_launch, len( vec_to_launch ), launch

#        for i_t in range(to_launch):
        for ae in vec_to_launch:
            pick = self.master_db.EnsembleConstructor(ae, init_db=True)
            self.lock.acquire()
#            pick = self.master_db.pick_ensemble()

            pick.test_run = self.test_run

            status = pick.get_updated_status()
            if status['process_not_run'] == 0:
                print "+D+ ----- %s " % (pick.full_name)
                self.master_db.query_master_db('UPDATE dbs SET status= ? WHERE full_name = ?', "D", pick.full_name)
                return

            self.lock.release()

            nt = SPGRunningAtom(pick, self.lock, self.active_processes)
#            nt.test_run = self.test_run
            # nt = SPGRunningAtom(pick, lock=self.get_lock( pick ) )

            nt.start()

    def active_threads(self):
        return threading.active_count() - 1











#rp = SPGRunningPool(50, 2)
#rp.run()