388 changes: 388 additions & 0 deletions mythtv/libs/libmythtv/restoredata.cpp
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,388 @@
/*
* $Id$
* vim: set expandtab tabstop=4 shiftwidth=4:
*
* Original Project
* MythTV http://www.mythtv.org
*
* Description:
* Restore data from deleted channels after scan for new channels
*
* When all channels are deleted and a new scan is done then
* all the non-scanned data is lost.
* This are the following fields:
* xmltvid Channel identification for XMLTV including Schedules Direct
* iconpath File name of icon for this channel
* visible Visible status
* When a channel is deleted it is not immediately deleted from the database;
* it is kept for a while with the "deleted" field set to the date at which it
* is deleted.
* This makes it possible to retrieve this data from the deleted records
* when the same channel is found again in a new scan.
*
* 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.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
* Or, point your browser to http://www.gnu.org/copyleft/gpl.html
*
*/

// C++ includes
#include <vector>
#include <iostream>

// MythTV includes
#include "restoredata.h"
#include "videosource.h"
#include "sourceutil.h"
#include "mythcorecontext.h"
#include "mythdb.h"

#include "mythuiimage.h"
#include "mythuitext.h"
#include "themeinfo.h"
#include "mythdialogbox.h"

#define LOC QString("RestoreData: ")

class RestoreXMLTVID : public TransMythUICheckBoxSetting
{
public:
RestoreXMLTVID()
{
setLabel(QObject::tr("Restore XMLTV ID"));
setHelpText(
QObject::tr(
"If checked, copy the XMLTV ID in field \"xmltvid\" "
"from a deleted channel in this video source "
"or from a channel in another video source."));
setValue(false);
};
};

class RestoreVisible : public TransMythUICheckBoxSetting
{
public:
RestoreVisible()
{
setLabel(QObject::tr("Restore Visible status"));
setHelpText(
QObject::tr(
"If checked, copy the Visible status in field \"visible\" "
"from a deleted channel in this video source."));
setValue(false);
};
};

class RestoreIcon : public TransMythUICheckBoxSetting
{
public:
RestoreIcon()
{
setLabel(QObject::tr("Restore Icon filename"));
setHelpText(
QObject::tr(
"If checked, copy the Icon filename in field \"icon\" "
"from a deleted channel in this video source "
"or from a channel in another video source."));
setValue(false);
};
};

RestoreData::RestoreData(uint sourceid) :
m_sourceid(sourceid)
{
setLabel(tr("Restore Data"));

m_videosource = new VideoSourceShow(m_sourceid);
m_videosource->setHelpText(
QObject::tr(
"The video source is selected in the Channel Editor page. "
"Searching for non-scanned data is done for all channels in this video source."
));
addChild(m_videosource);

m_restoreXMLTVID = new RestoreXMLTVID();
addChild(m_restoreXMLTVID);

m_restoreVisible = new RestoreVisible();
addChild(m_restoreVisible);

m_restoreIcon = new RestoreIcon();
addChild(m_restoreIcon);

auto *newTransport = new ButtonStandardSetting(tr("Search"));
newTransport->setHelpText(
QObject::tr(
"Start searching for non-scanned data. The data is written to the database "
"when \'Save and Exit\' is selected in the \'Exit Settings?\' dialog box."
));
addChild(newTransport);
connect(newTransport, &ButtonStandardSetting::clicked, this, &RestoreData::Restore);
}

