Python Bank.load_session方法代码示例

# 需要导入模块: from biomaj.bank import Bank [as 别名]
# 或者: from biomaj.bank.Bank import load_session [as 别名]
 def test_new_session(self):
   '''
   Checks an empty session is created
   '''
   b = Bank('alu')
   b.load_session(UpdateWorkflow.FLOW)
   for key in b.session._session['status'].keys():
     self.assertFalse(b.session.get_status(key))

# 需要导入模块: from biomaj.bank import Bank [as 别名]
# 或者: from biomaj.bank.Bank import load_session [as 别名]
 def test_computed_nofile(self):
   b = Bank('computed2')
   b.load_session(UpdateWorkflow.FLOW)
   b.session.config.set('protocol', 'none')
   b.session.config.set('sub1.files.move', 'flat/test_.*')
   res = b.update(True)
   self.assertTrue(res)
   self.assertTrue(os.path.exists(b.session.get_full_release_directory()+'/sub1/flat/test_100.txt'))

# 需要导入模块: from biomaj.bank import Bank [as 别名]
# 或者: from biomaj.bank.Bank import load_session [as 别名]
 def test_extract_release_from_file_content(self):
   b = Bank('local')
   b.load_session(UpdateWorkflow.FLOW)
   b.session.config.set('release.file', 'test_100\.txt')
   b.session.config.set('release.regexp', 'Release\s*(\d+)')
   w = UpdateWorkflow(b)
   w.wf_release()
   self.assertTrue(b.session.get('release') == '103')

# 需要导入模块: from biomaj.bank import Bank [as 别名]
# 或者: from biomaj.bank.Bank import load_session [as 别名]
 def test_extract_release_from_file_content(self):
     b = Bank("local")
     b.load_session(UpdateWorkflow.FLOW)
     b.session.config.set("release.file", "test_100\.txt")
     b.session.config.set("release.regexp", "Release\s*(\d+)")
     w = UpdateWorkflow(b)
     w.wf_release()
     self.assertTrue(b.session.get("release") == "103")

# 需要导入模块: from biomaj.bank import Bank [as 别名]
# 或者: from biomaj.bank.Bank import load_session [as 别名]
 def test_new_session(self):
     """
 Checks an empty session is created
 """
     b = Bank("alu")
     b.load_session(UpdateWorkflow.FLOW)
     for key in b.session._session["status"].keys():
         self.assertFalse(b.session.get_status(key))

# 需要导入模块: from biomaj.bank import Bank [as 别名]
# 或者: from biomaj.bank.Bank import load_session [as 别名]
 def test_remoterelease_check(self):
     b = Bank('local')
     b.load_session(ReleaseCheckWorkflow.FLOW)
     b.session.config.set('release.file', 'test_(\d+)\.txt')
     b.session.config.set('release.regexp', '')
     workflow = ReleaseCheckWorkflow(b)
     res = workflow.start()
     remoterelease = b.session.get('remoterelease')
     self.assertTrue(remoterelease == '100')

# 需要导入模块: from biomaj.bank import Bank [as 别名]
# 或者: from biomaj.bank.Bank import load_session [as 别名]
 def test_get_release(self):
   '''
   Get release
   '''
   b = Bank('alu')
   b.load_session(UpdateWorkflow.FLOW)
   res = b.update()
   self.assertTrue(b.session.get('update'))
   self.assertTrue(res)
   self.assertTrue(b.session._session['release'] is not None)

# 需要导入模块: from biomaj.bank import Bank [as 别名]
# 或者: from biomaj.bank.Bank import load_session [as 别名]
 def test_get_release(self):
     """
 Get release
 """
     b = Bank("alu")
     b.load_session(UpdateWorkflow.FLOW)
     res = b.update()
     self.assertTrue(b.session.get("update"))
     self.assertTrue(res)
     self.assertTrue(b.session._session["release"] is not None)

# 需要导入模块: from biomaj.bank import Bank [as 别名]
# 或者: from biomaj.bank.Bank import load_session [as 别名]
  def test_session_reload_notover(self):
    '''
    Checks a session is used if present
    '''
    b = Bank('alu')
    for i in range(1,5):
      s = Session('alu', self.config, UpdateWorkflow.FLOW)
      s._session['status'][Workflow.FLOW_INIT] = True
      b.session = s
      b.save_session()

    b = Bank('alu')
    b.load_session(UpdateWorkflow.FLOW)
    self.assertTrue(b.session.get_status(Workflow.FLOW_INIT))

# 需要导入模块: from biomaj.bank import Bank [as 别名]
# 或者: from biomaj.bank.Bank import load_session [as 别名]
    def test_session_reload_notover(self):
        """
    Checks a session is used if present
    """
        b = Bank("alu")
        for i in range(1, 5):
            s = Session("alu", self.config, UpdateWorkflow.FLOW)
            s._session["status"][Workflow.FLOW_INIT] = True
            b.session = s
            b.save_session()

        b = Bank("alu")
        b.load_session(UpdateWorkflow.FLOW)
        self.assertTrue(b.session.get_status(Workflow.FLOW_INIT))

