diff options
Diffstat (limited to 'lib/extras.py')
| -rw-r--r-- | lib/extras.py | 43 |
1 files changed, 42 insertions, 1 deletions
diff --git a/lib/extras.py b/lib/extras.py index 2db28fd..91ab9ab 100644 --- a/lib/extras.py +++ b/lib/extras.py @@ -24,7 +24,8 @@ try: import logging except: logging = None - + +from psycopg2 import extensions as _ext from psycopg2.extensions import cursor as _cursor from psycopg2.extensions import connection as _connection from psycopg2.extensions import adapt as _A @@ -281,3 +282,43 @@ class MinTimeLoggingCursor(LoggingCursor): self.timestamp = time.time() return LoggingCursor.execute(self, procname, vars) + +# a dbtype and adapter for Python UUID type + +try: + import uuid + + class UUID_adapter(object): + """Adapt Python's uuid.UUID type to PostgreSQL's uuid.""" + + def __init__(self, uuid): + self._uuid = uuid + + def prepare(self, conn): + pass + + def getquoted(self): + return "'"+str(self._uuid)+"'::uuid" + + __str__ = getquoted + + def register_uuid(oid=None): + """Create the UUID type and an uuid.UUID adapter.""" + if not oid: oid = 2950 + _ext.UUID = _ext.new_type((oid, ), "UUID", + lambda data, cursor: uuid.UUID(data)) + _ext.register_type(_ext.UUID) + _ext.register_adapter(uuid.UUID, UUID_adapter) + return _ext.UUID + +except ImportError, e: + def register_uuid(oid=None): + """Create the UUID type and an uuid.UUID adapter. + + This is a fake function that will always raise an error because the + import of the uuid module failed. + """ + raise e + + +__all__ = [ k for k in locals().keys() if not k.startswith('_') ]
\ No newline at end of file |