void RestoreData::Restore()
{
bool do_xmltvid = m_restoreXMLTVID->boolValue();
bool do_icon = m_restoreIcon->boolValue();
bool do_visible = m_restoreVisible->boolValue();

if (do_xmltvid || do_icon || do_visible)
{
QString msg = QString("Restore data from deleted channels for fields ");
if (do_xmltvid)
{
msg += "xmltvid ";
}
if (do_icon)
{
msg += "icon ";
}
if (do_visible)
{
msg += "visible ";
}
LOG(VB_GENERAL, LOG_INFO, LOC + msg);
}

// Old Channel Data
m_ocd.clear();

MSqlQuery query(MSqlQuery::InitCon());
MSqlQuery query2(MSqlQuery::InitCon());

// For all channels in this videosource
query.prepare(
"SELECT chanid, channum, name, serviceid, transportid, networkid, "
" channel.xmltvid, "
" channel.icon, "
" channel.visible "
"FROM channel, dtv_multiplex "
"WHERE channel.sourceid = :SOURCEID "
" AND channel.mplexid = dtv_multiplex.mplexid "
" AND deleted IS NULL ");
query.bindValue(":SOURCEID", m_sourceid);
if (!query.exec() || !query.isActive())
{
MythDB::DBError("RestoreData::Restore(1)", query);
return;
}
while (query.next())
{
OldChannelData cd;
cd.chanid = query.value(0).toUInt();
cd.channum = query.value(1).toString();
cd.name = query.value(2).toString();
cd.serviceid = query.value(3).toUInt();
cd.transportid = query.value(4).toUInt();
cd.networkid = query.value(5).toUInt();
cd.xmltvid = query.value(6).toString();
cd.icon = query.value(7).toString();
cd.visible = query.value(8).toInt();

cd.found_xmltvid = false;
cd.found_icon = false;
cd.found_visible = false;

// Get xmltvid from the last deleted channel
// from any video source
if (do_xmltvid && cd.xmltvid.isEmpty())
{
query2.prepare(
"SELECT xmltvid "
"FROM channel, dtv_multiplex "
"WHERE serviceid = :SERVICEID "
" AND transportid = :TRANSPORTID "
" AND networkid = :NETWORKID "
" AND channel.mplexid = dtv_multiplex.mplexid "
" AND xmltvid != ''"
" AND deleted IS NOT NULL "
"ORDER BY deleted DESC;");
query2.bindValue(":SERVICEID", cd.serviceid);
query2.bindValue(":TRANSPORTID", cd.transportid);
query2.bindValue(":NETWORKID", cd.networkid);
if (!query2.exec() || !query2.isActive())
{
MythDB::DBError("RestoreData::Restore(2)", query);
return;
}
if (query2.next())
{
cd.xmltvid = query2.value(0).toString();
cd.found_xmltvid = true;
}
}

// Get icon from the last deleted channel
// from any video source
if (do_icon && cd.icon.isEmpty())
{
query2.prepare(
"SELECT icon "
"FROM channel, dtv_multiplex "
"WHERE serviceid = :SERVICEID "
" AND transportid = :TRANSPORTID "
" AND networkid = :NETWORKID "
" AND channel.mplexid = dtv_multiplex.mplexid "
" AND icon != ''"
" AND deleted IS NOT NULL "
"ORDER BY deleted DESC;");
query2.bindValue(":SERVICEID", cd.serviceid);
query2.bindValue(":TRANSPORTID", cd.transportid);
query2.bindValue(":NETWORKID", cd.networkid);
if (!query2.exec() || !query2.isActive())
{
MythDB::DBError("RestoreData::Restore(3)", query);
return;
}
if (query2.next())
{
cd.icon = query2.value(0).toString();
cd.found_icon = true;
}
}

// Get visible from the last deleted channel
// but only from the same video source
if (do_visible)
{
query2.prepare(
"SELECT channel.visible "
"FROM channel, dtv_multiplex "
"WHERE serviceid = :SERVICEID "
" AND transportid = :TRANSPORTID "
" AND networkid = :NETWORKID "
" AND channel.sourceid = :SOURCEID "
" AND channel.mplexid = dtv_multiplex.mplexid "
" AND deleted IS NOT NULL "
"ORDER BY deleted DESC;");
query2.bindValue(":SERVICEID", cd.serviceid);
query2.bindValue(":TRANSPORTID", cd.transportid);
query2.bindValue(":NETWORKID", cd.networkid);
query2.bindValue(":SOURCEID", m_sourceid);
if (!query2.exec() || !query2.isActive())
{
MythDB::DBError("RestoreData::Restore(4)", query);
return;
}
if (query2.next())
{
int visible = query2.value(0).toInt();
if (visible != cd.visible)
{
cd.visible = visible;
cd.found_visible = true;
}
}
}

if (cd.found_xmltvid || cd.found_icon || cd.found_visible)
{
m_ocd.push_back(cd);
}
}

// Header line for log of all changes
if (m_ocd.empty())
{
LOG(VB_GENERAL, LOG_INFO, LOC + "No data found in deleted channels or no data needed");
}
else
{
LOG(VB_GENERAL, LOG_INFO, LOC +
QString("Restoring data in %1 channels from deleted channels:")
.arg(m_ocd.size()));
}

// Log of all channels that will be updated
int num_xmltvid = 0;
int num_icon = 0;
int num_visible = 0;
for (auto & cd : m_ocd)
{
QString msg = QString("Channel %1 \'%2\' update ").arg(cd.channum).arg(cd.name);
if (cd.found_xmltvid)
{
msg += QString("xmltvid(%1) ").arg(cd.xmltvid);
num_xmltvid++;
}
if (cd.found_icon)
{
msg += QString("icon(%1) ").arg(cd.icon);
num_icon++;
}
if (cd.found_visible)
{
msg += QString("visible(%1) ").arg(cd.visible);
num_visible++;
}
LOG(VB_GENERAL, LOG_INFO, LOC + msg);
}

// Show totals of what has been found.
{
QString msg;
if (m_ocd.empty())
{
msg = "No data found";
}
else
{
msg = QString("Found data for %1 channels\n").arg(m_ocd.size());
if (num_xmltvid > 0)
{
msg += QString("xmltvid: %1 ").arg(num_xmltvid);
}
if (num_icon > 0)
{
msg += QString("icon: %1 ").arg(num_icon);
}
if (num_visible > 0)
{
msg += QString("visible: %1").arg(num_visible);
}
}
WaitFor(ShowOkPopup(msg));
}
}

