886 lines
35 KiB
Python
886 lines
35 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
import copy
|
|
import os
|
|
import json
|
|
import hashlib
|
|
from utils import Whatever
|
|
import psycopg2
|
|
|
|
from wcs_olap.wcs_api import WcsApiError
|
|
|
|
|
|
def slugify(s):
|
|
return s.replace('-', '_').replace(' ', '_')
|
|
|
|
|
|
class Context(object):
|
|
def __init__(self):
|
|
self.stack = []
|
|
|
|
def __getitem__(self, key):
|
|
if not self.stack:
|
|
raise KeyError(key)
|
|
for d in self.stack[::-1]:
|
|
try:
|
|
return d[key]
|
|
except KeyError:
|
|
pass
|
|
else:
|
|
raise KeyError(key)
|
|
|
|
def push(self, d):
|
|
self.stack.append(d)
|
|
|
|
def pop(self):
|
|
self.stack = self.stack[:-1]
|
|
|
|
def as_dict(self):
|
|
r = {}
|
|
for d in self.stack:
|
|
r.update(d)
|
|
return r
|
|
|
|
|
|
class WcsOlapFeeder(object):
|
|
def __init__(self, api, pg_dsn, schema, logger=None, config=None, do_feed=True, fake=False):
|
|
self.api = api
|
|
self.fake = fake
|
|
self.logger = logger or Whatever()
|
|
self.schema = schema
|
|
self.connection = psycopg2.connect(dsn=pg_dsn)
|
|
self.connection.autocommit = True
|
|
self.cur = self.connection.cursor()
|
|
self.formdefs = api.formdefs
|
|
self.roles = api.roles
|
|
self.categories = api.categories
|
|
self.do_feed = do_feed
|
|
self.ctx = Context()
|
|
self.ctx.push({
|
|
'schema': self.schema,
|
|
'schema_temp': self.schema + '_temp',
|
|
'role_table': 'role',
|
|
'channel_table': 'channel',
|
|
'category_table': 'category',
|
|
'form_table': 'formdef',
|
|
'generic_formdata_table': 'formdata',
|
|
'generic_status_table': 'status',
|
|
'generic_evolution_table': 'evolution',
|
|
'year_table': 'year',
|
|
'month_table': 'month',
|
|
'day_table': 'day',
|
|
'dow_table': 'dow',
|
|
'hour_table': 'hour',
|
|
'agent_table': 'agent',
|
|
})
|
|
self.config = config or {}
|
|
self.model = {
|
|
'label': self.config.get('cubes_label', schema),
|
|
'name': schema,
|
|
'search_path': [schema, 'public'],
|
|
'pg_dsn': pg_dsn,
|
|
'cubes': [],
|
|
}
|
|
cube = {
|
|
'name': 'all_formdata',
|
|
'label': u'Tous les formulaires',
|
|
'fact_table': 'formdata',
|
|
'key': 'id',
|
|
'joins': [
|
|
{
|
|
'name': 'receipt_time',
|
|
'table': 'dates',
|
|
'detail': 'date',
|
|
'master': 'receipt_time',
|
|
},
|
|
{
|
|
'name': 'channel',
|
|
'table': 'channel',
|
|
'master': 'channel_id',
|
|
'detail': 'id',
|
|
},
|
|
{
|
|
'name': 'formdef',
|
|
'table': 'formdef',
|
|
'master': 'formdef_id',
|
|
'detail': 'id',
|
|
},
|
|
{
|
|
'name': 'category',
|
|
'table': 'category',
|
|
'master': 'formdef.category_id',
|
|
'detail': 'id',
|
|
'kind': 'left',
|
|
},
|
|
{
|
|
'name': 'hour',
|
|
'table': 'hour',
|
|
'master': 'hour_id',
|
|
'detail': 'id',
|
|
},
|
|
{
|
|
'name': 'generic_status',
|
|
'table': 'status',
|
|
'master': 'generic_status_id',
|
|
'detail': 'id',
|
|
},
|
|
{
|
|
'name': 'agent',
|
|
'table': 'agent',
|
|
'master': 'first_agent_id',
|
|
'detail': 'id',
|
|
'kind': 'inner',
|
|
},
|
|
],
|
|
'dimensions': [
|
|
{
|
|
'name': 'receipt_time',
|
|
'label': 'date de la demande',
|
|
'join': ['receipt_time'],
|
|
'type': 'date',
|
|
'value': 'receipt_time.date',
|
|
},
|
|
{
|
|
'name': 'channel',
|
|
'label': 'canal',
|
|
'join': ['channel'],
|
|
'type': 'integer',
|
|
'value': 'channel.id',
|
|
'value_label': 'channel.label',
|
|
},
|
|
{
|
|
'name': 'category',
|
|
'label': 'catégorie',
|
|
'join': ['formdef', 'category'],
|
|
'type': 'integer',
|
|
'value': 'category.id',
|
|
'value_label': 'category.label',
|
|
'order_by': 'category.label',
|
|
},
|
|
{
|
|
'name': 'formdef',
|
|
'label': 'formulaire',
|
|
'join': ['formdef'],
|
|
'type': 'integer',
|
|
'value': 'formdef.id',
|
|
'value_label': 'formdef.label',
|
|
'order_by': 'formdef.label',
|
|
},
|
|
{
|
|
'name': 'generic_status',
|
|
'label': 'statut simplifié',
|
|
'join': ['generic_status'],
|
|
'type': 'integer',
|
|
'value': 'generic_status.id',
|
|
'value_label': 'generic_status.label',
|
|
},
|
|
{
|
|
'name': 'hour',
|
|
'label': 'heure',
|
|
'join': ['hour'],
|
|
'type': 'integer',
|
|
'value': 'hour.id',
|
|
'filter': False,
|
|
},
|
|
{
|
|
'name': 'agent',
|
|
'label': 'premier agent traitant',
|
|
'join': ['agent'],
|
|
'type': 'integer',
|
|
'value': 'agent.id',
|
|
'value_label': 'agent.label',
|
|
'order_by': 'agent.label',
|
|
},
|
|
],
|
|
'measures': [
|
|
{
|
|
'name': 'count',
|
|
'label': 'nombre de demandes',
|
|
'type': 'integer',
|
|
'expression': 'count({fact_table}.id)',
|
|
},
|
|
{
|
|
'name': 'avg_endpoint_delay',
|
|
'label': 'délai de traitement moyen',
|
|
'type': 'duration',
|
|
'expression': 'avg(endpoint_delay)',
|
|
},
|
|
{
|
|
'name': 'max_endpoint_delay',
|
|
'label': 'délai de traitement maximum',
|
|
'type': 'duration',
|
|
'expression': 'max(endpoint_delay)',
|
|
},
|
|
{
|
|
'name': 'min_endpoint_delay',
|
|
'label': 'délai de traitement minimum',
|
|
'type': 'duration',
|
|
'expression': 'min(endpoint_delay)',
|
|
},
|
|
{
|
|
'name': 'percent',
|
|
'label': 'pourcentage des demandes',
|
|
'type': 'percent',
|
|
"expression": 'case (select count({fact_table}.id) from {table_expression} '
|
|
'where {where_conditions}) when 0 then null else '
|
|
'count({fact_table}.id) * 100. / (select '
|
|
'count({fact_table}.id) from {table_expression} where '
|
|
'{where_conditions}) end',
|
|
},
|
|
{
|
|
'name': 'geolocation',
|
|
'label': 'localisation géographique',
|
|
'type': 'point',
|
|
'expression': 'array_agg("{fact_table}".geolocation_base) FILTER (WHERE "{fact_table}".geolocation_base IS NOT NULL)',
|
|
}
|
|
]
|
|
}
|
|
self.model['cubes'].append(cube)
|
|
self.base_cube = self.model['cubes'][0]
|
|
self.agents_mapping = {}
|
|
self.formdata_json_index = []
|
|
self.has_jsonb = self.detect_jsonb()
|
|
if self.has_jsonb:
|
|
cube['json_field'] = 'json_data'
|
|
|
|
def detect_jsonb(self):
|
|
self.cur.execute("SELECT 1 FROM pg_type WHERE typname = 'jsonb'")
|
|
return bool(self.cur.rowcount)
|
|
|
|
def hash_table_name(self, table_name):
|
|
table_name = table_name.format(**self.default_ctx)
|
|
if len(table_name) < 64:
|
|
return table_name
|
|
else:
|
|
return table_name[:57] + hashlib.md5(table_name).hexdigest()[:6]
|
|
|
|
@property
|
|
def default_ctx(self):
|
|
return self.ctx.as_dict()
|
|
|
|
def ex(self, query, ctx=None, vars=None):
|
|
ctx = ctx or {}
|
|
ctx.update(self.default_ctx)
|
|
sql = query.format(**(ctx or {}))
|
|
self.logger.debug('SQL: %s VARS: %s', sql, vars)
|
|
self.cur.execute(sql, vars=vars)
|
|
|
|
def exmany(self, query, varslist, ctx=None):
|
|
ctx = ctx or {}
|
|
ctx.update(self.default_ctx)
|
|
sql = query.format(**(ctx or {}))
|
|
self.logger.debug('SQL: %s VARSLIST: %s', sql, varslist)
|
|
self.cur.executemany(sql, varslist)
|
|
|
|
def do_schema(self):
|
|
self.ex('SET search_path = public')
|
|
self.logger.debug('dropping schema %s', self.schema + '_temp')
|
|
self.ex('DROP SCHEMA IF EXISTS {schema_temp} CASCADE')
|
|
self.logger.debug('creating schema %s', self.schema)
|
|
self.ex('CREATE SCHEMA {schema_temp}')
|
|
self.ex('SET search_path = {schema_temp},public')
|
|
|
|
def do_dates_table(self):
|
|
# test if public.dates exists
|
|
self.ex("SELECT * FROM information_schema.tables WHERE table_name = 'dates' AND"
|
|
" table_schema = 'public'")
|
|
if len(list(self.cur.fetchall())) < 1:
|
|
self.ex('''
|
|
CREATE TABLE public.dates AS (SELECT
|
|
the_date.the_date::date AS date,
|
|
to_char(the_date.the_date, 'TMday') AS day,
|
|
to_char(the_date.the_date, 'TMmonth') AS month
|
|
FROM
|
|
generate_series('2010-01-01'::date, '2020-01-01'::date, '1 day'::interval)
|
|
AS the_date(the_date));''')
|
|
|
|
channels = [
|
|
[1, 'web', u'web'],
|
|
[2, 'mail', u'courrier'],
|
|
[3, 'phone', u'téléphone'],
|
|
[4, 'counter', u'guichet'],
|
|
[5, 'backoffice', u'backoffice'],
|
|
[6, 'email', u'email'],
|
|
[7, 'fax', u'fax'],
|
|
]
|
|
channel_to_id = dict((c[1], c[0]) for c in channels)
|
|
id_to_channel = dict((c[0], c[1]) for c in channels)
|
|
|
|
status = [
|
|
[1, 'Nouveau'],
|
|
[2, 'En cours'],
|
|
[3, 'Terminé'],
|
|
]
|
|
status_to_id = dict((c[1], c[0]) for c in channels)
|
|
id_to_status = dict((c[0], c[1]) for c in channels)
|
|
|
|
def create_table(self, name, columns, inherits=None, comment=None):
|
|
sql = 'CREATE TABLE %s' % name
|
|
sql += '(' + ', '.join('%s %s' % (n, t) for n, t in columns) + ')'
|
|
if inherits:
|
|
sql += ' INHERITS (%s)' % inherits
|
|
self.ex(sql)
|
|
if comment:
|
|
self.ex('COMMENT ON TABLE %s IS %%s' % name, vars=(comment,))
|
|
|
|
def create_labeled_table(self, name, labels, serial=False, comment=None):
|
|
if serial:
|
|
id_type = 'serial primary key'
|
|
else:
|
|
id_type = 'smallint primary key'
|
|
self.create_table(name,
|
|
[
|
|
['id', id_type],
|
|
['label', 'varchar']
|
|
], comment=comment)
|
|
values = ', '.join(self.cur.mogrify('(%s, %s)', [_id, _label]) for _id, _label in labels)
|
|
if not values:
|
|
return
|
|
self.ex('INSERT INTO %s (id, label) VALUES %s' % (str(name), values))
|
|
|
|
def tpl(self, o, ctx=None):
|
|
ctx = ctx or {}
|
|
ctx.update(self.default_ctx)
|
|
|
|
def helper(o):
|
|
if isinstance(o, basestring):
|
|
return o.format(**ctx)
|
|
elif isinstance(o, dict):
|
|
return dict((k, helper(v)) for k, v in o.iteritems())
|
|
elif isinstance(o, list):
|
|
return [helper(v) for v in o]
|
|
elif isinstance(o, (bool, int, float)):
|
|
return o
|
|
else:
|
|
assert False, '%s is not a valid value for JSON' % o
|
|
return helper(o)
|
|
|
|
def add_dim(self, **kwargs):
|
|
self.dimensions.append(self.tpl(kwargs))
|
|
|
|
def do_base_table(self):
|
|
# channels
|
|
self.create_labeled_table('{channel_table}', [[c[0], c[2]] for c in self.channels],
|
|
comment=u'canal')
|
|
|
|
# roles
|
|
roles = dict((i, role.name) for i, role in enumerate(self.roles))
|
|
self.create_labeled_table('{role_table}', roles.items(), comment=u'role')
|
|
self.role_mapping = dict((role.id, i) for i, role in enumerate(self.roles))
|
|
|
|
# categories
|
|
self.create_labeled_table('{category_table}', enumerate(c.name for c in self.categories),
|
|
comment=u'catégorie')
|
|
self.categories_mapping = dict((c.id, i) for i, c in enumerate(self.categories))
|
|
|
|
self.create_labeled_table('{hour_table}', zip(range(0, 24), map(str, range(0, 24))),
|
|
comment=u'heures')
|
|
|
|
self.create_labeled_table('{generic_status_table}', self.status,
|
|
comment=u'statuts simplifiés')
|
|
self.ex('CREATE TABLE {form_table} (id serial PRIMARY KEY,'
|
|
' category_id integer REFERENCES {category_table} (id),'
|
|
' label varchar)')
|
|
self.ex('COMMENT ON TABLE {form_table} IS %s', vars=(u'types de formulaire',))
|
|
# agents
|
|
self.create_labeled_table('{agent_table}', [], serial=True, comment=u'agents')
|
|
|
|
self.columns = [
|
|
['id', 'serial primary key'],
|
|
['formdef_id', 'smallint REFERENCES {form_table} (id)'],
|
|
['receipt_time', 'date'],
|
|
['hour_id', 'smallint REFERENCES {hour_table} (id)'],
|
|
['channel_id', 'smallint REFERENCES {channel_table} (id)'],
|
|
['backoffice', 'boolean'],
|
|
['generic_status_id', 'smallint REFERENCES {generic_status_table} (id)'],
|
|
['endpoint_delay', 'interval'],
|
|
['first_agent_id', 'smallint REFERENCES {agent_table} (id)'],
|
|
['geolocation_base', 'POINT NULL'],
|
|
]
|
|
if self.has_jsonb:
|
|
self.columns.append(['json_data', 'JSONB NULL'])
|
|
self.comments = {
|
|
'formdef_id': u'formulaire',
|
|
'receipt_time': u'date de réception',
|
|
'hour_id': u'heure',
|
|
'channel_id': u'canal',
|
|
'backoffice': u'soumission backoffce',
|
|
'generic_status_id': u'statut simplifié',
|
|
'endpoint_delay': u'délai de traitement',
|
|
'geolocation_base': u'position géographique',
|
|
}
|
|
self.create_table('{generic_formdata_table}', self.columns)
|
|
for at, comment in self.comments.iteritems():
|
|
self.ex('COMMENT ON COLUMN {generic_formdata_table}.%s IS %%s' % at, vars=(comment,))
|
|
self.ex('COMMENT ON TABLE {generic_formdata_table} IS %s', vars=(u'tous les formulaires',))
|
|
# evolutions
|
|
self.create_table('{generic_evolution_table}', [
|
|
['id', 'serial primary key'],
|
|
['generic_status_id', 'smallint REFERENCES {generic_status_table} (id)'],
|
|
['formdata_id', 'integer'], # "REFERENCES {generic_formdata_table} (id)" is impossible because FK constraints do not work on inherited tables
|
|
['time', 'timestamp'],
|
|
['date', 'date'],
|
|
['hour_id', 'smallint REFERENCES {hour_table} (id)'],
|
|
])
|
|
self.ex('COMMENT ON TABLE {generic_evolution_table} IS %s', vars=(u'evolution générique',))
|
|
|
|
def feed(self):
|
|
try:
|
|
if self.do_feed:
|
|
self.do_schema()
|
|
self.do_dates_table()
|
|
self.do_base_table()
|
|
for formdef in self.formdefs:
|
|
self.api.cache = {}
|
|
try:
|
|
formdef_feeder = WcsFormdefFeeder(self, formdef, do_feed=self.do_feed)
|
|
formdef_feeder.feed()
|
|
except WcsApiError, e:
|
|
# ignore authorization errors
|
|
if (len(e.args) > 2 and getattr(e.args[2], 'response', None)
|
|
and e.args[2].response.status_code == 403):
|
|
continue
|
|
self.logger.error('failed to retrieve formdef %s (%s)', formdef.slug, e)
|
|
if 'cubes_model_dirs' in self.config:
|
|
model_path = os.path.join(self.config['cubes_model_dirs'], '%s.model' % self.schema)
|
|
with open(model_path, 'w') as f:
|
|
json.dump(self.model, f, indent=2, sort_keys=True)
|
|
except:
|
|
raise
|
|
else:
|
|
if self.do_feed:
|
|
if not self.fake:
|
|
self.logger.debug('dropping schema %s', self.schema)
|
|
self.ex('DROP SCHEMA IF EXISTS {schema} CASCADE')
|
|
self.logger.debug('dropping schema %s to %s', self.schema + '_temp', self.schema)
|
|
self.ex('ALTER SCHEMA {schema_temp} RENAME TO {schema}')
|
|
|
|
def insert_agent(self, name):
|
|
self.ex('INSERT INTO {agent_table} (label) VALUES (%s) RETURNING (id)', vars=[name])
|
|
return self.cur.fetchone()[0]
|
|
|
|
def get_agent(self, user):
|
|
assert user.name
|
|
assert user.id
|
|
if user.id not in self.agents_mapping:
|
|
self.agents_mapping[user.id] = self.insert_agent(user.name)
|
|
return self.agents_mapping[user.id]
|
|
|
|
def create_formdata_json_index(self, varname):
|
|
if varname in self.formdata_json_index:
|
|
return
|
|
index_name = self.hash_table_name('{formdata_table}_%s_json_idx' % varname)
|
|
self.ex('CREATE INDEX {index_name} ON {generic_formdata_table} (("json_data"->>%s))',
|
|
ctx={'index_name': index_name}, vars=[varname])
|
|
# prevent double creation
|
|
self.formdata_json_index.append(varname)
|
|
|
|
|
|
class WcsFormdefFeeder(object):
|
|
def __init__(self, olap_feeder, formdef, do_feed=True):
|
|
self.olap_feeder = olap_feeder
|
|
self.formdef = formdef
|
|
self.status_mapping = {}
|
|
self.items_mappings = {}
|
|
self.do_feed = do_feed
|
|
self.fields = []
|
|
self.item_id_mappings = {}
|
|
|
|
@property
|
|
def table_name(self):
|
|
return self.hash_table_name('formdata_%s' % self.formdef.slug.replace('-', '_'))
|
|
|
|
@property
|
|
def status_table_name(self):
|
|
return self.hash_table_name('status_%s' % self.formdef.slug.replace('-', '_'))
|
|
|
|
@property
|
|
def evolution_table_name(self):
|
|
return self.hash_table_name('evolution_%s' % self.formdef.slug.replace('-', '_'))
|
|
|
|
def __getattr__(self, name):
|
|
return getattr(self.olap_feeder, name)
|
|
|
|
def do_statuses(self):
|
|
statuses = self.formdef.schema.workflow.statuses
|
|
self.olap_feeder.create_labeled_table(self.status_table_name,
|
|
enumerate([s.name for s in statuses]),
|
|
comment=u'statuts du formulaire « %s »' %
|
|
self.formdef.schema.name)
|
|
self.status_mapping = dict((s.id, i) for i, s in enumerate(statuses))
|
|
|
|
def do_data_table(self):
|
|
self.ex('INSERT INTO {form_table} (category_id, label) VALUES (%s, %s) RETURNING (id)',
|
|
vars=[self.categories_mapping.get(self.formdef.schema.category_id),
|
|
self.formdef.schema.name])
|
|
self.formdef_sql_id = self.cur.fetchone()[0]
|
|
|
|
columns = [['status_id', 'smallint REFERENCES {status_table} (id)']]
|
|
|
|
comments = {}
|
|
|
|
duplicated_varnames = {}
|
|
field_varnames = {}
|
|
for field in self.formdef.schema.fields:
|
|
if field.varname in field_varnames:
|
|
duplicated_varnames[field.varname] = True
|
|
field_varnames[field.varname] = True
|
|
|
|
# add item fields
|
|
fields = self.formdef.schema.fields
|
|
if self.formdef.schema.workflow:
|
|
fields += self.formdef.schema.workflow.fields
|
|
for field in fields:
|
|
if not field.type == 'item':
|
|
continue
|
|
if field.anonymise is True:
|
|
continue
|
|
if not field.varname or '-' in field.varname:
|
|
continue
|
|
if field.varname in duplicated_varnames:
|
|
continue
|
|
self.fields.append(field)
|
|
comment = (u'valeurs du champ « %s » du formulaire %s'
|
|
% (field.label, self.formdef.schema.name))
|
|
table_name = self.hash_table_name('{formdata_table}_field_%s' % field.varname)
|
|
# create table and mapping
|
|
if field.items:
|
|
self.create_labeled_table(table_name, enumerate(field.items),
|
|
comment=comment)
|
|
self.items_mappings[field.varname] = dict(
|
|
(item, i) for i, item in enumerate(field.items))
|
|
elif field.options:
|
|
options = enumerate(field.options)
|
|
self.create_labeled_table(table_name, [(i, o['label']) for i, o in options],
|
|
comment=comment)
|
|
self.items_mappings[field.varname] = dict((o['value'], i) for i, o in options)
|
|
else:
|
|
# open item field, from data sources...
|
|
self.create_labeled_table(table_name, [], serial=True, comment=comment)
|
|
|
|
at = 'field_%s' % field.varname
|
|
columns.append([at, 'smallint REFERENCES %s (id)' % table_name])
|
|
comments[at] = field.label
|
|
|
|
# add geolocation fields
|
|
for geolocation, label in self.formdef.schema.geolocations:
|
|
at = 'geolocation_%s' % geolocation
|
|
columns.append([at, 'point'])
|
|
comments[at] = label
|
|
|
|
# add function fields
|
|
for function, name in self.formdef.schema.workflow.functions.iteritems():
|
|
at = 'function_%s' % slugify(function)
|
|
columns.append([at, 'smallint REFERENCES {role_table} (id)'])
|
|
comments[at] = u'fonction « %s »' % name
|
|
|
|
self.columns = ([name for name, _type in self.olap_feeder.columns]
|
|
+ [name for name, _type in columns])
|
|
self.columns.remove('geolocation_base')
|
|
|
|
self.create_table('{formdata_table}', columns, inherits='{generic_formdata_table}',
|
|
comment=u'formulaire %s' % self.formdef.schema.name)
|
|
for at, comment in comments.iteritems():
|
|
self.ex('COMMENT ON COLUMN {formdata_table}.%s IS %%s' % at, vars=(comment,))
|
|
|
|
# Creat index for JSON fields
|
|
if self.has_jsonb:
|
|
for field in fields:
|
|
if field.varname and not '-' in field.varname:
|
|
self.create_formdata_json_index(field.varname)
|
|
|
|
# PostgreSQL does not propagate foreign key constraints to child tables
|
|
# so we must recreate them manually
|
|
for name, _type in self.olap_feeder.columns:
|
|
if 'REFERENCES' not in _type:
|
|
continue
|
|
i = _type.index('REFERENCES')
|
|
constraint = '%s_fk_constraint FOREIGN KEY (%s) %s' % (name, name, _type[i:])
|
|
self.ex('ALTER TABLE {formdata_table} ADD CONSTRAINT %s' % constraint)
|
|
self.ex('ALTER TABLE {formdata_table} ADD PRIMARY KEY (id)')
|
|
# table des evolutions
|
|
self.create_table('{evolution_table}', [
|
|
['id', 'serial primary key'],
|
|
['status_id', 'smallint REFERENCES {status_table} (id)'],
|
|
['formdata_id', 'integer REFERENCES {formdata_table} (id)'],
|
|
['time', 'timestamp'],
|
|
['date', 'date'],
|
|
['hour_id', 'smallint REFERENCES {hour_table} (id)'],
|
|
])
|
|
self.ex('COMMENT ON TABLE {evolution_table} IS %s',
|
|
vars=(u'evolution des demandes %s' % self.formdef.schema.name,))
|
|
|
|
def insert_item_value(self, field, value):
|
|
table_name = self.hash_table_name('{formdata_table}_field_%s' % field.varname)
|
|
self.ex('INSERT INTO {item_table} (label) VALUES (%s) RETURNING (id)', vars=[value],
|
|
ctx={'item_table': table_name})
|
|
return self.cur.fetchone()[0]
|
|
|
|
def get_item_id(self, field, value):
|
|
assert field
|
|
assert field.varname
|
|
assert value
|
|
|
|
mapping = self.item_id_mappings.setdefault(field.varname, {})
|
|
if value not in mapping:
|
|
mapping[value] = self.insert_item_value(field, value)
|
|
return mapping[value]
|
|
|
|
def generic_status(self, status):
|
|
if status.endpoint:
|
|
generic_status = 3
|
|
elif status.startpoint:
|
|
generic_status = 1
|
|
else:
|
|
generic_status = 2
|
|
return generic_status
|
|
|
|
def do_data(self):
|
|
values = []
|
|
generic_evolution_values = []
|
|
evolution_values = []
|
|
for data in self.formdef.datas:
|
|
json_data = {}
|
|
|
|
# ignore formdata without status
|
|
if data.workflow.status:
|
|
status_id = data.workflow.status.id
|
|
elif data.evolution:
|
|
for evolution in reversed(data.evolution):
|
|
if evolution.status:
|
|
status_id = evolution.status
|
|
break
|
|
else:
|
|
continue
|
|
else:
|
|
continue
|
|
|
|
try:
|
|
status = data.formdef.schema.workflow.statuses_map[status_id]
|
|
except KeyError:
|
|
self.logger.warning('%s.%s unknown status status_id %s',
|
|
data.formdef.schema.name, data.id, status_id)
|
|
continue
|
|
|
|
channel = data.submission.channel.lower()
|
|
if channel == 'web' and data.submission.backoffice:
|
|
channel = 'backoffice'
|
|
row = {
|
|
'formdef_id': self.formdef_sql_id,
|
|
'receipt_time': data.receipt_time,
|
|
'hour_id': data.receipt_time.hour,
|
|
'channel_id': self.channel_to_id[channel],
|
|
'backoffice': data.submission.backoffice,
|
|
'generic_status_id': self.generic_status(status),
|
|
'status_id': self.status_mapping[status_id],
|
|
'endpoint_delay': data.endpoint_delay,
|
|
'first_agent_id': self.get_first_agent_in_evolution(data),
|
|
}
|
|
# add form fields value
|
|
for field in self.fields:
|
|
v = None
|
|
raw = None
|
|
if field.type == 'item':
|
|
if field.varname in data.fields:
|
|
raw = data.fields[field.varname]
|
|
elif field.varname in data.workflow.fields:
|
|
raw = data.workflow.fields[field.varname]
|
|
else:
|
|
raw = None
|
|
# map items to sql id
|
|
if field.items or field.options:
|
|
v = self.items_mappings[field.varname].get(raw)
|
|
else:
|
|
v = raw and self.get_item_id(field, raw)
|
|
# unstructured storage of field values
|
|
if field.varname and raw:
|
|
json_data[field.varname] = raw
|
|
|
|
row['field_%s' % field.varname] = v
|
|
if self.has_jsonb:
|
|
row['json_data'] = json.dumps(json_data)
|
|
# add geolocation fields value
|
|
for geolocation, label in self.formdef.schema.geolocations:
|
|
v = (data.geolocations or {}).get(geolocation)
|
|
if v:
|
|
v = '(%.6f, %.6f)' % (v.get('lon'), v.get('lat'))
|
|
row['geolocation_%s' % geolocation] = v
|
|
# add function fields value
|
|
for function, name in self.formdef.schema.workflow.functions.iteritems():
|
|
try:
|
|
v = data.functions[function]
|
|
except KeyError:
|
|
v = None
|
|
else:
|
|
v = v and self.olap_feeder.role_mapping[v.id]
|
|
at = 'function_%s' % slugify(function)
|
|
row[at] = v
|
|
|
|
tpl = '(' + ', '.join(['%s'] * len(self.columns[1:])) + ')'
|
|
value = self.cur.mogrify(tpl, [row[column] for column in self.columns[1:]])
|
|
values.append(value)
|
|
# inert evolutions
|
|
generic_evolution = []
|
|
evolution = []
|
|
last_status = None
|
|
for evo in data.evolution:
|
|
if not evo.status:
|
|
continue
|
|
try:
|
|
status = data.formdef.schema.workflow.statuses_map[evo.status]
|
|
except KeyError:
|
|
self.logger.warning('%s.%s unknown status in evolution %s',
|
|
data.formdef.schema.name, data.id, evo.status)
|
|
continue
|
|
status_id = self.status_mapping[status.id]
|
|
generic_status_id = self.generic_status(status)
|
|
evolution.append(
|
|
[0, status_id, evo.time, evo.time.date(), evo.time.hour])
|
|
if generic_status_id == last_status:
|
|
continue
|
|
generic_evolution.append(
|
|
[0, generic_status_id, evo.time, evo.time.date(), evo.time.hour])
|
|
last_status = generic_status_id
|
|
generic_evolution_values.append(generic_evolution)
|
|
evolution_values.append(evolution)
|
|
if not values:
|
|
self.logger.warning('no data')
|
|
return
|
|
self.ex('INSERT INTO {formdata_table} ({columns}) VALUES {values} RETURNING id',
|
|
ctx=dict(columns=', '.join(self.columns[1:]), values=', '.join(values)))
|
|
|
|
# insert generic evolutions
|
|
generic_evolutions = []
|
|
ids = list(self.cur.fetchall())
|
|
for evos, (formdata_id,) in zip(generic_evolution_values, ids):
|
|
for row in evos:
|
|
row[0] = formdata_id
|
|
generic_evolutions.append(tuple(row))
|
|
if len(generic_evolutions) == 500:
|
|
self.ex('INSERT INTO {generic_evolution_table} (%s) VALUES %s' % (
|
|
', '.join(['formdata_id', 'generic_status_id', 'time', 'date', 'hour_id']),
|
|
', '.join(['%s'] * len(generic_evolutions))), vars=generic_evolutions)
|
|
generic_evolutions = []
|
|
if generic_evolutions:
|
|
self.ex('INSERT INTO {generic_evolution_table} (%s) VALUES %s' % (
|
|
', '.join(['formdata_id', 'generic_status_id', 'time', 'date', 'hour_id']),
|
|
', '.join(['%s'] * len(generic_evolutions))), vars=generic_evolutions)
|
|
|
|
# insert evolutions
|
|
evolutions = []
|
|
for evos, (formdata_id,) in zip(evolution_values, ids):
|
|
for row in evos:
|
|
row[0] = formdata_id
|
|
evolutions.append(tuple(row))
|
|
if len(evolutions) == 500:
|
|
self.ex('INSERT INTO {evolution_table} (%s) VALUES %s' % (
|
|
', '.join(['formdata_id', 'status_id', 'time', 'date', 'hour_id']),
|
|
', '.join(['%s'] * len(evolutions))), vars=evolutions)
|
|
evolutions = []
|
|
if evolutions:
|
|
self.ex('INSERT INTO {evolution_table} (%s) VALUES %s' % (
|
|
', '.join(['formdata_id', 'status_id', 'time', 'date', 'hour_id']),
|
|
', '.join(['%s'] * len(evolutions))), vars=evolutions)
|
|
|
|
def get_first_agent_in_evolution(self, formdata):
|
|
for evo in formdata.evolution:
|
|
if evo.who:
|
|
return self.get_agent(evo.who)
|
|
|
|
def feed(self):
|
|
self.olap_feeder.ctx.push({
|
|
'formdata_table': self.table_name,
|
|
'status_table': self.status_table_name,
|
|
'evolution_table': self.evolution_table_name
|
|
})
|
|
|
|
# create cube
|
|
cube = self.cube = copy.deepcopy(self.base_cube)
|
|
# remove json field from formdef cubes
|
|
cube.pop('json_field', None)
|
|
cube.update({
|
|
'name': self.table_name,
|
|
'label': self.formdef.schema.name,
|
|
'fact_table': self.table_name,
|
|
'key': 'id',
|
|
})
|
|
cube['dimensions'] = [dimension for dimension in cube['dimensions']
|
|
if dimension['name'] not in ('category', 'formdef')]
|
|
|
|
# add dimension for status
|
|
cube['joins'].append({
|
|
'name': 'status',
|
|
'table': self.status_table_name,
|
|
'master': 'status_id',
|
|
'detail': 'id',
|
|
})
|
|
cube['dimensions'].append({
|
|
'name': 'status',
|
|
'label': 'statut',
|
|
'join': ['status'],
|
|
'type': 'integer',
|
|
'value': 'status.id',
|
|
'value_label': 'status.label',
|
|
})
|
|
|
|
# add dimension for function
|
|
for function, name in self.formdef.schema.workflow.functions.iteritems():
|
|
at = 'function_%s' % slugify(function)
|
|
cube['joins'].append({
|
|
'name': at,
|
|
'table': 'role',
|
|
'master': at,
|
|
'detail': 'id',
|
|
})
|
|
cube['dimensions'].append({
|
|
'name': at,
|
|
'label': u'fonction %s' % name.lower(),
|
|
'join': [at],
|
|
'type': 'integer',
|
|
'value': '"%s".id' % at,
|
|
'value_label': '"%s".label' % at,
|
|
'filter': False,
|
|
})
|
|
|
|
# add dimensions for item fields
|
|
fields = self.formdef.schema.fields
|
|
if self.formdef.schema.workflow:
|
|
fields += self.formdef.schema.workflow.fields
|
|
for field in fields:
|
|
if not field.type == 'item':
|
|
continue
|
|
if field.anonymise is True:
|
|
continue
|
|
if not field.varname:
|
|
continue
|
|
if '-' in field.varname:
|
|
continue
|
|
table_name = self.hash_table_name('{formdata_table}_field_%s' % field.varname)
|
|
cube['joins'].append({
|
|
'name': field.varname,
|
|
'table': table_name,
|
|
'master': 'field_%s' % field.varname,
|
|
'detail': 'id',
|
|
})
|
|
if not field.required:
|
|
cube['joins'][-1]['kind'] = 'full'
|
|
cube['dimensions'].append({
|
|
'name': field.varname,
|
|
'label': field.label.lower(),
|
|
'join': [field.varname],
|
|
'type': 'integer',
|
|
'value': '"%s".id' % field.varname,
|
|
'value_label': '"%s".label' % field.varname,
|
|
'filter': True,
|
|
})
|
|
|
|
self.model['cubes'].append(cube)
|
|
if self.do_feed:
|
|
try:
|
|
self.logger.info('feed formdef %s', self.formdef.slug)
|
|
self.do_statuses()
|
|
self.do_data_table()
|
|
self.do_data()
|
|
finally:
|
|
self.olap_feeder.ctx.pop()
|