/[thuban]/branches/WIP-pyshapelib-bramz/libraries/pyshapelib/dbflibmodule.c
ViewVC logotype

Annotation of /branches/WIP-pyshapelib-bramz/libraries/pyshapelib/dbflibmodule.c

Parent Directory Parent Directory | Revision Log Revision Log


Revision 2752 - (hide annotations)
Tue Apr 10 23:45:00 2007 UTC (17 years, 10 months ago) by bramz
File MIME type: text/plain
File size: 24718 byte(s)
attempt to add support for Unicode and Language Driver ID (LDID) support in dbflib
1 bramz 2752 /* Copyright (c) 2001, 2002, 2007 by Intevation GmbH
2     * Authors:
3     * Bram de Greve <[email protected]>
4     * Bernhard Herzog <[email protected]>
5     *
6     * This program is free software under the GPL (>=v2)
7     * Read the file COPYING coming with Thuban for details.
8     */
9    
10 bramz 2742 #include "pyshapelib_common.h"
11 jan 1611
12 bramz 2752 /* UNICODE & LANGUAGE DRIVER SUPPORT FOR DBFLIB
13     *
14     * When writing Unicode objects to a dbflib database, the unicode has to be
15     * encoded in 8-bit characters using a code page. This code page is indentified
16     * by the Language Driver ID (LDID) field in the database header.
17     *
18     * At this moment, this need unofficial modifications of the maptools shapelib
19     * library because they do not read the LDID. No patch has been submitted yet,
20     * but the version contained in the Thuban source tree incorporates the required
21     * modifications.
22     *
23     * pyshapelib is designed to compile with either the patched or unpatched shapelib
24     * by defining HAVE_LANGUAGE_DRIVER as true or false respectively. In the latter
25     * case, a Windows ANSI code page (cp1252) is assumed
26     */
27     #if HAVE_LANGUAGE_DRIVER
28    
29     #define PYSHAPELIB_NUM_LANGUAGE_DRIVERS 256
30    
31     #define PYSHAPELIB_ADD_LANGUAGE_DRIVER(ldid, codec, name)\
32     codecs[ldid] = codec;\
33     drivers[ldid] = "LDID_" name;\
34     PyModule_AddIntConstant(module, "LDID_" name, ldid)
35    
36     static char* codecs[PYSHAPELIB_NUM_LANGUAGE_DRIVERS];
37     static char* drivers[PYSHAPELIB_NUM_LANGUAGE_DRIVERS];
38    
39     #endif
40    
41    
42    
43     /** Determine name of Python's built-in codec
44     */
45     static char* get_codec(DBFHandle handle)
46     {
47     #if HAVE_LANGUAGE_DRIVER
48     if (!codecs[handle->nLanguageDriver])
49     {
50     PyErr_Format(PyExc_ValueError, "Language Driver ID %d not recognized", handle->nLanguageDriver);
51     }
52     return codecs[handle->nLanguageDriver];
53     #else
54     return "cp1252";
55     #endif
56     }
57    
58    
59    
60     /** decode to unicode object
61     */
62     static PyObject* decode_string(DBFHandle handle, const char* string)
63     {
64     char* codec = get_codec(handle);
65     if (!codec) return NULL;
66     return PyUnicode_Decode(string, strlen(string), codec, NULL);
67     }
68    
69     /** encode unicode object to normal Python string object
70     */
71     static PyObject* encode_string(DBFHandle handle, PyObject* string)
72     {
73     char* codec = get_codec(handle);
74     if (!codec) return NULL;
75    
76     if (PyString_Check(string))
77     {
78     return PyString_AsEncodedObject(string, codec, NULL);
79     }
80     if (PyUnicode_Check(string))
81     {
82     return PyUnicode_AsEncodedString(string, codec, NULL);
83     }
84    
85     PyErr_SetString(PyExc_TypeError, "value is neither a string or unicode object");
86     return NULL;
87     }
88    
89    
90    
91 bramz 2742 /* --- DBFFile ------------------------------------------------------------------------------------------------------- */
92 jan 1611
93 bramz 2742 typedef struct {
94     PyObject_HEAD
95     DBFHandle handle;
96     } DBFFileObject;
97 jan 1611
98    
99 bh 1917
100 bramz 2742 /* allocator
101     */
102     static PyObject* dbffile_new(PyTypeObject* type, PyObject* args, PyObject* kwds)
103 bh 1917 {
104 bramz 2742 DBFFileObject* self;
105     self = (DBFFileObject*) type->tp_alloc(type, 0);
106     self->handle = NULL;
107     return (PyObject*) self;
108     }
109 bh 1917
110 bramz 2742
111    
112     /* deallocator
113     */
114     static void dbffile_dealloc(DBFFileObject* self)
115     {
116     DBFClose(self->handle);
117     self->handle = NULL;
118     self->ob_type->tp_free((PyObject*)self);
119     }
120    
121    
122    
123     /* constructor
124     */
125     static int dbffile_init(DBFFileObject* self, PyObject* args, PyObject* kwds)
126     {
127 bramz 2751 char* file = NULL;
128 bramz 2742 char* mode = "rb";
129 bramz 2745 static char *kwlist[] = {"name", "mode", NULL};
130 bramz 2751
131     DBFClose(self->handle);
132     self->handle = NULL;
133    
134     #if defined(SHPAPI_HAS_WIDE) && defined(Py_WIN_WIDE_FILENAMES)
135     if (GetVersion() < 0x80000000) { /* On NT, so wide API available */
136     PyObject *wfile;
137     if (PyArg_ParseTupleAndKeywords(args, kwds, "U|s:DBFFile", kwlist, &wfile, &mode))
138     {
139     PyObject *wmode = PyUnicode_DecodeASCII(mode, strlen(mode), NULL);
140     if (!wmode) return -1;
141     self->handle = DBFOpenW(PyUnicode_AS_UNICODE(wfile), PyUnicode_AS_UNICODE(wmode));
142     Py_DECREF(wmode);
143     if (!self->handle)
144     {
145     PyErr_SetFromErrnoWithFilenameObject(PyExc_IOError, wfile);
146     return -1;
147     }
148     }
149     else
150     {
151     /* Drop the argument parsing error as narrow
152     strings are also valid. */
153     PyErr_Clear();
154     }
155     }
156     #endif
157    
158 bramz 2749 if (!self->handle)
159     {
160 bramz 2751 if (!PyArg_ParseTupleAndKeywords(args, kwds, "et|s:DBFFile", kwlist,
161     Py_FileSystemDefaultEncoding, &file, &mode)) return -1;
162     self->handle = DBFOpen(file, mode);
163    
164     if (!self->handle)
165     {
166     PyErr_SetFromErrnoWithFilename(PyExc_IOError, file);
167     PyMem_Free(file);
168     return -1;
169     }
170    
171     PyMem_Free(file);
172 bramz 2749 }
173    
174 bramz 2751 return 0;
175 bramz 2742 }
176 bh 1917
177    
178    
179 bramz 2742 static PyObject* dbffile_close(DBFFileObject* self)
180 bh 1917 {
181 bramz 2742 DBFClose(self->handle);
182     self->handle = NULL;
183     Py_RETURN_NONE;
184     }
185 bh 1917
186    
187 bramz 2742
188     static PyObject* dbffile_field_count(DBFFileObject* self)
189     {
190     return PyInt_FromLong((long)DBFGetFieldCount(self->handle));
191 bh 1917 }
192    
193 jan 1611
194 bramz 2742
195     static PyObject* dbffile_record_count(DBFFileObject* self)
196 jan 1611 {
197 bramz 2742 return PyInt_FromLong((long)DBFGetRecordCount(self->handle));
198     }
199 jan 1611
200    
201 bramz 2742
202     static PyObject* dbffile_field_info(DBFFileObject* self, PyObject* args)
203     {
204     char field_name[12];
205     int field, width = 0, decimals = 0, field_type;
206 jan 1611
207 bramz 2744 if (!PyArg_ParseTuple(args, "i:field_info", &field)) return NULL;
208 bramz 2742
209     field_name[0] = '\0';
210     field_type = DBFGetFieldInfo(self->handle, field, field_name, &width, &decimals);
211    
212     return Py_BuildValue("isii", field_type, field_name, width, decimals);
213     }
214 bh 1917
215 jan 1611
216 bh 1917
217 bramz 2742 static PyObject* dbffile_add_field(DBFFileObject* self, PyObject* args)
218     {
219     char* name;
220     int type, width, decimals;
221     int field;
222    
223 bramz 2744 if (!PyArg_ParseTuple(args, "siii:add_field", &name, &type, &width, &decimals)) return NULL;
224 bramz 2742
225     field = DBFAddField(self->handle, name, (DBFFieldType)type, width, decimals);
226    
227     if (field < 0)
228     {
229     PyErr_SetString(PyExc_ValueError, "Failed to add field due to inappropriate field definition");
230     return NULL;
231     }
232     return PyInt_FromLong((long)field);
233 jan 1611 }
234    
235    
236    
237 bramz 2742 /* Read one attribute from the dbf handle and return it as a new python object
238     *
239     * If an error occurs, set the appropriate Python exception and return
240     * NULL.
241     *
242     * Assume that the values of the record and field arguments are valid.
243     * The name argument will be passed to DBFGetFieldInfo as is and should
244     * thus be either NULL or a pointer to an array of at least 12 chars
245     */
246     static PyObject* do_read_attribute(DBFHandle handle, int record, int field, char * name)
247 jan 1611 {
248 bramz 2742 int type, width;
249 bramz 2752 const char* string;
250 bramz 2742 type = DBFGetFieldInfo(handle, field, name, &width, NULL);
251    
252     /* For strings NULL and the empty string are indistinguishable
253     * in DBF files. We prefer empty strings instead for backwards
254     * compatibility reasons because older wrapper versions returned
255     * emtpy strings as empty strings.
256     */
257     if (type != FTString && DBFIsAttributeNULL(handle, record, field))
258 jan 1611 {
259 bramz 2742 Py_RETURN_NONE;
260 jan 1611 }
261 bramz 2742 else
262 jan 1611 {
263 bramz 2742 switch (type)
264     {
265     case FTString:
266 bramz 2752 string = DBFReadStringAttribute(handle, record, field);
267     if (string) return decode_string(handle, string);
268 jan 1611
269 bramz 2742 case FTInteger:
270     return PyInt_FromLong((long)DBFReadIntegerAttribute(handle, record, field));
271 jan 1611
272 bramz 2742 case FTDouble:
273     return PyFloat_FromDouble(DBFReadDoubleAttribute(handle, record, field));
274 bramz 2745
275     case FTLogical:
276 bramz 2752 string = DBFReadLogicalAttribute(handle, record, field);
277     if (string)
278 bramz 2745 {
279 bramz 2752 switch (string[0])
280 bramz 2745 {
281     case 'F':
282     case 'N':
283     Py_RETURN_FALSE;
284     case 'T':
285     case 'Y':
286     Py_RETURN_TRUE;
287     }
288     }
289     break;
290 jan 1611
291 bramz 2742 default:
292     PyErr_Format(PyExc_TypeError, "Invalid field data type %d", type);
293     return NULL;
294     }
295 jan 1611 }
296 bramz 2745
297     PyErr_Format(PyExc_IOError, "Can't read value for row %d column %d", record, field);
298     return NULL;
299 bramz 2742 }
300 jan 1611
301    
302 bramz 2742
303     /* the read_attribute method. Return the value of the given record and
304     * field as a python object of the appropriate type.
305     */
306     static PyObject* dbffile_read_attribute(DBFFileObject* self, PyObject* args)
307 jan 1611 {
308 bramz 2742 int record, field;
309 jan 1611
310 bramz 2744 if (!PyArg_ParseTuple(args, "ii:read_field", &record, &field)) return NULL;
311 bramz 2742
312     if (record < 0 || record >= DBFGetRecordCount(self->handle))
313     {
314     PyErr_Format(PyExc_ValueError,
315     "record index %d out of bounds (record count: %d)",
316     record, DBFGetRecordCount(self->handle));
317     return NULL;
318     }
319 jan 1611
320 bramz 2742 if (field < 0 || field >= DBFGetFieldCount(self->handle))
321 jan 1611 {
322 bramz 2742 PyErr_Format(PyExc_ValueError,
323     "field index %d out of bounds (field count: %d)",
324     field, DBFGetFieldCount(self->handle));
325     return NULL;
326 jan 1611 }
327 bramz 2742
328     return do_read_attribute(self->handle, record, field, NULL);
329     }
330    
331    
332    
333     /* the read_record method. Return the record record as a dictionary with
334     * whose keys are the names of the fields, and their values as the
335     * appropriate Python type.
336     */
337     static PyObject* dbffile_read_record(DBFFileObject* self, PyObject* args)
338     {
339     int record;
340     int num_fields;
341     int i;
342     char name[12];
343     PyObject *dict;
344     PyObject *value = NULL;
345    
346 bramz 2744 if (!PyArg_ParseTuple(args, "i:read_record", &record)) return NULL;
347 bramz 2742
348     if (record < 0 || record >= DBFGetRecordCount(self->handle))
349 jan 1611 {
350 bramz 2742 PyErr_Format(PyExc_ValueError,
351     "record index %d out of bounds (record count: %d)",
352     record, DBFGetRecordCount(self->handle));
353     return NULL;
354 jan 1611 }
355 bramz 2742
356     dict = PyDict_New();
357     if (!dict) return NULL;
358    
359     num_fields = DBFGetFieldCount(self->handle);
360 jan 1611 for (i = 0; i < num_fields; i++)
361     {
362 bramz 2742 value = do_read_attribute(self->handle, record, i, name);
363     if (!value || PyDict_SetItemString(dict, name, value) < 0) goto fail;
364 jan 1611 Py_DECREF(value);
365 bramz 2742 value = NULL;
366 jan 1611 }
367    
368 bramz 2742 return dict;
369 jan 1611
370 bramz 2742 fail:
371     Py_XDECREF(value);
372     Py_DECREF(dict);
373     return NULL;
374 jan 1611 }
375    
376    
377 bramz 2742
378     /* write a single field of a record. */
379     static int do_write_field(DBFHandle handle, int record, int field, int type, PyObject* value)
380 bh 2212 {
381 bramz 2752 PyObject* encoded_string = NULL;
382 bramz 2742 char * string_value;
383     int int_value;
384     double double_value;
385 bramz 2745 int logical_value;
386 bh 2212
387 bramz 2742 if (value == Py_None)
388     {
389 bramz 2745 if (DBFWriteNULLAttribute(handle, record, field)) return 1;
390 bramz 2742 }
391     else
392     {
393     switch (type)
394     {
395     case FTString:
396 bramz 2752 encoded_string = encode_string(handle, value);
397     if (!encoded_string) return 0;
398     string_value = PyString_AsString(encoded_string);
399     if (!string_value)
400     {
401     Py_DECREF(encoded_string);
402     return 0;
403     }
404     if (DBFWriteStringAttribute(handle, record, field, string_value))
405     {
406     Py_DECREF(encoded_string);
407     return 1;
408     }
409     Py_DECREF(encoded_string);
410 bramz 2742 break;
411 bh 2212
412 bramz 2742 case FTInteger:
413     int_value = PyInt_AsLong(value);
414     if (int_value == -1 && PyErr_Occurred()) return 0;
415 bramz 2745 if (DBFWriteIntegerAttribute(handle, record, field, int_value)) return 1;
416 bramz 2742 break;
417 jan 1611
418 bramz 2742 case FTDouble:
419     double_value = PyFloat_AsDouble(value);
420     if (double_value == -1 && PyErr_Occurred()) return 0;
421 bramz 2745 if (DBFWriteDoubleAttribute(handle, record, field, double_value)) return 1;
422 bramz 2742 break;
423 bramz 2745
424     case FTLogical:
425     logical_value = PyObject_IsTrue(value);
426     if (logical_value == -1) return 0;
427     if (DBFWriteLogicalAttribute(handle, record, field, logical_value ? 'T' : 'F')) return 1;
428     break;
429 jan 1611
430 bramz 2742 default:
431     PyErr_Format(PyExc_TypeError, "Invalid field data type %d", type);
432     return 0;
433     }
434     }
435 jan 1611
436 bramz 2745 PyErr_Format(PyExc_IOError, "can't write field %d of record %d", field, record);
437     return 0;
438 bramz 2742 }
439 jan 1611
440    
441    
442 bramz 2742 static PyObject* dbffile_write_field(DBFFileObject* self, PyObject* args)
443     {
444     int record, field;
445     PyObject* value;
446     int type;
447    
448 bramz 2744 if (!PyArg_ParseTuple(args, "iiO:write_field", &record, &field, &value)) return NULL;
449 bramz 2742
450     if (field < 0 || field >= DBFGetFieldCount(self->handle))
451     {
452     PyErr_Format(PyExc_ValueError,
453     "field index %d out of bounds (field count: %d)",
454     field, DBFGetFieldCount(self->handle));
455     return NULL;
456     }
457    
458     type = DBFGetFieldInfo(self->handle, field, NULL, NULL, NULL);
459     if (!do_write_field(self->handle, record, field, type, value)) return NULL;
460     Py_RETURN_NONE;
461 jan 1611 }
462    
463    
464    
465 bramz 2742 static PyObject* dbffile_write_record(DBFFileObject* self, PyObject* args)
466     {
467     int record;
468     PyObject* record_object;
469     int i, num_fields;
470    
471     int type;
472     char name[12];
473     PyObject* value = NULL;
474    
475 bramz 2744 if (!PyArg_ParseTuple(args, "iO:write_record", &record, &record_object)) return NULL;
476 bramz 2742
477     num_fields = DBFGetFieldCount(self->handle);
478    
479     /* mimic ShapeFile functionality where id = -1 means appending */
480     if (record == -1)
481     {
482     record = num_fields;
483     }
484 jan 1611
485 bramz 2742 if (PySequence_Check(record_object))
486     {
487     /* It's a sequence object. Iterate through all items in the
488     * sequence and write them to the appropriate field.
489     */
490     if (PySequence_Length(record_object) != num_fields)
491     {
492     PyErr_SetString(PyExc_TypeError, "record must have one item for each field");
493     return NULL;
494     }
495     for (i = 0; i < num_fields; ++i)
496     {
497     type = DBFGetFieldInfo(self->handle, i, NULL, NULL, NULL);
498     value = PySequence_GetItem(record_object, i);
499     if (!value) return NULL;
500     if (!do_write_field(self->handle, record, i, type, value))
501     {
502     Py_DECREF(value);
503     return NULL;
504     }
505     Py_DECREF(value);
506     }
507     }
508     else
509     {
510     /* It's a dictionary-like object. Iterate over the names of the
511     * known fields and write the corresponding item
512     */
513     for (i = 0; i < num_fields; ++i)
514     {
515     name[0] = '\0';
516     type = DBFGetFieldInfo(self->handle, i, name, NULL, NULL);
517     value = PyDict_GetItemString(record_object, name);
518     if (value && !do_write_field(self->handle, record, i, type, value)) return NULL;
519     }
520     }
521    
522     return PyInt_FromLong((long)record);
523 bh 1761 }
524 jan 1611
525    
526    
527 bramz 2742 static PyObject* dbffile_repr(DBFFileObject* self)
528     {
529     /* TODO: it would be nice to do something like "dbflib.DBFFile(filename, mode)" instead */
530     return PyString_FromFormat("<dbflib.DBFFile object at %p>", self->handle);
531 jan 1611 }
532    
533    
534    
535 bramz 2742 /* The commit method implementation
536     *
537     * The method relies on the DBFUpdateHeader method which is not
538     * available in shapelib <= 1.2.10. setup.py defines
539     * HAVE_UPDATE_HEADER's value depending on whether the function is
540     * available in the shapelib version the code is compiled with.
541     */
542     #if HAVE_UPDATE_HEADER
543     static PyObject* dbffile_commit(DBFFileObject* self)
544 jan 1611 {
545 bramz 2742 DBFUpdateHeader(self->handle);
546     Py_RETURN_NONE;
547     }
548     #endif
549 jan 1611
550    
551 bramz 2752 #if HAVE_LANGUAGE_DRIVER
552 jan 1611
553 bramz 2752 static PyObject* dbffile_language_driver(DBFFileObject* self, void* closure)
554     {
555     return PyInt_FromLong((long)self->handle->nLanguageDriver);
556     }
557    
558     #endif
559    
560    
561 bramz 2742 static struct PyMethodDef dbffile_methods[] =
562     {
563 bramz 2744 {"close", (PyCFunction)dbffile_close, METH_NOARGS,
564 bramz 2745 "close() -> None\n\n"
565     "closes DBFFile"},
566 bramz 2744 {"field_count", (PyCFunction)dbffile_field_count, METH_NOARGS,
567 bramz 2745 "field_count() -> integer\n\n"
568 bramz 2744 "returns number of fields currently defined"},
569     {"record_count", (PyCFunction)dbffile_record_count, METH_NOARGS,
570 bramz 2745 "record_count() -> integer\n\n"
571 bramz 2744 "returns number of records that currently exist"},
572     {"field_info", (PyCFunction)dbffile_field_info, METH_VARARGS,
573 bramz 2745 "field_info(field_index) -> (type, name, width, decimals)\n\n"
574     "returns info of a field as a tuple with:\n"
575     "- type: the type of the field corresponding to the integer value of one "
576     " of the constants FTString, FTInteger, ...\n"
577     "- name: the name of the field as a string\n"
578     "- width: the width of the field as a number of characters\n"
579     "- decimals: the number of decimal digits" },
580 bramz 2742 {"add_field", (PyCFunction)dbffile_add_field, METH_VARARGS,
581 bramz 2745 "add_field(type, name, width, decimals) -> field_index\n\n"
582 bramz 2742 "adds a new field and returns field index if successful\n"
583 bramz 2745 "- type: the type of the field corresponding to the integer value of one "
584     " of the constants FTString, FTInteger, ...\n"
585     "- name: the name of the field as a string\n"
586     "- width: the width of the field as a number of characters\n"
587     "- decimals: the number of decimal digits" },
588 bramz 2744 {"read_attribute", (PyCFunction)dbffile_read_attribute, METH_VARARGS,
589 bramz 2745 "read_attribute(record_index, field_index) -> value\n\n"
590     "returns the value of one field of a record"},
591 bramz 2744 {"read_record", (PyCFunction)dbffile_read_record, METH_VARARGS,
592 bramz 2745 "read_record(record_index) -> dict\n\n"
593     "returns an entire record as a dictionary of field names and values"},
594 bramz 2744 {"write_field", (PyCFunction)dbffile_write_field, METH_VARARGS,
595     "write_field(record_index, field_index, new_value)\n"
596 bramz 2745 "writes a single field of a record"},
597 bramz 2744 {"write_record", (PyCFunction)dbffile_write_record, METH_VARARGS,
598 bramz 2745 "write_record(record_index, record) -> record_index\n\n"
599     "Writes an entire record as a dict or a sequence, and return index of record\n"
600     "Record can either be a dictionary in which case the keys are used as field names, "
601 bramz 2744 "or a sequence that must have an item for every field (length = field_count())"},
602 bramz 2742 #if HAVE_UPDATE_HEADER
603 bramz 2750 {"commit", (PyCFunction)dbffile_commit, METH_NOARGS,
604 bramz 2745 "commit() -> None"},
605 bramz 2742 #endif
606     {NULL}
607     };
608 jan 1611
609    
610 bh 1917
611 bramz 2742 static struct PyGetSetDef dbffile_getsetters[] =
612     {
613 bramz 2752 #if HAVE_LANGUAGE_DRIVER
614     {"language_driver", (getter)dbffile_language_driver, NULL, "Language Driver ID (read-only)" },
615     #endif
616 bramz 2742 {NULL}
617     };
618 jan 1611
619    
620    
621 bramz 2742 static PyTypeObject DBFFileType = PYSHAPELIB_DEFINE_TYPE(DBFFileObject, dbffile, "shapelib.DBFFile", 0);
622 bh 2453
623    
624 jan 1611
625 bramz 2742 /* --- dbflib -------------------------------------------------------------------------------------------------------- */
626 jan 1611
627 bramz 2742 static PyObject* dbflib_open(PyObject* module, PyObject* args)
628     {
629     return PyObject_CallObject((PyObject*)&DBFFileType, args);
630     }
631 jan 1611
632    
633    
634 bramz 2742 static PyObject* dbflib_create(PyObject* module, PyObject* args)
635     {
636     char* file;
637     DBFFileObject* result;
638 bramz 2751 DBFHandle handle = NULL;
639     int wideargument = 0;
640    
641     #if defined(SHPAPI_HAS_WIDE) && defined(Py_WIN_WIDE_FILENAMES)
642     if (GetVersion() < 0x80000000) { /* On NT, so wide API available */
643     PyObject *wfile;
644     if (PyArg_ParseTuple(args, "U:create", &wfile))
645     {
646     wideargument = 1;
647     handle = DBFCreateW(PyUnicode_AS_UNICODE(wfile));
648     if (!handle)
649     {
650     PyErr_SetFromErrnoWithFilenameObject(PyExc_IOError, wfile);
651     return NULL;
652     }
653     }
654     else
655     {
656     /* Drop the argument parsing error as narrow
657     strings are also valid. */
658     PyErr_Clear();
659     }
660     }
661     #endif
662 bramz 2742
663 bramz 2751 if (!handle)
664     {
665     if (!PyArg_ParseTuple(args, "et:create", Py_FileSystemDefaultEncoding, &file)) return NULL;
666     handle = DBFCreate(file);
667     if (!handle)
668     {
669     PyErr_SetFromErrnoWithFilename(PyExc_IOError, file);
670     PyMem_Free(file);
671     return NULL;
672     }
673     PyMem_Free(file);
674     }
675    
676 bramz 2742 result = PyObject_New(DBFFileObject, &DBFFileType);
677     if (!result)
678     {
679 bramz 2751 DBFClose(handle);
680 bramz 2742 return PyErr_NoMemory();
681     }
682    
683 bramz 2751 result->handle = handle;
684 bramz 2742 return (PyObject*) result;
685     }
686 jan 1611
687    
688    
689 bramz 2752 #if HAVE_LANGUAGE_DRIVER
690    
691     /** translate a numeric Language Driver ID to the name of Python's codec.
692     */
693     static PyObject* dbflib_language_driver_codec(PyObject* module, PyObject* args)
694     {
695     int ldid;
696     if (!PyArg_ParseTuple(args, "i:language_driver_name", &ldid)) return NULL;
697     if (ldid < 0 || ldid >= PYSHAPELIB_NUM_LANGUAGE_DRIVERS || !codecs[ldid])
698     {
699     PyErr_SetString(PyExc_ValueError, "invalid driver id");
700     return NULL;
701     }
702     return PyString_FromString(codecs[ldid]);
703     }
704    
705     /** translate a numeric Language Driver ID to a string represting its constant.
706     */
707     static PyObject* dbflib_language_driver_name(PyObject* module, PyObject* args)
708     {
709     int ldid;
710     if (!PyArg_ParseTuple(args, "i:language_driver_name", &ldid)) return NULL;
711     if (ldid < 0 || ldid >= PYSHAPELIB_NUM_LANGUAGE_DRIVERS || !drivers[ldid])
712     {
713     PyErr_SetString(PyExc_ValueError, "invalid driver id");
714     return NULL;
715     }
716     return PyString_FromString(drivers[ldid]);
717     }
718    
719     #endif
720    
721    
722    
723 bramz 2742 static struct PyMethodDef dbflib_methods[] =
724     {
725 bramz 2744 {"open", (PyCFunction)dbflib_open, METH_VARARGS,
726 bramz 2745 "open(name [, mode]) -> DBFFile\n\n"
727     "opens a DBFFile" },
728 bramz 2744 {"create", (PyCFunction)dbflib_create, METH_VARARGS,
729 bramz 2752 "create(name [, language_driver]) -> DBFFile\n\n"
730     "create a DBFFile " },
731     #if HAVE_LANGUAGE_DRIVER
732     {"language_driver_codec", (PyCFunction)dbflib_language_driver_codec, METH_VARARGS,
733     "language_driver_codec(driver_id) -> string\n\n"
734     "translate language driver id into the name of the Python's codec used as code page." },
735     {"language_driver_name", (PyCFunction)dbflib_language_driver_name, METH_VARARGS,
736     "language_driver_name(driver_id) -> string\n\n"
737     "translate language driver id into a string." },
738     #endif
739 bramz 2742 {NULL}
740     };
741 jan 1611
742 bh 2212
743 bramz 2742 PyMODINIT_FUNC initdbflib(void)
744     {
745 bramz 2752 int i;
746    
747 bramz 2742 PyObject* module = Py_InitModule("dbflib", dbflib_methods);
748     if (!module) return;
749    
750     PYSHAPELIB_ADD_TYPE(DBFFileType, "DBFFile");
751    
752     PYSHAPELIB_ADD_CONSTANT(FTString);
753     PYSHAPELIB_ADD_CONSTANT(FTInteger);
754     PYSHAPELIB_ADD_CONSTANT(FTDouble);
755 bramz 2745 PYSHAPELIB_ADD_CONSTANT(FTLogical);
756 bramz 2742 PYSHAPELIB_ADD_CONSTANT(FTInvalid);
757     PyModule_AddIntConstant(module, "_have_commit", HAVE_UPDATE_HEADER);
758 bramz 2752
759     #if HAVE_LANGUAGE_DRIVER
760     /* table compiled from these resources:
761     * http://www.clicketyclick.dk/databases/xbase/format/dbf.html
762     * http://www.esrinl.com/content/file.asp?id=307
763     * http://msdn2.microsoft.com/en-us/library/aa975345(VS.71).aspx
764     */
765     for (i = 0; i < PYSHAPELIB_NUM_LANGUAGE_DRIVERS; ++i)
766     {
767     codecs[i] = NULL;
768     drivers[i] = NULL;
769     }
770     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x00, "cp1252", "NOT_SET");
771     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x01, "cp437", "DOS_USA");
772     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x02, "cp850", "DOS_INTERNATIONAL");
773     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x03, "cp1252", "WINDOWS_ANSI");
774     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x04, "mac_roman", "STANDARD_MACINTOSH");
775     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x08, "cp865", "DANISH_OEM");
776     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x09, "cp437", "DUTCH_OEM");
777     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x0a, "cp850", "DUTCH_OEM_2");
778     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x0b, "cp437", "FINNISH_OEM");
779     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x0d, "cp437", "FRENCH_OEM");
780     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x0e, "cp850", "FRENCH_OEM_2");
781     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x0f, "cp437", "GERMAN_OEM");
782     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x10, "cp850", "GERMAN_OEM_2");
783     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x11, "cp437", "ITALIAN_OEM");
784     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x12, "cp850", "ITALIAN_OEM_2");
785     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x13, "cp932", "JAPANESE_SHIFT_JIS");
786     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x14, "cp850", "SPANISH_OEM_2");
787     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x15, "cp437", "SWEDISH_OEM");
788     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x16, "cp850", "SWEDISH_OEM_2");
789     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x17, "cp865", "NORWEGIAN_OEM");
790     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x18, "cp437", "SPANISH_OEM");
791     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x19, "cp437", "ENGLISH_BRITAIN_OEM");
792     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x1a, "cp850", "ENGLISH_BRITAIN_OEM_2");
793     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x0b, "cp437", "ENGLISH_US_OEM");
794     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x1c, "cp863", "FRENCH_CANADA_OEM");
795     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x1d, "cp850", "FRENCH_OEM_2");
796     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x1f, "cp852", "CZECH_OEM");
797     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x22, "cp852", "HUNGARIAN_OEM");
798     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x23, "cp852", "POLISH_OEM");
799     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x24, "cp860", "PORTUGUESE_OEM");
800     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x25, "cp850", "PORTUGUESE_OEM_2");
801     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x26, "cp866", "RUSSIAN_OEM");
802     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x37, "cp850", "ENGLISH_US_OEM_2");
803     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x40, "cp852", "ROMANIAN_OEM");
804     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x4d, "cp936", "CHINESE_GBK_PRC");
805     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x4e, "cp949", "KOREAN_ANSI_OEM);");
806     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x4f, "cp950", "CHINESE_BIG5_TAIWAN");
807     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x50, "cp874", "THAI_ANSI_OEM");
808     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x57, "cp1252", "ESRI_ANSI");
809     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x58, "cp1252", "WESTERN_EUROPEAN_ANSI");
810     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x59, "cp1252", "SPANISH_ANSI");
811     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x64, "cp852", "EASTERN_EUROPEAN_MSDOS");
812     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x65, "cp866", "RUSSIAN_MSDOS");
813     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x66, "cp865", "NORDIC_MSDOS");
814     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x67, "cp861", "ICELANDIC_MSDOS");
815     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x68, "cp895", "CZECH_MSDOS");
816     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x69, "cp620", "POLISH_MSDOS");
817     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x6a, "cp737", "GREEK_MSDOS");
818     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x6b, "cp857", "TURKISH_MSDOS");
819     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x6c, "cp863", "FRENCH_CANADA_MSDOS");
820     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x78, "cp950", "TAIWAN_BIG5");
821     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x79, "cp949", "HANGUL_WANSUG");
822     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x7a, "cp936", "PRC_GBK");
823     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x7b, "cp932", "JAPANESE_SHIFT_JIS");
824     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x7c, "cp874", "THAI_WINDOWS_MSDOS");
825     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x7d, "cp1255", "HEBREW_WINDOWS");
826     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x7e, "cp1256", "ARABIC_WINDOWS");
827     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x86, "cp737", "GREEK_OEM");
828     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x87, "cp852", "SLOVENIAN_OEM");
829     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x88, "cp857", "TURKISH_OEM");
830     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x96, "mac_cyrillic", "RUSSIAN_MACINTOSH");
831     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x97, "mac_latin2", "EASTERN_EUROPEAN_MACINTOSH");
832     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0x98, "mac_greek", "GREEK_MACINTOSH");
833     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0xc8, "cp1250", "EASTERN_EUROPEAN_WINDOWS");
834     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0xc9, "cp1251", "RUSSIAN_WINDOWS");
835     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0xca, "cp1254", "TURKISH_WINDOWS");
836     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0xcb, "cp1253", "GREEK_WINDOWS");
837     PYSHAPELIB_ADD_LANGUAGE_DRIVER(0xcc, "cp1257", "BALTIC_WINDOWS");
838     #endif
839    
840 bramz 2742 }

Properties

Name Value
svn:eol-style native
svn:keywords Author Date Id Revision

[email protected]
ViewVC Help
Powered by ViewVC 1.1.26