aboutsummaryrefslogtreecommitdiffstats
path: root/Source/SPDataImport.m
diff options
context:
space:
mode:
Diffstat (limited to 'Source/SPDataImport.m')
-rw-r--r--Source/SPDataImport.m74
1 files changed, 41 insertions, 33 deletions
diff --git a/Source/SPDataImport.m b/Source/SPDataImport.m
index f0d1a853..00876c8d 100644
--- a/Source/SPDataImport.m
+++ b/Source/SPDataImport.m
@@ -67,6 +67,8 @@
if ((self = [super init])) {
nibObjectsToRelease = [[NSMutableArray alloc] init];
+ geometryFields = [[NSMutableArray alloc] init];
+ geometryFieldsMapIndex = [[NSMutableIndexSet alloc] init];
fieldMappingArray = nil;
fieldMappingGlobalValueArray = nil;
fieldMappingTableColumnNames = nil;
@@ -82,6 +84,7 @@
insertRemainingRowsAfterUpdate = NO;
numberOfImportDataColumns = 0;
selectedTableTarget = nil;
+ targetTableDetails = nil;
prefs = nil;
lastFilename = nil;
@@ -682,6 +685,9 @@
fieldMappingArray = nil;
fieldMappingGlobalValueArray = nil;
+ [geometryFields removeAllObjects];
+ [geometryFieldsMapIndex removeAllIndexes];
+
// Start the notification timer to allow notifications to be shown even if frontmost for long queries
[[SPGrowlController sharedGrowlController] setVisibilityForNotificationName:@"Import Finished"];
@@ -709,7 +715,7 @@
[[singleProgressBar onMainThread] setIndeterminate:YES];
[[singleProgressBar onMainThread] setUsesThreadedAnimation:YES];
[[singleProgressBar onMainThread] startAnimation:self];
-
+
// Open the progress sheet
[[NSApp onMainThread] beginSheet:singleProgressSheet modalForWindow:[tableDocumentInstance parentWindow] modalDelegate:self didEndSelector:nil contextInfo:nil];
[[singleProgressSheet onMainThread] makeKeyWindow];
@@ -887,14 +893,19 @@
// Set up the field names import string for INSERT or REPLACE INTO
[insertBaseString appendString:csvImportHeaderString];
if(!importMethodIsUpdate) {
- [insertBaseString appendString:[selectedTableTarget backtickQuotedString]];
- [insertBaseString appendString:@" ("];
+ NSString *fieldName;
+ [insertBaseString appendFormat:@"%@ (", [selectedTableTarget backtickQuotedString]];
insertBaseStringHasEntries = NO;
for (i = 0; i < [fieldMappingArray count]; i++) {
if ([NSArrayObjectAtIndex(fieldMapperOperator, i) integerValue] == 0) {
- if (insertBaseStringHasEntries) [insertBaseString appendString:@","];
- else insertBaseStringHasEntries = YES;
- [insertBaseString appendString:[NSArrayObjectAtIndex(fieldMappingTableColumnNames, i) backtickQuotedString]];
+ if (insertBaseStringHasEntries)
+ [insertBaseString appendString:@","];
+ else
+ insertBaseStringHasEntries = YES;
+ // Store column index for each geometry field to be able to apply GeomFromText() while importing
+ if([geometryFields containsObject:fieldName = NSArrayObjectAtIndex(fieldMappingTableColumnNames, i) ])
+ [geometryFieldsMapIndex addIndex:i];
+ [insertBaseString appendString:[fieldName backtickQuotedString]];
}
}
[insertBaseString appendString:@") VALUES\n"];
@@ -949,8 +960,7 @@
} else {
if(insertRemainingRowsAfterUpdate) {
[insertRemainingBaseString setString:@"INSERT INTO "];
- [insertRemainingBaseString appendString:[selectedTableTarget backtickQuotedString]];
- [insertRemainingBaseString appendString:@" ("];
+ [insertRemainingBaseString appendFormat:@"%@ (", [selectedTableTarget backtickQuotedString]];
insertBaseStringHasEntries = NO;
for (i = 0; i < [fieldMappingArray count]; i++) {
if ([NSArrayObjectAtIndex(fieldMapperOperator, i) integerValue] == 0) {
@@ -1145,19 +1155,6 @@
}
fieldMappingImportArrayIsPreview = dataIsPreviewData;
- // If there's no tables to select, error
- // if (![[tablesListInstance allTableNames] count]) {
- // [self closeAndStopProgressSheet];
- // SPBeginAlertSheet(NSLocalizedString(@"Error", @"error"),
- // NSLocalizedString(@"OK", @"OK button"),
- // nil, nil,
- // [tableDocumentInstance parentWindow], self,
- // nil, nil,
- // NSLocalizedString(@"Can't import CSV data into a database without any tables!", @"error text when trying to import csv data, but we have no tables in the db")
- // );
- // return FALSE;
- // }
-
// Set the import array
if (fieldMappingImportArray) [fieldMappingImportArray release];
fieldMappingImportArray = [[NSArray alloc] initWithArray:importData];
@@ -1213,6 +1210,17 @@
return FALSE;
}
+ // Store target table definitions
+ SPTableData *selectedTableData = [[SPTableData alloc] init];
+ [selectedTableData setConnection:mySQLConnection];
+ targetTableDetails = [selectedTableData informationForTable:selectedTableTarget];
+ [selectedTableData release];
+
+ // Store all field names which are of typegrouping 'geometry'
+ for(NSDictionary *field in [targetTableDetails objectForKey:@"columns"])
+ if([[field objectForKey:@"typegrouping"] isEqualToString:@"geometry"])
+ [geometryFields addObject:[field objectForKey:@"name"]];
+
[importFieldNamesSwitch setState:[fieldMapperController importFieldNamesHeader]];
[prefs setBool:[importFieldNamesSwitch state] forKey:SPCSVImportFirstLineIsHeader];
@@ -1275,9 +1283,7 @@
if (cellData == [NSNull null]) {
[setString appendString:@"NULL"];
} else {
- [setString appendString:@"'"];
- [setString appendString:[mySQLConnection prepareString:cellData]];
- [setString appendString:@"'"];
+ [setString appendFormat:@"'%@'", [mySQLConnection prepareString:cellData]];
}
}
}
@@ -1290,8 +1296,7 @@
// - check for global values
if(fieldMappingArrayHasGlobalVariables && mapColumn >= numberOfImportDataColumns) {
// Global variables are coming wrapped in ' ' if there're not marked as SQL
- [whereString appendString:@"="];
- [whereString appendString:NSArrayObjectAtIndex(fieldMappingGlobalValueArray, mapColumn)];
+ [whereString appendFormat:@"=%@", NSArrayObjectAtIndex(fieldMappingGlobalValueArray, mapColumn)];
} else {
cellData = NSArrayObjectAtIndex(csvRowArray, mapColumn);
@@ -1302,10 +1307,7 @@
if (cellData == [NSNull null]) {
[whereString appendString:@" IS NULL"];
} else {
- [whereString appendString:@"="];
- [whereString appendString:@"'"];
- [whereString appendString:[mySQLConnection prepareString:cellData]];
- [whereString appendString:@"'"];
+ [whereString appendFormat:@"='%@'", [mySQLConnection prepareString:cellData]];
}
}
}
@@ -1350,9 +1352,12 @@
if (cellData == [NSNull null]) {
[valueString appendString:@"NULL"];
} else {
- [valueString appendString:@"'"];
- [valueString appendString:[mySQLConnection prepareString:cellData]];
- [valueString appendString:@"'"];
+ // Apply GeomFromText() for each geometry field
+ if([geometryFieldsMapIndex containsIndex:i]) {
+ [valueString appendFormat:@"GeomFromText('%@')", [mySQLConnection prepareString:cellData]];
+ } else {
+ [valueString appendFormat:@"'%@'", [mySQLConnection prepareString:cellData]];
+ }
}
}
}
@@ -1497,6 +1502,9 @@
- (void)dealloc
{
if (fieldMappingImportArray) [fieldMappingImportArray release];
+ if (geometryFields) [geometryFields release];
+ if (geometryFieldsMapIndex) [geometryFieldsMapIndex release];
+
if (lastFilename) [lastFilename release];
if (prefs) [prefs release];
if(selectedTableTarget) [selectedTableTarget release];