184 lines
7.3 KiB
Python
184 lines
7.3 KiB
Python
import os
|
|
import xml.etree.ElementTree as ET
|
|
|
|
from wcs.blocks import BlockDef
|
|
from wcs.carddef import CardDef
|
|
from wcs.categories import Category
|
|
from wcs.data_sources import NamedDataSource
|
|
from wcs.formdef import FormDef
|
|
from wcs.mail_templates import MailTemplate
|
|
from wcs.workflows import Workflow
|
|
from wcs.wscalls import NamedWsCall
|
|
|
|
from ..qommon.ctl import Command, make_option
|
|
|
|
|
|
class Cmd(Command):
|
|
name = 'imio_import_directory'
|
|
|
|
def __init__(self):
|
|
super().__init__(
|
|
[
|
|
make_option('-d', '--domain', action='store', dest='domain'),
|
|
]
|
|
)
|
|
|
|
def execute(self, base_options, sub_options, args):
|
|
from .. import publisher
|
|
|
|
publisher.WcsPublisher.configure(self.config)
|
|
publisher = publisher.WcsPublisher.create_publisher(register_tld_names=False)
|
|
publisher.set_tenant_by_hostname(sub_options.domain)
|
|
publisher.substitutions.feed(publisher)
|
|
self.directory = args[0]
|
|
self.import_categories()
|
|
self.import_datasources()
|
|
self.import_wscalls()
|
|
self.import_mail_templates()
|
|
self.import_workflows()
|
|
self.import_blocks()
|
|
self.import_carddefs()
|
|
self.import_formdefs()
|
|
|
|
def import_categories(self):
|
|
if not os.path.exists(os.path.join(self.directory, 'category')):
|
|
return
|
|
for filename in os.listdir(os.path.join(self.directory, 'category')):
|
|
category = Category.import_from_xml(open(os.path.join(self.directory, 'category', filename)))
|
|
try:
|
|
existing_category = Category.get_by_urlname(category.url_name)
|
|
except KeyError:
|
|
category.store()
|
|
else:
|
|
# replace
|
|
category.id = existing_category.id
|
|
category.store()
|
|
|
|
def import_datasources(self):
|
|
if not os.path.exists(os.path.join(self.directory, 'datasources')):
|
|
return
|
|
for filename in os.listdir(os.path.join(self.directory, 'datasources')):
|
|
datasource = NamedDataSource.import_from_xml(
|
|
open(os.path.join(self.directory, 'datasources', filename))
|
|
)
|
|
try:
|
|
existing_datasource = NamedDataSource.get_by_slug(datasource.slug, ignore_errors=False)
|
|
except KeyError:
|
|
datasource.store(comment='Indus Initial Import')
|
|
else:
|
|
# replace
|
|
datasource.id = existing_datasource.id
|
|
datasource.store(comment='Indus Update')
|
|
|
|
def import_mail_templates(self):
|
|
if not os.path.exists(os.path.join(self.directory, 'mail-templates')):
|
|
return
|
|
for filename in os.listdir(os.path.join(self.directory, 'mail-templates')):
|
|
mail_template = MailTemplate.import_from_xml(
|
|
open(os.path.join(self.directory, 'mail-templates', filename))
|
|
)
|
|
existing_mail_template = MailTemplate.get_by_slug(mail_template.slug)
|
|
if existing_mail_template is None:
|
|
mail_template.store()
|
|
else:
|
|
# replace
|
|
mail_template.id = existing_mail_template.id
|
|
mail_template.store()
|
|
|
|
def import_workflows(self):
|
|
if not os.path.exists(os.path.join(self.directory, 'workflows')):
|
|
return
|
|
for filename in os.listdir(os.path.join(self.directory, 'workflows')):
|
|
workflow = Workflow.import_from_xml(
|
|
open(os.path.join(self.directory, 'workflows', filename)),
|
|
include_id=False,
|
|
check_datasources=True,
|
|
)
|
|
try:
|
|
existing_workflow = [
|
|
x for x in Workflow.select(ignore_errors=True) if x and x.name == workflow.name
|
|
][0]
|
|
except IndexError:
|
|
workflow.store(comment='Indus Initial Import')
|
|
else:
|
|
# replace
|
|
workflow.id = existing_workflow.id
|
|
workflow.store(comment='Indus Update')
|
|
|
|
def import_wscalls(self):
|
|
if not os.path.exists(os.path.join(self.directory, 'wscalls')):
|
|
return
|
|
for filename in os.listdir(os.path.join(self.directory, 'wscalls')):
|
|
wscall = NamedWsCall.import_from_xml(open(os.path.join(self.directory, 'wscalls', filename)))
|
|
try:
|
|
existing_wscall = NamedWsCall.get(filename, ignore_errors=False)
|
|
except KeyError:
|
|
wscall.store(comment='Indus Initial Import')
|
|
else:
|
|
# replace
|
|
wscall.id = existing_wscall.id
|
|
wscall.store(comment='Indus Update')
|
|
|
|
def import_blocks(self):
|
|
if not os.path.exists(os.path.join(self.directory, 'blocks')):
|
|
return
|
|
for filename in os.listdir(os.path.join(self.directory, 'blocks')):
|
|
fd = open(os.path.join(self.directory, 'blocks', filename))
|
|
tree = ET.parse(fd)
|
|
# do not use import_from_xml to avoid autofixing url_name.
|
|
blockdef = BlockDef.import_from_xml_tree(tree, include_id=False)
|
|
try:
|
|
existing_blockdef = BlockDef.get_on_index(blockdef.slug, 'slug')
|
|
except KeyError:
|
|
blockdef.store(comment='Indus Initial Import')
|
|
else:
|
|
# replace
|
|
blockdef.id = existing_blockdef.id
|
|
blockdef.store(comment='Indus Update')
|
|
|
|
def import_carddefs(self):
|
|
if not os.path.exists(os.path.join(self.directory, 'carddefs')):
|
|
return
|
|
for filename in os.listdir(os.path.join(self.directory, 'carddefs')):
|
|
fd = open(os.path.join(self.directory, 'carddefs', filename))
|
|
tree = ET.parse(fd)
|
|
# do not use import_from_xml to avoid autofixing url_name.
|
|
carddef = CardDef.import_from_xml_tree(tree, include_id=False)
|
|
try:
|
|
existing_carddef = CardDef.get_by_urlname(carddef.url_name)
|
|
except KeyError:
|
|
carddef.store(comment='Indus Initial Import')
|
|
else:
|
|
# replace
|
|
carddef.id = existing_carddef.id
|
|
carddef.store(comment='Indus Update')
|
|
|
|
def import_formdefs(self):
|
|
if not os.path.exists(os.path.join(self.directory, 'forms')):
|
|
return
|
|
for filename in os.listdir(os.path.join(self.directory, 'forms')):
|
|
fd = open(os.path.join(self.directory, 'forms', filename))
|
|
tree = ET.parse(fd)
|
|
# do not use import_from_xml to avoid autofixing url_name.
|
|
formdef = FormDef.import_from_xml_tree(tree, include_id=False)
|
|
try:
|
|
existing_formdef = FormDef.get_by_urlname(formdef.url_name)
|
|
except KeyError:
|
|
formdef.store(comment='Indus Initial Import')
|
|
else:
|
|
# partial update
|
|
for attribute in (
|
|
'fields',
|
|
'digest_template',
|
|
'lateral_template',
|
|
'submission_lateral_template',
|
|
'user_support',
|
|
'geolocations',
|
|
):
|
|
if hasattr(formdef, attribute):
|
|
setattr(existing_formdef, attribute, getattr(formdef, attribute))
|
|
existing_formdef.store(comment='Indus Update')
|
|
|
|
|
|
Cmd.register()
|