pgui_action_export()

static void pgui_action_export ( GtkWidget *  widget, gpointer  data  )

static

Definition at line 1764 of file shp2pgsql-gui.c.

References _, shp_dumper_state::config, shp_dumper_config::conn, conn, connection_test(), dialog_folderchooser, dialog_progress, EXPORT_POINTER_COLUMN, export_table_list_store, FALSE, GUIMSG_LINE_MAXLEN, is_running, label_progress, malloc(), shp_dumper_state::message, shp_dumper_state::outshptype, shp_dumper_state::pgis_major_version, pgui_logf(), pgui_raise_error_dialogue(), pgui_seterr(), progress, shp_dumper_config::schema, shapetypename(), shp_dumper_config::shp_file, ShpDumperCloseTable(), ShpDumperConnectDatabase(), ShpDumperCreate(), ShpDumperDestroy(), SHPDUMPERERR, ShpDumperGetRecordCount(), SHPDUMPEROK, ShpDumperOpenTable(), ShpLoaderGenerateShapeRow(), shp_dumper_config::table, TRUE, update_conn_ui_from_conn_config(), and window_conn.

Referenced by pgui_create_main_window().

{
        SHPDUMPERCONFIG *dumper_table_config;
        SHPDUMPERSTATE *state;
        gint is_valid;
        gpointer gptr;
        GtkTreeIter iter;
        char *output_shapefile, *orig_shapefile;
  char progress_text[GUIMSG_LINE_MAXLEN+1];
        gchar *folder_path;

        int ret, success = FALSE, i = 0;

        /* Get the first row of the import list */
        is_valid = gtk_tree_model_get_iter_first(GTK_TREE_MODEL(export_table_list_store), &iter);
        if (!is_valid)
        {
                pgui_seterr(_("ERROR: You haven't specified any tables to export"));
                pgui_raise_error_dialogue();

                return;
        }

        /* Firstly make sure that we can connect to the database - if we can't then there isn't much
           point doing anything else... */
        if (!connection_test())
        {
                pgui_seterr(_("Unable to connect to the database - please check your connection settings"));
                pgui_raise_error_dialogue();

                /* Open the connections UI for the user */
                update_conn_ui_from_conn_config();

                gtk_widget_show_all(GTK_WIDGET(window_conn));
                return;
        }

        /* Now open the file selector dialog so the user can specify where they would like the output
           files to reside */
        if (gtk_dialog_run(GTK_DIALOG(dialog_folderchooser)) != GTK_RESPONSE_ACCEPT)
        {
                gtk_widget_hide(dialog_folderchooser);

                return;
        }

        gtk_widget_hide(dialog_folderchooser);
        folder_path = gtk_file_chooser_get_current_folder(GTK_FILE_CHOOSER(dialog_folderchooser));

        /* Now everything is set up, let's extract the tables */
        is_valid = gtk_tree_model_get_iter_first(GTK_TREE_MODEL(export_table_list_store), &iter);
        while (is_valid)
        {
                /* Grab the SHPDUMPERCONFIG for this row */
                gtk_tree_model_get(GTK_TREE_MODEL(export_table_list_store), &iter, EXPORT_POINTER_COLUMN, &gptr, -1);
                dumper_table_config = (SHPDUMPERCONFIG *)gptr;

                pgui_logf("\n==============================");
                pgui_logf("Exporting with configuration: %s, %s, %s", dumper_table_config->table, dumper_table_config->schema, dumper_table_config->shp_file);

                /* Export is running */
                is_running = TRUE;
                success = FALSE;

                /* Disable the button to prevent multiple imports running at the same time */
                gtk_widget_set_sensitive(widget, FALSE);

                /* Allow GTK events to get a look in */
                while (gtk_events_pending())
                        gtk_main_iteration();

                /* Create the state for each configuration */
                state = ShpDumperCreate(dumper_table_config);
                state->config->conn = conn;

                /* Save the original shapefile name, then create a temporary version containing the full path */
                orig_shapefile = dumper_table_config->shp_file;
                output_shapefile = malloc(strlen(folder_path) + strlen(dumper_table_config->shp_file) + 2);
                strcpy(output_shapefile, folder_path);
                strcat(output_shapefile, G_DIR_SEPARATOR_S);
                strcat(output_shapefile, dumper_table_config->shp_file);

                dumper_table_config->shp_file = output_shapefile;

                /* Connect to the database */
                ret = ShpDumperConnectDatabase(state);
                if (ret != SHPDUMPEROK)
                {
                        pgui_seterr("%s", state->message);
                        pgui_raise_error_dialogue();

                        goto export_cleanup;
                }

                /* Display the progress dialog */
                gtk_label_set_text(GTK_LABEL(label_progress), _("Initialising..."));
                gtk_progress_bar_set_fraction(GTK_PROGRESS_BAR(progress), 0.0);
                gtk_widget_show_all(dialog_progress);

                ret = ShpDumperOpenTable(state);
                if (ret != SHPDUMPEROK)
                {
                        pgui_logf("%s", state->message);

                        if (ret == SHPDUMPERERR)
                        {
                                gtk_widget_hide(dialog_progress);

                                pgui_seterr("%s", state->message);
                                pgui_raise_error_dialogue();

                                goto export_cleanup;
                        }
                }

                /* Update the text */
                snprintf(progress_text, GUIMSG_LINE_MAXLEN, _("Exporting table %s (%d records)..."), dumper_table_config->table, ShpDumperGetRecordCount(state));
                progress_text[GUIMSG_LINE_MAXLEN] = '\0';
                gtk_label_set_text(GTK_LABEL(label_progress), progress_text);

                /* Allow GTK events to get a look in */
                while (gtk_events_pending())
                        gtk_main_iteration();

                pgui_logf(_("Done (postgis major version: %d)"), state->pgis_major_version);
                pgui_logf(_("Output shape: %s"), shapetypename(state->outshptype));

                for (i = 0; i < ShpDumperGetRecordCount(state) && is_running == TRUE; i++)
                {
                        ret = ShpLoaderGenerateShapeRow(state);
                        if (ret != SHPDUMPEROK)
                        {
                                pgui_logf("%s", state->message);

                                if (ret == SHPDUMPERERR)
                                {
                                        gtk_widget_hide(dialog_progress);

                                        pgui_seterr("%s", state->message);
                                        pgui_raise_error_dialogue();

                                        goto export_cleanup;
                                }
                        }

                        /* Update the progress bar */
                        gtk_progress_bar_set_fraction(GTK_PROGRESS_BAR(progress), (float)i / ShpDumperGetRecordCount(state));

                        /* Allow GTK events to get a look in */
                        while (gtk_events_pending())
                                gtk_main_iteration();
                }

                /* Finish the dump */
                ret = ShpDumperCloseTable(state);
                if (ret != SHPDUMPEROK)
                {
                        pgui_logf("%s", state->message);

                        if (ret == SHPDUMPERERR)
                        {
                                gtk_widget_hide(dialog_progress);

                                pgui_seterr("%s", state->message);
                                pgui_raise_error_dialogue();
                        }
                }

                /* Indicate success */
                if (is_running)
                        success = TRUE;

export_cleanup:

                /* Tidy up everything */
                ShpDumperDestroy(state);

                /* Reset shapefile back to original form (without full path) */
                dumper_table_config->shp_file = orig_shapefile;

                /* Get next entry */
                is_valid = gtk_tree_model_iter_next(GTK_TREE_MODEL(export_table_list_store), &iter);
        }

        /* Export has definitely finished */
        is_running = FALSE;
        if (!success)
                pgui_logf(_("Table export failed."));
        else
                pgui_logf(_("Table export completed."));

        /* Enable the button once again */
        gtk_widget_set_sensitive(widget, TRUE);

        /* Silly GTK bug means we have to hide and show the button for it to work again! */
        gtk_widget_hide(widget);
        gtk_widget_show(widget);

        /* Hide the progress dialog */
        gtk_widget_hide(dialog_progress);

        /* Allow GTK events to get a look in */
        while (gtk_events_pending())
                gtk_main_iteration();

        return;
}

Here is the call graph for this function:

Here is the caller graph for this function: