diff options
| -rw-r--r-- | Modules/_localemodule.c | 355 |
1 files changed, 355 insertions, 0 deletions
diff --git a/Modules/_localemodule.c b/Modules/_localemodule.c new file mode 100644 index 0000000..14a22f3 --- /dev/null +++ b/Modules/_localemodule.c @@ -0,0 +1,355 @@ +/*********************************************************** +Copyright (C) 1997 Martin von Löwis + +Permission to use, copy, modify, and distribute this software and its +documentation for any purpose and without fee is hereby granted, +provided that the above copyright notice appear in all copies. + +This software comes with no warranty. Use at your own risk. +******************************************************************/ + +#include <stdio.h> +#include <errno.h> +#include <locale.h> +#include <string.h> +#include <limits.h> +#include "Python.h" + +static char locale__doc__[]="Support for POSIX locales."; + +static PyObject *Error; + +/* support functions for formatting floating point numbers */ + +static char setlocale__doc__[]= +"(integer,string=None) -> string. Activates/queries locale processing." +; + +/* to record the LC_NUMERIC settings */ +static PyObject* grouping=0; +static PyObject* thousands_sep=0; +static PyObject* decimal_point=0; +/* if non-null, indicates that LC_NUMERIC is different from "C" */ +static char* saved_numeric=0; + +/* the grouping is terminated by either 0 or CHAR_MAX */ +static PyObject* +copy_grouping(s) + char* s; +{ + int i; + PyObject *result,*val=0; + if(s[0]=='\0') + /* empty string: no grouping at all */ + return PyList_New(0); + for(i=0;s[i]!='\0' && s[i]!=CHAR_MAX;i++) + /* nothing */; + result = PyList_New(i+1); + if(!result)return NULL; + i=-1; + do{ + i++; + val=PyInt_FromLong(s[i]); + if(!val)break; + if(PyList_SetItem(result,i,val)){ + Py_DECREF(val); + val=0; + break; + } + }while(s[i]!='\0' && s[i]!=CHAR_MAX); + if(!val){ + Py_DECREF(result); + return NULL; + } + return result; +} + +static void +fixup_ulcase() +{ + PyObject *mods,*strop,*string,*ulo; + unsigned char ul[256]; + int n,c; + + /* finding sys.modules */ + mods=PyImport_GetModuleDict(); + if(!mods)return; + /* finding the module */ + string=PyDict_GetItemString(mods,"string"); + if(string) + string=PyModule_GetDict(string); + strop=PyDict_GetItemString(mods,"strop"); + if(strop) + strop=PyModule_GetDict(strop); + if(!string && !strop)return; + /* create uppercase */ + n = 0; + for (c = 0; c < 256; c++) { + if (isupper(c)) + ul[n++] = c; + } + ulo=PyString_FromStringAndSize(ul,n); + if(!ulo)return; + if(string) + PyDict_SetItemString(string,"uppercase",ulo); + if(strop) + PyDict_SetItemString(strop,"uppercase",ulo); + Py_DECREF(ulo); + /* create lowercase */ + n = 0; + for (c = 0; c < 256; c++) { + if (islower(c)) + ul[n++] = c; + } + ulo=PyString_FromStringAndSize(ul,n); + if(!ulo)return; + if(string) + PyDict_SetItemString(string,"lowercase",ulo); + if(strop) + PyDict_SetItemString(strop,"lowercase",ulo); + Py_DECREF(ulo); + /* create letters */ + n = 0; + for (c = 0; c < 256; c++) { + if (isalpha(c)) + ul[n++] = c; + } + ulo=PyString_FromStringAndSize(ul,n); + if(!ulo)return; + if(string) + PyDict_SetItemString(string,"letters",ulo); + Py_DECREF(ulo); +} + + +static PyObject* +PyLocale_setlocale(self,args) + PyObject *self; + PyObject *args; +{ + int category; + char *locale=0,*result; + PyObject *result_object; + struct lconv *lc; + if(!PyArg_ParseTuple(args,"i|z",&category,&locale))return 0; + if(locale){ + /* set locale */ + result=setlocale(category,locale); + if(!result){ + /* operation failed, no setting was changed */ + PyErr_SetString(Error,"locale setting not supported"); + return NULL; + } + result_object=PyString_FromString(result); + if(!result)return NULL; + /* record changes to LC_NUMERIC */ + if(category==LC_NUMERIC || category==LC_ALL){ + if(strcmp(locale,"C")==0 || strcmp(locale,"POSIX")==0){ + /* user just asked for default numeric locale */ + if(saved_numeric)free(saved_numeric); + saved_numeric=0; + }else{ + /* remember values */ + lc=localeconv(); + Py_XDECREF(grouping); + grouping=copy_grouping(lc->grouping); + Py_XDECREF(thousands_sep); + thousands_sep=PyString_FromString(lc->thousands_sep); + Py_XDECREF(decimal_point); + decimal_point=PyString_FromString(lc->decimal_point); + saved_numeric = strdup(locale); + + /* restore to "C" */ + setlocale(LC_NUMERIC,"C"); + } + } + /* record changes to LC_CTYPE */ + if(category==LC_CTYPE || category==LC_ALL) + fixup_ulcase(); + /* things that got wrong up to here are ignored */ + PyErr_Clear(); + }else{ + /* get locale */ + /* restore LC_NUMERIC first, if appropriate */ + if(saved_numeric) + setlocale(LC_NUMERIC,saved_numeric); + result=setlocale(category,NULL); + if(!result){ + PyErr_SetString(Error,"locale query failed"); + return NULL; + } + result_object=PyString_FromString(result); + /* restore back to "C" */ + if(saved_numeric) + setlocale(LC_NUMERIC,"C"); + } + return result_object; +} + +static char localeconv__doc__[]= +"() -> dict. Returns numeric and monetary locale-specific parameters." +; + +static PyObject* +PyLocale_localeconv(self,args) + PyObject *self; + PyObject *args; +{ + PyObject* result; + struct lconv *l; + PyObject *x; + if(!PyArg_NoArgs(args))return 0; + result = PyDict_New(); + if(!result)return 0; + /* if LC_NUMERIC is different in the C library, use saved value */ + l = localeconv(); + /* hopefully, the localeconv result survives the C library calls + involved herein */ +#define RESULT_STRING(s) \ + x=PyString_FromString(l->s);if(!x)goto failed;PyDict_SetItemString(result,#s,x);Py_XDECREF(x) +#define RESULT_INT(i) \ + x=PyInt_FromLong(l->i);if(!x)goto failed;PyDict_SetItemString(result,#i,x);Py_XDECREF(x) + + /* Numeric information */ + if(saved_numeric){ + /* cannot use localeconv results */ + PyDict_SetItemString(result,"decimal_point",decimal_point); + PyDict_SetItemString(result,"grouping",grouping); + PyDict_SetItemString(result,"thousands_sep",thousands_sep); + }else{ + RESULT_STRING(decimal_point); + RESULT_STRING(thousands_sep); + x=copy_grouping(l->grouping); + if(!x)goto failed; + PyDict_SetItemString(result,"grouping",x); + Py_XDECREF(x); + } + + /* Monetary information */ + RESULT_STRING(int_curr_symbol); + RESULT_STRING(currency_symbol); + RESULT_STRING(mon_decimal_point); + RESULT_STRING(mon_thousands_sep); + x=copy_grouping(l->mon_grouping); + if(!x)goto failed; + PyDict_SetItemString(result,"mon_grouping",x); + Py_XDECREF(x); + RESULT_STRING(positive_sign); + RESULT_STRING(negative_sign); + RESULT_INT(int_frac_digits); + RESULT_INT(frac_digits); + RESULT_INT(p_cs_precedes); + RESULT_INT(p_sep_by_space); + RESULT_INT(n_cs_precedes); + RESULT_INT(n_sep_by_space); + RESULT_INT(p_sign_posn); + RESULT_INT(n_sign_posn); + + return result; + failed: + Py_XDECREF(result); + Py_XDECREF(x); + return NULL; +} + +static char strcoll__doc__[]= +"string,string -> int. Compares two strings according to the locale." +; + +static PyObject* +PyLocale_strcoll(self,args) + PyObject *self; + PyObject *args; +{ + char *s1,*s2; + if(!PyArg_ParseTuple(args,"ss",&s1,&s2)) + return NULL; + return PyInt_FromLong(strcoll(s1,s2)); +} + +static char strxfrm__doc__[]= +"string -> string. Returns a string that behaves for cmp locale-aware." +; + +static PyObject* +PyLocale_strxfrm(self,args) + PyObject* self; + PyObject* args; +{ + char *s,*buf; + int n1,n2; + PyObject *result; + if(!PyArg_ParseTuple(args,"s",&s)) + return NULL; + /* assume no change in size, first */ + n1=strlen(s)+1; + buf=Py_Malloc(n1); + if(!buf)return NULL; + n2=strxfrm(buf,s,n1); + if(n2>n1){ + /* more space needed */ + buf=Py_Realloc(buf,n2); + if(!buf)return NULL; + strxfrm(buf,s,n2); + } + result=PyString_FromString(buf); + Py_Free(buf); + return result; +} + +static struct PyMethodDef PyLocale_Methods[] = { + {"setlocale",(PyCFunction)PyLocale_setlocale,1,setlocale__doc__}, + {"localeconv",(PyCFunction)PyLocale_localeconv,0,localeconv__doc__}, + {"strcoll",(PyCFunction)PyLocale_strcoll,1,strcoll__doc__}, + {"strxfrm",(PyCFunction)PyLocale_strxfrm,1,strxfrm__doc__}, + {NULL, NULL} +}; + +void +init_locale() +{ + PyObject *m,*d,*x; + m=Py_InitModule("_locale",PyLocale_Methods); + d = PyModule_GetDict(m); + x=PyInt_FromLong(LC_CTYPE); + PyDict_SetItemString(d,"LC_CTYPE",x); + Py_XDECREF(x); + + x=PyInt_FromLong(LC_TIME); + PyDict_SetItemString(d,"LC_TIME",x); + Py_XDECREF(x); + + x=PyInt_FromLong(LC_COLLATE); + PyDict_SetItemString(d,"LC_COLLATE",x); + Py_XDECREF(x); + + x=PyInt_FromLong(LC_MONETARY); + PyDict_SetItemString(d,"LC_MONETARY",x); + Py_XDECREF(x); + + x=PyInt_FromLong(LC_MESSAGES); + PyDict_SetItemString(d,"LC_MESSAGES",x); + Py_XDECREF(x); + + x=PyInt_FromLong(LC_NUMERIC); + PyDict_SetItemString(d,"LC_NUMERIC",x); + Py_XDECREF(x); + + x=PyInt_FromLong(LC_ALL); + PyDict_SetItemString(d,"LC_ALL",x); + Py_XDECREF(x); + + x=PyInt_FromLong(CHAR_MAX); + PyDict_SetItemString(d,"CHAR_MAX",x); + Py_XDECREF(x); + + Error = PyErr_NewException("locale.Error", NULL, NULL); + PyDict_SetItemString(d, "Error", Error); + + x=PyString_FromString(locale__doc__); + PyDict_SetItemString(d,"__doc__",x); + Py_XDECREF(x); + + if(PyErr_Occurred()) + Py_FatalError("Can't initialize module locale"); +} |