Changeset 580


Ignore:
Timestamp:
Nov 21, 2015, 6:40:19 PM (4 years ago)
Author:
cito
Message:

Return proper class and module attrs

We do not need to customize class attribute (anyway it was wrong
since it should return a class, not a class name).

Also, return the proper names of the PyGreSQL modules.

Location:
trunk/module
Files:
4 edited

Legend:

Unmodified
Added
Removed
  • trunk/module/TEST_PyGreSQL_classic_connection.py

    r576 r580  
    7676        except pg.InternalError:
    7777            pass
     78
     79    def testClassName(self):
     80        self.assertEqual(self.connection.__class__.__name__, 'pgconnobject')
     81
     82    def testModuleName(self):
     83        self.assertEqual(self.connection.__module__, 'pg')
    7884
    7985    def testAllConnectAttributes(self):
  • trunk/module/TEST_PyGreSQL_classic_dbwrapper.py

    r571 r580  
    293293    def tearDown(self):
    294294        self.db.close()
     295
     296    def testClassName(self):
     297        self.assertEqual(self.db.__class__.__name__, 'DB')
     298
     299    def testModuleName(self):
     300        self.assertEqual(self.db.__module__, 'pg')
    295301
    296302    def testEscapeLiteral(self):
  • trunk/module/TEST_PyGreSQL_classic_largeobj.py

    r559 r580  
    158158        self.pgcnx.query('end')
    159159        self.pgcnx.close()
     160
     161    def testClassName(self):
     162        self.assertEqual(self.obj.__class__.__name__, 'pglarge')
     163
     164    def testModuleName(self):
     165        self.assertEqual(self.obj.__module__, 'pg')
    160166
    161167    def testOid(self):
  • trunk/module/pgmodule.c

    r579 r580  
    5555
    5656/* default values */
    57 #define MODULE_NAME                     "pgsql"
     57#define MODULE_NAME                     "pg"
     58#define MODULE_NAME_DB          "pgdb"
    5859#define PG_ARRAYSIZE                    1
    5960
     
    963964        if (!strcmp(name, "__module__"))
    964965                return PyStr_FromString(MODULE_NAME);
    965 
    966         /* class name */
    967         if (!strcmp(name, "__class__"))
    968                 return PyStr_FromString("pglarge");
    969966
    970967        /* seeks name in methods (fallback) */
     
    22572254#endif
    22582255
     2256        /* module name */
     2257        if (!strcmp(name, "__module__"))
     2258                return PyStr_FromString(MODULE_NAME);
     2259
    22592260        return PyObject_GenericGetAttr((PyObject *) self, nameobj);
    22602261}
     
    28702871        /* module name */
    28712872        if (!strcmp(name, "__module__"))
    2872                 return PyStr_FromString(MODULE_NAME);
    2873 
    2874         /* class name */
    2875         if (!strcmp(name, "__class__"))
    2876                 return PyStr_FromString("pgsource");
     2873                return PyStr_FromString(MODULE_NAME_DB);
    28772874
    28782875        /* seeks name in methods (fallback) */
     
    35353532        }
    35363533
     3534        /* module name */
     3535        if (!strcmp(name, "__module__"))
     3536                return PyStr_FromString(MODULE_NAME);
     3537
    35373538        return PyObject_GenericGetAttr((PyObject *) self, nameobj);
    35383539}
     
    42374238#endif
    42384239
    4239         if (PyType_Ready(&connType) ||
    4240                 PyType_Ready(&noticeType) ||
    4241                 PyType_Ready(&queryType) ||
    4242                 PyType_Ready(&sourceType)) return NULL;
     4240        if (PyType_Ready(&connType)
     4241                || PyType_Ready(&noticeType)
     4242                || PyType_Ready(&queryType)
     4243                || PyType_Ready(&sourceType)
     4244#ifdef LARGE_OBJECTS
     4245                || PyType_Ready(largeType.ob_type)
     4246#endif
     4247                ) return NULL;
    42434248
    42444249        dict = PyModule_GetDict(mod);
Note: See TracChangeset for help on using the changeset viewer.