Skip to content

gh-71587: Establish global state in _datetime #110475

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
erlend-aasland merged 21 commits into from
Oct 12, 2023
Merged

gh-71587: Establish global state in _datetime #110475

Changes from 1 commit
Commits
Show all changes
21 commits
Select commit Hold shift + click to select a range
cacf1e8
Consistency: use explicit initialiser for m_base
erlend-aasland Feb 15, 2023
c2eba8f
Add module state stub; establish global state on stack
erlend-aasland Feb 15, 2023
8d54ecd
Put conversion factors in state struct
erlend-aasland Feb 15, 2023
360397a
Move PyDateTime_TimeZone_UTC to state
erlend-aasland Feb 15, 2023
4f509f9
Move PyDateTime_Epoch to state struct
erlend-aasland Feb 15, 2023
6b14c8c
Remove datetime_strptime globals entry
erlend-aasland Feb 15, 2023
2eb3315
Apply Victor's suggestions from gh-102995
erlend-aasland Oct 6, 2023
f5ede1e
Apply PEP-7 to modified code
erlend-aasland Oct 6, 2023
f6ac875
Merge branch 'main' into isolate-datetime/part1-establish-global-state
erlend-aasland Oct 6, 2023
7139569
Fix check-c-globals
erlend-aasland Oct 6, 2023
26e6454
Merge PR updates
erlend-aasland Oct 6, 2023
227dd2c
Pull in main
erlend-aasland Oct 10, 2023
beb0977
Address review: use shorter names
erlend-aasland Oct 10, 2023
8525be6
Pull in main
erlend-aasland Oct 11, 2023
6863f76
Address review: rename global_state => _datetime_global_state
erlend-aasland Oct 11, 2023
0738205
Pull in main
erlend-aasland Oct 11, 2023
c56b5e1
Address review
erlend-aasland Oct 11, 2023
3ee48d2
Fix ref leak in init
erlend-aasland Oct 11, 2023
0a1d770
Pull in main
erlend-aasland Oct 11, 2023
a9e8d1a
Clean up initialisation
erlend-aasland Oct 11, 2023
c3ba7fe
Preserve comment
erlend-aasland Oct 11, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Address review
  • Loading branch information
@erlend-aasland
erlend-aasland committed Oct 11, 2023
commit c56b5e1fa7673705a1806abb8779058f1deca5e2
57 changes: 34 additions & 23 deletions Modules/_datetimemodule.c
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -53,13 +53,13 @@ typedef struct {
/* The interned UTC timezone instance */
PyObject *utc;

/* The interned Epoch datetime instance */
/* The interned Unix epoch datetime instance */
PyObject *epoch;
} datetime_state;

static datetime_state _datetime_global_state;

#define GLOBAL_STATE() (&_datetime_global_state)
#define STATIC_STATE() (&_datetime_global_state)

/*[clinic input]
module datetime
Expand Down Expand Up @@ -1193,7 +1193,7 @@ new_timezone(PyObject *offset, PyObject *name)
assert(name == NULL || PyUnicode_Check(name));

if (name == NULL && delta_bool((PyDateTime_Delta *)offset) == 0) {
datetime_state *st = GLOBAL_STATE();
datetime_state *st = STATIC_STATE();
return Py_NewRef(st->utc);
}
if ((GET_TD_DAYS(offset) == -1 &&
Expand Down Expand Up @@ -1407,7 +1407,7 @@ tzinfo_from_isoformat_results(int rv, int tzoffset, int tz_useconds)
if (rv == 1) {
// Create a timezone from offset in seconds (0 returns UTC)
if (tzoffset == 0) {
datetime_state *st = GLOBAL_STATE();
datetime_state *st = STATIC_STATE();
return Py_NewRef(st->utc);
}

Expand Down Expand Up @@ -1850,7 +1850,7 @@ delta_to_microseconds(PyDateTime_Delta *self)
x1 = PyLong_FromLong(GET_TD_DAYS(self));
if (x1 == NULL)
goto Done;
datetime_state *st = GLOBAL_STATE();
datetime_state *st = STATIC_STATE();
x2 = PyNumber_Multiply(x1, st->seconds_per_day); /* days in seconds */
if (x2 == NULL)
goto Done;
Expand Down Expand Up @@ -1923,7 +1923,7 @@ microseconds_to_delta_ex(PyObject *pyus, PyTypeObject *type)
PyObject *num = NULL;
PyObject *result = NULL;

