Page Menu
Home
c4science
Search
Configure Global Search
Log In
Files
F79228058
client.py
No One
Temporary
Actions
Download File
Edit File
Delete File
View Transforms
Subscribe
Mute Notifications
Award Token
Subscribers
None
File Metadata
Details
File Info
Storage
Attached
Created
Sun, Aug 25, 09:18
Size
5 KB
Mime Type
text/x-python
Expires
Tue, Aug 27, 09:18 (1 d, 23 h)
Engine
blob
Format
Raw Data
Handle
20163400
Attached To
R3600 invenio-infoscience
client.py
View Options
# -*- coding: utf-8 -*-
## This file is part of Invenio.
## Copyright (C) 2012, 2013 CERN.
##
## Invenio is free software; you can redistribute it and/or
## modify it under the terms of the GNU General Public License as
## published by the Free Software Foundation; either version 2 of the
## License, or (at your option) any later version.
##
## Invenio is distributed in the hope that it will be useful, but
## WITHOUT ANY WARRANTY; without even the implied warranty of
## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
## General Public License for more details.
##
## You should have received a copy of the GNU General Public License
## along with Invenio; if not, write to the Free Software Foundation, Inc.,
## 59 Temple Place, Suite 330, Boston, MA 02111-1307, USA.
import
traceback
from
workflow.engine
import
HaltProcessing
from
invenio.bibworkflow_config
import
CFG_OBJECT_VERSION
from
invenio.bibworkflow_config
import
CFG_WORKFLOW_STATUS
def
run_workflow
(
wfe
,
data
,
stop_on_halt
=
False
,
stop_on_error
=
False
,
**
kwargs
):
"""
Main function running the workflow.
"""
initial_run
=
True
while
True
:
try
:
if
initial_run
:
initial_run
=
False
wfe
.
process
(
data
)
# We processed the workflow. We're done.
break
else
:
wfe
.
_unpickled
=
True
wfe
.
restart
(
'current'
,
'current'
)
# We processed the restarted workflow. We're done.
break
except
HaltProcessing
as
e
:
# Processing was halted. Lets save current object and continue.
wfe
.
log_error
(
message
=
"Processing halted!"
,
error_msg
=
str
(
e
))
wfe
.
_objects
[
wfe
.
getCurrObjId
()]
.
save
(
CFG_OBJECT_VERSION
.
HALTED
,
wfe
.
getCurrTaskId
(),
id_workflow
=
wfe
.
uuid
)
wfe
.
save
(
CFG_WORKFLOW_STATUS
.
HALTED
)
wfe
.
setPosition
(
wfe
.
getCurrObjId
()
+
1
,
[
0
,
0
])
if
stop_on_halt
:
break
except
Exception
as
e
:
# Processing generated an exception.
# We print the stacktrace, save the object and continue
wfe
.
log_error
(
message
=
"Processing error!
%r
"
%
(
e
,),
error_msg
=
traceback
.
format_exc
())
# Changing counter should be moved to wfe object
# together with default exception handling
wfe
.
increase_counter_error
()
wfe
.
_objects
[
wfe
.
getCurrObjId
()]
.
save
(
CFG_OBJECT_VERSION
.
HALTED
,
wfe
.
getCurrTaskId
(),
id_workflow
=
wfe
.
uuid
)
wfe
.
save
(
CFG_WORKFLOW_STATUS
.
ERROR
)
wfe
.
setPosition
(
wfe
.
getCurrObjId
()
+
1
,
[
0
,
0
])
if
stop_on_halt
or
stop_on_error
:
raise
e
def
continue_execution
(
wfe
,
data
,
restart_point
=
"restart_task"
,
stop_on_halt
=
False
,
stop_on_error
=
False
,
**
kwargs
):
"""
Continue execution of workflow for given object (wfe) from "restart_point".
restart_point can be one of:
* restart_prev: will restart from the previous task
* continue_next: will continue to the next task
* restart_task: will restart the current task
You can use stop_on_error to raise exception's and stop the processing.
Use stop_on_halt to stop processing the workflow
if HaltProcessing is raised.
"""
wfe
.
log_info
(
"Continue execution from: "
+
str
(
restart_point
))
pos
=
data
[
0
]
.
get_current_task
()
if
restart_point
==
"restart_prev"
:
pos
[
-
1
]
=
pos
[
-
1
]
-
1
wfe
.
setPosition
(
wfe
.
db_obj
.
current_object
,
pos
)
elif
restart_point
==
"continue_next"
:
pos
[
-
1
]
=
pos
[
-
1
]
+
1
wfe
.
setPosition
(
wfe
.
db_obj
.
current_object
,
pos
)
else
:
# restart_task
wfe
.
setPosition
(
wfe
.
db_obj
.
current_object
,
pos
)
wfe
.
_unpickled
=
True
initial_run
=
True
wfe
.
_objects
=
data
while
True
:
try
:
if
initial_run
:
initial_run
=
False
wfe
.
restart
(
'current'
,
'current'
)
# We processed the workflow. We're done.
break
else
:
wfe
.
_unpickled
=
True
wfe
.
restart
(
'current'
,
'current'
)
# We processed the restarted workflow. We're done.
break
except
HaltProcessing
as
e
:
# Processing was halted. Lets save current object and continue.
wfe
.
log_error
(
message
=
"Processing halted!"
,
error_msg
=
str
(
e
))
wfe
.
_objects
[
wfe
.
getCurrObjId
()]
.
save
(
2
,
wfe
.
getCurrTaskId
())
wfe
.
save
(
CFG_WORKFLOW_STATUS
.
HALTED
)
wfe
.
setPosition
(
wfe
.
getCurrObjId
()
+
1
,
[
0
,
0
])
if
stop_on_halt
:
break
except
Exception
as
e
:
# Processing generated an exception. We print the stacktrace,
# save the object and continue
wfe
.
log_error
(
message
=
"Processing error!
%r
"
%
(
e
,),
error_msg
=
traceback
.
format_exc
())
# Changing counter should be moved to wfe object together
# with default exception handling
wfe
.
increase_counter_error
()
wfe
.
_objects
[
wfe
.
getCurrObjId
()]
.
save
(
2
,
wfe
.
getCurrTaskId
())
wfe
.
save
(
CFG_WORKFLOW_STATUS
.
ERROR
)
wfe
.
setPosition
(
wfe
.
getCurrObjId
()
+
1
,
[
0
,
0
])
if
stop_on_halt
or
stop_on_error
:
raise
e
Event Timeline
Log In to Comment