datetime_safe.py 2.7 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798
  1. # Python's datetime strftime doesn't handle dates before 1900.
  2. # These classes override date and datetime to support the formatting of a date
  3. # through its full "proleptic Gregorian" date range.
  4. #
  5. # Based on code submitted to comp.lang.python by Andrew Dalke
  6. #
  7. # >>> datetime_safe.date(1850, 8, 2).strftime("%Y/%m/%d was a %A")
  8. # '1850/08/02 was a Friday'
  9. from datetime import date as real_date, datetime as real_datetime
  10. import re
  11. import time
  12. class date(real_date):
  13. def strftime(self, fmt):
  14. return strftime(self, fmt)
  15. class datetime(real_datetime):
  16. def strftime(self, fmt):
  17. return strftime(self, fmt)
  18. @classmethod
  19. def combine(cls, date, time):
  20. return cls(date.year, date.month, date.day,
  21. time.hour, time.minute, time.second,
  22. time.microsecond, time.tzinfo)
  23. def date(self):
  24. return date(self.year, self.month, self.day)
  25. def new_date(d):
  26. "Generate a safe date from a datetime.date object."
  27. return date(d.year, d.month, d.day)
  28. def new_datetime(d):
  29. """
  30. Generate a safe datetime from a datetime.date or datetime.datetime object.
  31. """
  32. kw = [d.year, d.month, d.day]
  33. if isinstance(d, real_datetime):
  34. kw.extend([d.hour, d.minute, d.second, d.microsecond, d.tzinfo])
  35. return datetime(*kw)
  36. # This library does not support strftime's "%s" or "%y" format strings.
  37. # Allowed if there's an even number of "%"s because they are escaped.
  38. _illegal_formatting = re.compile(r"((^|[^%])(%%)*%[sy])")
  39. def _findall(text, substr):
  40. # Also finds overlaps
  41. sites = []
  42. i = 0
  43. while 1:
  44. j = text.find(substr, i)
  45. if j == -1:
  46. break
  47. sites.append(j)
  48. i = j + 1
  49. return sites
  50. def strftime(dt, fmt):
  51. if dt.year >= 1900:
  52. return super(type(dt), dt).strftime(fmt)
  53. illegal_formatting = _illegal_formatting.search(fmt)
  54. if illegal_formatting:
  55. raise TypeError("strftime of dates before 1900 does not handle" + illegal_formatting.group(0))
  56. year = dt.year
  57. # For every non-leap year century, advance by
  58. # 6 years to get into the 28-year repeat cycle
  59. delta = 2000 - year
  60. off = 6 * (delta // 100 + delta // 400)
  61. year = year + off
  62. # Move to around the year 2000
  63. year = year + ((2000 - year) // 28) * 28
  64. timetuple = dt.timetuple()
  65. s1 = time.strftime(fmt, (year,) + timetuple[1:])
  66. sites1 = _findall(s1, str(year))
  67. s2 = time.strftime(fmt, (year + 28,) + timetuple[1:])
  68. sites2 = _findall(s2, str(year + 28))
  69. sites = []
  70. for site in sites1:
  71. if site in sites2:
  72. sites.append(site)
  73. s = s1
  74. syear = "%04d" % (dt.year,)
  75. for site in sites:
  76. s = s[:site] + syear + s[site + 4:]
  77. return s