datetime_state *st = GLOBAL_STATE();
datetime_state *st = STATIC_STATE();
tuple = checked_divmod(pyus, st->us_per_second);
if (tuple == NULL) {
goto Done;
Expand Down Expand Up @@ -2542,7 +2542,7 @@ delta_new(PyTypeObject *type, PyObject *args, PyObject *kw)
y = accum("microseconds", x, us, _PyLong_GetOne(), &leftover_us);
CLEANUP;
}
datetime_state *st = GLOBAL_STATE();
datetime_state *st = STATIC_STATE();
if (ms) {
y = accum("milliseconds", x, ms, st->us_per_ms, &leftover_us);
CLEANUP;
Expand Down Expand Up @@ -2719,7 +2719,7 @@ delta_total_seconds(PyObject *self, PyObject *Py_UNUSED(ignored))
if (total_microseconds == NULL)
return NULL;

datetime_state *st = GLOBAL_STATE();
datetime_state *st = STATIC_STATE();
total_seconds = PyNumber_TrueDivide(total_microseconds, st->us_per_second);

Py_DECREF(total_microseconds);
Expand Down Expand Up @@ -3952,7 +3952,7 @@ timezone_repr(PyDateTime_TimeZone *self)
to use Py_TYPE(self)->tp_name here. */
const char *type_name = Py_TYPE(self)->tp_name;

datetime_state *st = GLOBAL_STATE();
datetime_state *st = STATIC_STATE();
if (((PyObject *)self) == st->utc) {
return PyUnicode_FromFormat("%s.utc", type_name);
}
Expand All @@ -3975,7 +3975,7 @@ timezone_str(PyDateTime_TimeZone *self)
if (self->name != NULL) {
return Py_NewRef(self->name);
}
datetime_state *st = GLOBAL_STATE();
datetime_state *st = STATIC_STATE();
if ((PyObject *)self == st->utc ||
(GET_TD_DAYS(self->offset) == 0 &&
GET_TD_SECONDS(self->offset) == 0 &&
Expand Down Expand Up @@ -6148,7 +6148,7 @@ local_timezone(PyDateTime_DateTime *utc_time)
PyObject *one_second;
PyObject *seconds;

datetime_state *st = GLOBAL_STATE();
datetime_state *st = STATIC_STATE();
delta = datetime_subtract((PyObject *)utc_time, st->epoch);
if (delta == NULL)
return NULL;
Expand Down Expand Up @@ -6261,7 +6261,7 @@ datetime_astimezone(PyDateTime_DateTime *self, PyObject *args, PyObject *kw)
if (result == NULL)
return NULL;

datetime_state *st = GLOBAL_STATE();
datetime_state *st = STATIC_STATE();
/* Make sure result is aware and UTC. */
if (!HASTZINFO(result)) {
temp = (PyObject *)result;
Expand Down Expand Up @@ -6386,7 +6386,7 @@ datetime_timestamp(PyDateTime_DateTime *self, PyObject *Py_UNUSED(ignored))
PyObject *result;

if (HASTZINFO(self) && self->tzinfo != Py_None) {
datetime_state *st = GLOBAL_STATE();
datetime_state *st = STATIC_STATE();
PyObject *delta;
delta = datetime_subtract((PyObject *)self, st->epoch);
if (delta == NULL)
Expand Down Expand Up @@ -6711,7 +6711,7 @@ get_datetime_capi(void)
capi->Time_FromTimeAndFold = new_time_ex2;
// Make sure this function is called after utc has
// been initialized.
datetime_state *st = GLOBAL_STATE();
datetime_state *st = STATIC_STATE();
assert(st->utc != NULL);
capi->TimeZone_UTC = st->utc; // borrowed ref
return capi;
Expand All @@ -6724,6 +6724,23 @@ datetime_destructor(PyObject *op)
PyMem_Free(ptr);
}

static int
datetime_clear(PyObject *module)
{
datetime_state *st = STATIC_STATE();

Py_CLEAR(st->us_per_ms);
Py_CLEAR(st->us_per_second);
Py_CLEAR(st->us_per_minute);
Py_CLEAR(st->us_per_hour);
Py_CLEAR(st->us_per_day);
Py_CLEAR(st->us_per_week);
Py_CLEAR(st->seconds_per_day);
Py_CLEAR(st->utc);
Py_CLEAR(st->epoch);
return 0;
}

static int
_datetime_exec(PyObject *module)
{
Expand Down Expand Up @@ -6810,8 +6827,8 @@ _datetime_exec(PyObject *module)
return -1;
}

datetime_state *st = GLOBAL_STATE();
st->utc = utc;
datetime_state *st = STATIC_STATE();
Py_SETREF(st->utc, utc);

/* bpo-37642: These attributes are rounded to the nearest minute for backwards
* compatibility, even though the constructor will accept a wider range of
Expand Down Expand Up @@ -6920,13 +6937,7 @@ _datetime_exec(PyObject *module)
return 0;

error:
Py_CLEAR(st->us_per_ms);
Py_CLEAR(st->us_per_second);
Py_CLEAR(st->us_per_minute);
Py_CLEAR(st->us_per_hour);
Py_CLEAR(st->us_per_day);
Py_CLEAR(st->us_per_week);
Py_CLEAR(st->seconds_per_day);
datetime_clear(module);
return -1;
}

Expand Down