Add FuzzyDateTime/FuzzyNaiveDateTime.

1 files changed, 28 insertions, 0 deletions

diff --git a/factory/compat.py b/factory/compat.py
index 84f31b7..9594550 100644
--- a/factory/compat.py
+++ b/factory/compat.py
@@ -23,6 +23,7 @@
 
 """Compatibility tools"""
 
+import datetime
 import sys
 
 PY2 = (sys.version_info[0] == 2)
@@ -33,3 +34,30 @@ if PY2:
 else:
     def is_string(obj):
         return isinstance(obj, str)
+
+try:
+    # Python >= 3.2
+    UTC = datetime.timezone.utc
+except AttributeError:
+    try:
+        # Fallback to pytz
+        from pytz import UTC
+    except ImportError:
+
+        # Ok, let's write our own.
+        class _UTC(datetime.tzinfo):
+            """The UTC tzinfo."""
+
+            def utcoffset(self, dt):
+                return datetime.timedelta(0)
+
+            def tzname(self, dt):
+                return "UTC"
+
+            def dst(self, dt):
+                return datetime.timedelta(0)
+
+            def localize(self, dt):
+                dt.astimezone(self)
+
+        UTC = _UTC()