# 需要导入模块: from biomaj.bank import Bank [as 别名]
# 或者: from biomaj.bank.Bank import load_session [as 别名]
  def test_session_reload_over(self):
    """
    Checks a session if is not over
    """
    b = Bank('alu')
    for i in range(1,5):
      s = Session('alu', self.config, UpdateWorkflow.FLOW)
      s._session['status'][Workflow.FLOW_INIT] = True
      s._session['status'][Workflow.FLOW_OVER] = True
      b.session = s
      b.save_session()

    b = Bank('alu')
    b.load_session(UpdateWorkflow.FLOW)
    self.assertFalse(b.session.get_status(Workflow.FLOW_INIT))

# 需要导入模块: from biomaj.bank import Bank [as 别名]
# 或者: from biomaj.bank.Bank import load_session [as 别名]
  def test_fromscratch_update_with_release(self):
      '''
      Try updating twice, at second time, bank should  be updated (force with fromscratch)

      Use case with release defined in release file
      '''
      b = Bank('local')
      b.load_session(UpdateWorkflow.FLOW)
      b.session.config.set('release.file', 'test_(\d+)\.txt')
      b.session.config.set('release.regexp', '')
      w = UpdateWorkflow(b)
      w.wf_release()
      self.assertTrue(b.session.get('release') == '100')
      os.makedirs(b.session.get_full_release_directory())
      w = UpdateWorkflow(b)
      # Reset release
      b.session.set('release', None)
      w.options.fromscratch = True
      w.wf_release()
      self.assertTrue(b.session.get('release') == '100__1')

# 需要导入模块: from biomaj.bank import Bank [as 别名]
# 或者: from biomaj.bank.Bank import load_session [as 别名]
    def test_fromscratch_update_with_release(self):
        """
      Try updating twice, at second time, bank should  be updated (force with fromscratch)

      Use case with release defined in release file
      """
        b = Bank("local")
        b.load_session(UpdateWorkflow.FLOW)
        b.session.config.set("release.file", "test_(\d+)\.txt")
        b.session.config.set("release.regexp", "")
        w = UpdateWorkflow(b)
        w.wf_release()
        self.assertTrue(b.session.get("release") == "100")
        os.makedirs(b.session.get_full_release_directory())
        w = UpdateWorkflow(b)
        # Reset release
        b.session.set("release", None)
        w.options.fromscratch = True
        w.wf_release()
        self.assertTrue(b.session.get("release") == "100__1")

# 需要导入模块: from biomaj.bank import Bank [as 别名]
# 或者: from biomaj.bank.Bank import load_session [as 别名]

#.........这里部分代码省略.........
                print("Maintenance set to Off")
                sys.exit(0)

        if options.owner:
            if not options.bank:
                print("Bank option is missing")
                sys.exit(1)
            bank = Bank(options.bank, no_log=True)
            bank.set_owner(options.owner)
            sys.exit(0)

        if options.visibility:
            if not options.bank:
                print("Bank option is missing")
                sys.exit(1)
            if options.visibility not in ['public', 'private']:
                print("Valid values are public|private")
                sys.exit(1)
            bank = Bank(options.bank, no_log=True)
            bank.set_visibility(options.visibility)
            print("Do not forget to update accordingly the visibility.default parameter in the configuration file")
            sys.exit(0)

        if options.newdir:
            if not options.bank:
                print("Bank option is missing")
                sys.exit(1)
            if not os.path.exists(options.newdir):
                print("Destination directory does not exists")
            bank = Bank(options.bank, options=options, no_log=True)
            if not bank.bank['production']:
                print("Nothing to move, no production directory")
                sys.exit(0)
            bank.load_session(Workflow.FLOW, None)
            w = Workflow(bank)
            res = w.wf_init()
            if not res:
                sys.exit(1)
            for prod in bank.bank['production']:
                session = bank.get_session_from_release(prod['release'])
                bank.load_session(Workflow.FLOW, session)
                prod_path = bank.session.get_full_release_directory()
                if os.path.exists(prod_path):
                    shutil.move(prod_path, options.newdir)
                prod['data_dir'] = options.newdir
            bank.banks.update({'name': options.bank}, {'$set' : { 'production': bank.bank['production'] }})
            print("Bank production directories moved to " + options.newdir)
            print("WARNING: do not forget to update accordingly the data.dir and dir.version properties")
            w.wf_over()
            sys.exit(0)

        if options.newbank:
            if not options.bank:
                print("Bank option is missing")
                sys.exit(1)
            bank = Bank(options.bank, no_log=True)
            conf_dir = BiomajConfig.global_config.get('GENERAL', 'conf.dir')
            bank_prop_file = os.path.join(conf_dir,options.bank+'.properties')
            config_bank = configparser.SafeConfigParser()
            config_bank.read([os.path.join(conf_dir,options.bank+'.properties')])
            config_bank.set('GENERAL', 'db.name', options.newbank)
            newbank_prop_file = open(os.path.join(conf_dir,options.newbank+'.properties'),'w')
            config_bank.write(newbank_prop_file)
            newbank_prop_file.close()
            bank.banks.update({'name': options.bank}, {'$set' : { 'name': options.newbank }})
            os.remove(bank_prop_file)