mirror of
https://github.com/mdbtools/mdbtools.git
synced 2025-11-26 10:29:27 +08:00
missed added files
This commit is contained in:
brianb
240
src/util/mdb-import.c
Normal file
240
src/util/mdb-import.c
Normal file
@@ -0,0 +1,240 @@
|
||||
/* MDB Tools - A library for reading MS Access database file
|
||||
* Copyright (C) 2000 Brian Bruns
|
||||
*
|
||||
*
|
||||
* This library is free software; you can redistribute it and/or
|
||||
* modify it under the terms of the GNU Library General Public
|
||||
* License as published by the Free Software Foundation; either
|
||||
* version 2 of the License, or (at your option) any later version.
|
||||
*
|
||||
* This library is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
||||
* Library General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU Library General Public
|
||||
* License along with this library; if not, write to the
|
||||
* Free Software Foundation, Inc., 59 Temple Place - Suite 330,
|
||||
* Boston, MA 02111-1307, USA.
|
||||
*/
|
||||
|
||||
#include "mdbtools.h"
|
||||
|
||||
#define MAX_ROW_SIZE 4096
|
||||
|
||||
char *delimiter = ",";
|
||||
char header_rows = 0;
|
||||
|
||||
void
|
||||
free_values(MdbField *fields, int num_fields)
|
||||
{
|
||||
int i;
|
||||
|
||||
for (i=0;i<num_fields;i++) {
|
||||
if (!fields[i].is_null)
|
||||
g_free(fields[i].value);
|
||||
}
|
||||
}
|
||||
int
|
||||
convert_field(MdbColumn *col, char *s, MdbField *field)
|
||||
{
|
||||
char *c;
|
||||
MdbAny any;
|
||||
|
||||
switch (col->col_type) {
|
||||
case MDB_TEXT:
|
||||
field->value = g_strdup(s);
|
||||
field->siz = strlen(s);
|
||||
break;
|
||||
case MDB_BYTE:
|
||||
any.i = strtol(s, &c, 16);
|
||||
if (*c) return 1;
|
||||
field->siz = mdb_col_fixed_size(col);
|
||||
field->value = g_malloc(field->siz);
|
||||
((unsigned char *)field->value)[0] = any.i;
|
||||
break;
|
||||
case MDB_INT:
|
||||
any.i = strtol(s, &c, 16);
|
||||
if (*c) return 1;
|
||||
field->siz = mdb_col_fixed_size(col);
|
||||
field->value = g_malloc(field->siz);
|
||||
_mdb_put_int16(field->value, 0, any.i);
|
||||
break;
|
||||
case MDB_LONGINT:
|
||||
any.i = strtol(s, &c, 16);
|
||||
if (*c) return 1;
|
||||
field->siz = mdb_col_fixed_size(col);
|
||||
field->value = g_malloc(field->siz);
|
||||
_mdb_put_int32(field->value, 0, any.i);
|
||||
break;
|
||||
case MDB_BOOL:
|
||||
if (*s == '1') {
|
||||
field->is_null = 1;
|
||||
} else if (*s == '0') {
|
||||
field->is_null = 0;
|
||||
} else {
|
||||
fprintf(stderr, "%c is not a valid value for type BOOLEAN\n", *s);
|
||||
return 1;
|
||||
}
|
||||
/*
|
||||
case MDB_FLOAT:
|
||||
any.d = strtod(s, &c);
|
||||
if (*c) return 1;
|
||||
field.value = g_malloc(mdb_col_fixed_size(col));
|
||||
_mdb_put_single(field.value, 0, any.i);
|
||||
break;
|
||||
case MDB_DOUBLE:
|
||||
any.d = strtod(s, &c);
|
||||
if (*c) return 1;
|
||||
field.value = g_malloc(mdb_col_fixed_size(col));
|
||||
_mdb_put_double(field.value, 0, any.i);
|
||||
break;
|
||||
*/
|
||||
/*
|
||||
case MDB_MONEY:
|
||||
case MDB_SDATETIME:
|
||||
case MDB_OLE:
|
||||
case MDB_MEMO:
|
||||
case MDB_REPID:
|
||||
case MDB_NUMERIC:
|
||||
*/
|
||||
default:
|
||||
fprintf(stderr,"Conversion of type %02x not supported yet.\n", col->col_type);
|
||||
return 1;
|
||||
break;
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
int
|
||||
prep_row(MdbTableDef *table, unsigned char *line, MdbField *fields)
|
||||
{
|
||||
MdbColumn *col;
|
||||
char delims[20];
|
||||
char *s;
|
||||
int file_cols;
|
||||
|
||||
/*
|
||||
* pull apart the row and turn it into something respectable
|
||||
*/
|
||||
sprintf(delims, "%c\n", delimiter[0]);
|
||||
s = strtok(line, delims);
|
||||
file_cols = 0;
|
||||
do {
|
||||
if (file_cols >= table->num_cols) {
|
||||
fprintf(stderr, "Number of columns in file exceeds number in table.\n");
|
||||
return 0;
|
||||
}
|
||||
if (s[0]=='"' && s[strlen(s)-1]=='"') {
|
||||
s[strlen(s)-1] = '\0';
|
||||
s++;
|
||||
}
|
||||
printf("field = %s\n", s);
|
||||
col = g_ptr_array_index (table->columns, file_cols);
|
||||
if (convert_field(col, s, &fields[file_cols])) {
|
||||
fprintf(stderr, "Format error in column %d\n", file_cols+1);
|
||||
return 0;
|
||||
}
|
||||
file_cols++;
|
||||
} while (s = strtok(NULL, delims));
|
||||
|
||||
/*
|
||||
* verify number of columns in table against number in row
|
||||
*/
|
||||
if (file_cols < table->num_cols) {
|
||||
fprintf(stderr, "Row has %d columns, but table has %d\n", file_cols, table->num_cols);
|
||||
free_values(fields, file_cols);
|
||||
return 0;
|
||||
}
|
||||
return file_cols;
|
||||
}
|
||||
int
|
||||
main(int argc, char **argv)
|
||||
{
|
||||
int i, row;
|
||||
MdbHandle *mdb;
|
||||
MdbCatalogEntry *entry;
|
||||
MdbTableDef *table;
|
||||
MdbField fields[256];
|
||||
unsigned char line[MAX_ROW_SIZE];
|
||||
int num_fields;
|
||||
/* doesn't handle tables > 256 columns. Can that happen? */
|
||||
int opt;
|
||||
char *s;
|
||||
FILE *in;
|
||||
|
||||
while ((opt=getopt(argc, argv, "H:d:"))!=-1) {
|
||||
switch (opt) {
|
||||
case 'H':
|
||||
header_rows = atol(optarg);
|
||||
break;
|
||||
case 'd':
|
||||
delimiter = (char *) malloc(strlen(optarg)+1);
|
||||
strcpy(delimiter, optarg);
|
||||
break;
|
||||
default:
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
/*
|
||||
** optind is now the position of the first non-option arg,
|
||||
** see getopt(3)
|
||||
*/
|
||||
if (argc-optind < 3) {
|
||||
fprintf(stderr,"Usage: %s [options] <database> <table> <csv file>\n",argv[0]);
|
||||
fprintf(stderr,"where options are:\n");
|
||||
fprintf(stderr," -H <rows> skip <rows> header rows\n");
|
||||
fprintf(stderr," -d <delimiter> specify a column delimiter\n");
|
||||
exit(1);
|
||||
}
|
||||
|
||||
mdb_init();
|
||||
|
||||
if (!(mdb = _mdb_open(argv[optind], TRUE))) {
|
||||
exit(1);
|
||||
}
|
||||
|
||||
mdb_read_catalog(mdb, MDB_TABLE);
|
||||
|
||||
for (i=0;i<mdb->num_catalog;i++) {
|
||||
entry = g_ptr_array_index(mdb->catalog,i);
|
||||
if (entry->object_type == MDB_TABLE &&
|
||||
!strcmp(entry->object_name,argv[argc-2])) {
|
||||
table = mdb_read_table(entry);
|
||||
mdb_read_columns(table);
|
||||
mdb_read_indices(table);
|
||||
mdb_rewind_table(table);
|
||||
break;
|
||||
}
|
||||
}
|
||||
/*
|
||||
* open the CSV file and read any header rows
|
||||
*/
|
||||
in = fopen(argv[argc-1], "r");
|
||||
if (!in) {
|
||||
fprintf(stderr, "Can not open file %s\n", argv[argc-1]);
|
||||
exit(1);
|
||||
}
|
||||
for (i=0;i<header_rows;i++)
|
||||
fgets(line, MAX_ROW_SIZE, in);
|
||||
|
||||
row = 1;
|
||||
while (fgets(line, MAX_ROW_SIZE, in)) {
|
||||
num_fields = prep_row(table, line, fields);
|
||||
if (!num_fields) {
|
||||
fprintf(stderr, "Aborting import at row %d\n", row);
|
||||
exit(1);
|
||||
}
|
||||
/*
|
||||
* all the prep work is done, let's add the row
|
||||
*/
|
||||
mdb_insert_row(table, num_fields, fields);
|
||||
}
|
||||
|
||||
fclose(in);
|
||||
mdb_free_handle(mdb);
|
||||
mdb_exit();
|
||||
|
||||
exit(0);
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user