// Load value of selected video source (for display only)
void RestoreData::Load(void)
{
GroupSetting::Load();
}

// Do the actual updating if "Save and Exit" is selected in the "Exit Settings?" dialog.
void RestoreData::Save(void)
{
MSqlQuery query(MSqlQuery::InitCon());
for (auto & cd : m_ocd)
{
query.prepare(
"UPDATE channel "
"SET icon = :ICON, "
" xmltvid = :XMLTVID, "
" visible = :VISIBLE "
"WHERE chanid = :CHANID");
query.bindValue(":ICON", cd.icon);
query.bindValue(":XMLTVID", cd.xmltvid);
query.bindValue(":VISIBLE", cd.visible);
query.bindValue(":CHANID", cd.chanid);
if (!query.exec() || !query.isActive())
{
MythDB::DBError("RestoreData::Restore(5)", query);
return;
}
}
LOG(VB_GENERAL, LOG_INFO, LOC + QString("Restored data for %1 channels").arg(m_ocd.size()));
}
103 changes: 103 additions & 0 deletions mythtv/libs/libmythtv/restoredata.h
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,103 @@
/*
* $Id$
* vim: set expandtab tabstop=4 shiftwidth=4:
*
* Original Project
* MythTV http://www.mythtv.org
*
*
* Description:
* Postprocessing of channelscan result
*
* When all channels are deleted and a new scan is done then
* all the non-scanned data is lost.
* This are the following fields:
* xmltvid Channel identification for XMLTV including Schedules Direct
* iconpath File name of icon for this channel
* visible Visible status
* When a channel is deleted it is not immediately deleted from the database;
* it is kept for a while with the "deleted" field set to the date at which it
* is deleted.
* This makes it possible to retrieve this data from the deleted records
* when the same channel is found again in a new scan.
*
*
* 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.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
* Or, point your browser to http://www.gnu.org/copyleft/gpl.html
*
*/

#ifndef RESTOREDATA_H
#define RESTOREDATA_H

#include <vector>

#include "mythuigroup.h"
#include "mythtvexp.h"
#include "standardsettings.h"
#include "cardutil.h"
#include "xmlparsebase.h"
#include "videosource.h"

class VideoSourceShow;
class MythUIType;
class RestoreXMLTVID;
class RestoreVisible;
class RestoreIcon;

struct OldChannelData
{
uint chanid;
uint sourceid;
uint serviceid;
uint transportid;
uint networkid;
QString channum;
QString name;
QString xmltvid;
QString icon;
int visible;
QString deleted;
bool found_xmltvid;
bool found_icon;
bool found_visible;
};

// Page for updating the fields with non-scanned data
// from deleted channel data.
//
class MTV_PUBLIC RestoreData : public GroupSetting
{
Q_OBJECT
public:
explicit RestoreData(uint sourceid);
void Load(void) override; // StandardSetting
void Save(void) override; // StandardSetting

public slots:
void Restore(void);

private:
VideoSourceShow *m_videosource {nullptr};
RestoreXMLTVID *m_restoreXMLTVID {nullptr};
RestoreVisible *m_restoreVisible {nullptr};
RestoreIcon *m_restoreIcon {nullptr};

uint m_sourceid {0};

std::vector<OldChannelData> m_ocd;
};

