]> git.ipfire.org Git - thirdparty/psycopg.git/commitdiff
Use static attributes to declare dumpers oid where possible
authorDaniele Varrazzo <daniele.varrazzo@gmail.com>
Thu, 29 Oct 2020 21:00:02 +0000 (22:00 +0100)
committerDaniele Varrazzo <daniele.varrazzo@gmail.com>
Thu, 29 Oct 2020 21:38:24 +0000 (22:38 +0100)
psycopg3/psycopg3/types/date.py
psycopg3/psycopg3/types/text.py

index 05b73039d87c0b69bef638f2b8b1240432c4fbe3..eadc031135da25ec42c765cd082d93ce42e18efa 100644 (file)
@@ -22,49 +22,38 @@ _decode_ascii = codecs.lookup("ascii").decode
 @Dumper.text(date)
 class DateDumper(Dumper):
 
-    DATE_OID = builtins["date"].oid
+    oid = builtins["date"].oid
 
     def dump(self, obj: date, __encode: EncodeFunc = _encode_ascii) -> bytes:
         # NOTE: whatever the PostgreSQL DateStyle input format (DMY, MDY, YMD)
         # the YYYY-MM-DD is always understood correctly.
         return __encode(str(obj))[0]
 
-    @property
-    def oid(self) -> int:
-        return self.DATE_OID
-
 
 @Dumper.text(time)
 class TimeDumper(Dumper):
 
-    TIMETZ_OID = builtins["timetz"].oid
+    oid = builtins["timetz"].oid
 
     def dump(self, obj: time, __encode: EncodeFunc = _encode_ascii) -> bytes:
         return __encode(str(obj))[0]
 
-    @property
-    def oid(self) -> int:
-        return self.TIMETZ_OID
-
 
 @Dumper.text(datetime)
 class DateTimeDumper(Dumper):
 
-    TIMESTAMPTZ_OID = builtins["timestamptz"].oid
+    oid = builtins["timestamptz"].oid
 
     def dump(self, obj: date, __encode: EncodeFunc = _encode_ascii) -> bytes:
         # NOTE: whatever the PostgreSQL DateStyle input format (DMY, MDY, YMD)
         # the YYYY-MM-DD is always understood correctly.
         return __encode(str(obj))[0]
 
-    @property
-    def oid(self) -> int:
-        return self.TIMESTAMPTZ_OID
-
 
 @Dumper.text(timedelta)
 class TimeDeltaDumper(Dumper):
-    INTERVAL_OID = builtins["interval"].oid
+
+    oid = builtins["interval"].oid
 
     def __init__(self, src: type, context: AdaptContext = None):
         super().__init__(src, context)
@@ -89,10 +78,6 @@ class TimeDeltaDumper(Dumper):
             obj.microseconds,
         )
 
-    @property
-    def oid(self) -> int:
-        return self.INTERVAL_OID
-
 
 @Loader.text(builtins["date"].oid)
 class DateLoader(Loader):
index 90123e8088b2ef9b87184d37b2ba5d089beb11e8..19608579858542d5d1c825890c4e447150b4b35b 100644 (file)
@@ -16,10 +16,6 @@ if TYPE_CHECKING:
     from ..pq.proto import Escaping as EscapingProto
 
 
-TEXT_OID = builtins["text"].oid
-BYTEA_OID = builtins["bytea"].oid
-
-
 @Dumper.text(str)
 @Dumper.binary(str)
 class StringDumper(Dumper):
@@ -87,6 +83,9 @@ class UnknownLoader(Loader):
 
 @Dumper.text(bytes)
 class BytesDumper(Dumper):
+
+    oid = builtins["bytea"].oid
+
     def __init__(self, src: type, context: AdaptContext = None):
         super().__init__(src, context)
         self.esc = Escaping(
@@ -96,20 +95,15 @@ class BytesDumper(Dumper):
     def dump(self, obj: bytes) -> bytes:
         return self.esc.escape_bytea(obj)
 
-    @property
-    def oid(self) -> int:
-        return BYTEA_OID
-
 
 @Dumper.binary(bytes)
 class BytesBinaryDumper(Dumper):
+
+    oid = builtins["bytea"].oid
+
     def dump(self, b: bytes) -> bytes:
         return b
 
-    @property
-    def oid(self) -> int:
-        return BYTEA_OID
-
 
 @Loader.text(builtins["bytea"].oid)
 class ByteaLoader(Loader):