search fix

[toast/confclerk.git] / src / orm / ormrecord.h

diff --git a/src/orm/ormrecord.h b/src/orm/ormrecord.h
index 22e19f485e47d4f496e32e36b151d01a275681d1..a815c8e08287565b074a2fc771616e5bd38fd274 100644 (file)
--- a/src/orm/ormrecord.h
+++ b/src/orm/ormrecord.h
@@ -11,20 +11,26 @@
 
 class OrmException
 {
+public:
+    OrmException(const QString& text) : mText(text) {};
+    virtual ~OrmException(){};
+    virtual const QString& text() const { return mText; }
+private:
+    QString mText;
 };
 
-class OrmNoObjectException : OrmException
+class OrmNoObjectException : public OrmException
 {
+public:
+    OrmNoObjectException() : OrmException("No object exception"){};
+    ~OrmNoObjectException(){};
 };
 
-class OrmSqlException : OrmException
+class OrmSqlException : public OrmException
 {
 public:
-    OrmSqlException(const QString& text) : mText(text) {}
-    QString text() const { return mText; }
-
-private:
-    QString mText;
+    OrmSqlException(const QString& text) : OrmException( QString("Sql error: ") + text ) {};
+    ~OrmSqlException(){};
 };
 
 template <typename T>
@@ -33,6 +39,7 @@ class OrmRecord : protected QSqlRecord
 public:
     OrmRecord();
     static T hydrate(const QSqlRecord& record);
+    void update(QString col, QVariant value = QVariant()); // updates specified column 'col'
 
 protected:
     QVariant value(QString col) const;
@@ -42,9 +49,11 @@ protected:
     static QList<T> load(QSqlQuery query);
 
     // auxiliary methods
+    static QSqlRecord toRecord(const QList<QSqlField> & columnList);
+    // all record items/columns are in one table
     static QString columnsForSelect(const QString& prefix = QString());
     static QString selectQuery();
-    static QSqlRecord toRecord(const QList<QSqlField> & columnList);
+    static QString updateQuery();
 
     static QVariant convertToC(QVariant value, QVariant::Type colType);
     static QVariant convertToDb(QVariant value, QVariant::Type colType);
@@ -64,6 +73,24 @@ T OrmRecord<T>::hydrate(const QSqlRecord& record)
     return object;
 }
 
+// updates specified column 'col'
+// if the value is not specified  as an argument,
+// it's taken from the reford itself
+// see also: setValue() method for more details
+template <typename T>
+void OrmRecord<T>::update(QString col, QVariant value)
+{
+    QSqlQuery query;
+    query.prepare(QString(updateQuery() + "SET %1 = :col WHERE id = :id").arg(col));
+    if(value.isValid()) // take 'col' value from the method's arguments
+        query.bindValue(":col", value);
+    else // take 'col' value from the record; see setValue()
+        query.bindValue(":col", convertToDb(this->value(col), this->value(col).type()));
+    query.bindValue(":id", this->value("id"));
+    //query.bindValue(":id", convertToDb(value("id"), QVariant::Int));
+    query.exec();
+}
+
 template <typename T>
 QVariant OrmRecord<T>::value(QString col) const
 {
@@ -83,13 +110,13 @@ T OrmRecord<T>::loadOne(QSqlQuery query)
     {
         if (!query.exec())
         {
-            throw new OrmSqlException(query.lastError().text());
+            throw OrmSqlException(query.lastError().text());
         }
     }
 
     if (!query.next())
     {
-        throw new OrmNoObjectException();
+        throw OrmNoObjectException();
     }
 
     return hydrate(query.record());
@@ -102,7 +129,12 @@ QList<T> OrmRecord<T>::load(QSqlQuery query)
     {
         if (!query.exec())
         {
-            throw new OrmSqlException(query.lastError().text());
+            qDebug() << "Error: " << query.lastError().driverText() << "; Type: " << query.lastError().type();
+            throw OrmSqlException(query.lastError().text());
+        }
+        else
+        {
+            qDebug() << "SQL OK";
         }
     }
 
@@ -111,7 +143,7 @@ QList<T> OrmRecord<T>::load(QSqlQuery query)
     {
         objects << hydrate(query.record());
     }
-
+    qDebug() << "Fetch done";
     return objects;
 }
 
@@ -134,6 +166,12 @@ QString OrmRecord<T>::selectQuery()
     return QString("SELECT %1 FROM %2 ").arg(columnsForSelect(), T::sTableName);
 }
 
+template <typename T>
+QString OrmRecord<T>::updateQuery()
+{
+    return QString("UPDATE %1 ").arg(T::sTableName);
+}
+
 template <typename T>
 QSqlRecord OrmRecord<T>::toRecord(const QList<QSqlField> & columnList)
 {
@@ -148,10 +186,10 @@ QSqlRecord OrmRecord<T>::toRecord(const QList<QSqlField> & columnList)
 template <typename T>
 QVariant OrmRecord<T>::convertToC(QVariant value, QVariant::Type colType)
 {
-    if (colType == QVariant::DateTime &&
-        (value.type() == QVariant::UInt || value.type() == QVariant::Int))
+    if (colType == QVariant::DateTime && value.canConvert<uint>())
     {
         QDateTime date;
+        date.setTimeSpec(Qt::UTC);
         date.setTime_t(value.toUInt());
         return date;
     }
@@ -162,7 +200,7 @@ QVariant OrmRecord<T>::convertToC(QVariant value, QVariant::Type colType)
 template <typename T>
 QVariant OrmRecord<T>::convertToDb(QVariant value, QVariant::Type colType)
 {
-    if (colType == QVariant::DateTime && value.type() == QVariant::DateTime)
+    if (colType == QVariant::DateTime && value.canConvert<QDateTime>())
     {
         return value.toDateTime().toTime_t();
     }
@@ -171,3 +209,4 @@ QVariant OrmRecord<T>::convertToDb(QVariant value, QVariant::Type colType)
 }
 
 #endif // ORMRECORD_H
+