82 lines
3.4 KiB
Python
82 lines
3.4 KiB
Python
# Copyright © 2023 Lee Yingtong Li (RunasSudo)
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as published by
|
|
# the Free Software Foundation, either version 3 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 Affero General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
import itertools
|
|
import sqlite3
|
|
|
|
# Open database
|
|
con = sqlite3.connect('database.db')
|
|
con.row_factory = sqlite3.Row
|
|
cur = con.cursor()
|
|
|
|
# Populate pbs_restriction.criteria_rendered
|
|
# sql.js is too slow to do the database JOINs at runtime
|
|
|
|
class Operator:
|
|
def __init__(self, operator, nodes):
|
|
self.operator = operator
|
|
self.nodes = nodes
|
|
|
|
def flatten(self):
|
|
# If contains just 1 element, return it as itself, not as an Operator
|
|
if len(self.nodes) == 1:
|
|
return self.nodes[0]
|
|
|
|
# Otherwise flatten recursively
|
|
flattened = [n.flatten() if isinstance(n, Operator) else n for n in self.nodes]
|
|
return Operator(self.operator, flattened)
|
|
|
|
@staticmethod
|
|
def render(op):
|
|
if isinstance(op, str):
|
|
return '<p>' + op + '</p>'
|
|
elif isinstance(op, Operator):
|
|
if len(op.nodes) == 1:
|
|
# Single simple string
|
|
return '<p>' + op.nodes[0] + '</p>'
|
|
if not any(isinstance(n, Operator) for n in op.nodes):
|
|
# All simple strings
|
|
return '<ul><li><p>' + ', <i>{}</i></p></li><li><p>'.format({'all': 'AND', 'any': 'OR', 'one-of': 'OR'}[op.operator]).join(n.rstrip('.') for n in op.nodes) + '</p></li></ul>'
|
|
else:
|
|
# Mix of Operator +/- simple strings
|
|
return '<p><i>{}</i></p>'.format({'all': 'AND', 'any': 'OR', 'one-of': 'OR'}[op.operator]).join(Operator.render(n) for n in op.nodes)
|
|
else:
|
|
# Must be a list of a single item
|
|
return Operator.render(op[0])
|
|
|
|
cur.execute('SELECT * FROM pbs_restriction INNER JOIN pbs_restriction_criteria ON pbs_restriction.code = pbs_restriction_criteria.restriction_code INNER JOIN pbs_criteria ON pbs_restriction_criteria.criteria_code = pbs_criteria.code INNER JOIN pbs_criteria_parameter ON pbs_criteria.code = pbs_criteria_parameter.criteria_code')
|
|
flat_parameters = cur.fetchall()
|
|
|
|
parameters_by_restriction = itertools.groupby(flat_parameters, lambda p: p['restriction_code'])
|
|
for restriction_code, restriction_parameters in parameters_by_restriction:
|
|
restriction_parameters = list(restriction_parameters)
|
|
|
|
# Build tree of nodes representing parameters
|
|
restriction_operator = Operator(restriction_parameters[0]['criteria_operator'], [])
|
|
|
|
parameters_by_criteria = itertools.groupby(restriction_parameters, lambda p: p['criteria_code'])
|
|
for criteria_code, criteria_parameters in parameters_by_criteria:
|
|
criteria_parameters = list(criteria_parameters)
|
|
criteria_operator = Operator(criteria_parameters[0]['parameters_operator'], [p['text'] for p in criteria_parameters])
|
|
restriction_operator.nodes.append(criteria_operator)
|
|
|
|
# Flatten and render the tree
|
|
restriction_operator = restriction_operator.flatten()
|
|
rendered = Operator.render(restriction_operator)
|
|
|
|
cur.execute('UPDATE pbs_restriction SET criteria_rendered = ? WHERE code = ?', (rendered, restriction_code))
|
|
|
|
con.commit()
|