~ubuntu-branches/ubuntu/trusty/tomahawk/trusty-proposed

« back to all changes in this revision

Viewing changes to src/libtomahawk/database/DatabaseCommand_SetTrackAttributes.cpp

  • Committer: Package Import Robot
  • Author(s): Harald Sitter
  • Date: 2013-03-07 21:50:13 UTC
  • Revision ID: package-import@ubuntu.com-20130307215013-6gdjkdds7i9uenvs
Tags: upstream-0.6.0+dfsg
ImportĀ upstreamĀ versionĀ 0.6.0+dfsg

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
/* === This file is part of Tomahawk Player - <http://tomahawk-player.org> ===
 
2
 *
 
3
 *   Copyright 2010-2011, Leo Franchi <lfranchi@kde.org>
 
4
 *
 
5
 *   Tomahawk is free software: you can redistribute it and/or modify
 
6
 *   it under the terms of the GNU General Public License as published by
 
7
 *   the Free Software Foundation, either version 3 of the License, or
 
8
 *   (at your option) any later version.
 
9
 *
 
10
 *   Tomahawk is distributed in the hope that it will be useful,
 
11
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of
 
12
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 
13
 *   GNU General Public License for more details.
 
14
 *
 
15
 *   You should have received a copy of the GNU General Public License
 
16
 *   along with Tomahawk. If not, see <http://www.gnu.org/licenses/>.
 
17
 */
 
18
 
 
19
#include "DatabaseCommand_SetTrackAttributes.h"
 
20
#include "TomahawkSqlQuery.h"
 
21
#include "DatabaseImpl.h"
 
22
#include "Source.h"
 
23
#include "utils/Logger.h"
 
24
 
 
25
using namespace Tomahawk;
 
26
 
 
27
DatabaseCommand_SetTrackAttributes::DatabaseCommand_SetTrackAttributes( DatabaseCommand_SetTrackAttributes::AttributeType type, QList< QPair< QID, QString > > ids, bool toDelete )
 
28
    : DatabaseCommandLoggable()
 
29
    , m_loggable( false )
 
30
    , m_delete( toDelete )
 
31
    , m_type( type )
 
32
    , m_tracks( ids )
 
33
{
 
34
}
 
35
 
 
36
DatabaseCommand_SetTrackAttributes::DatabaseCommand_SetTrackAttributes( DatabaseCommand_SetTrackAttributes::AttributeType type )
 
37
    : DatabaseCommandLoggable()
 
38
    , m_loggable( false )
 
39
    , m_delete( true )
 
40
    , m_type( type )
 
41
{
 
42
 
 
43
}
 
44
 
 
45
 
 
46
void
 
47
DatabaseCommand_SetTrackAttributes::exec( DatabaseImpl* dbi )
 
48
{
 
49
    TomahawkSqlQuery checkquery = dbi->newquery();
 
50
    TomahawkSqlQuery delquery = dbi->newquery();
 
51
    TomahawkSqlQuery insertquery = dbi->newquery();
 
52
 
 
53
    QString k;
 
54
    switch ( m_type )
 
55
    {
 
56
    case EchonestCatalogId:
 
57
        k = "echonestcatalogid";
 
58
        break;
 
59
    }
 
60
 
 
61
    if ( m_delete && m_tracks.isEmpty() )
 
62
    {
 
63
        //delete all
 
64
        TomahawkSqlQuery delAll = dbi->newquery();
 
65
        delAll.prepare( "DELETE FROM track_attributes WHERE k = ?" );
 
66
        delAll.bindValue( 0, k );
 
67
        delAll.exec();
 
68
        return;
 
69
    }
 
70
 
 
71
    checkquery.prepare( "SELECT id, sortname FROM track WHERE id = ?" );
 
72
    delquery.prepare( "DELETE FROM track_attributes WHERE id = ? AND k = ?" );
 
73
    insertquery.prepare( "INSERT INTO track_attributes ( id, k, v ) VALUES( ?, ?, ? )" );
 
74
 
 
75
    QPair< QID, QString > track;
 
76
    foreach ( track, m_tracks )
 
77
    {
 
78
        checkquery.bindValue( 0, track.first );
 
79
        if ( !checkquery.exec() )
 
80
        {
 
81
            tLog() << "No track in track table for set track attribute command...aborting:" << track.first;
 
82
            continue;
 
83
        }
 
84
 
 
85
        delquery.bindValue( 0, track.first );
 
86
        delquery.bindValue( 1, k );
 
87
        delquery.exec();
 
88
 
 
89
        if ( m_delete )
 
90
            continue; // stop at deleting, don't insert
 
91
 
 
92
        insertquery.bindValue( 0, track.first );
 
93
        insertquery.bindValue( 1, k );
 
94
        insertquery.bindValue( 2, track.second );
 
95
        if ( !insertquery.exec() )
 
96
            tLog() << "Failed to insert track attribute:" << k << track.first << track.second;
 
97
 
 
98
    }
 
99
}