[SQL-CVS] r135 - trunk/SQLObject

[<sub...@co...>]

Author: ahmedmo
Date: 2004-06-14 03:59:20 -0400 (Mon, 14 Jun 2004)
New Revision: 135

Added:
   trunk/SQLObject/converters.py
Log:


Added: trunk/SQLObject/converters.py
=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=
=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=
=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D
--- trunk/SQLObject/converters.py	2004-06-14 07:58:45 UTC (rev 134)
+++ trunk/SQLObject/converters.py	2004-06-14 07:59:20 UTC (rev 135)
@@ -0,0 +1,183 @@
+try:
+    import mx.DateTime.ISO
+    origISOStr =3D mx.DateTime.ISO.strGMT
+    from mx.DateTime import DateTimeType
+except ImportError:
+    try:
+        import DateTime.ISO
+        origISOStr =3D DateTime.ISO.strGMT
+        from DateTime import DateTimeType
+    except ImportError:
+        origISOStr =3D None
+        DateTimeType =3D None
+import time
+try:
+    import datetime
+except ImportError:
+    datetime =3D None
+
+try:
+    import Sybase
+    NumericType=3DSybase.NumericType
+except ImportError:
+    NumericType =3D None=20
+
+if type(1=3D=3D1) =3D=3D type(1):
+    class BOOL(object):
+        def __init__(self, value):
+            self.value =3D not not value
+        def __nonzero__(self):
+            return self.value
+        def __repr__(self):
+            if self:
+                return 'TRUE'
+            else:
+                return 'FALSE'
+    TRUE =3D BOOL(1)
+    FALSE =3D BOOL(0)
+else:
+    TRUE =3D 1=3D=3D1
+    FALSE =3D 0=3D=3D1
+
+from types import InstanceType, ClassType, TypeType
+
+########################################
+## Quoting
+########################################
+
+sqlStringReplace =3D [
+    ('\\', '\\\\'),
+    ("'", "''"),
+    ('\000', '\\0'),
+    ('\b', '\\b'),
+    ('\n', '\\n'),
+    ('\r', '\\r'),
+    ('\t', '\\t'),
+    ]
+
+def isoStr(val):
+    """
+    Gets rid of time zone information
+    (@@: should we convert to GMT?)
+    """
+    val =3D origISOStr(val)
+    if val.find('+') =3D=3D -1:
+        return val
+    else:
+        return val[:val.find('+')]
+
+class ConverterRegistry:
+
+    def __init__(self):
+        self.basic =3D {}
+        self.klass =3D {}
+
+    def registerConverter(self, typ, func):
+        if type(typ) is ClassType:
+            self.klass[typ] =3D func
+        else:
+            self.basic[typ] =3D func
+
+    def lookupConverter(self, value, default=3DNone):
+        if type(value) =3D=3D InstanceType:
+            # lookup on klasses dict
+            return self.klass.get(value.__class__, default)
+        return self.basic.get(type(value), default)
+
+converters =3D ConverterRegistry()
+registerConverter =3D converters.registerConverter
+lookupConverter =3D converters.lookupConverter
+
+def StringLikeConverter(value, db):
+    if db in ('mysql', 'postgres', 'sybase'):
+        for orig, repl in sqlStringReplace:
+            value =3D value.replace(orig, repl)
+    elif db in ('sqlite', 'firebird','maxdb'):
+        value =3D value.replace("'", "''")
+    else:
+        assert 0, "Database %s unknown" % db
+    return "'%s'" % value
+
+registerConverter(type(""), StringLikeConverter)
+registerConverter(type(u""), StringLikeConverter)
+
+def IntConverter(value, db):
+    return repr(int(value))
+
+registerConverter(type(1), IntConverter)
+registerConverter(type(0L), IntConverter)
+
+if NumericType:
+    registerConverter(NumericType, IntConverter)
+
+def BoolConverter(value, db):
+    if db in ('postgres',):
+        if value:
+            return "'t'"
+        else:
+            return "'f'"
+    else:
+        if value:
+            return '1'
+        else:
+            return '0'
+
+registerConverter(type(TRUE), BoolConverter)
+
+def FloatConverter(value, db):
+    return repr(value)
+
+registerConverter(type(1.0), FloatConverter)
+
+if DateTimeType:
+    def DateTimeConverter(value, db):
+        return "'%s'" % isoStr(value)
+
+    registerConverter(DateTimeType, DateTimeConverter)
+
+def NoneConverter(value, db):
+    return "NULL"
+
+registerConverter(type(None), NoneConverter)
+
+def SequenceConverter(value, db):
+    return "(%s)" % ", ".join([sqlrepr(v, db) for v in value])
+
+registerConverter(type(()), SequenceConverter)
+registerConverter(type([]), SequenceConverter)
+
+if hasattr(time, 'struct_time'):
+    def StructTimeConverter(value, db):
+        return time.strftime("'%Y-%m-%d %H:%M:%S'", value)
+
+    registerConverter(time.struct_time, StructTimeConverter)
+
+if datetime:
+    def DateTimeConverter(value, db):
+        return value.strftime("'%Y-%m-%d %H:%M:%s'")
+
+    registerConverter(datetime.datetime, DateTimeConverter)
+
+    def TimeConverter(value, db):
+        return value.strftime("'%H:%M:%s'")
+
+    registerConverter(datetime.time, TimeConverter)
+
+    def DateConverter(value, db):
+        return value.strftime("'%Y-%m-%d'")
+
+    registerConverter(datetime.date, DateConverter)
+
+def sqlrepr(obj, db=3DNone):
+    try:
+        reprFunc =3D obj.__sqlrepr__
+    except AttributeError:
+        converter =3D lookupConverter(obj)
+        if converter is None:
+            raise ValueError, "Unknown SQL builtin type: %s for %s" % \
+                  (type(obj), repr(obj))
+        return converter(obj, db)
+    else:
+        return reprFunc(db)
+
+