diff options
Diffstat (limited to 'src/backends/PostgreSQL')
-rw-r--r-- | src/backends/PostgreSQL/Makefile.am | 17 | ||||
-rw-r--r-- | src/backends/PostgreSQL/psqlrecipedb.cpp | 560 | ||||
-rw-r--r-- | src/backends/PostgreSQL/psqlrecipedb.h | 64 |
3 files changed, 641 insertions, 0 deletions
diff --git a/src/backends/PostgreSQL/Makefile.am b/src/backends/PostgreSQL/Makefile.am new file mode 100644 index 0000000..dac2623 --- /dev/null +++ b/src/backends/PostgreSQL/Makefile.am @@ -0,0 +1,17 @@ +## Makefile.am for krecipes + +# this is the program that gets installed. it's name is used for all +# of the other Makefile.am variables + +# set the include path for X, tqt and TDE +INCLUDES = -I$(srcdir) -I$(srcdir)/.. -I$(srcdir)/../.. $(all_includes) + + +# Instructions for building the convenience library +noinst_LTLIBRARIES=libkrecpsql.la +libkrecpsql_la_SOURCES=psqlrecipedb.cpp +libkrecpsql_la_METASOURCES=AUTO + + +#the library search path. +libkrecpsql_la_LDFLAGS = $(KDE_RPATH) $(all_libraries) diff --git a/src/backends/PostgreSQL/psqlrecipedb.cpp b/src/backends/PostgreSQL/psqlrecipedb.cpp new file mode 100644 index 0000000..c5e2129 --- /dev/null +++ b/src/backends/PostgreSQL/psqlrecipedb.cpp @@ -0,0 +1,560 @@ +/*************************************************************************** +* Copyright (C) 2003 by * +* Unai Garro ([email protected]) * +* Cyril Bosselut ([email protected]) * +* Jason Kivlighn ([email protected]) * +* * +* Copyright (C) 2006 Jason Kivlighn ([email protected]) * +* * +* This program is free software; you can redistribute it and/or modify * +* it under the terms of the GNU General Public License as published by * +* the Free Software Foundation; either version 2 of the License, or * +* (at your option) any later version. * +***************************************************************************/ + +#include "psqlrecipedb.h" + +#include <kdebug.h> +#include <kstandarddirs.h> +#include <tdetempfile.h> +#include <tdelocale.h> +#include <tdeconfig.h> + +#include <tqvariant.h> + +//Note: PostgreSQL's database names are always lowercase +PSqlRecipeDB::PSqlRecipeDB( const TQString& host, const TQString& user, const TQString& pass, const TQString& DBname, int port ) : TQSqlRecipeDB( host, user, pass, DBname.lower(), port ) +{} + +PSqlRecipeDB::~PSqlRecipeDB() +{} + +void PSqlRecipeDB::createDB() +{ + TQString real_db_name = database->databaseName(); + + //we have to be connected to some database in order to create the Krecipes database + //so long as the permissions given are allowed access to "template1', this works + database->setDatabaseName( "template1" ); + if ( database->open() ) { + TQSqlQuery query( TQString( "CREATE DATABASE %1" ).arg( real_db_name ), database ); + if ( !query.isActive() ) + kdDebug() << "create query failed: " << database->lastError().databaseText() << endl; + + database->close(); + } + else + kdDebug() << "create open failed: " << database->lastError().databaseText() << endl; + + database->setDatabaseName( real_db_name ); +} + +TQStringList PSqlRecipeDB::backupCommand() const +{ + TDEConfig *config = TDEGlobal::config(); + config->setGroup("Server"); + + TQStringList command; + command<<config->readEntry( "PgDumpPath", "pg_dump" )<<"-d"<<database->databaseName() + <<"-U"<<config->readEntry( "Username" ); + + int port = config->readNumEntry( "Port", 0 ); + if ( port > 0 ) + command<<"-p"<<TQString::number(port); + + return command; +} + +TQStringList PSqlRecipeDB::restoreCommand() const +{ + TDEConfig *config = TDEGlobal::config(); + config->setGroup("Server"); + + TQStringList command; + command<<config->readEntry( "PsqlPath", "psql" )<<database->databaseName() + <<"-U"<<config->readEntry( "Username" ); + + int port = config->readNumEntry( "Port", 0 ); + if ( port > 0 ) + command<<"-p"<<TQString::number(port); + + return command; +} + +void PSqlRecipeDB::createTable( const TQString &tableName ) +{ + + TQStringList commands; + + if ( tableName == "recipes" ) + commands << "CREATE TABLE recipes (id SERIAL NOT NULL PRIMARY KEY,title CHARACTER VARYING, yield_amount FLOAT, yield_amount_offset FLOAT, yield_type_id INTEGER DEFAULT '-1', instructions TEXT, photo TEXT, prep_time TIME, ctime TIMESTAMP, mtime TIMESTAMP, atime TIMESTAMP );"; + + else if ( tableName == "ingredients" ) + commands << "CREATE TABLE ingredients (id SERIAL NOT NULL PRIMARY KEY, name CHARACTER VARYING);"; + + else if ( tableName == "ingredient_list" ) { + commands << "CREATE TABLE ingredient_list (id SERIAL NOT NULL PRIMARY KEY, recipe_id INTEGER, ingredient_id INTEGER, amount FLOAT, amount_offset FLOAT, unit_id INTEGER, order_index INTEGER, group_id INTEGER, substitute_for INTEGER);"; + commands << "CREATE INDEX ridil_index ON ingredient_list USING BTREE (recipe_id);"; + commands << "CREATE INDEX iidil_index ON ingredient_list USING BTREE (ingredient_id);"; + commands << "CREATE INDEX gidil_index ON ingredient_list USING BTREE (group_id);"; + } + + else if ( tableName == "unit_list" ) + commands << "CREATE TABLE unit_list (ingredient_id INTEGER, unit_id INTEGER);"; + + else if ( tableName == "units" ) + commands << "CREATE TABLE units (id SERIAL NOT NULL PRIMARY KEY, name CHARACTER VARYING, name_abbrev CHARACTER VARYING, plural CHARACTER VARYING, plural_abbrev CHARACTER VARYING, type INTEGER NOT NULL DEFAULT '0' );"; + + else if ( tableName == "prep_methods" ) + commands << "CREATE TABLE prep_methods (id SERIAL NOT NULL PRIMARY KEY, name CHARACTER VARYING);"; + else if ( tableName == "prep_method_list" ) { + commands << "CREATE TABLE prep_method_list (ingredient_list_id INTEGER NOT NULL,prep_method_id INTEGER NOT NULL, order_index INTEGER);"; + commands << "CREATE INDEX iid_index ON prep_method_list USING BTREE (ingredient_list_id);"; + commands << "CREATE INDEX pid_index ON prep_method_list USING BTREE (prep_method_id);"; + } + else if ( tableName == "ingredient_info" ) + commands << "CREATE TABLE ingredient_info (ingredient_id INTEGER, property_id INTEGER, amount FLOAT, per_units INTEGER);"; + + else if ( tableName == "ingredient_properties" ) + commands << "CREATE TABLE ingredient_properties (id SERIAL NOT NULL,name CHARACTER VARYING, units CHARACTER VARYING);"; + + else if ( tableName == "ingredient_weights" ) { + commands << "CREATE TABLE ingredient_weights (id SERIAL NOT NULL PRIMARY KEY, ingredient_id INTEGER NOT NULL, amount FLOAT, unit_id INTEGER, weight FLOAT, weight_unit_id INTEGER, prep_method_id INTEGER );" + << "CREATE INDEX weight_wid_index ON ingredient_weights USING BTREE (weight_unit_id)" + << "CREATE INDEX weight_pid_index ON ingredient_weights USING BTREE (prep_method_id)" + << "CREATE INDEX weight_uid_index ON ingredient_weights USING BTREE (unit_id)" + << "CREATE INDEX weight_iid_index ON ingredient_weights USING BTREE (ingredient_id)"; + } + + else if ( tableName == "units_conversion" ) + commands << "CREATE TABLE units_conversion (unit1_id INTEGER, unit2_id INTEGER, ratio FLOAT);"; + + else if ( tableName == "categories" ) { + commands << "CREATE TABLE categories (id SERIAL NOT NULL PRIMARY KEY, name CHARACTER VARYING default NULL, parent_id INTEGER NOT NULL default -1);"; + commands << "CREATE index parent_id_index ON categories USING BTREE(parent_id);"; + } + else if ( tableName == "category_list" ) { + commands << "CREATE TABLE category_list (recipe_id INTEGER NOT NULL,category_id INTEGER NOT NULL);"; + commands << "CREATE INDEX rid_index ON category_list USING BTREE (recipe_id);"; + commands << "CREATE INDEX cid_index ON category_list USING BTREE (category_id);"; + } + + else if ( tableName == "authors" ) + commands << "CREATE TABLE authors (id SERIAL NOT NULL PRIMARY KEY, name CHARACTER VARYING default NULL);"; + + else if ( tableName == "author_list" ) + commands << "CREATE TABLE author_list (recipe_id INTEGER NOT NULL,author_id INTEGER NOT NULL);"; + + else if ( tableName == "db_info" ) { + commands << "CREATE TABLE db_info (ver FLOAT NOT NULL,generated_by CHARACTER VARYING default NULL);"; + commands << TQString( "INSERT INTO db_info VALUES(%1,'Krecipes %2');" ).arg( latestDBVersion() ).arg( krecipes_version() ); + } + else if ( tableName == "ingredient_groups" ) { + commands << "CREATE TABLE ingredient_groups (id SERIAL NOT NULL PRIMARY KEY, name CHARACTER VARYING);"; + } + else if ( tableName == "yield_types" ) { + commands << "CREATE TABLE yield_types (id SERIAL NOT NULL PRIMARY KEY, name CHARACTER VARYING);"; + } + + else if ( tableName == "ratings" ) + commands << "CREATE TABLE ratings (id SERIAL NOT NULL PRIMARY KEY, recipe_id INTEGER NOT NULL, comment CHARACTER VARYING, rater CHARACTER VARYING, created TIMESTAMP);"; + + else if ( tableName == "rating_criteria" ) + commands << "CREATE TABLE rating_criteria (id SERIAL NOT NULL PRIMARY KEY, name CHARACTER VARYING);"; + + else if ( tableName == "rating_criterion_list" ) + commands << "CREATE TABLE rating_criterion_list (rating_id INTEGER NOT NULL, rating_criterion_id INTEGER, stars FLOAT);"; + else + return ; + + TQSqlQuery databaseToCreate( TQString::null, database ); + + // execute the queries + for ( TQStringList::const_iterator it = commands.begin(); it != commands.end(); ++it ) + databaseToCreate.exec( *it ); +} + +void PSqlRecipeDB::initializeData() +{ + TQSqlRecipeDB::initializeData(); + + TQSqlQuery updateSeq( "SELECT setval('units_id_seq',(SELECT COUNT(1) FROM units))", database ); + updateSeq.exec( "SELECT setval('categories_id_seq',(SELECT COUNT(1) FROM categories))" ); +} + +void PSqlRecipeDB::portOldDatabases( float version ) +{ + kdDebug() << "Current database version is..." << version << "\n"; + TQString command; + + if ( tqRound(version*10) < 7 ) { + //version added + } + + if ( tqRound(version*100) < 81 ) { + database->transaction(); + + addColumn("CREATE TABLE %1 (recipe_id INTEGER, ingredient_id INTEGER, amount FLOAT, %2 unit_id INTEGER, prep_method_id INTEGER, order_index INTEGER, group_id INTEGER);","amount_offset FLOAT","'0'","ingredient_list",3); + + TQSqlQuery query(TQString::null,database); + query.exec( "CREATE INDEX ridil_index ON ingredient_list USING BTREE (recipe_id);" ); + query.exec( "CREATE INDEX iidil_index ON ingredient_list USING BTREE (ingredient_id);"); + + query.exec( "UPDATE db_info SET ver='0.81',generated_by='Krecipes SVN (20050816)';" ); + + if ( !database->commit() ) + kdDebug()<<"Update to 0.81 failed. Maybe you should try again."<<endl; + } + + if ( tqRound(version*100) < 82 ) { + database->transaction(); + + //==================add a columns to 'recipes' to allow yield range + yield type + database->exec( "CREATE TABLE recipes_copy (id SERIAL NOT NULL PRIMARY KEY,title CHARACTER VARYING, persons INTEGER, instructions TEXT, photo TEXT, prep_time TIME);" ); + TQSqlQuery copyQuery = database->exec( "SELECT id,title,persons,instructions,photo,prep_time FROM recipes;" ); + if ( copyQuery.isActive() ) { + while ( copyQuery.next() ) { + TQSqlQuery query(TQString::null,database); + query.prepare( "INSERT INTO recipes_copy VALUES (?, ?, ?, ?, ?, ?)" ); + query.addBindValue( copyQuery.value( 0 ) ); + query.addBindValue( copyQuery.value( 1 ) ); + query.addBindValue( copyQuery.value( 2 ) ); + query.addBindValue( copyQuery.value( 3 ) ); + query.addBindValue( copyQuery.value( 4 ) ); + query.addBindValue( copyQuery.value( 5 ) ); + query.exec(); + + emit progress(); + } + } + database->exec( "DROP TABLE recipes" ); + database->exec( "CREATE TABLE recipes (id SERIAL NOT NULL PRIMARY KEY,title CHARACTER VARYING, yield_amount FLOAT, yield_amount_offset FLOAT, yield_type_id INTEGER DEFAULT '-1', instructions TEXT, photo TEXT, prep_time TIME);" ); + copyQuery = database->exec( "SELECT id,title,persons,instructions,photo,prep_time FROM recipes_copy" ); + if ( copyQuery.isActive() ) { + while ( copyQuery.next() ) { + TQSqlQuery query(TQString::null,database); + query.prepare( "INSERT INTO recipes VALUES (?, ?, ?, ?, ?, ?, ?, ?)" ); + query.addBindValue( copyQuery.value( 0 ) ); //id + query.addBindValue( copyQuery.value( 1 ) ); //title + query.addBindValue( copyQuery.value( 2 ) ); //persons, now yield_amount + query.addBindValue( 0 ); //yield_amount_offset + query.addBindValue( -1 ); //yield_type_id + query.addBindValue( copyQuery.value( 3 ) ); //instructions + query.addBindValue( copyQuery.value( 4 ) ); //photo + query.addBindValue( copyQuery.value( 5 ) ); //prep_time + query.exec(); + + emit progress(); + } + } + database->exec( "DROP TABLE recipes_copy" ); + + database->exec( "UPDATE db_info SET ver='0.82',generated_by='Krecipes SVN (20050902)';" ); + + if ( !database->commit() ) + kdDebug()<<"Update to 0.82 failed. Maybe you should try again."<<endl; + } + + if ( tqRound(version*100) < 83 ) { + database->transaction(); + + //====add a id columns to 'ingredient_list' to identify it for the prep method list + database->exec( "ALTER TABLE ingredient_list RENAME TO ingredient_list_copy;" ); + + database->exec( "CREATE TABLE ingredient_list (id SERIAL NOT NULL PRIMARY KEY, recipe_id INTEGER, ingredient_id INTEGER, amount FLOAT, amount_offset FLOAT, unit_id INTEGER, order_index INTEGER, group_id INTEGER);" ); + + TQSqlQuery copyQuery = database->exec( "SELECT recipe_id,ingredient_id,amount,amount_offset,unit_id,prep_method_id,order_index,group_id FROM ingredient_list_copy" ); + if ( copyQuery.isActive() ) { + while ( copyQuery.next() ) { + int ing_list_id = getNextInsertID("ingredient_list","id"); + + TQSqlQuery query(TQString::null,database); + query.prepare( "INSERT INTO ingredient_list VALUES (?, ?, ?, ?, ?, ?, ?, ?)" ); + query.addBindValue( ing_list_id ); + query.addBindValue( copyQuery.value( 0 ) ); + query.addBindValue( copyQuery.value( 1 ) ); + query.addBindValue( copyQuery.value( 2 ) ); + query.addBindValue( copyQuery.value( 3 ) ); + query.addBindValue( copyQuery.value( 4 ) ); + query.addBindValue( copyQuery.value( 6 ) ); + query.addBindValue( copyQuery.value( 7 ) ); + query.exec(); + + int prep_method_id = copyQuery.value( 5 ).toInt(); + if ( prep_method_id != -1 ) { + query.prepare( "INSERT INTO prep_method_list VALUES (?, ?, ?);" ); + query.addBindValue( ing_list_id ); + query.addBindValue( prep_method_id ); + query.addBindValue( 1 ); + query.exec(); + } + + emit progress(); + } + } + database->exec( "DROP TABLE ingredient_list_copy" ); + + database->exec( "CREATE INDEX ridil_index ON ingredient_list USING BTREE (recipe_id);" ); + database->exec( "CREATE INDEX iidil_index ON ingredient_list USING BTREE (ingredient_id);" ); + + database->exec( "UPDATE db_info SET ver='0.83',generated_by='Krecipes SVN (20050909)';" ); + + if ( !database->commit() ) { + kdDebug()<<"Update to 0.83 failed. Maybe you should try again."<<endl; + return; + } + } + + if ( tqRound(version*100) < 84 ) { + database->transaction(); + + database->exec( "ALTER TABLE recipes ADD COLUMN ctime TIMESTAMP" ); + database->exec( "ALTER TABLE recipes ADD COLUMN mtime TIMESTAMP" ); + database->exec( "ALTER TABLE recipes ADD COLUMN atime TIMESTAMP" ); + + database->exec( "UPDATE recipes SET ctime=CURRENT_TIMESTAMP, mtime=CURRENT_TIMESTAMP, atime=CURRENT_TIMESTAMP;" ); + + database->exec( "UPDATE db_info SET ver='0.84',generated_by='Krecipes SVN (20050913)';" ); + + if ( !database->commit() ) { + kdDebug()<<"Update to 0.84 failed. Maybe you should try again."<<endl; + return; + } + } + + if ( tqRound(version*100) < 85 ) { //this change altered the photo format, but this backend already used the newer format + database->transaction(); + + database->exec( "UPDATE db_info SET ver='0.85',generated_by='Krecipes SVN (20050926)';" ); + + if ( !database->commit() ) { + kdDebug()<<"Update to 0.85 failed. Maybe you should try again."<<endl; + return; + } + } + + if ( tqRound(version*100) < 86 ) { + database->transaction(); + + database->exec( "CREATE INDEX gidil_index ON ingredient_list USING BTREE (group_id);" ); + + TQSqlQuery query( "SELECT id,name FROM ingredient_groups ORDER BY name", database ); + + TQString last; + int lastID; + if ( query.isActive() ) { + while ( query.next() ) { + TQString name = query.value(1).toString(); + int id = query.value(0).toInt(); + if ( last == name ) { + TQString command = TQString("UPDATE ingredient_list SET group_id=%1 WHERE group_id=%2").arg(lastID).arg(id); + database->exec(command); + + command = TQString("DELETE FROM ingredient_groups WHERE id=%1").arg(id); + database->exec(command); + } + last = name; + lastID = id; + + emit progress(); + } + } + + database->exec( "UPDATE db_info SET ver='0.86',generated_by='Krecipes SVN (20050928)';" ); + if ( !database->commit() ) + kdDebug()<<"Update to 0.86 failed. Maybe you should try again."<<endl; + } + + if ( tqRound(version*100) < 87 ) { + //Load this default data so the user knows what rating criteria is + database->exec( TQString("INSERT INTO rating_criteria VALUES (1,'%1')").arg(i18n("Overall")) ); + database->exec( TQString("INSERT INTO rating_criteria VALUES (2,'%1')").arg(i18n("Taste") ) ); + database->exec( TQString("INSERT INTO rating_criteria VALUES (3,'%1')").arg(i18n("Appearance") ) ); + database->exec( TQString("INSERT INTO rating_criteria VALUES (4,'%1')").arg(i18n("Originality") ) ); + database->exec( TQString("INSERT INTO rating_criteria VALUES (5,'%1')").arg(i18n("Ease of Preparation") ) ); + + database->exec( "UPDATE db_info SET ver='0.87',generated_by='Krecipes SVN (20051014)'" ); + } + + if ( tqRound(version*100) < 90 ) { + database->exec("UPDATE db_info SET ver='0.9',generated_by='Krecipes 0.9'"); + } + + if ( tqRound(version*100) < 91 ) { + database->exec("CREATE index parent_id_index ON categories USING BTREE(parent_id)"); + database->exec("UPDATE db_info SET ver='0.91',generated_by='Krecipes SVN (20060526)'"); + } + + if ( tqRound(version*100) < 92 ) { + database->transaction(); + + //==================add a columns to 'units' to allow unit abbreviations + database->exec( "ALTER TABLE units RENAME TO units_copy" ); + + int nextval = -1; + TQSqlQuery getID( "SELECT nextval('units_id_seq')", database ); + if ( getID.isActive() && getID.first() ) + nextval = getID.value( 0 ).toInt(); + if ( nextval == -1 ) + kdDebug() << "Database update failed! Unable to update units sequence." << endl; + + database->exec( "CREATE TABLE units (id SERIAL NOT NULL PRIMARY KEY, name CHARACTER VARYING, name_abbrev CHARACTER VARYING, plural CHARACTER VARYING, plural_abbrev CHARACTER VARYING )" ); + TQSqlQuery copyQuery = database->exec( "SELECT id,name,plural FROM units_copy" ); + if ( copyQuery.isActive() ) { + while ( copyQuery.next() ) { + TQSqlQuery query(TQString::null,database); + query.prepare( "INSERT INTO units VALUES(?, ?, ?, ?, ?)" ); + query.addBindValue( copyQuery.value( 0 ) ); + query.addBindValue( copyQuery.value( 1 ) ); + query.addBindValue( TQVariant() ); + query.addBindValue( copyQuery.value( 2 ) ); + query.addBindValue( TQVariant() ); + query.exec(); + + emit progress(); + } + } + database->exec( "DROP TABLE units_copy" ); + + database->exec( "ALTER TABLE units_id_seq1 RENAME TO units_id_seq" ); + database->exec( "ALTER SEQUENCE units_id_seq RESTART WITH "+TQString::number(nextval) ); + + database->exec("UPDATE db_info SET ver='0.92',generated_by='Krecipes SVN (20060609)'"); + if ( !database->commit() ) + kdDebug()<<"Update to 0.92 failed. Maybe you should try again."<<endl; + } + + if ( tqRound(version*100) < 93 ) { + database->transaction(); + + database->exec( "ALTER TABLE ingredient_list ADD COLUMN substitute_for INTEGER" ); + + database->exec("UPDATE db_info SET ver='0.93',generated_by='Krecipes SVN (20060616)'"); + if ( !database->commit() ) + kdDebug()<<"Update to 0.93 failed. Maybe you should try again."<<endl; + } + + if ( tqRound(version*100) < 94 ) { + database->transaction(); + + database->exec( "ALTER TABLE units ADD COLUMN type INTEGER NOT NULL DEFAULT '0'" ); + + database->exec("UPDATE db_info SET ver='0.94',generated_by='Krecipes SVN (20060712)'"); + if ( !database->commit() ) + kdDebug()<<"Update to 0.94 failed. Maybe you should try again."<<endl; + } + + if ( tqRound(version*100) < 95 ) { + database->exec( "DROP TABLE ingredient_weights" ); + createTable( "ingredient_weights" ); + database->exec( "UPDATE db_info SET ver='0.95',generated_by='Krecipes SVN (20060726)'" ); + } +} + +void PSqlRecipeDB::addColumn( const TQString &new_table_sql, const TQString &new_col_info, const TQString &default_value, const TQString &table_name, int col_index ) +{ + TQString command; + + command = TQString(new_table_sql).arg(table_name+"_copy").arg(TQString::null); + kdDebug()<<"calling: "<<command<<endl; + TQSqlQuery query( command, database ); + + command = "SELECT * FROM "+table_name+";"; + query.exec( command ); + if ( query.isActive() ) { + while ( query.next() ) { + TQStringList dataList; + for ( int i = 0 ;; ++i ) { + TQVariant variant = query.value(i); + if ( variant.type() == TQVariant::Invalid ) break; + + dataList << "'"+variant.toString()+"'"; + } + command = "INSERT INTO "+table_name+"_copy VALUES("+dataList.join(",")+");"; + kdDebug()<<"calling: "<<command<<endl; + TQSqlQuery insert_query( command, database ); + + emit progress(); + } + } + query.exec( "DROP TABLE "+table_name+";" ); + query.exec( TQString(new_table_sql).arg(table_name).arg(new_col_info+",") ); + query.exec( "SELECT * FROM "+table_name+"_copy;" ); + if ( query.isActive() ) { + while ( query.next() ) { + TQStringList dataList; + for ( int i = 0 ;; ++i ) { + if ( i == col_index ) + dataList << default_value; + + TQVariant variant = query.value(i); + if ( variant.type() == TQVariant::Invalid ) break; + + dataList << "'"+variant.toString()+"'"; + } + command = "INSERT INTO "+table_name+" VALUES(" +dataList.join(",")+");"; + TQSqlQuery insert_query( command, database ); + kdDebug()<<"calling: "<<command<<endl; + + emit progress(); + } + } + query.exec( "DROP TABLE "+table_name+"_copy;" ); +} + +int PSqlRecipeDB::lastInsertID() +{ + return last_insert_id; +} + +int PSqlRecipeDB::getNextInsertID( const TQString &table, const TQString &column ) +{ + TQString command = TQString( "SELECT nextval('%1_%2_seq');" ).arg( table ).arg( column ); + TQSqlQuery getID( command, database ); + + if ( getID.isActive() && getID.first() ) { + last_insert_id = getID.value( 0 ).toInt(); + } + else + last_insert_id = -1; + + return last_insert_id; +} + +void PSqlRecipeDB::givePermissions( const TQString & /*dbName*/, const TQString &username, const TQString &password, const TQString & /*clientHost*/ ) +{ + TQStringList tables; + tables << "ingredient_info" << "ingredient_list" << "ingredient_properties" << "ingredients" << "recipes" << "unit_list" << "units" << "units_conversion" << "categories" << "category_list" << "authors" << "author_list" << "prep_methods" << "db_info" << "ingredient_groups" << "ingredient_weights" << "prep_method_list" << "yield_types" << "ratings" << "rating_criteria" << "rating_criterion_list"; + + //we also have to grant permissions on the sequences created + tables << "authors_id_seq" << "categories_id_seq" << "ingredient_properties_id_seq" << "ingredient_weights_id_seq" << "ingredients_id_seq" << "prep_methods_id_seq" << "recipes_id_seq" << "units_id_seq" << "ingredient_groups_id_seq" << "yield_types_id_seq" << "ingredient_list_id_seq" << "ratings_id_seq" << "rating_criteria_id_seq"; + + TQString command; + + kdDebug() << "I'm doing the query to create the new user" << endl; + command = "CREATE USER " + username; + if ( !password.isEmpty() ) + command.append( "WITH PASSWORD '" + password + "'" ); + command.append( ";" ); + TQSqlQuery permissionsToSet( command, database ); + + kdDebug() << "I'm doing the query to setup permissions\n"; + command = TQString( "GRANT ALL ON %1 TO %2;" ).arg( tables.join( "," ) ).arg( username ); + permissionsToSet.exec( command ); +} + +void PSqlRecipeDB::empty( void ) +{ + TQSqlRecipeDB::empty(); + + TQStringList tables; + tables << "authors_id_seq" << "categories_id_seq" << "ingredient_properties_id_seq" << "ingredient_weights_id_seq" << "ingredients_id_seq" << "prep_methods_id_seq" << "recipes_id_seq" << "units_id_seq" << "ingredient_groups_id_seq" << "yield_types_id_seq" << "ingredient_list_id_seq" << "prep_method_list_id_seq" << "ratings_id_seq" << "rating_criteria_id_seq"; + + TQSqlQuery tablesToEmpty( TQString::null, database ); + for ( TQStringList::Iterator it = tables.begin(); it != tables.end(); ++it ) { + TQString command = TQString( "DELETE FROM %1;" ).arg( *it ); + tablesToEmpty.exec( command ); + } +} + +#include "psqlrecipedb.moc" diff --git a/src/backends/PostgreSQL/psqlrecipedb.h b/src/backends/PostgreSQL/psqlrecipedb.h new file mode 100644 index 0000000..feaee9a --- /dev/null +++ b/src/backends/PostgreSQL/psqlrecipedb.h @@ -0,0 +1,64 @@ +/*************************************************************************** +* Copyright (C) 2003 by * +* Unai Garro ([email protected]) * +* Cyril Bosselut ([email protected]) * +* Jason Kivlighn ([email protected]) * +* * +* This program is free software; you can redistribute it and/or modify * +* it under the terms of the GNU General Public License as published by * +* the Free Software Foundation; either version 2 of the License, or * +* (at your option) any later version. * +***************************************************************************/ + + +#ifndef PSQLRECIPEDB_H +#define PSQLRECIPEDB_H + +#define PSQL_DRIVER "TQPSQL7" + +#include "qsqlrecipedb.h" + +/** +@author Jason Kivlighn +*/ +class PSqlRecipeDB : public TQSqlRecipeDB +{ + + TQ_OBJECT + +private: + void createDB( void ); + +public: + PSqlRecipeDB( const TQString& host, const TQString& user = TQString::null, const TQString& pass = TQString::null, const TQString& DBName = DEFAULT_DB_NAME, int port = 0 ); + ~PSqlRecipeDB( void ); + + int lastInsertID(); + void initializeData(); + + void createTable( const TQString &tableName ); + void givePermissions( const TQString &dbName, const TQString &username, const TQString &password, const TQString &clientHost ); + +protected: + virtual TQString qsqlDriverPlugin() const + { + return PSQL_DRIVER; + } + virtual int getNextInsertID( const TQString &table, const TQString &column ); + + virtual void empty( void ); + +private: + void portOldDatabases( float version ); + TQStringList backupCommand() const; + TQStringList restoreCommand() const; + + void addColumn( const TQString &new_table_sql, const TQString &new_col_info, const TQString &default_value, const TQString &table_name, int col_index ); + + int last_insert_id; +}; + + + + +#endif |