134 lines
5.0 KiB
Python
134 lines
5.0 KiB
Python
# w.c.s. - web application for online forms
|
|
# Copyright (C) 2005-2017 Entr'ouvert
|
|
#
|
|
# This program 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.
|
|
#
|
|
# This program 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 this program; if not, see <http://www.gnu.org/licenses/>.
|
|
|
|
from __future__ import print_function
|
|
|
|
import json
|
|
import os
|
|
|
|
from django.core.management.base import CommandError
|
|
|
|
from quixote import get_publisher
|
|
|
|
from wcs.formdef import FormDef
|
|
from wcs.workflows import Workflow
|
|
from wcs.wf.jump import JumpWorkflowStatusItem, jump_and_perform as wcs_jump_and_perform
|
|
|
|
from . import TenantCommand
|
|
|
|
|
|
class Command(TenantCommand):
|
|
'''Triggers all "jump trigger" for a formdef, given host publisher context
|
|
|
|
source.json file format:
|
|
[
|
|
{
|
|
"data": { "info_for_wf_data": 42, ... },
|
|
"select": { "form_number": 1, ... }
|
|
},
|
|
...
|
|
]
|
|
'''
|
|
|
|
def add_arguments(self, parser):
|
|
parser.add_argument('-d', '--domain', metavar='DOMAIN', required=True)
|
|
parser.add_argument('--trigger', metavar='TRIGGER', required=True)
|
|
parser.add_argument('--workflow-id', metavar='WORKFLOW_ID')
|
|
parser.add_argument('--formdef-id', metavar='FORMDEF_ID')
|
|
parser.add_argument('--all-formdata', action='store_true')
|
|
parser.add_argument('filenames', metavar='FILENAME', nargs='+')
|
|
|
|
def handle(self, filenames, domain, trigger, workflow_id, formdef_id,
|
|
all_formdata, **options):
|
|
|
|
self.init_tenant_publisher(domain)
|
|
|
|
if not all_formdata:
|
|
rows = list(get_rows(filenames))
|
|
else:
|
|
rows = '__all__'
|
|
|
|
if formdef_id:
|
|
formdef = FormDef.get(id=formdef_id, ignore_errors=True)
|
|
if not formdef:
|
|
raise CommandError('formdef-id does not exist')
|
|
select_and_jump_formdata(formdef, trigger, rows)
|
|
else:
|
|
if workflow_id:
|
|
workflow = Workflow.get(id=workflow_id, ignore_errors=True)
|
|
if not workflow:
|
|
raise CommandError('workflow does not exist')
|
|
workflows = [workflow]
|
|
else:
|
|
workflows = Workflow.select()
|
|
for workflow in workflows:
|
|
status_ids = list(get_status_ids_accepting_trigger(workflow, trigger))
|
|
if status_ids:
|
|
for formdef in [f for f in FormDef.select() if f.workflow_id == workflow.id]:
|
|
select_and_jump_formdata(formdef, trigger, rows, status_ids)
|
|
|
|
|
|
def get_rows(args):
|
|
for arg in args:
|
|
for row in json.load(file(arg)):
|
|
yield row
|
|
|
|
def get_status_ids_accepting_trigger(workflow, trigger):
|
|
for status in workflow.possible_status:
|
|
for item in status.items:
|
|
if isinstance(item, JumpWorkflowStatusItem) and \
|
|
item.trigger == trigger:
|
|
yield 'wf-%s' % status.id, item
|
|
break
|
|
|
|
def get_formdata_accepting_trigger(formdef, trigger, status_ids=None):
|
|
if status_ids is None:
|
|
workflow = formdef.get_workflow()
|
|
status_ids = get_status_ids_accepting_trigger(workflow, trigger)
|
|
formdata_ids = []
|
|
|
|
data_class = formdef.data_class()
|
|
for status_id, action_item in status_ids:
|
|
formdata_ids = data_class.get_ids_with_indexed_value('status', status_id)
|
|
for formdata_id in formdata_ids:
|
|
yield data_class.get(id=formdata_id), action_item
|
|
|
|
def match_row(substitution_variables, row):
|
|
select = row['select']
|
|
for key, value in select.items():
|
|
if str(substitution_variables.get(key)) != str(value):
|
|
return False
|
|
return True
|
|
|
|
def jump_and_perform(formdata, action, workflow_data=None):
|
|
get_publisher().substitutions.reset()
|
|
get_publisher().substitutions.feed(get_publisher())
|
|
get_publisher().substitutions.feed(formdata.formdef)
|
|
get_publisher().substitutions.feed(formdata)
|
|
print('formdata %s jumps to status %s' % (formdata, action.status))
|
|
wcs_jump_and_perform(formdata, action, workflow_data=workflow_data)
|
|
|
|
def select_and_jump_formdata(formdef, trigger, rows, status_ids=None):
|
|
for formdata, action_item in get_formdata_accepting_trigger(formdef, trigger, status_ids):
|
|
if rows == '__all__':
|
|
jump_and_perform(formdata, action=action_item)
|
|
else:
|
|
substitution_variables = formdata.get_substitution_variables()
|
|
for row in rows:
|
|
if match_row(substitution_variables, row):
|
|
jump_and_perform(formdata, action_item, row.get('data'))
|
|
break # next formdata
|