4 PyDoc_STRVAR(operator_doc
,
5 "Operator interface.\n\
7 This module exports a set of functions implemented in C corresponding\n\
8 to the intrinsic operators of Python. For example, operator.add(x, y)\n\
9 is equivalent to the expression x+y. The function names are those\n\
10 used for special class methods; variants without leading and trailing\n\
11 '__' are also provided for convenience.");
13 #define spam1(OP,AOP) static PyObject *OP(PyObject *s, PyObject *a1) { \
16 #define spam2(OP,AOP) static PyObject *OP(PyObject *s, PyObject *a) { \
18 if(! PyArg_UnpackTuple(a,#OP,2,2,&a1,&a2)) return NULL; \
21 #define spamoi(OP,AOP) static PyObject *OP(PyObject *s, PyObject *a) { \
22 PyObject *a1; int a2; \
23 if(! PyArg_ParseTuple(a,"Oi:" #OP,&a1,&a2)) return NULL; \
26 #define spam2n(OP,AOP) static PyObject *OP(PyObject *s, PyObject *a) { \
28 if(! PyArg_UnpackTuple(a,#OP,2,2,&a1,&a2)) return NULL; \
29 if(-1 == AOP(a1,a2)) return NULL; \
33 #define spam3n(OP,AOP) static PyObject *OP(PyObject *s, PyObject *a) { \
34 PyObject *a1, *a2, *a3; \
35 if(! PyArg_UnpackTuple(a,#OP,3,3,&a1,&a2,&a3)) return NULL; \
36 if(-1 == AOP(a1,a2,a3)) return NULL; \
40 #define spami(OP,AOP) static PyObject *OP(PyObject *s, PyObject *a1) { \
42 if(-1 == (r=AOP(a1))) return NULL; \
43 return PyBool_FromLong(r); }
45 #define spami2(OP,AOP) static PyObject *OP(PyObject *s, PyObject *a) { \
46 PyObject *a1, *a2; long r; \
47 if(! PyArg_UnpackTuple(a,#OP,2,2,&a1,&a2)) return NULL; \
48 if(-1 == (r=AOP(a1,a2))) return NULL; \
49 return PyInt_FromLong(r); }
51 #define spamn2(OP,AOP) static PyObject *OP(PyObject *s, PyObject *a) { \
52 PyObject *a1, *a2; Py_ssize_t r; \
53 if(! PyArg_UnpackTuple(a,#OP,2,2,&a1,&a2)) return NULL; \
54 if(-1 == (r=AOP(a1,a2))) return NULL; \
55 return PyInt_FromSsize_t(r); }
57 #define spami2b(OP,AOP) static PyObject *OP(PyObject *s, PyObject *a) { \
58 PyObject *a1, *a2; long r; \
59 if(! PyArg_UnpackTuple(a,#OP,2,2,&a1,&a2)) return NULL; \
60 if(-1 == (r=AOP(a1,a2))) return NULL; \
61 return PyBool_FromLong(r); }
63 #define spamrc(OP,A) static PyObject *OP(PyObject *s, PyObject *a) { \
65 if(! PyArg_UnpackTuple(a,#OP,2,2,&a1,&a2)) return NULL; \
66 return PyObject_RichCompare(a1,a2,A); }
68 spami(isCallable
, PyCallable_Check
)
69 spami(isNumberType
, PyNumber_Check
)
70 spami(truth
, PyObject_IsTrue
)
71 spam2(op_add
, PyNumber_Add
)
72 spam2(op_sub
, PyNumber_Subtract
)
73 spam2(op_mul
, PyNumber_Multiply
)
74 spam2(op_div
, PyNumber_Divide
)
75 spam2(op_floordiv
, PyNumber_FloorDivide
)
76 spam2(op_truediv
, PyNumber_TrueDivide
)
77 spam2(op_mod
, PyNumber_Remainder
)
78 spam1(op_neg
, PyNumber_Negative
)
79 spam1(op_pos
, PyNumber_Positive
)
80 spam1(op_abs
, PyNumber_Absolute
)
81 spam1(op_inv
, PyNumber_Invert
)
82 spam1(op_invert
, PyNumber_Invert
)
83 spam2(op_lshift
, PyNumber_Lshift
)
84 spam2(op_rshift
, PyNumber_Rshift
)
85 spami(op_not_
, PyObject_Not
)
86 spam2(op_and_
, PyNumber_And
)
87 spam2(op_xor
, PyNumber_Xor
)
88 spam2(op_or_
, PyNumber_Or
)
89 spam2(op_iadd
, PyNumber_InPlaceAdd
)
90 spam2(op_isub
, PyNumber_InPlaceSubtract
)
91 spam2(op_imul
, PyNumber_InPlaceMultiply
)
92 spam2(op_idiv
, PyNumber_InPlaceDivide
)
93 spam2(op_ifloordiv
, PyNumber_InPlaceFloorDivide
)
94 spam2(op_itruediv
, PyNumber_InPlaceTrueDivide
)
95 spam2(op_imod
, PyNumber_InPlaceRemainder
)
96 spam2(op_ilshift
, PyNumber_InPlaceLshift
)
97 spam2(op_irshift
, PyNumber_InPlaceRshift
)
98 spam2(op_iand
, PyNumber_InPlaceAnd
)
99 spam2(op_ixor
, PyNumber_InPlaceXor
)
100 spam2(op_ior
, PyNumber_InPlaceOr
)
101 spami(isSequenceType
, PySequence_Check
)
102 spam2(op_concat
, PySequence_Concat
)
103 spamoi(op_repeat
, PySequence_Repeat
)
104 spam2(op_iconcat
, PySequence_InPlaceConcat
)
105 spamoi(op_irepeat
, PySequence_InPlaceRepeat
)
106 spami2b(op_contains
, PySequence_Contains
)
107 spami2b(sequenceIncludes
, PySequence_Contains
)
108 spamn2(indexOf
, PySequence_Index
)
109 spamn2(countOf
, PySequence_Count
)
110 spami(isMappingType
, PyMapping_Check
)
111 spam2(op_getitem
, PyObject_GetItem
)
112 spam2n(op_delitem
, PyObject_DelItem
)
113 spam3n(op_setitem
, PyObject_SetItem
)
114 spamrc(op_lt
, Py_LT
)
115 spamrc(op_le
, Py_LE
)
116 spamrc(op_eq
, Py_EQ
)
117 spamrc(op_ne
, Py_NE
)
118 spamrc(op_gt
, Py_GT
)
119 spamrc(op_ge
, Py_GE
)
122 op_pow(PyObject
*s
, PyObject
*a
)
125 if (PyArg_UnpackTuple(a
,"pow", 2, 2, &a1
, &a2
))
126 return PyNumber_Power(a1
, a2
, Py_None
);
131 op_ipow(PyObject
*s
, PyObject
*a
)
134 if (PyArg_UnpackTuple(a
,"ipow", 2, 2, &a1
, &a2
))
135 return PyNumber_InPlacePower(a1
, a2
, Py_None
);
140 op_index(PyObject
*s
, PyObject
*a
)
142 return PyNumber_Index(a
);
146 is_(PyObject
*s
, PyObject
*a
)
148 PyObject
*a1
, *a2
, *result
= NULL
;
149 if (PyArg_UnpackTuple(a
,"is_", 2, 2, &a1
, &a2
)) {
150 result
= (a1
== a2
) ? Py_True
: Py_False
;
157 is_not(PyObject
*s
, PyObject
*a
)
159 PyObject
*a1
, *a2
, *result
= NULL
;
160 if (PyArg_UnpackTuple(a
,"is_not", 2, 2, &a1
, &a2
)) {
161 result
= (a1
!= a2
) ? Py_True
: Py_False
;
168 op_getslice(PyObject
*s
, PyObject
*a
)
173 if (!PyArg_ParseTuple(a
, "Onn:getslice", &a1
, &a2
, &a3
))
175 return PySequence_GetSlice(a1
, a2
, a3
);
179 op_setslice(PyObject
*s
, PyObject
*a
)
184 if (!PyArg_ParseTuple(a
, "OnnO:setslice", &a1
, &a2
, &a3
, &a4
))
187 if (-1 == PySequence_SetSlice(a1
, a2
, a3
, a4
))
194 op_delslice(PyObject
*s
, PyObject
*a
)
199 if (!PyArg_ParseTuple(a
, "Onn:delslice", &a1
, &a2
, &a3
))
202 if (-1 == PySequence_DelSlice(a1
, a2
, a3
))
212 #define spam1(OP,DOC) {#OP, OP, METH_VARARGS, PyDoc_STR(DOC)},
213 #define spam2(OP,ALTOP,DOC) {#OP, op_##OP, METH_VARARGS, PyDoc_STR(DOC)}, \
214 {#ALTOP, op_##OP, METH_VARARGS, PyDoc_STR(DOC)},
215 #define spam1o(OP,DOC) {#OP, OP, METH_O, PyDoc_STR(DOC)},
216 #define spam2o(OP,ALTOP,DOC) {#OP, op_##OP, METH_O, PyDoc_STR(DOC)}, \
217 {#ALTOP, op_##OP, METH_O, PyDoc_STR(DOC)},
219 static struct PyMethodDef operator_methods
[] = {
222 "isCallable(a) -- Same as callable(a).")
224 "isNumberType(a) -- Return True if a has a numeric type, False otherwise.")
225 spam1o(isSequenceType
,
226 "isSequenceType(a) -- Return True if a has a sequence type, False otherwise.")
228 "truth(a) -- Return True if a is true, False otherwise.")
229 spam2(contains
,__contains__
,
230 "contains(a, b) -- Same as b in a (note reversed operands).")
231 spam1(sequenceIncludes
,
232 "sequenceIncludes(a, b) -- Same as b in a (note reversed operands; deprecated).")
234 "indexOf(a, b) -- Return the first index of b in a.")
236 "countOf(a, b) -- Return the number of times b occurs in a.")
237 spam1o(isMappingType
,
238 "isMappingType(a) -- Return True if a has a mapping type, False otherwise.")
240 spam1(is_
, "is_(a, b) -- Same as a is b.")
241 spam1(is_not
, "is_not(a, b) -- Same as a is not b.")
242 spam2o(index
, __index__
, "index(a) -- Same as a.__index__()")
243 spam2(add
,__add__
, "add(a, b) -- Same as a + b.")
244 spam2(sub
,__sub__
, "sub(a, b) -- Same as a - b.")
245 spam2(mul
,__mul__
, "mul(a, b) -- Same as a * b.")
246 spam2(div
,__div__
, "div(a, b) -- Same as a / b when __future__.division is not in effect.")
247 spam2(floordiv
,__floordiv__
, "floordiv(a, b) -- Same as a // b.")
248 spam2(truediv
,__truediv__
, "truediv(a, b) -- Same as a / b when __future__.division is in effect.")
249 spam2(mod
,__mod__
, "mod(a, b) -- Same as a % b.")
250 spam2o(neg
,__neg__
, "neg(a) -- Same as -a.")
251 spam2o(pos
,__pos__
, "pos(a) -- Same as +a.")
252 spam2o(abs
,__abs__
, "abs(a) -- Same as abs(a).")
253 spam2o(inv
,__inv__
, "inv(a) -- Same as ~a.")
254 spam2o(invert
,__invert__
, "invert(a) -- Same as ~a.")
255 spam2(lshift
,__lshift__
, "lshift(a, b) -- Same as a << b.")
256 spam2(rshift
,__rshift__
, "rshift(a, b) -- Same as a >> b.")
257 spam2o(not_
,__not__
, "not_(a) -- Same as not a.")
258 spam2(and_
,__and__
, "and_(a, b) -- Same as a & b.")
259 spam2(xor,__xor__
, "xor(a, b) -- Same as a ^ b.")
260 spam2(or_
,__or__
, "or_(a, b) -- Same as a | b.")
261 spam2(iadd
,__iadd__
, "iadd(a, b) -- Same as a += b.")
262 spam2(isub
,__isub__
, "isub(a, b) -- Same as a -= b.")
263 spam2(imul
,__imul__
, "imul(a, b) -- Same as a *= b.")
264 spam2(idiv
,__idiv__
, "idiv(a, b) -- Same as a /= b when __future__.division is not in effect.")
265 spam2(ifloordiv
,__ifloordiv__
, "ifloordiv(a, b) -- Same as a //= b.")
266 spam2(itruediv
,__itruediv__
, "itruediv(a, b) -- Same as a /= b when __future__.division is in effect.")
267 spam2(imod
,__imod__
, "imod(a, b) -- Same as a %= b.")
268 spam2(ilshift
,__ilshift__
, "ilshift(a, b) -- Same as a <<= b.")
269 spam2(irshift
,__irshift__
, "irshift(a, b) -- Same as a >>= b.")
270 spam2(iand
,__iand__
, "iand(a, b) -- Same as a &= b.")
271 spam2(ixor
,__ixor__
, "ixor(a, b) -- Same as a ^= b.")
272 spam2(ior
,__ior__
, "ior(a, b) -- Same as a |= b.")
273 spam2(concat
,__concat__
,
274 "concat(a, b) -- Same as a + b, for a and b sequences.")
275 spam2(repeat
,__repeat__
,
276 "repeat(a, b) -- Return a * b, where a is a sequence, and b is an integer.")
277 spam2(iconcat
,__iconcat__
,
278 "iconcat(a, b) -- Same as a += b, for a and b sequences.")
279 spam2(irepeat
,__irepeat__
,
280 "irepeat(a, b) -- Same as a *= b, where a is a sequence, and b is an integer.")
281 spam2(getitem
,__getitem__
,
282 "getitem(a, b) -- Same as a[b].")
283 spam2(setitem
,__setitem__
,
284 "setitem(a, b, c) -- Same as a[b] = c.")
285 spam2(delitem
,__delitem__
,
286 "delitem(a, b) -- Same as del a[b].")
287 spam2(pow
,__pow__
, "pow(a, b) -- Same as a ** b.")
288 spam2(ipow
,__ipow__
, "ipow(a, b) -- Same as a **= b.")
289 spam2(getslice
,__getslice__
,
290 "getslice(a, b, c) -- Same as a[b:c].")
291 spam2(setslice
,__setslice__
,
292 "setslice(a, b, c, d) -- Same as a[b:c] = d.")
293 spam2(delslice
,__delslice__
,
294 "delslice(a, b, c) -- Same as del a[b:c].")
295 spam2(lt
,__lt__
, "lt(a, b) -- Same as a<b.")
296 spam2(le
,__le__
, "le(a, b) -- Same as a<=b.")
297 spam2(eq
,__eq__
, "eq(a, b) -- Same as a==b.")
298 spam2(ne
,__ne__
, "ne(a, b) -- Same as a!=b.")
299 spam2(gt
,__gt__
, "gt(a, b) -- Same as a>b.")
300 spam2(ge
,__ge__
, "ge(a, b) -- Same as a>=b.")
302 {NULL
, NULL
} /* sentinel */
306 /* itemgetter object **********************************************************/
314 static PyTypeObject itemgetter_type
;
317 itemgetter_new(PyTypeObject
*type
, PyObject
*args
, PyObject
*kwds
)
319 itemgetterobject
*ig
;
323 if (!_PyArg_NoKeywords("itemgetter()", kwds
))
326 nitems
= PyTuple_GET_SIZE(args
);
328 if (!PyArg_UnpackTuple(args
, "itemgetter", 1, 1, &item
))
333 /* create itemgetterobject structure */
334 ig
= PyObject_GC_New(itemgetterobject
, &itemgetter_type
);
342 PyObject_GC_Track(ig
);
343 return (PyObject
*)ig
;
347 itemgetter_dealloc(itemgetterobject
*ig
)
349 PyObject_GC_UnTrack(ig
);
350 Py_XDECREF(ig
->item
);
355 itemgetter_traverse(itemgetterobject
*ig
, visitproc visit
, void *arg
)
362 itemgetter_call(itemgetterobject
*ig
, PyObject
*args
, PyObject
*kw
)
364 PyObject
*obj
, *result
;
365 Py_ssize_t i
, nitems
=ig
->nitems
;
367 if (!PyArg_UnpackTuple(args
, "itemgetter", 1, 1, &obj
))
370 return PyObject_GetItem(obj
, ig
->item
);
372 assert(PyTuple_Check(ig
->item
));
373 assert(PyTuple_GET_SIZE(ig
->item
) == nitems
);
375 result
= PyTuple_New(nitems
);
379 for (i
=0 ; i
< nitems
; i
++) {
380 PyObject
*item
, *val
;
381 item
= PyTuple_GET_ITEM(ig
->item
, i
);
382 val
= PyObject_GetItem(obj
, item
);
387 PyTuple_SET_ITEM(result
, i
, val
);
392 PyDoc_STRVAR(itemgetter_doc
,
393 "itemgetter(item, ...) --> itemgetter object\n\
395 Return a callable object that fetches the given item(s) from its operand.\n\
396 After, f=itemgetter(2), the call f(r) returns r[2].\n\
397 After, g=itemgetter(2,5,3), the call g(r) returns (r[2], r[5], r[3])");
399 static PyTypeObject itemgetter_type
= {
400 PyObject_HEAD_INIT(NULL
)
402 "operator.itemgetter", /* tp_name */
403 sizeof(itemgetterobject
), /* tp_basicsize */
406 (destructor
)itemgetter_dealloc
, /* tp_dealloc */
412 0, /* tp_as_number */
413 0, /* tp_as_sequence */
414 0, /* tp_as_mapping */
416 (ternaryfunc
)itemgetter_call
, /* tp_call */
418 PyObject_GenericGetAttr
, /* tp_getattro */
420 0, /* tp_as_buffer */
421 Py_TPFLAGS_DEFAULT
| Py_TPFLAGS_HAVE_GC
, /* tp_flags */
422 itemgetter_doc
, /* tp_doc */
423 (traverseproc
)itemgetter_traverse
, /* tp_traverse */
425 0, /* tp_richcompare */
426 0, /* tp_weaklistoffset */
434 0, /* tp_descr_get */
435 0, /* tp_descr_set */
436 0, /* tp_dictoffset */
439 itemgetter_new
, /* tp_new */
444 /* attrgetter object **********************************************************/
452 static PyTypeObject attrgetter_type
;
455 attrgetter_new(PyTypeObject
*type
, PyObject
*args
, PyObject
*kwds
)
457 attrgetterobject
*ag
;
461 if (!_PyArg_NoKeywords("attrgetter()", kwds
))
464 nattrs
= PyTuple_GET_SIZE(args
);
466 if (!PyArg_UnpackTuple(args
, "attrgetter", 1, 1, &attr
))
471 /* create attrgetterobject structure */
472 ag
= PyObject_GC_New(attrgetterobject
, &attrgetter_type
);
480 PyObject_GC_Track(ag
);
481 return (PyObject
*)ag
;
485 attrgetter_dealloc(attrgetterobject
*ag
)
487 PyObject_GC_UnTrack(ag
);
488 Py_XDECREF(ag
->attr
);
493 attrgetter_traverse(attrgetterobject
*ag
, visitproc visit
, void *arg
)
500 attrgetter_call(attrgetterobject
*ag
, PyObject
*args
, PyObject
*kw
)
502 PyObject
*obj
, *result
;
503 Py_ssize_t i
, nattrs
=ag
->nattrs
;
505 if (!PyArg_UnpackTuple(args
, "attrgetter", 1, 1, &obj
))
508 return PyObject_GetAttr(obj
, ag
->attr
);
510 assert(PyTuple_Check(ag
->attr
));
511 assert(PyTuple_GET_SIZE(ag
->attr
) == nattrs
);
513 result
= PyTuple_New(nattrs
);
517 for (i
=0 ; i
< nattrs
; i
++) {
518 PyObject
*attr
, *val
;
519 attr
= PyTuple_GET_ITEM(ag
->attr
, i
);
520 val
= PyObject_GetAttr(obj
, attr
);
525 PyTuple_SET_ITEM(result
, i
, val
);
530 PyDoc_STRVAR(attrgetter_doc
,
531 "attrgetter(attr, ...) --> attrgetter object\n\
533 Return a callable object that fetches the given attribute(s) from its operand.\n\
534 After, f=attrgetter('name'), the call f(r) returns r.name.\n\
535 After, g=attrgetter('name', 'date'), the call g(r) returns (r.name, r.date).");
537 static PyTypeObject attrgetter_type
= {
538 PyObject_HEAD_INIT(NULL
)
540 "operator.attrgetter", /* tp_name */
541 sizeof(attrgetterobject
), /* tp_basicsize */
544 (destructor
)attrgetter_dealloc
, /* tp_dealloc */
550 0, /* tp_as_number */
551 0, /* tp_as_sequence */
552 0, /* tp_as_mapping */
554 (ternaryfunc
)attrgetter_call
, /* tp_call */
556 PyObject_GenericGetAttr
, /* tp_getattro */
558 0, /* tp_as_buffer */
559 Py_TPFLAGS_DEFAULT
| Py_TPFLAGS_HAVE_GC
, /* tp_flags */
560 attrgetter_doc
, /* tp_doc */
561 (traverseproc
)attrgetter_traverse
, /* tp_traverse */
563 0, /* tp_richcompare */
564 0, /* tp_weaklistoffset */
572 0, /* tp_descr_get */
573 0, /* tp_descr_set */
574 0, /* tp_dictoffset */
577 attrgetter_new
, /* tp_new */
580 /* Initialization function for the module (*must* be called initoperator) */
587 /* Create the module and add the functions */
588 m
= Py_InitModule4("operator", operator_methods
, operator_doc
,
589 (PyObject
*)NULL
, PYTHON_API_VERSION
);
593 if (PyType_Ready(&itemgetter_type
) < 0)
595 Py_INCREF(&itemgetter_type
);
596 PyModule_AddObject(m
, "itemgetter", (PyObject
*)&itemgetter_type
);
598 if (PyType_Ready(&attrgetter_type
) < 0)
600 Py_INCREF(&attrgetter_type
);
601 PyModule_AddObject(m
, "attrgetter", (PyObject
*)&attrgetter_type
);