Guido van Rossum | 00efe7e | 2002-07-19 17:04:46 +0000 | [diff] [blame] | 1 | """Strptime-related classes and functions. |
| 2 | |
| 3 | CLASSES: |
Brett Cannon | 474335c | 2003-08-05 04:02:49 +0000 | [diff] [blame] | 4 | LocaleTime -- Discovers and stores locale-specific time information |
Barry Warsaw | 4d895fa | 2002-09-23 22:46:49 +0000 | [diff] [blame] | 5 | TimeRE -- Creates regexes for pattern matching a string of text containing |
Brett Cannon | 474335c | 2003-08-05 04:02:49 +0000 | [diff] [blame] | 6 | time information |
Guido van Rossum | 00efe7e | 2002-07-19 17:04:46 +0000 | [diff] [blame] | 7 | |
| 8 | FUNCTIONS: |
Raymond Hettinger | 1fdb633 | 2003-03-09 07:44:42 +0000 | [diff] [blame] | 9 | _getlang -- Figure out what language is being used for the locale |
Guido van Rossum | 00efe7e | 2002-07-19 17:04:46 +0000 | [diff] [blame] | 10 | strptime -- Calculates the time struct represented by the passed-in string |
| 11 | |
Guido van Rossum | 00efe7e | 2002-07-19 17:04:46 +0000 | [diff] [blame] | 12 | """ |
| 13 | import time |
| 14 | import locale |
| 15 | import calendar |
| 16 | from re import compile as re_compile |
| 17 | from re import IGNORECASE |
Raymond Hettinger | 1fdb633 | 2003-03-09 07:44:42 +0000 | [diff] [blame] | 18 | from datetime import date as datetime_date |
Brett Cannon | 474335c | 2003-08-05 04:02:49 +0000 | [diff] [blame] | 19 | try: |
| 20 | from thread import allocate_lock as _thread_allocate_lock |
| 21 | except: |
| 22 | from dummy_thread import allocate_lock as _thread_allocate_lock |
Guido van Rossum | 00efe7e | 2002-07-19 17:04:46 +0000 | [diff] [blame] | 23 | |
Guido van Rossum | 00efe7e | 2002-07-19 17:04:46 +0000 | [diff] [blame] | 24 | __author__ = "Brett Cannon" |
Raymond Hettinger | 1fdb633 | 2003-03-09 07:44:42 +0000 | [diff] [blame] | 25 | __email__ = "brett@python.org" |
Guido van Rossum | 00efe7e | 2002-07-19 17:04:46 +0000 | [diff] [blame] | 26 | |
| 27 | __all__ = ['strptime'] |
| 28 | |
Tim Peters | 80cebc1 | 2003-01-19 04:40:44 +0000 | [diff] [blame] | 29 | def _getlang(): |
| 30 | # Figure out what the current language is set to. |
Brett Cannon | 175ddb5 | 2003-07-24 06:27:17 +0000 | [diff] [blame] | 31 | return locale.getlocale(locale.LC_TIME) |
Barry Warsaw | 35816e6 | 2002-08-29 16:24:50 +0000 | [diff] [blame] | 32 | |
Guido van Rossum | 00efe7e | 2002-07-19 17:04:46 +0000 | [diff] [blame] | 33 | class LocaleTime(object): |
| 34 | """Stores and handles locale-specific information related to time. |
| 35 | |
Brett Cannon | 474335c | 2003-08-05 04:02:49 +0000 | [diff] [blame] | 36 | ATTRIBUTES: |
Guido van Rossum | 00efe7e | 2002-07-19 17:04:46 +0000 | [diff] [blame] | 37 | f_weekday -- full weekday names (7-item list) |
| 38 | a_weekday -- abbreviated weekday names (7-item list) |
Brett Cannon | f5c96fb | 2003-08-08 01:53:05 +0000 | [diff] [blame] | 39 | f_month -- full month names (13-item list; dummy value in [0], which |
Guido van Rossum | 00efe7e | 2002-07-19 17:04:46 +0000 | [diff] [blame] | 40 | is added by code) |
Brett Cannon | f5c96fb | 2003-08-08 01:53:05 +0000 | [diff] [blame] | 41 | a_month -- abbreviated month names (13-item list, dummy value in |
Guido van Rossum | 00efe7e | 2002-07-19 17:04:46 +0000 | [diff] [blame] | 42 | [0], which is added by code) |
| 43 | am_pm -- AM/PM representation (2-item list) |
| 44 | LC_date_time -- format string for date/time representation (string) |
| 45 | LC_date -- format string for date representation (string) |
| 46 | LC_time -- format string for time representation (string) |
Tim Peters | 469cdad | 2002-08-08 20:19:19 +0000 | [diff] [blame] | 47 | timezone -- daylight- and non-daylight-savings timezone representation |
Brett Cannon | 474335c | 2003-08-05 04:02:49 +0000 | [diff] [blame] | 48 | (2-item list of sets) |
| 49 | lang -- Language used by instance (2-item tuple) |
Guido van Rossum | 00efe7e | 2002-07-19 17:04:46 +0000 | [diff] [blame] | 50 | """ |
| 51 | |
Brett Cannon | 474335c | 2003-08-05 04:02:49 +0000 | [diff] [blame] | 52 | def __init__(self): |
| 53 | """Set all attributes. |
Raymond Hettinger | 6b59f5f | 2003-10-16 05:53:16 +0000 | [diff] [blame] | 54 | |
Brett Cannon | 474335c | 2003-08-05 04:02:49 +0000 | [diff] [blame] | 55 | Order of methods called matters for dependency reasons. |
| 56 | |
| 57 | The locale language is set at the offset and then checked again before |
| 58 | exiting. This is to make sure that the attributes were not set with a |
| 59 | mix of information from more than one locale. This would most likely |
| 60 | happen when using threads where one thread calls a locale-dependent |
| 61 | function while another thread changes the locale while the function in |
| 62 | the other thread is still running. Proper coding would call for |
| 63 | locks to prevent changing the locale while locale-dependent code is |
| 64 | running. The check here is done in case someone does not think about |
| 65 | doing this. |
Brett Cannon | 5187a3b | 2003-08-11 07:24:05 +0000 | [diff] [blame] | 66 | |
| 67 | Only other possible issue is if someone changed the timezone and did |
| 68 | not call tz.tzset . That is an issue for the programmer, though, |
| 69 | since changing the timezone is worthless without that call. |
Raymond Hettinger | 6b59f5f | 2003-10-16 05:53:16 +0000 | [diff] [blame] | 70 | |
Brett Cannon | 474335c | 2003-08-05 04:02:49 +0000 | [diff] [blame] | 71 | """ |
| 72 | self.lang = _getlang() |
| 73 | self.__calc_weekday() |
| 74 | self.__calc_month() |
| 75 | self.__calc_am_pm() |
| 76 | self.__calc_timezone() |
| 77 | self.__calc_date_time() |
| 78 | if _getlang() != self.lang: |
| 79 | raise ValueError("locale changed during initialization") |
Guido van Rossum | 00efe7e | 2002-07-19 17:04:46 +0000 | [diff] [blame] | 80 | |
| 81 | def __pad(self, seq, front): |
Brett Cannon | 474335c | 2003-08-05 04:02:49 +0000 | [diff] [blame] | 82 | # Add '' to seq to either the front (is True), else the back. |
Guido van Rossum | 00efe7e | 2002-07-19 17:04:46 +0000 | [diff] [blame] | 83 | seq = list(seq) |
Barry Warsaw | 35816e6 | 2002-08-29 16:24:50 +0000 | [diff] [blame] | 84 | if front: |
| 85 | seq.insert(0, '') |
| 86 | else: |
| 87 | seq.append('') |
Guido van Rossum | 00efe7e | 2002-07-19 17:04:46 +0000 | [diff] [blame] | 88 | return seq |
| 89 | |
Guido van Rossum | 00efe7e | 2002-07-19 17:04:46 +0000 | [diff] [blame] | 90 | def __calc_weekday(self): |
Brett Cannon | 474335c | 2003-08-05 04:02:49 +0000 | [diff] [blame] | 91 | # Set self.a_weekday and self.f_weekday using the calendar |
Barry Warsaw | 35816e6 | 2002-08-29 16:24:50 +0000 | [diff] [blame] | 92 | # module. |
Brett Cannon | 474335c | 2003-08-05 04:02:49 +0000 | [diff] [blame] | 93 | a_weekday = [calendar.day_abbr[i].lower() for i in range(7)] |
| 94 | f_weekday = [calendar.day_name[i].lower() for i in range(7)] |
| 95 | self.a_weekday = a_weekday |
| 96 | self.f_weekday = f_weekday |
Tim Peters | 469cdad | 2002-08-08 20:19:19 +0000 | [diff] [blame] | 97 | |
Guido van Rossum | 00efe7e | 2002-07-19 17:04:46 +0000 | [diff] [blame] | 98 | def __calc_month(self): |
Brett Cannon | 474335c | 2003-08-05 04:02:49 +0000 | [diff] [blame] | 99 | # Set self.f_month and self.a_month using the calendar module. |
| 100 | a_month = [calendar.month_abbr[i].lower() for i in range(13)] |
| 101 | f_month = [calendar.month_name[i].lower() for i in range(13)] |
| 102 | self.a_month = a_month |
| 103 | self.f_month = f_month |
Guido van Rossum | 00efe7e | 2002-07-19 17:04:46 +0000 | [diff] [blame] | 104 | |
| 105 | def __calc_am_pm(self): |
Brett Cannon | 474335c | 2003-08-05 04:02:49 +0000 | [diff] [blame] | 106 | # Set self.am_pm by using time.strftime(). |
Tim Peters | 469cdad | 2002-08-08 20:19:19 +0000 | [diff] [blame] | 107 | |
Barry Warsaw | 35816e6 | 2002-08-29 16:24:50 +0000 | [diff] [blame] | 108 | # The magic date (1999,3,17,hour,44,55,2,76,0) is not really that |
| 109 | # magical; just happened to have used it everywhere else where a |
| 110 | # static date was needed. |
Guido van Rossum | 00efe7e | 2002-07-19 17:04:46 +0000 | [diff] [blame] | 111 | am_pm = [] |
| 112 | for hour in (01,22): |
| 113 | time_tuple = time.struct_time((1999,3,17,hour,44,55,2,76,0)) |
Brett Cannon | 474335c | 2003-08-05 04:02:49 +0000 | [diff] [blame] | 114 | am_pm.append(time.strftime("%p", time_tuple).lower()) |
| 115 | self.am_pm = am_pm |
Guido van Rossum | 00efe7e | 2002-07-19 17:04:46 +0000 | [diff] [blame] | 116 | |
| 117 | def __calc_date_time(self): |
Brett Cannon | 474335c | 2003-08-05 04:02:49 +0000 | [diff] [blame] | 118 | # Set self.date_time, self.date, & self.time by using |
Barry Warsaw | 35816e6 | 2002-08-29 16:24:50 +0000 | [diff] [blame] | 119 | # time.strftime(). |
Tim Peters | 469cdad | 2002-08-08 20:19:19 +0000 | [diff] [blame] | 120 | |
Barry Warsaw | 35816e6 | 2002-08-29 16:24:50 +0000 | [diff] [blame] | 121 | # Use (1999,3,17,22,44,55,2,76,0) for magic date because the amount of |
| 122 | # overloaded numbers is minimized. The order in which searches for |
| 123 | # values within the format string is very important; it eliminates |
| 124 | # possible ambiguity for what something represents. |
Guido van Rossum | 00efe7e | 2002-07-19 17:04:46 +0000 | [diff] [blame] | 125 | time_tuple = time.struct_time((1999,3,17,22,44,55,2,76,0)) |
| 126 | date_time = [None, None, None] |
Brett Cannon | 474335c | 2003-08-05 04:02:49 +0000 | [diff] [blame] | 127 | date_time[0] = time.strftime("%c", time_tuple).lower() |
| 128 | date_time[1] = time.strftime("%x", time_tuple).lower() |
| 129 | date_time[2] = time.strftime("%X", time_tuple).lower() |
| 130 | replacement_pairs = [('%', '%%'), (self.f_weekday[2], '%A'), |
Barry Warsaw | 4d895fa | 2002-09-23 22:46:49 +0000 | [diff] [blame] | 131 | (self.f_month[3], '%B'), (self.a_weekday[2], '%a'), |
| 132 | (self.a_month[3], '%b'), (self.am_pm[1], '%p'), |
Barry Warsaw | 4d895fa | 2002-09-23 22:46:49 +0000 | [diff] [blame] | 133 | ('1999', '%Y'), ('99', '%y'), ('22', '%H'), |
| 134 | ('44', '%M'), ('55', '%S'), ('76', '%j'), |
| 135 | ('17', '%d'), ('03', '%m'), ('3', '%m'), |
| 136 | # '3' needed for when no leading zero. |
Brett Cannon | 474335c | 2003-08-05 04:02:49 +0000 | [diff] [blame] | 137 | ('2', '%w'), ('10', '%I')] |
| 138 | replacement_pairs.extend([(tz, "%Z") for tz_values in self.timezone |
| 139 | for tz in tz_values]) |
| 140 | for offset,directive in ((0,'%c'), (1,'%x'), (2,'%X')): |
| 141 | current_format = date_time[offset] |
| 142 | for old, new in replacement_pairs: |
Jack Jansen | 62fe755 | 2003-01-15 22:59:39 +0000 | [diff] [blame] | 143 | # Must deal with possible lack of locale info |
| 144 | # manifesting itself as the empty string (e.g., Swedish's |
| 145 | # lack of AM/PM info) or a platform returning a tuple of empty |
| 146 | # strings (e.g., MacOS 9 having timezone as ('','')). |
| 147 | if old: |
Barry Warsaw | 4d895fa | 2002-09-23 22:46:49 +0000 | [diff] [blame] | 148 | current_format = current_format.replace(old, new) |
Guido van Rossum | 00efe7e | 2002-07-19 17:04:46 +0000 | [diff] [blame] | 149 | time_tuple = time.struct_time((1999,1,3,1,1,1,6,3,0)) |
| 150 | if time.strftime(directive, time_tuple).find('00'): |
| 151 | U_W = '%U' |
| 152 | else: |
| 153 | U_W = '%W' |
| 154 | date_time[offset] = current_format.replace('11', U_W) |
Brett Cannon | 474335c | 2003-08-05 04:02:49 +0000 | [diff] [blame] | 155 | self.LC_date_time = date_time[0] |
| 156 | self.LC_date = date_time[1] |
Raymond Hettinger | 6b59f5f | 2003-10-16 05:53:16 +0000 | [diff] [blame] | 157 | self.LC_time = date_time[2] |
Guido van Rossum | 00efe7e | 2002-07-19 17:04:46 +0000 | [diff] [blame] | 158 | |
| 159 | def __calc_timezone(self): |
Brett Cannon | 474335c | 2003-08-05 04:02:49 +0000 | [diff] [blame] | 160 | # Set self.timezone by using time.tzname. |
Brett Cannon | 5187a3b | 2003-08-11 07:24:05 +0000 | [diff] [blame] | 161 | # Do not worry about possibility of time.tzname[0] == timetzname[1] |
| 162 | # and time.daylight; handle that in strptime . |
Brett Cannon | abe8eb0 | 2003-05-13 20:28:15 +0000 | [diff] [blame] | 163 | try: |
| 164 | time.tzset() |
| 165 | except AttributeError: |
| 166 | pass |
Raymond Hettinger | a690a99 | 2003-11-16 16:17:49 +0000 | [diff] [blame] | 167 | no_saving = frozenset(["utc", "gmt", time.tzname[0].lower()]) |
Brett Cannon | 172d9ef | 2003-05-11 06:23:36 +0000 | [diff] [blame] | 168 | if time.daylight: |
Raymond Hettinger | a690a99 | 2003-11-16 16:17:49 +0000 | [diff] [blame] | 169 | has_saving = frozenset([time.tzname[1].lower()]) |
Brett Cannon | 172d9ef | 2003-05-11 06:23:36 +0000 | [diff] [blame] | 170 | else: |
Raymond Hettinger | a690a99 | 2003-11-16 16:17:49 +0000 | [diff] [blame] | 171 | has_saving = frozenset() |
Brett Cannon | 474335c | 2003-08-05 04:02:49 +0000 | [diff] [blame] | 172 | self.timezone = (no_saving, has_saving) |
Guido van Rossum | 00efe7e | 2002-07-19 17:04:46 +0000 | [diff] [blame] | 173 | |
Guido van Rossum | 00efe7e | 2002-07-19 17:04:46 +0000 | [diff] [blame] | 174 | |
| 175 | class TimeRE(dict): |
| 176 | """Handle conversion from format directives to regexes.""" |
| 177 | |
Brett Cannon | 2c24d42 | 2003-07-24 20:02:28 +0000 | [diff] [blame] | 178 | def __init__(self, locale_time=None): |
Brett Cannon | 474335c | 2003-08-05 04:02:49 +0000 | [diff] [blame] | 179 | """Create keys/values. |
Raymond Hettinger | 6b59f5f | 2003-10-16 05:53:16 +0000 | [diff] [blame] | 180 | |
Brett Cannon | 474335c | 2003-08-05 04:02:49 +0000 | [diff] [blame] | 181 | Order of execution is important for dependency reasons. |
Raymond Hettinger | 6b59f5f | 2003-10-16 05:53:16 +0000 | [diff] [blame] | 182 | |
Brett Cannon | 474335c | 2003-08-05 04:02:49 +0000 | [diff] [blame] | 183 | """ |
| 184 | if locale_time: |
| 185 | self.locale_time = locale_time |
| 186 | else: |
| 187 | self.locale_time = LocaleTime() |
Neal Norwitz | 5efc50d | 2002-12-30 22:23:12 +0000 | [diff] [blame] | 188 | base = super(TimeRE, self) |
| 189 | base.__init__({ |
Brett Cannon | 474335c | 2003-08-05 04:02:49 +0000 | [diff] [blame] | 190 | # The " \d" part of the regex is to make %c from ANSI C work |
Neal Norwitz | 5efc50d | 2002-12-30 22:23:12 +0000 | [diff] [blame] | 191 | 'd': r"(?P<d>3[0-1]|[1-2]\d|0[1-9]|[1-9]| [1-9])", |
Guido van Rossum | 00efe7e | 2002-07-19 17:04:46 +0000 | [diff] [blame] | 192 | 'H': r"(?P<H>2[0-3]|[0-1]\d|\d)", |
Neal Norwitz | 5efc50d | 2002-12-30 22:23:12 +0000 | [diff] [blame] | 193 | 'I': r"(?P<I>1[0-2]|0[1-9]|[1-9])", |
| 194 | 'j': r"(?P<j>36[0-6]|3[0-5]\d|[1-2]\d\d|0[1-9]\d|00[1-9]|[1-9]\d|0[1-9]|[1-9])", |
| 195 | 'm': r"(?P<m>1[0-2]|0[1-9]|[1-9])", |
Guido van Rossum | 00efe7e | 2002-07-19 17:04:46 +0000 | [diff] [blame] | 196 | 'M': r"(?P<M>[0-5]\d|\d)", |
| 197 | 'S': r"(?P<S>6[0-1]|[0-5]\d|\d)", |
| 198 | 'U': r"(?P<U>5[0-3]|[0-4]\d|\d)", |
| 199 | 'w': r"(?P<w>[0-6])", |
Neal Norwitz | 5efc50d | 2002-12-30 22:23:12 +0000 | [diff] [blame] | 200 | # W is set below by using 'U' |
Guido van Rossum | 00efe7e | 2002-07-19 17:04:46 +0000 | [diff] [blame] | 201 | 'y': r"(?P<y>\d\d)", |
Brett Cannon | 474335c | 2003-08-05 04:02:49 +0000 | [diff] [blame] | 202 | #XXX: Does 'Y' need to worry about having less or more than |
| 203 | # 4 digits? |
| 204 | 'Y': r"(?P<Y>\d\d\d\d)", |
| 205 | 'A': self.__seqToRE(self.locale_time.f_weekday, 'A'), |
| 206 | 'a': self.__seqToRE(self.locale_time.a_weekday, 'a'), |
| 207 | 'B': self.__seqToRE(self.locale_time.f_month[1:], 'B'), |
| 208 | 'b': self.__seqToRE(self.locale_time.a_month[1:], 'b'), |
| 209 | 'p': self.__seqToRE(self.locale_time.am_pm, 'p'), |
| 210 | 'Z': self.__seqToRE([tz for tz_names in self.locale_time.timezone |
| 211 | for tz in tz_names], |
| 212 | 'Z'), |
| 213 | '%': '%'}) |
Neal Norwitz | 5efc50d | 2002-12-30 22:23:12 +0000 | [diff] [blame] | 214 | base.__setitem__('W', base.__getitem__('U')) |
Brett Cannon | 474335c | 2003-08-05 04:02:49 +0000 | [diff] [blame] | 215 | base.__setitem__('c', self.pattern(self.locale_time.LC_date_time)) |
| 216 | base.__setitem__('x', self.pattern(self.locale_time.LC_date)) |
| 217 | base.__setitem__('X', self.pattern(self.locale_time.LC_time)) |
Tim Peters | 469cdad | 2002-08-08 20:19:19 +0000 | [diff] [blame] | 218 | |
Guido van Rossum | 00efe7e | 2002-07-19 17:04:46 +0000 | [diff] [blame] | 219 | def __seqToRE(self, to_convert, directive): |
Brett Cannon | 474335c | 2003-08-05 04:02:49 +0000 | [diff] [blame] | 220 | """Convert a list to a regex string for matching a directive. |
Raymond Hettinger | 6b59f5f | 2003-10-16 05:53:16 +0000 | [diff] [blame] | 221 | |
Brett Cannon | 474335c | 2003-08-05 04:02:49 +0000 | [diff] [blame] | 222 | Want possible matching values to be from longest to shortest. This |
| 223 | prevents the possibility of a match occuring for a value that also |
| 224 | a substring of a larger value that should have matched (e.g., 'abc' |
| 225 | matching when 'abcdef' should have been the match). |
Raymond Hettinger | 6b59f5f | 2003-10-16 05:53:16 +0000 | [diff] [blame] | 226 | |
Brett Cannon | 474335c | 2003-08-05 04:02:49 +0000 | [diff] [blame] | 227 | """ |
Jack Jansen | 62fe755 | 2003-01-15 22:59:39 +0000 | [diff] [blame] | 228 | for value in to_convert: |
| 229 | if value != '': |
| 230 | break |
| 231 | else: |
| 232 | return '' |
Raymond Hettinger | 6b59f5f | 2003-10-16 05:53:16 +0000 | [diff] [blame] | 233 | to_convert = to_convert[:] |
| 234 | to_convert.sort(key=len, reverse=True) |
Barry Warsaw | 4d895fa | 2002-09-23 22:46:49 +0000 | [diff] [blame] | 235 | regex = '|'.join(to_convert) |
| 236 | regex = '(?P<%s>%s' % (directive, regex) |
Guido van Rossum | 00efe7e | 2002-07-19 17:04:46 +0000 | [diff] [blame] | 237 | return '%s)' % regex |
| 238 | |
| 239 | def pattern(self, format): |
Brett Cannon | 474335c | 2003-08-05 04:02:49 +0000 | [diff] [blame] | 240 | """Return regex pattern for the format string. |
Tim Peters | 0eadaac | 2003-04-24 16:02:54 +0000 | [diff] [blame] | 241 | |
Brett Cannon | 1e91d8e | 2003-04-19 04:00:56 +0000 | [diff] [blame] | 242 | Need to make sure that any characters that might be interpreted as |
Brett Cannon | 5187a3b | 2003-08-11 07:24:05 +0000 | [diff] [blame] | 243 | regex syntax are escaped. |
Tim Peters | 0eadaac | 2003-04-24 16:02:54 +0000 | [diff] [blame] | 244 | |
Brett Cannon | 1e91d8e | 2003-04-19 04:00:56 +0000 | [diff] [blame] | 245 | """ |
Guido van Rossum | 00efe7e | 2002-07-19 17:04:46 +0000 | [diff] [blame] | 246 | processed_format = '' |
Brett Cannon | 1e91d8e | 2003-04-19 04:00:56 +0000 | [diff] [blame] | 247 | # The sub() call escapes all characters that might be misconstrued |
| 248 | # as regex syntax. |
Brett Cannon | 953c6f5 | 2003-08-29 02:28:54 +0000 | [diff] [blame] | 249 | regex_chars = re_compile(r"([\\.^$*+?\(\){}\[\]|])") |
Brett Cannon | 1e91d8e | 2003-04-19 04:00:56 +0000 | [diff] [blame] | 250 | format = regex_chars.sub(r"\\\1", format) |
Tim Peters | 80cebc1 | 2003-01-19 04:40:44 +0000 | [diff] [blame] | 251 | whitespace_replacement = re_compile('\s+') |
| 252 | format = whitespace_replacement.sub('\s*', format) |
Raymond Hettinger | bac788a | 2004-05-04 09:21:43 +0000 | [diff] [blame] | 253 | while '%' in format: |
Guido van Rossum | 00efe7e | 2002-07-19 17:04:46 +0000 | [diff] [blame] | 254 | directive_index = format.index('%')+1 |
Tim Peters | 469cdad | 2002-08-08 20:19:19 +0000 | [diff] [blame] | 255 | processed_format = "%s%s%s" % (processed_format, |
Barry Warsaw | 35816e6 | 2002-08-29 16:24:50 +0000 | [diff] [blame] | 256 | format[:directive_index-1], |
| 257 | self[format[directive_index]]) |
Guido van Rossum | 00efe7e | 2002-07-19 17:04:46 +0000 | [diff] [blame] | 258 | format = format[directive_index+1:] |
| 259 | return "%s%s" % (processed_format, format) |
| 260 | |
| 261 | def compile(self, format): |
| 262 | """Return a compiled re object for the format string.""" |
Guido van Rossum | 00efe7e | 2002-07-19 17:04:46 +0000 | [diff] [blame] | 263 | return re_compile(self.pattern(format), IGNORECASE) |
| 264 | |
Brett Cannon | 474335c | 2003-08-05 04:02:49 +0000 | [diff] [blame] | 265 | _cache_lock = _thread_allocate_lock() |
| 266 | # DO NOT modify _TimeRE_cache or _regex_cache without acquiring the cache lock |
| 267 | # first! |
| 268 | _TimeRE_cache = TimeRE() |
Brett Cannon | 5187a3b | 2003-08-11 07:24:05 +0000 | [diff] [blame] | 269 | _CACHE_MAX_SIZE = 5 # Max number of regexes stored in _regex_cache |
Brett Cannon | 474335c | 2003-08-05 04:02:49 +0000 | [diff] [blame] | 270 | _regex_cache = {} |
Guido van Rossum | 00efe7e | 2002-07-19 17:04:46 +0000 | [diff] [blame] | 271 | |
| 272 | def strptime(data_string, format="%a %b %d %H:%M:%S %Y"): |
Brett Cannon | 5187a3b | 2003-08-11 07:24:05 +0000 | [diff] [blame] | 273 | """Return a time struct based on the input string and the format string.""" |
Brett Cannon | 474335c | 2003-08-05 04:02:49 +0000 | [diff] [blame] | 274 | global _TimeRE_cache |
| 275 | _cache_lock.acquire() |
| 276 | try: |
| 277 | time_re = _TimeRE_cache |
| 278 | locale_time = time_re.locale_time |
| 279 | if _getlang() != locale_time.lang: |
| 280 | _TimeRE_cache = TimeRE() |
| 281 | if len(_regex_cache) > _CACHE_MAX_SIZE: |
| 282 | _regex_cache.clear() |
| 283 | format_regex = _regex_cache.get(format) |
| 284 | if not format_regex: |
| 285 | format_regex = time_re.compile(format) |
| 286 | _regex_cache[format] = format_regex |
| 287 | finally: |
| 288 | _cache_lock.release() |
Tim Peters | 80cebc1 | 2003-01-19 04:40:44 +0000 | [diff] [blame] | 289 | found = format_regex.match(data_string) |
Tim Peters | 08e5427 | 2003-01-18 03:53:49 +0000 | [diff] [blame] | 290 | if not found: |
Raymond Hettinger | 4a6302b | 2003-07-13 01:31:38 +0000 | [diff] [blame] | 291 | raise ValueError("time data did not match format: data=%s fmt=%s" % |
| 292 | (data_string, format)) |
Brett Cannon | 2b6dfec | 2003-04-28 21:30:13 +0000 | [diff] [blame] | 293 | if len(data_string) != found.end(): |
| 294 | raise ValueError("unconverted data remains: %s" % |
| 295 | data_string[found.end():]) |
Tim Peters | 08e5427 | 2003-01-18 03:53:49 +0000 | [diff] [blame] | 296 | year = 1900 |
| 297 | month = day = 1 |
| 298 | hour = minute = second = 0 |
| 299 | tz = -1 |
Raymond Hettinger | 1fdb633 | 2003-03-09 07:44:42 +0000 | [diff] [blame] | 300 | # weekday and julian defaulted to -1 so as to signal need to calculate values |
Tim Peters | 08e5427 | 2003-01-18 03:53:49 +0000 | [diff] [blame] | 301 | weekday = julian = -1 |
| 302 | found_dict = found.groupdict() |
| 303 | for group_key in found_dict.iterkeys(): |
| 304 | if group_key == 'y': |
| 305 | year = int(found_dict['y']) |
| 306 | # Open Group specification for strptime() states that a %y |
| 307 | #value in the range of [00, 68] is in the century 2000, while |
| 308 | #[69,99] is in the century 1900 |
| 309 | if year <= 68: |
| 310 | year += 2000 |
| 311 | else: |
| 312 | year += 1900 |
| 313 | elif group_key == 'Y': |
| 314 | year = int(found_dict['Y']) |
| 315 | elif group_key == 'm': |
| 316 | month = int(found_dict['m']) |
| 317 | elif group_key == 'B': |
Brett Cannon | 474335c | 2003-08-05 04:02:49 +0000 | [diff] [blame] | 318 | month = locale_time.f_month.index(found_dict['B'].lower()) |
Tim Peters | 08e5427 | 2003-01-18 03:53:49 +0000 | [diff] [blame] | 319 | elif group_key == 'b': |
Brett Cannon | 474335c | 2003-08-05 04:02:49 +0000 | [diff] [blame] | 320 | month = locale_time.a_month.index(found_dict['b'].lower()) |
Tim Peters | 08e5427 | 2003-01-18 03:53:49 +0000 | [diff] [blame] | 321 | elif group_key == 'd': |
| 322 | day = int(found_dict['d']) |
Neal Norwitz | 77290f2 | 2003-06-29 04:16:49 +0000 | [diff] [blame] | 323 | elif group_key == 'H': |
Tim Peters | 08e5427 | 2003-01-18 03:53:49 +0000 | [diff] [blame] | 324 | hour = int(found_dict['H']) |
| 325 | elif group_key == 'I': |
| 326 | hour = int(found_dict['I']) |
| 327 | ampm = found_dict.get('p', '').lower() |
| 328 | # If there was no AM/PM indicator, we'll treat this like AM |
Brett Cannon | 474335c | 2003-08-05 04:02:49 +0000 | [diff] [blame] | 329 | if ampm in ('', locale_time.am_pm[0]): |
Tim Peters | 08e5427 | 2003-01-18 03:53:49 +0000 | [diff] [blame] | 330 | # We're in AM so the hour is correct unless we're |
| 331 | # looking at 12 midnight. |
| 332 | # 12 midnight == 12 AM == hour 0 |
| 333 | if hour == 12: |
| 334 | hour = 0 |
Brett Cannon | 474335c | 2003-08-05 04:02:49 +0000 | [diff] [blame] | 335 | elif ampm == locale_time.am_pm[1]: |
Tim Peters | 08e5427 | 2003-01-18 03:53:49 +0000 | [diff] [blame] | 336 | # We're in PM so we need to add 12 to the hour unless |
| 337 | # we're looking at 12 noon. |
| 338 | # 12 noon == 12 PM == hour 12 |
| 339 | if hour != 12: |
| 340 | hour += 12 |
| 341 | elif group_key == 'M': |
| 342 | minute = int(found_dict['M']) |
| 343 | elif group_key == 'S': |
| 344 | second = int(found_dict['S']) |
| 345 | elif group_key == 'A': |
Brett Cannon | 474335c | 2003-08-05 04:02:49 +0000 | [diff] [blame] | 346 | weekday = locale_time.f_weekday.index(found_dict['A'].lower()) |
Tim Peters | 08e5427 | 2003-01-18 03:53:49 +0000 | [diff] [blame] | 347 | elif group_key == 'a': |
Brett Cannon | 474335c | 2003-08-05 04:02:49 +0000 | [diff] [blame] | 348 | weekday = locale_time.a_weekday.index(found_dict['a'].lower()) |
Tim Peters | 08e5427 | 2003-01-18 03:53:49 +0000 | [diff] [blame] | 349 | elif group_key == 'w': |
| 350 | weekday = int(found_dict['w']) |
| 351 | if weekday == 0: |
| 352 | weekday = 6 |
| 353 | else: |
| 354 | weekday -= 1 |
| 355 | elif group_key == 'j': |
| 356 | julian = int(found_dict['j']) |
| 357 | elif group_key == 'Z': |
Brett Cannon | 172d9ef | 2003-05-11 06:23:36 +0000 | [diff] [blame] | 358 | # Since -1 is default value only need to worry about setting tz if |
| 359 | # it can be something other than -1. |
Tim Peters | 08e5427 | 2003-01-18 03:53:49 +0000 | [diff] [blame] | 360 | found_zone = found_dict['Z'].lower() |
Brett Cannon | 5187a3b | 2003-08-11 07:24:05 +0000 | [diff] [blame] | 361 | for value, tz_values in enumerate(locale_time.timezone): |
| 362 | if found_zone in tz_values: |
| 363 | # Deal with bad locale setup where timezone names are the |
| 364 | # same and yet time.daylight is true; too ambiguous to |
| 365 | # be able to tell what timezone has daylight savings |
Brett Cannon | 8172ac3 | 2004-03-07 23:16:27 +0000 | [diff] [blame] | 366 | if (time.tzname[0] == time.tzname[1] and |
| 367 | time.daylight and found_zone not in ("utc", "gmt")): |
Tim Peters | 58eb11c | 2004-01-18 20:29:55 +0000 | [diff] [blame] | 368 | break |
Brett Cannon | 5187a3b | 2003-08-11 07:24:05 +0000 | [diff] [blame] | 369 | else: |
Brett Cannon | 474335c | 2003-08-05 04:02:49 +0000 | [diff] [blame] | 370 | tz = value |
Brett Cannon | 5187a3b | 2003-08-11 07:24:05 +0000 | [diff] [blame] | 371 | break |
Raymond Hettinger | 1fdb633 | 2003-03-09 07:44:42 +0000 | [diff] [blame] | 372 | # Cannot pre-calculate datetime_date() since can change in Julian |
| 373 | #calculation and thus could have different value for the day of the week |
| 374 | #calculation |
Tim Peters | 08e5427 | 2003-01-18 03:53:49 +0000 | [diff] [blame] | 375 | if julian == -1: |
Raymond Hettinger | 1fdb633 | 2003-03-09 07:44:42 +0000 | [diff] [blame] | 376 | # Need to add 1 to result since first day of the year is 1, not 0. |
| 377 | julian = datetime_date(year, month, day).toordinal() - \ |
| 378 | datetime_date(year, 1, 1).toordinal() + 1 |
| 379 | else: # Assume that if they bothered to include Julian day it will |
Tim Peters | 08e5427 | 2003-01-18 03:53:49 +0000 | [diff] [blame] | 380 | #be accurate |
Raymond Hettinger | 1fdb633 | 2003-03-09 07:44:42 +0000 | [diff] [blame] | 381 | datetime_result = datetime_date.fromordinal((julian - 1) + datetime_date(year, 1, 1).toordinal()) |
| 382 | year = datetime_result.year |
| 383 | month = datetime_result.month |
| 384 | day = datetime_result.day |
Tim Peters | 08e5427 | 2003-01-18 03:53:49 +0000 | [diff] [blame] | 385 | if weekday == -1: |
Raymond Hettinger | 1fdb633 | 2003-03-09 07:44:42 +0000 | [diff] [blame] | 386 | weekday = datetime_date(year, month, day).weekday() |
Tim Peters | 08e5427 | 2003-01-18 03:53:49 +0000 | [diff] [blame] | 387 | return time.struct_time((year, month, day, |
| 388 | hour, minute, second, |
| 389 | weekday, julian, tz)) |