1# This file is part of ctrl_bps. 

2# 

3# Developed for the LSST Data Management System. 

4# This product includes software developed by the LSST Project 

5# (https://www.lsst.org). 

6# See the COPYRIGHT file at the top-level directory of this distribution 

7# for details of code ownership. 

8# 

9# This program is free software: you can redistribute it and/or modify 

10# it under the terms of the GNU General Public License as published by 

11# the Free Software Foundation, either version 3 of the License, or 

12# (at your option) any later version. 

13# 

14# This program is distributed in the hope that it will be useful, 

15# but WITHOUT ANY WARRANTY; without even the implied warranty of 

16# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 

17# GNU General Public License for more details. 

18# 

19# You should have received a copy of the GNU General Public License 

20# along with this program. If not, see <https://www.gnu.org/licenses/>. 

21 

22"""Driver for submitting a prepared WMS-specific workflow 

23""" 

24import logging 

25import time 

26 

27from lsst.utils import doImport 

28from lsst.ctrl.bps.bps_utils import _create_execution_butler 

29 

30_LOG = logging.getLogger(__name__) 

31 

32 

33def submit(config, wms_workflow, wms_service=None): 

34 """Convert generic workflow to a workflow for a particular WMS. 

35 

36 Parameters 

37 ---------- 

38 config : `lsst.ctrl.bps.BpsConfig` 

39 Configuration values to be used by submission. 

40 wms_workflow : `lsst.ctrl.bps.BaseWmsWorkflow` 

41 The workflow to submit. 

42 wms_service : `lsst.ctrl.bps.BaseWmsService`, optional 

43 The workflow management service to which the workflow should be 

44 submitted. 

45 

46 Returns 

47 ------- 

48 wms_workflow : `lsst.ctrl.bps.BaseWmsWorkflow` 

49 WMS-specific workflow. 

50 """ 

51 if wms_service is None: 

52 wms_service_class = doImport(config["wmsServiceClass"]) 

53 wms_service = wms_service_class(config) 

54 

55 _, when_create = config.search(".executionButler.whenCreate") 

56 if when_create.upper() == "SUBMIT": 

57 _LOG.info("Creating execution butler") 

58 stime = time.time() 

59 _, execution_butler_dir = config.search(".bps_defined.executionButlerDir") 

60 _create_execution_butler(config, config["runQgraphFile"], execution_butler_dir, config["submitPath"]) 

61 _LOG.info("Creating execution butler took %.2f seconds", time.time() - stime) 

62 

63 return wms_service.submit(wms_workflow)