Как вызвать postgresql функцию из django с передачей параметров?

Question

[ellz]

Есть такая функция в views.py:

def getNearestNotes(request, longitude, latitude):
	if request.method == 'GET':
		#вызов функции
		return HttpResponse('')
	else:
		return HttpResponse('needGetMethod')

Как вызвать хранимую процедуру(функцию) из базы данных postgersql, с передачей в нее параметров, и последующим внесения ответа в объект? БД подключил. Все работает.

Искал решения в интернете, но там мне кажется совсем не то, то нужно.

Answer 1

[Сергей Горностаев]

from django.db import connection

some_arg = 42

c = connection.cursor()
try:
    c.callproc('some_stored_procedure', (some_arg,))
    r = c.fetchone()
    ...
finally:
    c.close()

Comments

[ellz]

def getNearestNotes(request, longitude, latitude):
	if request.method == 'GET':
		c = connection.cursor()
		r = None
		try:
			c.callproc('GetAllNotes', (longitude, latitude))
			r = c.fetchone()
		finally:
			c.close()
		return HttpResponse(str(r))
	else:
		return HttpResponse('needGetMethod')

Выдает

ОШИБКА: функция getallnotes(unknown, unknown) не существует
LINE 1: SELECT * FROM GetAllNotes('28.7','23.2')
^
HINT: Функция с данными именем и типами аргументов не найдена. Возможно, вам следует добавить явные приведения типов.

Функция выглядит так:

create function "GetAllNotes"(long double precision, lat double precision) returns TABLE(userid integer, username character varying, notename character varying, notelong double precision, notelat double precision)
  language plpgsql
as
$$
BEGIN
RETURN query  (SELECT notes."userid", users."name", notes."name", notes."longitude", notes."latitude" FROM notes INNER JOIN users ON notes."userid" = users."id" WHERE  (point(long, lat) <@> point(notes."longitude", notes."latitude") <= 0.124274) );
END
$$;

[Сергей Горностаев]

ellz, попробуйте привести аргументы к числу с плавающей точкой

c.callproc('GetAllNotes', (float(longitude), float(latitude)))

[ellz]

Сергей Горностаев, к сожалению тоже самое.

[Сергей Горностаев]

ellz, это странно. Тогда придётся закостылить

c.execute('select * from GetAllNotes(%s::double, %s::double)', (longitude, latitude))

[ellz]

Сергей Горностаев,

ОШИБКА: тип "double" не существует
LINE 1: select * from GetAllNotes('28.7'::double, '23.2'::double)

Мне кажется, что в самой БД/процедуре проблема, раз столько способов не работает. Буду в эту сторону копать.

[Сергей Горностаев]

ellz, поиграйте с типами. Попробуйте написать double precision после двоеточия.

[ellz]

Сергей Горностаев, тоже самое. Решил посмотреть какие django функции вообще видит таким способом:

def getNearestNotes(request, longitude, latitude):
	if request.method == 'GET':
		c = connection.cursor()
		r = None
		try:
			#c.callproc('GetAllNotes', (float(longitude), float(latitude)))
			c.execute("SELECT routine_name FROM information_schema.routines WHERE routine_type='FUNCTION' AND specific_schema='public'")
			r = c.fetchone()
		finally:
			c.close()
		return HttpResponse(str(r))
	else:
		return HttpResponse('needGetMethod')

и вот, что он выдал -

('earth_distance',)

, хотя у меня их больше 20 точно. Но моя одна. Все остальные (и earth_distance) создались, когда extentions добавил.

[Сергей Горностаев]

ellz, во-первых, метод fetchone возвращает только один результат, как следует из его названия. Надо заменить на fetchall, если надо много. Во-вторых, ошибки возникают из-за того, что не совпадают типы передаваемых параметром с ожидаемыми.

[ellz]

Сергей Горностаев,

c.callproc('"GetAllNotes"', (float(longitude), float(latitude)))

- вот так работает. Получаю вот такой результат - "[(1, 'Max', 'АВ', Decimal('55.8694248'), Decimal('37.6647453')), (2, 'Nick', 'afsd', Decimal('55.8691788'), Decimal('37.6650843')), (3, 'Jason', 'выаф', Decimal('55.8697598'), Decimal('37.6647674'))]". Вы случайно не знаете как конвертировать это в json. Пытаюсь вот таким способом:
r = json.load(c.fetchall())
и получаю ошибку -

'list' object has no attribute 'read'

[Сергей Горностаев]

ellz,

import json
from decimal import Decimal

def to_json(obj):
    if isinstance(obj, Decimal):
        return float(obj)
    return str(obj)

filed_names = ['a', 'b', 'c', 'd', 'e']
data = [(1, 'Max', 'АВ', Decimal('55.8694248'), Decimal('37.6647453')), (2, 'Nick', 'afsd', Decimal('55.8691788'), Decimal('37.6650843')), (3, 'Jason', 'выаф', Decimal('55.8697598'), Decimal('37.6647674'))]

json_str = json.dumps([dict(zip(field_names, i)) for i in data)], ensure_ascii=False, default=to_json)

[ellz]

Сергей Горностаев, Спасибо

[ellz]

Сергей Горностаев,

c.callproc('GetAllNotes', (float(longitude), float(latitude)))

- преобразует переменные во float. А есть какой ни будь аналог, чтобы в varchar(postgres тип данных) конвертировать?
А то опять получаю ошибку -

HINT: No function matches the given name and argument types. You might need to add explicit type casts.

Сделал вот такой запрос, чтобы определить тип переменных которые я передаю:

SELECT pg_typeof('https://i.imgur.com/DPalXFO.jpg'); --unknown
SELECT pg_typeof('https://i.imgur.com/DPalXFO.jpg'::varchar); --character varying

получается нужно конвертировать данные в самой postgres функции? Но ведь этого нельзя сделать на этапе передачи параметров(add_imgae(image_url character varying, user_password character varying)). Пробовал еще так, чтобы по аналогии с float(var) было:

c.callproc('add_image', (str(uploadet_link), str(request.user.password)))

, но не помогло. Или как гулить такое? По запросу

No function matches the given name and argument types. You might need to add explicit type casts django

ничего не нашел.
Еще так пробовал:

c.execute('select * from add_imgae(%s::varchar, %s::varchar)', (uploadet_link, request.user.password))

, как вы раньше писали, но не помогло.

[Сергей Горностаев]

ellz, это странно, так как python-тип str свободно конвертируется в postgresql-тип varchar.