← Back to branch summary

~psycopg/psycopg/2.0.x

« back to all changes in this revision

Viewing changes to psycopg/typecast_basic.c.old

  • Committer: Federico Di Gregorio
  • Date: 2004-10-19 03:17:12 UTC
  • Revision ID: fog-3331e1822980e428b2fe291ebc794e704e32642a
Initial psycopg 2 import after SVN crash.

Show diffs side-by-side

added added

removed removed

Lines of Context:
psycopg/typecast_basic.c.old
 
1
/* pgcasts_basic.c - basic typecasting functions to python types
 
2
 *
 
3
 * Copyright (C) 2001-2003 Federico Di Gregorio <fog@debian.org>
 
4
 *
 
5
 * This file is part of the psycopg module.
 
6
 *
 
7
 * This program is free software; you can redistribute it and/or
 
8
 * modify it under the terms of the GNU General Public License
 
9
 * as published by the Free Software Foundation; either version 2,
 
10
 * or (at your option) any later version.
 
11
 *
 
12
 * This program is distributed in the hope that it will be useful,
 
13
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 
14
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
15
 * GNU General Public License for more details.
 
16
 *
 
17
 * You should have received a copy of the GNU General Public License
 
18
 * along with this program; if not, write to the Free Software
 
19
 * Foundation, 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
 
20
 */
 
21
 
 
22
#include <libpq-fe.h>
 
23
 
 
24
/** INTEGER - cast normal integers (4 bytes) to python int **/
 
25
 
 
26
static PyObject *
 
27
typecast_INTEGER_cast(PyObject *s, PyObject *curs)
 
28
{
 
29
    if (s == Py_None) {Py_INCREF(s); return s;}
 
30
    return PyNumber_Int(s);
 
31
}
 
32
 
 
33
/** LONGINTEGER - cast long integers (8 bytes) to python long **/
 
34
 
 
35
static PyObject *
 
36
typecast_LONGINTEGER_cast(PyObject *s, PyObject *curs)
 
37
{
 
38
    if (s == Py_None) {Py_INCREF(s); return s;}
 
39
    return PyNumber_Long(s);
 
40
}
 
41
 
 
42
/** FLOAT - cast floating point numbers to python float **/
 
43
 
 
44
static PyObject *
 
45
typecast_FLOAT_cast(PyObject *s, PyObject *curs)
 
46
{
 
47
    if (s == Py_None) {Py_INCREF(s); return s;}
 
48
    return PyNumber_Float(s);
 
49
}
 
50
 
 
51
/** STRING - cast strings of any type to python string **/
 
52
 
 
53
static PyObject *
 
54
typecast_STRING_cast(PyObject *s, PyObject *curs)
 
55
{
 
56
    Py_INCREF(s);
 
57
    return s;
 
58
}
 
59
 
 
60
/** BINARY - cast a binary string into a python string **/
 
61
 
 
62
/* the function typecast_BINARY_cast_unescape is used when libpq does not
 
63
   provide PQunescapeBytea: it convert all the \xxx octal sequences to the
 
64
   proper byte value */
 
65
 
 
66
#ifdef PSYCOPG_OWN_QUOTING
 
67
static unsigned char *
 
68
typecast_BINARY_cast_unescape(unsigned char *str, size_t *to_length)
 
69
{
 
70
    char *dstptr, *dststr;
 
71
    int len, i;
 
72
 
 
73
    len = strlen(str);
 
74
    dststr = (char*)calloc(len, sizeof(char));
 
75
    dstptr = dststr;
 
76
 
 
77
    if (dststr == NULL) return NULL;
 
78
 
 
79
    Py_BEGIN_ALLOW_THREADS;
 
80
   
 
81
    for (i = 0; i < len; i++) {
 
82
        if (str[i] == '\\') {
 
83
            if ( ++i < len) {
 
84
                if (str[i] == '\\') {
 
85
                    *dstptr = '\\';
 
86
                }
 
87
                else {
 
88
                    *dstptr = 0;
 
89
                    *dstptr |= (str[i++] & 7) << 6;
 
90
                    *dstptr |= (str[i++] & 7) << 3;
 
91
                    *dstptr |= (str[i] & 7);
 
92
                }
 
93
            }
 
94
        }
 
95
        else {
 
96
            *dstptr = str[i];
 
97
        }
 
98
        dstptr++;
 
99
    }
 
100
    
 
101
    Py_END_ALLOW_THREADS;
 
102
 
 
103
    *to_length = (size_t)(dstptr-dststr);
 
104
    
 
105
    return dststr;
 
106
}
 
107
 
 
108
#define PQunescapeBytea typecast_BINARY_cast_unescape
 
109
#endif
 
110
    
 
111
static PyObject *
 
112
typecast_BINARY_cast(PyObject *s, PyObject *curs)
 
113
{
 
114
    PyObject *res;
 
115
    unsigned char *str;
 
116
    size_t len;
 
117
 
 
118
    if (s == Py_None) {Py_INCREF(s); return s;}
 
119
   
 
120
    str = PQunescapeBytea(PyString_AS_STRING(s), &len);
 
121
    res = PyBuffer_FromMemory((void*)str, len);
 
122
    free(str);
 
123
    
 
124
    return res;
 
125
}
 
126
 
 
127
/** BOOLEAN - cast boolean value into right python object **/
 
128
 
 
129
static PyObject *
 
130
typecast_BOOLEAN_cast(PyObject *s, PyObject *curs)
 
131
{
 
132
    PyObject *res;
 
133
    
 
134
    if (s == Py_None) {Py_INCREF(s); return s;}
 
135
 
 
136
    if (PyString_AS_STRING(s)[0] == 't')
 
137
        res = Py_True;
 
138
    else
 
139
        res = Py_False;
 
140
 
 
141
    Py_INCREF(res);
 
142
    return res;
 
143
}
 
144
 
 
145
/* some needed aliases */
 
146
#define typecast_NUMBER_cast   typecast_FLOAT_cast
 
147
#define typecast_ROWID_cast    typecast_INTEGER_cast