2019-09-12 22:56:28 +08:00
|
|
|
/* Implementation helper: a struct that looks like a tuple.
|
|
|
|
See timemodule and posixmodule for example uses.
|
|
|
|
|
|
|
|
The structseq helper is considered an internal CPython implementation
|
|
|
|
detail. Docs for modules using structseqs should call them
|
|
|
|
"named tuples" (be sure to include a space between the two
|
|
|
|
words and add a link back to the term in Docs/glossary.rst).
|
|
|
|
*/
|
2001-10-19 04:47:51 +08:00
|
|
|
|
|
|
|
#include "Python.h"
|
2023-10-17 20:30:31 +08:00
|
|
|
#include "pycore_initconfig.h" // _PyStatus_OK()
|
|
|
|
#include "pycore_modsupport.h" // _PyArg_NoPositional()
|
2020-06-22 23:27:35 +08:00
|
|
|
#include "pycore_object.h" // _PyObject_GC_TRACK()
|
2021-05-03 22:50:24 +08:00
|
|
|
#include "pycore_structseq.h" // PyStructSequence_InitType()
|
2023-10-17 20:30:31 +08:00
|
|
|
#include "pycore_tuple.h" // _PyTuple_FromArray()
|
2001-10-19 04:47:51 +08:00
|
|
|
|
2015-12-26 01:53:18 +08:00
|
|
|
static const char visible_length_key[] = "n_sequence_fields";
|
|
|
|
static const char real_length_key[] = "n_fields";
|
|
|
|
static const char unnamed_fields_key[] = "n_unnamed_fields";
|
2021-03-04 08:03:44 +08:00
|
|
|
static const char match_args_key[] = "__match_args__";
|
2001-10-19 04:47:51 +08:00
|
|
|
|
2010-05-09 23:52:27 +08:00
|
|
|
/* Fields with this name have only a field index, not a field name.
|
2002-10-17 02:27:39 +08:00
|
|
|
They are only allowed for indices < n_visible_fields. */
|
2019-11-17 00:55:29 +08:00
|
|
|
const char * const PyStructSequence_UnnamedField = "unnamed field";
|
2020-10-26 14:43:39 +08:00
|
|
|
|
2021-03-01 19:18:33 +08:00
|
|
|
static Py_ssize_t
|
2022-02-09 04:39:07 +08:00
|
|
|
get_type_attr_as_size(PyTypeObject *tp, PyObject *name)
|
2020-10-26 14:43:39 +08:00
|
|
|
{
|
2023-05-02 10:34:43 +08:00
|
|
|
PyObject *v = PyDict_GetItemWithError(_PyType_GetDict(tp), name);
|
2020-10-26 14:43:39 +08:00
|
|
|
if (v == NULL && !PyErr_Occurred()) {
|
|
|
|
PyErr_Format(PyExc_TypeError,
|
|
|
|
"Missed attribute '%U' of type %s",
|
|
|
|
name, tp->tp_name);
|
2023-04-28 08:28:51 +08:00
|
|
|
return -1;
|
2020-10-26 14:43:39 +08:00
|
|
|
}
|
|
|
|
return PyLong_AsSsize_t(v);
|
|
|
|
}
|
2001-10-19 04:47:51 +08:00
|
|
|
|
2020-10-26 14:43:39 +08:00
|
|
|
#define VISIBLE_SIZE(op) Py_SIZE(op)
|
2022-02-09 04:39:07 +08:00
|
|
|
#define VISIBLE_SIZE_TP(tp) \
|
|
|
|
get_type_attr_as_size(tp, &_Py_ID(n_sequence_fields))
|
|
|
|
#define REAL_SIZE_TP(tp) \
|
|
|
|
get_type_attr_as_size(tp, &_Py_ID(n_fields))
|
2024-08-03 00:11:44 +08:00
|
|
|
#define REAL_SIZE(op) get_real_size((PyObject *)op)
|
2001-10-19 04:47:51 +08:00
|
|
|
|
2022-02-09 04:39:07 +08:00
|
|
|
#define UNNAMED_FIELDS_TP(tp) \
|
|
|
|
get_type_attr_as_size(tp, &_Py_ID(n_unnamed_fields))
|
2007-12-19 10:45:37 +08:00
|
|
|
#define UNNAMED_FIELDS(op) UNNAMED_FIELDS_TP(Py_TYPE(op))
|
2002-10-17 03:10:03 +08:00
|
|
|
|
2024-08-03 00:11:44 +08:00
|
|
|
static Py_ssize_t
|
|
|
|
get_real_size(PyObject *op)
|
|
|
|
{
|
|
|
|
// Compute the real size from the visible size (i.e., Py_SIZE()) and the
|
|
|
|
// number of non-sequence fields accounted for in tp_basicsize.
|
|
|
|
Py_ssize_t hidden = Py_TYPE(op)->tp_basicsize - offsetof(PyStructSequence, ob_item);
|
|
|
|
return Py_SIZE(op) + hidden / sizeof(PyObject *);
|
|
|
|
}
|
2001-10-19 04:47:51 +08:00
|
|
|
|
|
|
|
PyObject *
|
|
|
|
PyStructSequence_New(PyTypeObject *type)
|
|
|
|
{
|
2010-05-09 23:52:27 +08:00
|
|
|
PyStructSequence *obj;
|
2010-07-08 04:54:01 +08:00
|
|
|
Py_ssize_t size = REAL_SIZE_TP(type), i;
|
2020-10-26 14:43:39 +08:00
|
|
|
if (size < 0) {
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
Py_ssize_t vsize = VISIBLE_SIZE_TP(type);
|
|
|
|
if (vsize < 0) {
|
|
|
|
return NULL;
|
|
|
|
}
|
Merged revisions 59933-59951 via svnmerge from
svn+ssh://pythondev@svn.python.org/python/trunk
........
r59935 | raymond.hettinger | 2008-01-13 07:15:15 +0100 (Sun, 13 Jan 2008) | 1 line
Named tuple is a concept, not a specific type.
........
r59936 | raymond.hettinger | 2008-01-13 07:18:07 +0100 (Sun, 13 Jan 2008) | 1 line
Fix spelling.
........
r59937 | georg.brandl | 2008-01-13 10:36:18 +0100 (Sun, 13 Jan 2008) | 2 lines
Clarify the effect of text mode.
........
r59938 | thomas.heller | 2008-01-13 12:19:43 +0100 (Sun, 13 Jan 2008) | 1 line
Make Modules/socketobject.c compile for Windows again.
........
r59939 | ka-ping.yee | 2008-01-13 12:25:13 +0100 (Sun, 13 Jan 2008) | 9 lines
Check in the patch proposed by Ben Hayden (benjhayden) for issue
#1550: help('modules') broken by several 3rd party libraries.
Tested with Python build: trunk:54235:59936M -- the reported error
occurs with Django installed (or with any __init__.py present on
the path that raises an exception), and such errors indeed go away
when this change is applied.
........
r59940 | georg.brandl | 2008-01-13 16:04:05 +0100 (Sun, 13 Jan 2008) | 2 lines
Back out r59931 - test_ctypes fails with it.
........
r59943 | amaury.forgeotdarc | 2008-01-14 01:22:44 +0100 (Mon, 14 Jan 2008) | 6 lines
As discussed in issue 1700288:
ctypes takes some liberties when creating python types: it modifies the types'
__dict__ directly, bypassing all the machinery of type objects which deal with
special methods. And this broke recent optimisations of method lookup.
Now we try to modify the type with more "official" functions.
........
r59944 | amaury.forgeotdarc | 2008-01-14 01:29:41 +0100 (Mon, 14 Jan 2008) | 5 lines
Re-apply patch #1700288 (first applied in r59931, rolled back in r59940)
now that ctypes uses a more supported method to create types:
Method cache optimization, by Armin Rigo, ported to 2.6 by Kevin Jacobs.
........
r59946 | amaury.forgeotdarc | 2008-01-14 02:07:27 +0100 (Mon, 14 Jan 2008) | 4 lines
?Why did my tests not notice this before?
Slots inheritance is very different from OO inheritance.
This code lead to infinite recursion on classes derived from StructType.
........
r59947 | christian.heimes | 2008-01-14 04:33:52 +0100 (Mon, 14 Jan 2008) | 1 line
Added new an better structseq representation. E.g. repr(time.gmtime(0)) now returns 'time.struct_time(tm_year=1970, tm_mon=1, tm_mday=1, tm_hour=0, tm_min=0, tm_sec=0, tm_wday=3, tm_yday=1, tm_isdst=0)' instead of '(1970, 1, 1, 0, 0, 0, 3, 1, 0)'. The feature is part of #1816: sys.flags
........
r59948 | christian.heimes | 2008-01-14 04:35:38 +0100 (Mon, 14 Jan 2008) | 1 line
I missed the most important file
........
r59949 | christian.heimes | 2008-01-14 04:42:48 +0100 (Mon, 14 Jan 2008) | 1 line
Applied patch #1816: sys.flags patch
........
r59950 | christian.heimes | 2008-01-14 05:13:37 +0100 (Mon, 14 Jan 2008) | 2 lines
Now that I've learnt about structseq objects I felt like converting sys.float_info to a structseq. It's
readonly and help(sys.float_info) explains the attributes nicely.
........
r59951 | christian.heimes | 2008-01-14 07:06:19 +0100 (Mon, 14 Jan 2008) | 1 line
Added more comments to the new structseq repr code and implemented several of Neal's suggestions.
........
2008-01-15 02:49:24 +08:00
|
|
|
|
2010-07-08 04:54:01 +08:00
|
|
|
obj = PyObject_GC_NewVar(PyStructSequence, type, size);
|
2010-05-09 23:52:27 +08:00
|
|
|
if (obj == NULL)
|
|
|
|
return NULL;
|
2010-07-09 06:33:03 +08:00
|
|
|
/* Hack the size of the variable object, so invisible fields don't appear
|
|
|
|
to Python code. */
|
2020-10-26 14:43:39 +08:00
|
|
|
Py_SET_SIZE(obj, vsize);
|
2010-07-08 04:54:01 +08:00
|
|
|
for (i = 0; i < size; i++)
|
|
|
|
obj->ob_item[i] = NULL;
|
2001-10-19 04:47:51 +08:00
|
|
|
|
2010-07-08 04:54:01 +08:00
|
|
|
return (PyObject*)obj;
|
Merge the trunk changes in. Breaks socket.ssl for now.
Merged revisions 57392-57619 via svnmerge from
svn+ssh://pythondev@svn.python.org/python/trunk
........
r57395 | georg.brandl | 2007-08-24 19:23:23 +0200 (Fri, 24 Aug 2007) | 2 lines
Bug #1011: fix rfc822.Message.getheader docs.
........
r57397 | georg.brandl | 2007-08-24 19:38:49 +0200 (Fri, 24 Aug 2007) | 2 lines
Patch #1006: port test_winreg to unittest.
........
r57398 | georg.brandl | 2007-08-24 19:46:54 +0200 (Fri, 24 Aug 2007) | 2 lines
Fix #1012: wrong URL to :mod:`site` in install/index.rst.
........
r57399 | georg.brandl | 2007-08-24 20:07:52 +0200 (Fri, 24 Aug 2007) | 2 lines
Patch #1008: port test_signal to unittest.
........
r57400 | georg.brandl | 2007-08-24 20:22:54 +0200 (Fri, 24 Aug 2007) | 2 lines
Port test_frozen to unittest.
........
r57401 | georg.brandl | 2007-08-24 20:27:43 +0200 (Fri, 24 Aug 2007) | 2 lines
Document new utility functions in test_support.
........
r57402 | georg.brandl | 2007-08-24 20:30:06 +0200 (Fri, 24 Aug 2007) | 2 lines
Remove test_rgbimg output file, there is no test_rgbimg.py.
........
r57403 | georg.brandl | 2007-08-24 20:35:27 +0200 (Fri, 24 Aug 2007) | 2 lines
Remove output file for test_ossaudiodev, also properly close the dsp object.
........
r57404 | georg.brandl | 2007-08-24 20:46:27 +0200 (Fri, 24 Aug 2007) | 2 lines
Convert test_linuxaudiodev to unittest. Fix a wrong finally clause in test_ossaudiodev.
........
r57406 | collin.winter | 2007-08-24 21:13:58 +0200 (Fri, 24 Aug 2007) | 1 line
Convert test_pkg to use unittest.
........
r57408 | georg.brandl | 2007-08-24 21:22:34 +0200 (Fri, 24 Aug 2007) | 2 lines
Catch the correct errors.
........
r57409 | georg.brandl | 2007-08-24 21:33:53 +0200 (Fri, 24 Aug 2007) | 2 lines
Port test_class to unittest. Patch #1671298.
........
r57415 | collin.winter | 2007-08-24 23:09:42 +0200 (Fri, 24 Aug 2007) | 1 line
Make test_structmembers pass when run with regrtests's -R flag.
........
r57455 | nick.coghlan | 2007-08-25 06:32:07 +0200 (Sat, 25 Aug 2007) | 1 line
Revert misguided attempt at fixing incompatibility between -m and -i switches (better fix coming soon)
........
r57456 | nick.coghlan | 2007-08-25 06:35:54 +0200 (Sat, 25 Aug 2007) | 1 line
Revert compile.c changes that shouldn't have been included in previous checkin
........
r57461 | nick.coghlan | 2007-08-25 12:50:41 +0200 (Sat, 25 Aug 2007) | 1 line
Fix bug 1764407 - the -i switch now does the right thing when using the -m switch
........
r57464 | guido.van.rossum | 2007-08-25 17:08:43 +0200 (Sat, 25 Aug 2007) | 4 lines
Server-side SSL and certificate validation, by Bill Janssen.
While cleaning up Bill's C style, I may have cleaned up some code
he didn't touch as well (in _ssl.c).
........
r57465 | neal.norwitz | 2007-08-25 18:41:36 +0200 (Sat, 25 Aug 2007) | 3 lines
Try to get this to build with Visual Studio by moving all the variable
declarations to the beginning of a scope.
........
r57466 | neal.norwitz | 2007-08-25 18:54:38 +0200 (Sat, 25 Aug 2007) | 1 line
Fix test so it is skipped properly if there is no SSL support.
........
r57467 | neal.norwitz | 2007-08-25 18:58:09 +0200 (Sat, 25 Aug 2007) | 2 lines
Fix a few more variables to try to get this to compile with Visual Studio.
........
r57473 | neal.norwitz | 2007-08-25 19:25:17 +0200 (Sat, 25 Aug 2007) | 1 line
Try to get this test to pass for systems that do not have SO_REUSEPORT
........
r57482 | gregory.p.smith | 2007-08-26 02:26:00 +0200 (Sun, 26 Aug 2007) | 7 lines
keep setup.py from listing unneeded hash modules (_md5, _sha*) as
missing when they were not built because _hashlib with openssl provided
their functionality instead.
don't build bsddb185 if bsddb was built.
........
r57483 | neal.norwitz | 2007-08-26 03:08:16 +0200 (Sun, 26 Aug 2007) | 1 line
Fix typo in docstring (missing c in reacquire)
........
r57484 | neal.norwitz | 2007-08-26 03:42:03 +0200 (Sun, 26 Aug 2007) | 2 lines
Spell check (also americanify behaviour, it's almost 3 times as common)
........
r57503 | neal.norwitz | 2007-08-26 08:29:57 +0200 (Sun, 26 Aug 2007) | 4 lines
Reap children before the test starts so hopefully SocketServer
won't find any old children left around which causes an exception
in collect_children() and the test to fail.
........
r57510 | neal.norwitz | 2007-08-26 20:50:39 +0200 (Sun, 26 Aug 2007) | 1 line
Fail gracefully if the cert files cannot be created
........
r57513 | guido.van.rossum | 2007-08-26 21:35:09 +0200 (Sun, 26 Aug 2007) | 4 lines
Bill Janssen wrote:
Here's a patch which makes test_ssl a better player in the buildbots
environment. I deep-ended on "try-except-else" clauses.
........
r57518 | neal.norwitz | 2007-08-26 23:40:16 +0200 (Sun, 26 Aug 2007) | 1 line
Get the test passing by commenting out some writes (should they be removed?)
........
r57522 | neal.norwitz | 2007-08-27 00:16:23 +0200 (Mon, 27 Aug 2007) | 3 lines
Catch IOError for when the device file doesn't exist or the user doesn't have
permission to write to the device.
........
r57524 | neal.norwitz | 2007-08-27 00:20:03 +0200 (Mon, 27 Aug 2007) | 5 lines
Another patch from Bill Janssen that:
1) Fixes the bug that two class names are initial-lower-case.
2) Replaces the poll waiting for the server to become ready with
a threading.Event signal.
........
r57536 | neal.norwitz | 2007-08-27 02:58:33 +0200 (Mon, 27 Aug 2007) | 1 line
Stop using string.join (from the module) to ease upgrade to py3k
........
r57537 | neal.norwitz | 2007-08-27 03:03:18 +0200 (Mon, 27 Aug 2007) | 1 line
Make a utility function for handling (printing) an error
........
r57538 | neal.norwitz | 2007-08-27 03:15:33 +0200 (Mon, 27 Aug 2007) | 4 lines
If we can't create a certificate, print a warning, but don't fail the test.
Modified patch from what Bill Janssen sent on python-3000.
........
r57539 | facundo.batista | 2007-08-27 03:15:34 +0200 (Mon, 27 Aug 2007) | 7 lines
Ignore test failures caused by 'resource temporarily unavailable'
exceptions raised in the test server thread, since SimpleXMLRPCServer
does not gracefully handle them. Changed number of requests handled
by tests server thread to one (was 2) because no tests require more
than one request. [GSoC - Alan McIntyre]
........
r57561 | guido.van.rossum | 2007-08-27 19:19:42 +0200 (Mon, 27 Aug 2007) | 8 lines
> Regardless, building a fixed test certificate and checking it in sounds like
> the better option. Then the openssl command in the test code can be turned
> into a comment describing how the test data was pregenerated.
Here's a patch that does that.
Bill
........
r57568 | guido.van.rossum | 2007-08-27 20:42:23 +0200 (Mon, 27 Aug 2007) | 26 lines
> Some of the code sets the error string in this directly before
> returning NULL, and other pieces of the code call PySSL_SetError,
> which creates the error string. I think some of the places which set
> the string directly probably shouldn't; instead, they should call
> PySSL_SetError to cons up the error name directly from the err code.
> However, PySSL_SetError only works after the construction of an ssl
> object, which means it can't be used there... I'll take a longer look
> at it and see if there's a reasonable fix.
Here's a patch which addresses this. It also fixes the indentation in
PySSL_SetError, bringing it into line with PEP 7, fixes a compile warning
about one of the OpenSSL macros, and makes the namespace a bit more
consistent. I've tested it on FC 7 and OS X 10.4.
% ./python ./Lib/test/regrtest.py -R :1: -u all test_ssl
test_ssl
beginning 6 repetitions
123456
......
1 test OK.
[29244 refs]
%
[GvR: slightly edited to enforce 79-char line length, even if it required
violating the style guide.]
........
r57570 | guido.van.rossum | 2007-08-27 21:11:11 +0200 (Mon, 27 Aug 2007) | 2 lines
Patch 10124 by Bill Janssen, docs for the new ssl code.
........
r57574 | guido.van.rossum | 2007-08-27 22:51:00 +0200 (Mon, 27 Aug 2007) | 3 lines
Patch # 1739906 by Christian Heimes -- add reduce to functools (importing
it from __builtin__).
........
r57575 | guido.van.rossum | 2007-08-27 22:52:10 +0200 (Mon, 27 Aug 2007) | 2 lines
News about functools.reduce.
........
r57611 | georg.brandl | 2007-08-28 10:29:08 +0200 (Tue, 28 Aug 2007) | 2 lines
Document rev. 57574.
........
r57612 | sean.reifschneider | 2007-08-28 11:07:54 +0200 (Tue, 28 Aug 2007) | 2 lines
Adding basic imputil documentation.
........
r57614 | georg.brandl | 2007-08-28 12:48:18 +0200 (Tue, 28 Aug 2007) | 2 lines
Fix some glitches.
........
r57616 | lars.gustaebel | 2007-08-28 14:31:09 +0200 (Tue, 28 Aug 2007) | 5 lines
TarFile.__init__() no longer fails if no name argument is passed and
the fileobj argument has no usable name attribute (e.g. StringIO).
(will backport to 2.5)
........
r57619 | thomas.wouters | 2007-08-28 17:28:19 +0200 (Tue, 28 Aug 2007) | 22 lines
Improve extended slicing support in builtin types and classes. Specifically:
- Specialcase extended slices that amount to a shallow copy the same way as
is done for simple slices, in the tuple, string and unicode case.
- Specialcase step-1 extended slices to optimize the common case for all
involved types.
- For lists, allow extended slice assignment of differing lengths as long
as the step is 1. (Previously, 'l[:2:1] = []' failed even though
'l[:2] = []' and 'l[:2:None] = []' do not.)
- Implement extended slicing for buffer, array, structseq, mmap and
UserString.UserString.
- Implement slice-object support (but not non-step-1 slice assignment) for
UserString.MutableString.
- Add tests for all new functionality.
........
2007-08-29 05:37:11 +08:00
|
|
|
}
|
|
|
|
|
2010-12-04 04:14:31 +08:00
|
|
|
void
|
2023-06-28 09:45:57 +08:00
|
|
|
PyStructSequence_SetItem(PyObject *op, Py_ssize_t index, PyObject *value)
|
2010-12-04 04:14:31 +08:00
|
|
|
{
|
2023-06-28 09:45:57 +08:00
|
|
|
PyTupleObject *tuple = _PyTuple_CAST(op);
|
|
|
|
assert(0 <= index);
|
|
|
|
#ifndef NDEBUG
|
|
|
|
Py_ssize_t n_fields = REAL_SIZE(op);
|
|
|
|
assert(n_fields >= 0);
|
|
|
|
assert(index < n_fields);
|
|
|
|
#endif
|
|
|
|
tuple->ob_item[index] = value;
|
2010-12-04 04:14:31 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
PyObject*
|
2023-06-28 09:45:57 +08:00
|
|
|
PyStructSequence_GetItem(PyObject *op, Py_ssize_t index)
|
2010-12-04 04:14:31 +08:00
|
|
|
{
|
2023-06-28 09:45:57 +08:00
|
|
|
assert(0 <= index);
|
|
|
|
#ifndef NDEBUG
|
|
|
|
Py_ssize_t n_fields = REAL_SIZE(op);
|
|
|
|
assert(n_fields >= 0);
|
|
|
|
assert(index < n_fields);
|
|
|
|
#endif
|
|
|
|
return PyTuple_GET_ITEM(op, index);
|
2010-12-04 04:14:31 +08:00
|
|
|
}
|
|
|
|
|
2019-06-02 22:45:13 +08:00
|
|
|
|
|
|
|
static int
|
|
|
|
structseq_traverse(PyStructSequence *obj, visitproc visit, void *arg)
|
|
|
|
{
|
2020-05-27 17:03:38 +08:00
|
|
|
if (Py_TYPE(obj)->tp_flags & Py_TPFLAGS_HEAPTYPE) {
|
|
|
|
Py_VISIT(Py_TYPE(obj));
|
|
|
|
}
|
2019-06-02 22:45:13 +08:00
|
|
|
Py_ssize_t i, size;
|
|
|
|
size = REAL_SIZE(obj);
|
|
|
|
for (i = 0; i < size; ++i) {
|
|
|
|
Py_VISIT(obj->ob_item[i]);
|
|
|
|
}
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2010-07-09 06:33:03 +08:00
|
|
|
static void
|
|
|
|
structseq_dealloc(PyStructSequence *obj)
|
|
|
|
{
|
|
|
|
Py_ssize_t i, size;
|
2019-06-03 04:52:49 +08:00
|
|
|
PyObject_GC_UnTrack(obj);
|
2013-07-17 07:22:45 +08:00
|
|
|
|
2022-01-22 06:33:43 +08:00
|
|
|
PyTypeObject *tp = Py_TYPE(obj);
|
2024-08-03 00:11:44 +08:00
|
|
|
// gh-122527: We can't use REAL_SIZE_TP() or any macros that access the
|
|
|
|
// type's dictionary here, because the dictionary may have already been
|
|
|
|
// cleared by the garbage collector.
|
2010-07-09 06:33:03 +08:00
|
|
|
size = REAL_SIZE(obj);
|
|
|
|
for (i = 0; i < size; ++i) {
|
|
|
|
Py_XDECREF(obj->ob_item[i]);
|
|
|
|
}
|
|
|
|
PyObject_GC_Del(obj);
|
2020-09-23 20:08:38 +08:00
|
|
|
if (_PyType_HasFeature(tp, Py_TPFLAGS_HEAPTYPE)) {
|
2019-03-27 19:52:18 +08:00
|
|
|
Py_DECREF(tp);
|
|
|
|
}
|
2010-07-09 06:33:03 +08:00
|
|
|
}
|
|
|
|
|
2017-03-19 14:51:07 +08:00
|
|
|
/*[clinic input]
|
|
|
|
class structseq "PyStructSequence *" "NULL"
|
|
|
|
[clinic start generated code]*/
|
|
|
|
/*[clinic end generated code: output=da39a3ee5e6b4b0d input=9d781c6922c77752]*/
|
|
|
|
|
|
|
|
#include "clinic/structseq.c.h"
|
|
|
|
|
|
|
|
/*[clinic input]
|
|
|
|
@classmethod
|
|
|
|
structseq.__new__ as structseq_new
|
|
|
|
sequence as arg: object
|
2019-09-14 17:24:05 +08:00
|
|
|
dict: object(c_default="NULL") = {}
|
2017-03-19 14:51:07 +08:00
|
|
|
[clinic start generated code]*/
|
|
|
|
|
2001-10-19 04:47:51 +08:00
|
|
|
static PyObject *
|
2017-03-19 14:51:07 +08:00
|
|
|
structseq_new_impl(PyTypeObject *type, PyObject *arg, PyObject *dict)
|
2019-09-14 17:24:05 +08:00
|
|
|
/*[clinic end generated code: output=baa082e788b171da input=90532511101aa3fb]*/
|
2001-10-19 04:47:51 +08:00
|
|
|
{
|
2010-05-09 23:52:27 +08:00
|
|
|
PyStructSequence *res = NULL;
|
|
|
|
Py_ssize_t len, min_len, max_len, i, n_unnamed_fields;
|
|
|
|
|
2020-10-26 14:43:39 +08:00
|
|
|
min_len = VISIBLE_SIZE_TP(type);
|
|
|
|
if (min_len < 0) {
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
max_len = REAL_SIZE_TP(type);
|
|
|
|
if (max_len < 0) {
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
n_unnamed_fields = UNNAMED_FIELDS_TP(type);
|
|
|
|
if (n_unnamed_fields < 0) {
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
2010-05-09 23:52:27 +08:00
|
|
|
arg = PySequence_Fast(arg, "constructor requires a sequence");
|
|
|
|
|
|
|
|
if (!arg) {
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (dict && !PyDict_Check(dict)) {
|
|
|
|
PyErr_Format(PyExc_TypeError,
|
|
|
|
"%.500s() takes a dict as second arg, if any",
|
|
|
|
type->tp_name);
|
|
|
|
Py_DECREF(arg);
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
len = PySequence_Fast_GET_SIZE(arg);
|
|
|
|
if (min_len != max_len) {
|
|
|
|
if (len < min_len) {
|
|
|
|
PyErr_Format(PyExc_TypeError,
|
2012-02-15 09:52:58 +08:00
|
|
|
"%.500s() takes an at least %zd-sequence (%zd-sequence given)",
|
|
|
|
type->tp_name, min_len, len);
|
|
|
|
Py_DECREF(arg);
|
|
|
|
return NULL;
|
2010-05-09 23:52:27 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
if (len > max_len) {
|
|
|
|
PyErr_Format(PyExc_TypeError,
|
2012-02-15 09:52:58 +08:00
|
|
|
"%.500s() takes an at most %zd-sequence (%zd-sequence given)",
|
|
|
|
type->tp_name, max_len, len);
|
|
|
|
Py_DECREF(arg);
|
|
|
|
return NULL;
|
2010-05-09 23:52:27 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
else {
|
|
|
|
if (len != min_len) {
|
|
|
|
PyErr_Format(PyExc_TypeError,
|
2012-02-15 09:52:58 +08:00
|
|
|
"%.500s() takes a %zd-sequence (%zd-sequence given)",
|
|
|
|
type->tp_name, min_len, len);
|
|
|
|
Py_DECREF(arg);
|
|
|
|
return NULL;
|
2010-05-09 23:52:27 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
res = (PyStructSequence*) PyStructSequence_New(type);
|
|
|
|
if (res == NULL) {
|
2012-02-15 09:51:43 +08:00
|
|
|
Py_DECREF(arg);
|
2010-05-09 23:52:27 +08:00
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
for (i = 0; i < len; ++i) {
|
|
|
|
PyObject *v = PySequence_Fast_GET_ITEM(arg, i);
|
2022-11-11 06:58:07 +08:00
|
|
|
res->ob_item[i] = Py_NewRef(v);
|
2010-05-09 23:52:27 +08:00
|
|
|
}
|
2020-10-26 14:43:39 +08:00
|
|
|
Py_DECREF(arg);
|
2023-10-05 00:44:17 +08:00
|
|
|
if (dict != NULL && PyDict_GET_SIZE(dict) > 0) {
|
|
|
|
Py_ssize_t n_found_keys = 0;
|
|
|
|
for (i = len; i < max_len; ++i) {
|
|
|
|
PyObject *ob = NULL;
|
|
|
|
const char *name = type->tp_members[i - n_unnamed_fields].name;
|
2023-08-24 04:59:00 +08:00
|
|
|
if (PyDict_GetItemStringRef(dict, name, &ob) < 0) {
|
|
|
|
Py_DECREF(res);
|
|
|
|
return NULL;
|
2020-10-26 14:43:39 +08:00
|
|
|
}
|
2023-10-05 00:44:17 +08:00
|
|
|
if (ob == NULL) {
|
|
|
|
ob = Py_NewRef(Py_None);
|
|
|
|
}
|
|
|
|
else {
|
|
|
|
++n_found_keys;
|
|
|
|
}
|
|
|
|
res->ob_item[i] = ob;
|
|
|
|
}
|
|
|
|
if (PyDict_GET_SIZE(dict) > n_found_keys) {
|
|
|
|
PyErr_Format(PyExc_TypeError,
|
|
|
|
"%.500s() got duplicate or unexpected field name(s)",
|
|
|
|
type->tp_name);
|
|
|
|
Py_DECREF(res);
|
|
|
|
return NULL;
|
2010-05-09 23:52:27 +08:00
|
|
|
}
|
2023-10-05 00:44:17 +08:00
|
|
|
} else {
|
|
|
|
for (i = len; i < max_len; ++i) {
|
|
|
|
res->ob_item[i] = Py_NewRef(Py_None);
|
2023-08-24 04:59:00 +08:00
|
|
|
}
|
2010-05-09 23:52:27 +08:00
|
|
|
}
|
|
|
|
|
2019-06-02 22:45:13 +08:00
|
|
|
_PyObject_GC_TRACK(res);
|
2010-05-09 23:52:27 +08:00
|
|
|
return (PyObject*) res;
|
2001-10-19 04:47:51 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
static PyObject *
|
|
|
|
structseq_repr(PyStructSequence *obj)
|
|
|
|
{
|
2010-05-09 23:52:27 +08:00
|
|
|
PyTypeObject *typ = Py_TYPE(obj);
|
2019-03-20 07:05:51 +08:00
|
|
|
|
2024-10-10 06:04:50 +08:00
|
|
|
// count 5 characters per item: "x=1, "
|
|
|
|
Py_ssize_t type_name_len = strlen(typ->tp_name);
|
|
|
|
Py_ssize_t prealloc = (type_name_len + 1
|
|
|
|
+ VISIBLE_SIZE(obj) * 5 + 1);
|
|
|
|
PyUnicodeWriter *writer = PyUnicodeWriter_Create(prealloc);
|
|
|
|
if (writer == NULL) {
|
2019-03-22 15:24:34 +08:00
|
|
|
return NULL;
|
2019-03-20 07:05:51 +08:00
|
|
|
}
|
|
|
|
|
2024-10-10 06:04:50 +08:00
|
|
|
// Write "typename("
|
|
|
|
if (PyUnicodeWriter_WriteUTF8(writer, typ->tp_name, type_name_len) < 0) {
|
2019-03-20 07:05:51 +08:00
|
|
|
goto error;
|
|
|
|
}
|
2024-10-10 06:04:50 +08:00
|
|
|
if (PyUnicodeWriter_WriteChar(writer, '(') < 0) {
|
2019-03-20 07:05:51 +08:00
|
|
|
goto error;
|
|
|
|
}
|
|
|
|
|
|
|
|
for (Py_ssize_t i=0; i < VISIBLE_SIZE(obj); i++) {
|
|
|
|
if (i > 0) {
|
2024-10-10 06:04:50 +08:00
|
|
|
// Write ", "
|
|
|
|
if (PyUnicodeWriter_WriteChar(writer, ',') < 0) {
|
|
|
|
goto error;
|
|
|
|
}
|
|
|
|
if (PyUnicodeWriter_WriteChar(writer, ' ') < 0) {
|
2019-03-20 07:05:51 +08:00
|
|
|
goto error;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2024-10-10 06:04:50 +08:00
|
|
|
// Write name
|
2019-03-20 07:05:51 +08:00
|
|
|
const char *name_utf8 = typ->tp_members[i].name;
|
|
|
|
if (name_utf8 == NULL) {
|
2024-10-10 06:04:50 +08:00
|
|
|
PyErr_Format(PyExc_SystemError,
|
|
|
|
"In structseq_repr(), member %zd name is NULL"
|
2010-07-09 06:33:03 +08:00
|
|
|
" for type %.500s", i, typ->tp_name);
|
2019-03-20 07:05:51 +08:00
|
|
|
goto error;
|
2010-07-09 06:33:03 +08:00
|
|
|
}
|
2024-10-10 06:04:50 +08:00
|
|
|
if (PyUnicodeWriter_WriteUTF8(writer, name_utf8, -1) < 0) {
|
2019-03-20 07:05:51 +08:00
|
|
|
goto error;
|
|
|
|
}
|
2010-05-09 23:52:27 +08:00
|
|
|
|
2024-10-10 06:04:50 +08:00
|
|
|
// Write "=" + repr(value)
|
|
|
|
if (PyUnicodeWriter_WriteChar(writer, '=') < 0) {
|
2019-03-20 07:05:51 +08:00
|
|
|
goto error;
|
2010-05-09 23:52:27 +08:00
|
|
|
}
|
2023-06-28 09:45:57 +08:00
|
|
|
PyObject *value = PyStructSequence_GetItem((PyObject*)obj, i);
|
2019-03-20 07:05:51 +08:00
|
|
|
assert(value != NULL);
|
2024-10-10 06:04:50 +08:00
|
|
|
if (PyUnicodeWriter_WriteRepr(writer, value) < 0) {
|
2019-03-20 07:05:51 +08:00
|
|
|
goto error;
|
2010-05-09 23:52:27 +08:00
|
|
|
}
|
|
|
|
}
|
2019-03-20 07:05:51 +08:00
|
|
|
|
2024-10-10 06:04:50 +08:00
|
|
|
if (PyUnicodeWriter_WriteChar(writer, ')') < 0) {
|
2019-03-20 07:05:51 +08:00
|
|
|
goto error;
|
2010-05-09 23:52:27 +08:00
|
|
|
}
|
|
|
|
|
2024-10-10 06:04:50 +08:00
|
|
|
return PyUnicodeWriter_Finish(writer);
|
2019-03-20 07:05:51 +08:00
|
|
|
|
|
|
|
error:
|
2024-10-10 06:04:50 +08:00
|
|
|
PyUnicodeWriter_Discard(writer);
|
2019-03-20 07:05:51 +08:00
|
|
|
return NULL;
|
2001-10-19 04:47:51 +08:00
|
|
|
}
|
|
|
|
|
2019-03-20 07:05:51 +08:00
|
|
|
|
2002-03-05 21:27:58 +08:00
|
|
|
static PyObject *
|
2018-04-30 02:59:33 +08:00
|
|
|
structseq_reduce(PyStructSequence* self, PyObject *Py_UNUSED(ignored))
|
2002-03-05 21:27:58 +08:00
|
|
|
{
|
2013-07-17 19:41:39 +08:00
|
|
|
PyObject* tup = NULL;
|
|
|
|
PyObject* dict = NULL;
|
2010-05-09 23:52:27 +08:00
|
|
|
PyObject* result;
|
2015-09-07 02:25:30 +08:00
|
|
|
Py_ssize_t n_fields, n_visible_fields, n_unnamed_fields, i;
|
2010-05-09 23:52:27 +08:00
|
|
|
|
|
|
|
n_fields = REAL_SIZE(self);
|
2020-10-26 14:43:39 +08:00
|
|
|
if (n_fields < 0) {
|
|
|
|
return NULL;
|
|
|
|
}
|
2010-05-09 23:52:27 +08:00
|
|
|
n_visible_fields = VISIBLE_SIZE(self);
|
|
|
|
n_unnamed_fields = UNNAMED_FIELDS(self);
|
2020-10-26 14:43:39 +08:00
|
|
|
if (n_unnamed_fields < 0) {
|
|
|
|
return NULL;
|
|
|
|
}
|
2019-02-26 00:59:12 +08:00
|
|
|
tup = _PyTuple_FromArray(self->ob_item, n_visible_fields);
|
2013-07-17 19:41:39 +08:00
|
|
|
if (!tup)
|
|
|
|
goto error;
|
2010-05-09 23:52:27 +08:00
|
|
|
|
|
|
|
dict = PyDict_New();
|
2013-07-17 19:41:39 +08:00
|
|
|
if (!dict)
|
|
|
|
goto error;
|
2010-05-09 23:52:27 +08:00
|
|
|
|
2019-02-26 00:59:12 +08:00
|
|
|
for (i = n_visible_fields; i < n_fields; i++) {
|
2016-11-22 13:58:08 +08:00
|
|
|
const char *n = Py_TYPE(self)->tp_members[i-n_unnamed_fields].name;
|
2013-07-17 19:41:39 +08:00
|
|
|
if (PyDict_SetItemString(dict, n, self->ob_item[i]) < 0)
|
|
|
|
goto error;
|
2010-05-09 23:52:27 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
result = Py_BuildValue("(O(OO))", Py_TYPE(self), tup, dict);
|
|
|
|
|
|
|
|
Py_DECREF(tup);
|
|
|
|
Py_DECREF(dict);
|
|
|
|
|
|
|
|
return result;
|
2013-07-17 19:41:39 +08:00
|
|
|
|
|
|
|
error:
|
|
|
|
Py_XDECREF(tup);
|
|
|
|
Py_XDECREF(dict);
|
|
|
|
return NULL;
|
2002-03-05 21:27:58 +08:00
|
|
|
}
|
|
|
|
|
2023-10-05 00:47:41 +08:00
|
|
|
|
|
|
|
static PyObject *
|
|
|
|
structseq_replace(PyStructSequence *self, PyObject *args, PyObject *kwargs)
|
|
|
|
{
|
|
|
|
PyStructSequence *result = NULL;
|
|
|
|
Py_ssize_t n_fields, n_unnamed_fields, i;
|
|
|
|
|
|
|
|
if (!_PyArg_NoPositional("__replace__", args)) {
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
n_fields = REAL_SIZE(self);
|
|
|
|
if (n_fields < 0) {
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
n_unnamed_fields = UNNAMED_FIELDS(self);
|
|
|
|
if (n_unnamed_fields < 0) {
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
if (n_unnamed_fields > 0) {
|
|
|
|
PyErr_Format(PyExc_TypeError,
|
|
|
|
"__replace__() is not supported for %.500s "
|
|
|
|
"because it has unnamed field(s)",
|
|
|
|
Py_TYPE(self)->tp_name);
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
result = (PyStructSequence *) PyStructSequence_New(Py_TYPE(self));
|
|
|
|
if (!result) {
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (kwargs != NULL) {
|
|
|
|
// We do not support types with unnamed fields, so we can iterate over
|
|
|
|
// i >= n_visible_fields case without slicing with (i - n_unnamed_fields).
|
|
|
|
for (i = 0; i < n_fields; ++i) {
|
2023-11-14 20:51:00 +08:00
|
|
|
PyObject *ob;
|
|
|
|
if (PyDict_PopString(kwargs, Py_TYPE(self)->tp_members[i].name,
|
|
|
|
&ob) < 0) {
|
2023-10-05 00:47:41 +08:00
|
|
|
goto error;
|
|
|
|
}
|
2023-11-14 20:51:00 +08:00
|
|
|
if (ob == NULL) {
|
|
|
|
ob = Py_NewRef(self->ob_item[i]);
|
2023-10-05 00:47:41 +08:00
|
|
|
}
|
|
|
|
result->ob_item[i] = ob;
|
|
|
|
}
|
|
|
|
// Check if there are any unexpected fields.
|
|
|
|
if (PyDict_GET_SIZE(kwargs) > 0) {
|
|
|
|
PyObject *names = PyDict_Keys(kwargs);
|
|
|
|
if (names) {
|
|
|
|
PyErr_Format(PyExc_TypeError, "Got unexpected field name(s): %R", names);
|
|
|
|
Py_DECREF(names);
|
|
|
|
}
|
|
|
|
goto error;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
// Just create a copy of the original.
|
|
|
|
for (i = 0; i < n_fields; ++i) {
|
|
|
|
result->ob_item[i] = Py_NewRef(self->ob_item[i]);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return (PyObject *)result;
|
|
|
|
|
|
|
|
error:
|
|
|
|
Py_DECREF(result);
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
2002-03-05 21:27:58 +08:00
|
|
|
static PyMethodDef structseq_methods[] = {
|
2010-07-08 04:54:01 +08:00
|
|
|
{"__reduce__", (PyCFunction)structseq_reduce, METH_NOARGS, NULL},
|
2024-04-12 16:46:20 +08:00
|
|
|
{"__replace__", _PyCFunction_CAST(structseq_replace), METH_VARARGS | METH_KEYWORDS,
|
|
|
|
PyDoc_STR("__replace__($self, /, **changes)\n--\n\n"
|
|
|
|
"Return a copy of the structure with new values for the specified fields.")},
|
2023-10-05 00:47:41 +08:00
|
|
|
{NULL, NULL} // sentinel
|
2002-03-05 21:27:58 +08:00
|
|
|
};
|
|
|
|
|
2018-11-13 20:09:31 +08:00
|
|
|
static Py_ssize_t
|
|
|
|
count_members(PyStructSequence_Desc *desc, Py_ssize_t *n_unnamed_members) {
|
|
|
|
Py_ssize_t i;
|
|
|
|
|
|
|
|
*n_unnamed_members = 0;
|
|
|
|
for (i = 0; desc->fields[i].name != NULL; ++i) {
|
|
|
|
if (desc->fields[i].name == PyStructSequence_UnnamedField) {
|
|
|
|
(*n_unnamed_members)++;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return i;
|
|
|
|
}
|
|
|
|
|
|
|
|
static int
|
|
|
|
initialize_structseq_dict(PyStructSequence_Desc *desc, PyObject* dict,
|
|
|
|
Py_ssize_t n_members, Py_ssize_t n_unnamed_members) {
|
|
|
|
PyObject *v;
|
|
|
|
|
|
|
|
#define SET_DICT_FROM_SIZE(key, value) \
|
|
|
|
do { \
|
|
|
|
v = PyLong_FromSsize_t(value); \
|
|
|
|
if (v == NULL) { \
|
|
|
|
return -1; \
|
|
|
|
} \
|
|
|
|
if (PyDict_SetItemString(dict, key, v) < 0) { \
|
|
|
|
Py_DECREF(v); \
|
|
|
|
return -1; \
|
|
|
|
} \
|
|
|
|
Py_DECREF(v); \
|
|
|
|
} while (0)
|
|
|
|
|
|
|
|
SET_DICT_FROM_SIZE(visible_length_key, desc->n_in_sequence);
|
|
|
|
SET_DICT_FROM_SIZE(real_length_key, n_members);
|
|
|
|
SET_DICT_FROM_SIZE(unnamed_fields_key, n_unnamed_members);
|
2021-03-04 08:03:44 +08:00
|
|
|
|
|
|
|
// Prepare and set __match_args__
|
|
|
|
Py_ssize_t i, k;
|
|
|
|
PyObject* keys = PyTuple_New(desc->n_in_sequence);
|
|
|
|
if (keys == NULL) {
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
for (i = k = 0; i < desc->n_in_sequence; ++i) {
|
|
|
|
if (desc->fields[i].name == PyStructSequence_UnnamedField) {
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
PyObject* new_member = PyUnicode_FromString(desc->fields[i].name);
|
|
|
|
if (new_member == NULL) {
|
|
|
|
goto error;
|
|
|
|
}
|
|
|
|
PyTuple_SET_ITEM(keys, k, new_member);
|
|
|
|
k++;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (_PyTuple_Resize(&keys, k) == -1) {
|
|
|
|
goto error;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (PyDict_SetItemString(dict, match_args_key, keys) < 0) {
|
|
|
|
goto error;
|
|
|
|
}
|
|
|
|
|
|
|
|
Py_DECREF(keys);
|
2018-11-13 20:09:31 +08:00
|
|
|
return 0;
|
2021-03-04 08:03:44 +08:00
|
|
|
|
|
|
|
error:
|
|
|
|
Py_DECREF(keys);
|
|
|
|
return -1;
|
2018-11-13 20:09:31 +08:00
|
|
|
}
|
|
|
|
|
2022-07-26 02:47:31 +08:00
|
|
|
static PyMemberDef *
|
|
|
|
initialize_members(PyStructSequence_Desc *desc,
|
2023-05-02 05:08:34 +08:00
|
|
|
Py_ssize_t n_members, Py_ssize_t n_unnamed_members)
|
2022-07-26 02:47:31 +08:00
|
|
|
{
|
|
|
|
PyMemberDef *members;
|
|
|
|
|
|
|
|
members = PyMem_NEW(PyMemberDef, n_members - n_unnamed_members + 1);
|
|
|
|
if (members == NULL) {
|
|
|
|
PyErr_NoMemory();
|
|
|
|
return NULL;
|
|
|
|
}
|
2018-11-13 20:09:31 +08:00
|
|
|
|
2022-07-26 02:47:31 +08:00
|
|
|
Py_ssize_t i, k;
|
2018-11-13 20:09:31 +08:00
|
|
|
for (i = k = 0; i < n_members; ++i) {
|
|
|
|
if (desc->fields[i].name == PyStructSequence_UnnamedField) {
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* The names and docstrings in these MemberDefs are statically */
|
|
|
|
/* allocated so it is expected that they'll outlive the MemberDef */
|
|
|
|
members[k].name = desc->fields[i].name;
|
2023-07-25 21:28:30 +08:00
|
|
|
members[k].type = _Py_T_OBJECT;
|
2018-11-13 20:09:31 +08:00
|
|
|
members[k].offset = offsetof(PyStructSequence, ob_item)
|
|
|
|
+ i * sizeof(PyObject*);
|
2023-07-25 21:28:30 +08:00
|
|
|
members[k].flags = Py_READONLY;
|
2018-11-13 20:09:31 +08:00
|
|
|
members[k].doc = desc->fields[i].doc;
|
|
|
|
k++;
|
|
|
|
}
|
|
|
|
members[k].name = NULL;
|
2022-07-26 02:47:31 +08:00
|
|
|
|
|
|
|
return members;
|
2018-11-13 20:09:31 +08:00
|
|
|
}
|
2001-10-19 04:47:51 +08:00
|
|
|
|
2021-04-30 18:46:15 +08:00
|
|
|
|
2022-07-26 02:47:31 +08:00
|
|
|
static void
|
|
|
|
initialize_static_fields(PyTypeObject *type, PyStructSequence_Desc *desc,
|
2024-08-03 00:11:44 +08:00
|
|
|
PyMemberDef *tp_members, Py_ssize_t n_members,
|
|
|
|
unsigned long tp_flags)
|
2001-10-19 04:47:51 +08:00
|
|
|
{
|
2010-05-09 23:52:27 +08:00
|
|
|
type->tp_name = desc->name;
|
2024-08-03 00:11:44 +08:00
|
|
|
// Account for hidden members in tp_basicsize because they are not
|
|
|
|
// included in the variable size.
|
|
|
|
Py_ssize_t n_hidden = n_members - desc->n_in_sequence;
|
|
|
|
type->tp_basicsize = sizeof(PyStructSequence) + (n_hidden - 1) * sizeof(PyObject *);
|
2018-11-13 20:09:31 +08:00
|
|
|
type->tp_itemsize = sizeof(PyObject *);
|
|
|
|
type->tp_dealloc = (destructor)structseq_dealloc;
|
|
|
|
type->tp_repr = (reprfunc)structseq_repr;
|
2010-05-09 23:52:27 +08:00
|
|
|
type->tp_doc = desc->doc;
|
2018-11-13 20:09:31 +08:00
|
|
|
type->tp_base = &PyTuple_Type;
|
|
|
|
type->tp_methods = structseq_methods;
|
|
|
|
type->tp_new = structseq_new;
|
2021-04-30 18:46:15 +08:00
|
|
|
type->tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC | tp_flags;
|
2019-06-02 22:45:13 +08:00
|
|
|
type->tp_traverse = (traverseproc) structseq_traverse;
|
2022-07-26 02:47:31 +08:00
|
|
|
type->tp_members = tp_members;
|
|
|
|
}
|
2010-05-09 23:52:27 +08:00
|
|
|
|
2022-07-26 02:47:31 +08:00
|
|
|
static int
|
|
|
|
initialize_static_type(PyTypeObject *type, PyStructSequence_Desc *desc,
|
|
|
|
Py_ssize_t n_members, Py_ssize_t n_unnamed_members) {
|
|
|
|
/* initialize_static_fields() should have been called already. */
|
2018-11-13 20:09:31 +08:00
|
|
|
if (PyType_Ready(type) < 0) {
|
2013-07-23 04:24:54 +08:00
|
|
|
return -1;
|
2018-11-13 20:09:31 +08:00
|
|
|
}
|
2010-05-09 23:52:27 +08:00
|
|
|
Py_INCREF(type);
|
|
|
|
|
2018-11-13 20:09:31 +08:00
|
|
|
if (initialize_structseq_dict(
|
2023-05-02 10:34:43 +08:00
|
|
|
desc, _PyType_GetDict(type), n_members, n_unnamed_members) < 0) {
|
2018-11-13 20:09:31 +08:00
|
|
|
Py_DECREF(type);
|
|
|
|
return -1;
|
|
|
|
}
|
2013-07-23 04:24:54 +08:00
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2022-07-26 02:47:31 +08:00
|
|
|
int
|
2023-05-02 09:36:00 +08:00
|
|
|
_PyStructSequence_InitBuiltinWithFlags(PyInterpreterState *interp,
|
|
|
|
PyTypeObject *type,
|
2022-07-26 02:47:31 +08:00
|
|
|
PyStructSequence_Desc *desc,
|
|
|
|
unsigned long tp_flags)
|
|
|
|
{
|
2024-03-20 16:54:42 +08:00
|
|
|
if (Py_TYPE(type) == NULL) {
|
|
|
|
Py_SET_TYPE(type, &PyType_Type);
|
|
|
|
}
|
2023-05-02 05:08:34 +08:00
|
|
|
Py_ssize_t n_unnamed_members;
|
|
|
|
Py_ssize_t n_members = count_members(desc, &n_unnamed_members);
|
|
|
|
PyMemberDef *members = NULL;
|
|
|
|
|
|
|
|
if ((type->tp_flags & Py_TPFLAGS_READY) == 0) {
|
|
|
|
assert(type->tp_name == NULL);
|
|
|
|
assert(type->tp_members == NULL);
|
|
|
|
assert(type->tp_base == NULL);
|
|
|
|
|
|
|
|
members = initialize_members(desc, n_members, n_unnamed_members);
|
|
|
|
if (members == NULL) {
|
|
|
|
goto error;
|
2023-04-28 06:19:43 +08:00
|
|
|
}
|
2024-08-03 00:11:44 +08:00
|
|
|
initialize_static_fields(type, desc, members, n_members, tp_flags);
|
2023-04-28 06:19:43 +08:00
|
|
|
|
2024-03-20 16:54:42 +08:00
|
|
|
_Py_SetImmortal((PyObject *)type);
|
2023-05-02 05:08:34 +08:00
|
|
|
}
|
|
|
|
#ifndef NDEBUG
|
|
|
|
else {
|
|
|
|
// Ensure that the type was initialized.
|
|
|
|
assert(type->tp_name != NULL);
|
|
|
|
assert(type->tp_members != NULL);
|
|
|
|
assert(type->tp_base == &PyTuple_Type);
|
|
|
|
assert((type->tp_flags & _Py_TPFLAGS_STATIC_BUILTIN));
|
|
|
|
assert(_Py_IsImmortal(type));
|
2022-07-26 02:47:31 +08:00
|
|
|
}
|
2023-05-02 05:08:34 +08:00
|
|
|
#endif
|
2023-04-28 06:19:43 +08:00
|
|
|
|
2023-05-02 09:36:00 +08:00
|
|
|
if (_PyStaticType_InitBuiltin(interp, type) < 0) {
|
2023-05-02 05:08:34 +08:00
|
|
|
PyErr_Format(PyExc_RuntimeError,
|
|
|
|
"Can't initialize builtin type %s",
|
|
|
|
desc->name);
|
|
|
|
goto error;
|
|
|
|
}
|
2023-04-28 06:19:43 +08:00
|
|
|
|
|
|
|
if (initialize_structseq_dict(
|
2023-05-03 11:30:03 +08:00
|
|
|
desc, _PyType_GetDict(type), n_members, n_unnamed_members) < 0)
|
|
|
|
{
|
2023-05-02 05:08:34 +08:00
|
|
|
goto error;
|
2022-07-26 02:47:31 +08:00
|
|
|
}
|
2023-05-02 05:08:34 +08:00
|
|
|
|
2022-07-26 02:47:31 +08:00
|
|
|
return 0;
|
2023-04-28 06:19:43 +08:00
|
|
|
|
2023-05-02 05:08:34 +08:00
|
|
|
error:
|
|
|
|
if (members != NULL) {
|
|
|
|
PyMem_Free(members);
|
|
|
|
}
|
2023-04-28 06:19:43 +08:00
|
|
|
return -1;
|
2022-07-26 02:47:31 +08:00
|
|
|
}
|
|
|
|
|
2021-04-30 18:46:15 +08:00
|
|
|
int
|
|
|
|
PyStructSequence_InitType2(PyTypeObject *type, PyStructSequence_Desc *desc)
|
|
|
|
{
|
2022-07-26 02:47:31 +08:00
|
|
|
PyMemberDef *members;
|
|
|
|
Py_ssize_t n_members, n_unnamed_members;
|
|
|
|
|
|
|
|
#ifdef Py_TRACE_REFS
|
2023-09-01 00:33:34 +08:00
|
|
|
/* if the type object was traced, remove it first
|
2022-07-26 02:47:31 +08:00
|
|
|
before overwriting its storage */
|
2023-09-01 00:33:34 +08:00
|
|
|
PyInterpreterState *interp = _PyInterpreterState_GET();
|
|
|
|
if (_PyRefchain_IsTraced(interp, (PyObject *)type)) {
|
2022-07-26 02:47:31 +08:00
|
|
|
_Py_ForgetReference((PyObject *)type);
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
|
|
|
/* PyTypeObject has already been initialized */
|
|
|
|
if (Py_REFCNT(type) != 0) {
|
|
|
|
PyErr_BadInternalCall();
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
2023-05-02 05:08:34 +08:00
|
|
|
n_members = count_members(desc, &n_unnamed_members);
|
|
|
|
members = initialize_members(desc, n_members, n_unnamed_members);
|
2022-07-26 02:47:31 +08:00
|
|
|
if (members == NULL) {
|
|
|
|
return -1;
|
|
|
|
}
|
2024-08-03 00:11:44 +08:00
|
|
|
initialize_static_fields(type, desc, members, n_members, 0);
|
2022-07-26 02:47:31 +08:00
|
|
|
if (initialize_static_type(type, desc, n_members, n_unnamed_members) < 0) {
|
|
|
|
PyMem_Free(members);
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
return 0;
|
2021-04-30 18:46:15 +08:00
|
|
|
}
|
|
|
|
|
2013-07-23 04:24:54 +08:00
|
|
|
void
|
|
|
|
PyStructSequence_InitType(PyTypeObject *type, PyStructSequence_Desc *desc)
|
|
|
|
{
|
|
|
|
(void)PyStructSequence_InitType2(type, desc);
|
2001-10-19 04:47:51 +08:00
|
|
|
}
|
2010-12-04 04:14:31 +08:00
|
|
|
|
2022-01-21 08:42:25 +08:00
|
|
|
|
2023-05-02 05:08:34 +08:00
|
|
|
/* This is exposed in the internal API, not the public API.
|
|
|
|
It is only called on builtin static types, which are all
|
|
|
|
initialized via _PyStructSequence_InitBuiltinWithFlags(). */
|
|
|
|
|
2022-01-21 08:42:25 +08:00
|
|
|
void
|
2023-05-02 09:36:00 +08:00
|
|
|
_PyStructSequence_FiniBuiltin(PyInterpreterState *interp, PyTypeObject *type)
|
2022-01-21 08:42:25 +08:00
|
|
|
{
|
|
|
|
// Ensure that the type is initialized
|
|
|
|
assert(type->tp_name != NULL);
|
|
|
|
assert(type->tp_base == &PyTuple_Type);
|
2023-05-02 05:08:34 +08:00
|
|
|
assert((type->tp_flags & _Py_TPFLAGS_STATIC_BUILTIN));
|
2024-09-03 00:17:48 +08:00
|
|
|
assert(_Py_IsImmortalLoose(type));
|
2022-01-21 08:42:25 +08:00
|
|
|
|
|
|
|
// Cannot delete a type if it still has subclasses
|
2022-08-05 09:26:59 +08:00
|
|
|
if (_PyType_HasSubclasses(type)) {
|
2023-05-02 05:08:34 +08:00
|
|
|
// XXX Shouldn't this be an error?
|
2022-01-21 08:42:25 +08:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2024-06-04 07:09:18 +08:00
|
|
|
_PyStaticType_FiniBuiltin(interp, type);
|
2022-01-21 08:42:25 +08:00
|
|
|
|
2023-05-02 09:36:00 +08:00
|
|
|
if (_Py_IsMainInterpreter(interp)) {
|
|
|
|
// Undo _PyStructSequence_InitBuiltinWithFlags().
|
|
|
|
type->tp_name = NULL;
|
|
|
|
PyMem_Free(type->tp_members);
|
|
|
|
type->tp_members = NULL;
|
|
|
|
type->tp_base = NULL;
|
|
|
|
}
|
2022-01-21 08:42:25 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2018-11-13 20:09:31 +08:00
|
|
|
PyTypeObject *
|
2022-01-21 10:30:20 +08:00
|
|
|
_PyStructSequence_NewType(PyStructSequence_Desc *desc, unsigned long tp_flags)
|
2010-12-04 04:14:31 +08:00
|
|
|
{
|
2018-11-13 20:09:31 +08:00
|
|
|
PyMemberDef *members;
|
|
|
|
PyTypeObject *type;
|
2019-06-02 22:45:13 +08:00
|
|
|
PyType_Slot slots[8];
|
2018-11-13 20:09:31 +08:00
|
|
|
PyType_Spec spec;
|
|
|
|
Py_ssize_t n_members, n_unnamed_members;
|
|
|
|
|
|
|
|
/* Initialize MemberDefs */
|
2023-05-02 05:08:34 +08:00
|
|
|
n_members = count_members(desc, &n_unnamed_members);
|
|
|
|
members = initialize_members(desc, n_members, n_unnamed_members);
|
2018-11-13 20:09:31 +08:00
|
|
|
if (members == NULL) {
|
2013-07-23 04:24:54 +08:00
|
|
|
return NULL;
|
2018-11-13 20:09:31 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
/* Initialize Slots */
|
|
|
|
slots[0] = (PyType_Slot){Py_tp_dealloc, (destructor)structseq_dealloc};
|
|
|
|
slots[1] = (PyType_Slot){Py_tp_repr, (reprfunc)structseq_repr};
|
|
|
|
slots[2] = (PyType_Slot){Py_tp_doc, (void *)desc->doc};
|
|
|
|
slots[3] = (PyType_Slot){Py_tp_methods, structseq_methods};
|
|
|
|
slots[4] = (PyType_Slot){Py_tp_new, structseq_new};
|
|
|
|
slots[5] = (PyType_Slot){Py_tp_members, members};
|
2019-06-02 22:45:13 +08:00
|
|
|
slots[6] = (PyType_Slot){Py_tp_traverse, (traverseproc)structseq_traverse};
|
|
|
|
slots[7] = (PyType_Slot){0, 0};
|
2018-11-13 20:09:31 +08:00
|
|
|
|
|
|
|
/* Initialize Spec */
|
|
|
|
/* The name in this PyType_Spec is statically allocated so it is */
|
|
|
|
/* expected that it'll outlive the PyType_Spec */
|
|
|
|
spec.name = desc->name;
|
2024-08-03 00:11:44 +08:00
|
|
|
Py_ssize_t hidden = n_members - desc->n_in_sequence;
|
|
|
|
spec.basicsize = (int)(sizeof(PyStructSequence) + (hidden - 1) * sizeof(PyObject *));
|
2018-11-13 20:09:31 +08:00
|
|
|
spec.itemsize = sizeof(PyObject *);
|
2022-01-21 10:30:20 +08:00
|
|
|
spec.flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC | tp_flags;
|
2018-11-13 20:09:31 +08:00
|
|
|
spec.slots = slots;
|
|
|
|
|
2020-11-22 19:25:02 +08:00
|
|
|
type = (PyTypeObject *)PyType_FromSpecWithBases(&spec, (PyObject *)&PyTuple_Type);
|
2020-12-01 16:56:42 +08:00
|
|
|
PyMem_Free(members);
|
2018-11-13 20:09:31 +08:00
|
|
|
if (type == NULL) {
|
2013-07-23 04:24:54 +08:00
|
|
|
return NULL;
|
2012-08-19 17:20:41 +08:00
|
|
|
}
|
2018-11-13 20:09:31 +08:00
|
|
|
|
|
|
|
if (initialize_structseq_dict(
|
2023-05-02 10:34:43 +08:00
|
|
|
desc, _PyType_GetDict(type), n_members, n_unnamed_members) < 0) {
|
2018-11-15 10:03:57 +08:00
|
|
|
Py_DECREF(type);
|
2018-11-13 20:09:31 +08:00
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
return type;
|
2010-12-04 04:14:31 +08:00
|
|
|
}
|
2013-07-17 07:22:45 +08:00
|
|
|
|
2021-12-10 03:59:26 +08:00
|
|
|
|
2022-01-21 10:30:20 +08:00
|
|
|
PyTypeObject *
|
|
|
|
PyStructSequence_NewType(PyStructSequence_Desc *desc)
|
|
|
|
{
|
|
|
|
return _PyStructSequence_NewType(desc, 0);
|
|
|
|
}
|