]> git.ipfire.org Git - thirdparty/psycopg.git/commitdiff
Fix row factory docs examples
authorDaniele Varrazzo <daniele.varrazzo@gmail.com>
Wed, 24 Feb 2021 02:09:26 +0000 (03:09 +0100)
committerDaniele Varrazzo <daniele.varrazzo@gmail.com>
Wed, 24 Feb 2021 02:09:26 +0000 (03:09 +0100)
docs/row-factories.rst

index ec504bcd984ea2fe6f675699ab06d52e25505983..c404c2fb81356c0b6ce61cd7211594ab6f98c21e 100644 (file)
@@ -18,19 +18,19 @@ This can be implemented as a class, for instance:
 
    class DictRowFactory:
        def __init__(self, cursor):
-           self.cursor = cursor
+           self.fields = [c.name for c in cursor.description]
 
        def __call__(self, values):
-           fields = (c.name for c in self.cursor.description)
-           return dict(zip(fields, values))
+           return dict(zip(self.fields, values))
 
 or as a plain function:
 
 .. code:: python
 
    def dict_row_factory(cursor):
+       fields = [c.name for c in cursor.description]
+
        def make_row(values):
-           fields = (c.name for c in cursor.description)
            return dict(zip(fields, values))
 
        return make_row