You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
60 lines
2.1 KiB
60 lines
2.1 KiB
from django.core.management.base import BaseCommand
|
|
from service.utils import run_sql_command, log_to_telegram
|
|
|
|
CMD_PIVOT_DIST = """CREATE OR REPLACE procedure pivot_dist()
|
|
--RETURNS SET OF record
|
|
AS $BODY$
|
|
DECLARE columnNames TEXT;
|
|
BEGIN
|
|
DROP VIEW IF EXISTS points_with_dist;
|
|
SELECT 'placement_point bigint, ' || string_agg(c, ', ') FROM (SELECT distinct pvz_postamates_group_id, 'd' || pvz_postamates_group_id || ' double precision' as c from service_placementpointpvzdistance order by 1) as asd
|
|
INTO columnNames;
|
|
EXECUTE format('CREATE OR REPLACE VIEW points_with_dist AS SELECT *
|
|
FROM CROSSTAB(
|
|
$$
|
|
SELECT placement_point_id, pvz_postamates_group_id, dist
|
|
FROM service_placementpointpvzdistance
|
|
ORDER BY 1, 2
|
|
$$
|
|
) AS ct(%s)
|
|
LEFT JOIN service_placementpoint ON placement_point_id=id'
|
|
,columnNames);
|
|
END;
|
|
$BODY$
|
|
LANGUAGE plpgsql;
|
|
CALL public.pivot_dist();"""
|
|
|
|
CMD_PIVOT_DIST_PRE = """CREATE OR REPLACE procedure prepivot_dist()
|
|
AS $BODY$
|
|
DECLARE columnNames TEXT;
|
|
BEGIN
|
|
DROP VIEW IF EXISTS prepoints_with_dist;
|
|
SELECT 'placement_point_id bigint, ' || string_agg(c, ', ') FROM (SELECT distinct pvz_postamates_group_id, 'd' || pvz_postamates_group_id || ' double precision' as c from service_preplacementpointpvzdistance order by 1) as asd
|
|
INTO columnNames;
|
|
EXECUTE format('CREATE OR REPLACE VIEW prepoints_with_dist AS SELECT *
|
|
FROM CROSSTAB(
|
|
$$
|
|
SELECT placement_point_id, pvz_postamates_group_id, dist
|
|
FROM service_preplacementpointpvzdistance
|
|
ORDER BY 1, 2
|
|
$$
|
|
) AS ct(%s)
|
|
LEFT JOIN service_preplacementpoint ON placement_point_id=id'
|
|
,columnNames);
|
|
END;
|
|
$BODY$
|
|
LANGUAGE plpgsql;
|
|
CALL public.prepivot_dist();"""
|
|
class Command(BaseCommand):
|
|
help = 'Create procedures'
|
|
|
|
def handle(self, *args, **kwargs):
|
|
try:
|
|
log_to_telegram('Creating views')
|
|
run_sql_command(CMD_PIVOT_DIST)
|
|
log_to_telegram('pivot_dist created')
|
|
run_sql_command(CMD_PIVOT_DIST_PRE)
|
|
log_to_telegram('prepivot_dist created')
|
|
except Exception as e:
|
|
log_to_telegram('Error creating views: ' + str(e))
|