Modules/future_builtins.c

   1
   2 /* future_builtins module */
   3
   4 /* This module provides functions that will be builtins in Python 3.0,
   5    but that conflict with builtins that already exist in Python
   6    2.x. */
   7
   8
   9 #include "Python.h"
  10
  11 PyDoc_STRVAR(module_doc,
  12 "This module provides functions that will be builtins in Python 3.0,\n\
  13 but that conflict with builtins that already exist in Python 2.x.\n\
  14 \n\
  15 Functions:\n\
  16 \n\
  17 hex(arg) -- Returns the hexadecimal representation of an integer\n\
  18 oct(arg) -- Returns the octal representation of an integer\n\
  19 \n\
  20 The typical usage of this module is to replace existing builtins in a\n\
  21 module's namespace:\n \n\
  22 from future_builtins import hex, oct\n");
  23
  24 static PyObject *
  25 builtin_hex(PyObject *self, PyObject *v)
  26 {
  27         return PyNumber_ToBase(v, 16);
  28 }
  29
  30 PyDoc_STRVAR(hex_doc,
  31 "hex(number) -> string\n\
  32 \n\
  33 Return the hexadecimal representation of an integer or long integer.");
  34
  35
  36 static PyObject *
  37 builtin_oct(PyObject *self, PyObject *v)
  38 {
  39         return PyNumber_ToBase(v, 8);
  40 }
  41
  42 PyDoc_STRVAR(oct_doc,
  43 "oct(number) -> string\n\
  44 \n\
  45 Return the octal representation of an integer or long integer.");
  46
  47
  48 /* List of functions exported by this module */
  49
  50 static PyMethodDef module_functions[] = {
  51         {"hex",         builtin_hex,        METH_O, hex_doc},
  52         {"oct",         builtin_oct,        METH_O, oct_doc},
  53         {NULL,          NULL}   /* Sentinel */
  54 };
  55
  56
  57 /* Initialize this module. */
  58
  59 PyMODINIT_FUNC
  60 initfuture_builtins(void)
  61 {
  62         PyObject *m, *itertools, *iter_func;
  63         char *it_funcs[] = {"imap", "ifilter", "izip", NULL};
  64         char **cur_func;
  65
  66         m = Py_InitModule3("future_builtins", module_functions, module_doc);
  67         if (m == NULL)
  68                 return;
  69
  70         itertools = PyImport_ImportModuleNoBlock("itertools");
  71         if (itertools == NULL)
  72                 return;
  73
  74         for (cur_func = it_funcs; *cur_func; ++cur_func){
  75                 iter_func = PyObject_GetAttrString(itertools, *cur_func);
  76                 if (iter_func == NULL)
  77                         return;
  78                 PyModule_AddObject(m, *cur_func+1, iter_func);
  79         }
  80         Py_DECREF(itertools);
  81         /* any other initialization needed */
  82 }