#endif // RESTOREDATA_H
31 changes: 29 additions & 2 deletions mythtv/programs/mythtv-setup/channeleditor.cpp
View file
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@
#include "mythuibuttonlist.h"
#include "channelsettings.h"
#include "transporteditor.h"
#include "restoredata.h"
#include "mythuicheckbox.h"
#include "mythuitextedit.h"
#include "channeleditor.h"
Expand Down Expand Up @@ -87,16 +88,16 @@ bool ChannelEditor::Create()

MythUIButton *deleteButton = dynamic_cast<MythUIButton *>(GetChild("delete"));
MythUIButton *scanButton = dynamic_cast<MythUIButton *>(GetChild("scan"));
MythUIButton *restoreDataButton = dynamic_cast<MythUIButton *>(GetChild("restoredata"));
MythUIButton *importIconButton = dynamic_cast<MythUIButton *>(GetChild("importicons"));
MythUIButton *transportEditorButton = dynamic_cast<MythUIButton *>(GetChild("edittransport"));

MythUICheckBox *hideCheck = dynamic_cast<MythUICheckBox *>(GetChild("nochannum"));

if (!sortList || !m_sourceList || !m_channelList || !deleteButton ||
!scanButton || !importIconButton || !transportEditorButton ||
!scanButton || !restoreDataButton || !importIconButton || !transportEditorButton ||
!hideCheck)
{

return false;
}

Expand Down Expand Up @@ -141,6 +142,11 @@ bool ChannelEditor::Create()
scanButton->SetHelpText(tr("Starts the channel scanner."));
scanButton->SetEnabled(SourceUtil::IsAnySourceScanable());

// Restore Data button
restoreDataButton->SetHelpText(tr("Restore Data from deleted channels."));
restoreDataButton->SetEnabled(true);
connect(restoreDataButton, &MythUIButton::Clicked, this, &ChannelEditor::restoreData);

// Import Icons Button
importIconButton->SetHelpText(tr("Starts the icon downloader"));
importIconButton->SetEnabled(true);
Expand Down Expand Up @@ -627,6 +633,27 @@ void ChannelEditor::scan(void)
delete ssd;
}

void ChannelEditor::restoreData(void)
{
// Check that we have a videosource and a connected capture card
if (!check_cardsource(m_sourceFilter, m_sourceFilterName))
return;

// Create the dialog now that we have a video source and a capture card
MythScreenStack *mainStack = GetMythMainWindow()->GetMainStack();

auto *ssd = new StandardSettingDialog(mainStack, "restoredata",
new RestoreData((uint)m_sourceFilter));
if (ssd->Create())
{
// Reload channel list with fillList after Restore
connect(ssd, &MythScreenType::Exiting, this, &ChannelEditor::fillList);
mainStack->AddScreen(ssd);
}
else
delete ssd;
}

void ChannelEditor::transportEditor(void)
{
// Check that we have a videosource and a connected capture card
Expand Down
1 change: 1 addition & 0 deletions mythtv/programs/mythtv-setup/channeleditor.h
View file
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,7 @@ class ChannelEditor : public MythScreenType
void del(void);
void edit(MythUIButtonListItem *item = nullptr);
void scan(void);
void restoreData(void);
void transportEditor(void);
void channelIconImport(void);
void deleteChannels(void);
Expand Down
21 changes: 18 additions & 3 deletions mythtv/themes/MythCenter-wide/config-ui.xml
View file
Original file line number Diff line number Diff line change
Expand Up @@ -208,20 +208,35 @@
</checkbox>

<button name="scan" from="basewidebutton">
<position>90,644</position>
<position>40,644</position>
<value>Channel Scan</value>
</button>

<button name="restoredata" from="basewidebutton">
<position>340,644</position>
<value>Restore Data</value>
</button>

<button name="importicons" from="basewidebutton">
<position>490,644</position>
<position>640,644</position>
<value>Icon Download</value>
</button>

<button name="edittransport" from="basewidebutton">
<position>890,644</position>
<position>940,644</position>
<value>Edit Transports</value>
</button>

</window>

<window name="restoredata">

<textarea name="pagetitle">
<area>20,4,570,40</area>
<font>baselarge</font>
<value>Restore Data</value>
</textarea>

</window>

</mythuitheme>