создайте оболочку пользовательского класса db.connection, продолжайте получать "django.db.utils.InterfaceError: курсор уже закрыт" - PullRequest
Новая
       12

создайте оболочку пользовательского класса db.connection, продолжайте получать "django.db.utils.InterfaceError: курсор уже закрыт"

0 голосов
/ 31 марта 2019

я пытаюсь создать пользовательскую оболочку класса для "django.db.connection", но я продолжаю получать "django.db.utils.InterfaceError: курсор уже закрыт".

рабочий код выглядит примерно такесли без обертки пользовательского класса (скопировано из ./manage.py shell): 

>>> from django.db import (connection as con)
>>> with con.cursor() as q:
...     q.execute('select * from master.years a where a.years = %s', [str(2019)])
...     f = [f.name for f in q.description]
...     for b in q:
...             print(dict(zip(f,b)))

моя обертка webapp/mine/db.py: 

class query_00:
    def __init__(self, db_con, db_sql, db_stmt = []):
        self.db_con = db_con
        self.db_sql = db_sql
        self.fields = []

        with db_con.cursor() as self.query:
            self.query.execute(db_sql, db_stmt)
            self.fields = [f.name for f in self.query.description]

        # for entry in self.query:
            # yield dict(zip(self.fields, entry))

        # self.query = db_con.cursor()

    # must return self, because
    # AttributeError: 'NoneType' object has no attribute 'result'
    def __enter__(self):
        return self
        # self.query.execute(self.db_sql, self.db_stmt)
        # self.fields = [f.name for f in self.query.description]
        # pass

    def __exit__(self, exc_type, exc_value, traceback):
        # is this necessary?
        # self.db_con.close()
        pass

    def result(self):
        for entry in self.query.fetchall():
            yield dict(zip(self.fields, entry))

        # not working
        # for entry in self.query:
            # yield dict(zip(self.fields, entry))

        # not working
        # yield dict(zip(self.fields, entry))

        # pass

, тогда я пробую ./manage.py shell, набрав 

from django.db import (connection as con)
from mine.db import query_00

# expected usage, attempt 1
q = query_00(con, 'SELECT * FROM master.years a where a.year = %s', [str(2019),])
for b in q.result():
    print(b)

# expected usage, attempt 2
with query_00(con, 'SELECT * FROM master.years a where a.year = %s', [str(2019),]) as q:
    for b in q.result():
        print(b)

python-3, django-2, postgresql-9

(извините за мой английский)

1 Ответ

0 голосов
/ 01 апреля 2019

итак, прочитав документацию, я понимаю, что соединение было закрыто внутри моего метода init. 

class query_00:
    def __init__(self, db_con, db_sql, db_stmt = []):
        self.db_con = db_con
        self.db_sql = db_sql
        self.fields = []

        with db_con.cursor() as self.query:
            self.query.execute(db_sql, db_stmt)
            self.fields = [f.name for f in self.query.description]
        # the connection only happend inside "with:" block
        # the connection, was already closed here (after with:)

Итак, я просто должен держать курсор () на переменной 

class query_01:
    def __init__(self, db_sql, db_stmt):
        self.db_sql = db_sql
        self.db_stmt = db_stmt
        self.query = connection.cursor()
...

поставить query.execute() на другой метод 

...
    def execute(self):
        self.query.execute(self.db_sql, self.db_stmt)
        self.fields = [f.name for f in self.query.description]
...

в результате () 

...
    def result(self):
        self.execute()
        for entry in self.query:
            yield self.keyvalue(entry)
...

тогда мой класс работает 

with query_00('SELECT * FROM master.years a where a.year = %s', [str(2019),]) as q:
    for b in q.result():
        print(b)
...