Extract setting up the rename environment into a separate function

diff --git a/localmylist/include/LocalMyList/RenameUtils b/localmylist/include/LocalMyList/RenameUtils
new file mode 100644 (file)
index 0000000..a2ef5ac
--- /dev/null
+++ b/localmylist/include/LocalMyList/RenameUtils
@@ -0,0 +1,2 @@
+#include "../../renameutils.h"
+

diff --git a/localmylist/localmylist.pro b/localmylist/localmylist.pro
index e5149a7fcd504ed9bbac45b1308e831574fa7b1a..998b4c48c46f56b1b14b04bc8ab89c9c8bc13a57 100644 (file)
--- a/localmylist/localmylist.pro
+++ b/localmylist/localmylist.pro
@@ -23,7 +23,8 @@ SOURCES += \
        mylistexportparsetask.cpp \
        settings.cpp \
        renamehandler.cpp \
-       unknownfilelookuptask.cpp
+       unknownfilelookuptask.cpp \
+       renameutils.cpp
 
 HEADERS += \
        localmylist_global.h \
@@ -40,7 +41,8 @@ HEADERS += \
        mylistexportparsetask.h \
        settings.h \
        renamehandler.h \
-       unknownfilelookuptask.h
+       unknownfilelookuptask.h \
+       renameutils.h
 
 CONV_HEADERS += \
        include/LocalMyList/AbstractTask \
@@ -52,7 +54,8 @@ CONV_HEADERS += \
        include/LocalMyList/MyListExportParseTask \
        include/LocalMyList/MyListModel \
        include/LocalMyList/Settings \
-       include/LocalMyList/UnknownFileLookupTask
+       include/LocalMyList/UnknownFileLookupTask \
+       include/LocalMyList/RenameUtils
 
 SHARE += \
        share/schema/schema.sql \

diff --git a/localmylist/renameutils.cpp b/localmylist/renameutils.cpp
new file mode 100644 (file)
index 0000000..edde002
--- /dev/null
+++ b/localmylist/renameutils.cpp
@@ -0,0 +1,62 @@
+#include "renameutils.h"
+
+#include "mylist.h"
+#include <QVariant>
+#include <QDateTime>
+#include <QFileInfo>
+
+namespace LocalMyList {
+
+void setupRenameEnv(const QSqlRecord &record, RenameParser::Environment &env)
+{
+
+       env["ATr"] = record.value("anime_title_romaji").toString();
+       env["ATe"] = record.value("anime_title_english").toString();
+       env["ATk"] = record.value("anime_title_kanji").toString();
+
+       env["ETr"] = record.value("episode_title_romaji").toString();
+       env["ETe"] = record.value("episode_title_english").toString();
+       env["ETk"] = record.value("episode_title_kanji").toString();
+
+       env["GTs"] = record.value("group_name_short").toString();
+       env["GTl"] = record.value("group_name").toString();
+
+       env["EpNo"] = record.value("epno").toString();
+       env["EpHiNo"] = record.value("episode_count").toString();
+       env["EpCount"] = "0"; // TODO This is missing from the data
+
+       QString year = record.value("year").toString().trimmed();
+       if (!year.contains('-'))
+       {
+               env["AYearBegin"] = year;
+               env["AYearEnd"] = "";
+       }
+       else
+       {
+               QStringList years = year.split('-');
+               env["AYearBegin"] = years[0].trimmed();
+               env["AYearEnd"] = years[1].trimmed();
+       }
+
+       env["Type"] = record.value("anime_type").toString();
+       env["Depr"] = /*r.value("")*/ false ? "1" : ""; // Data missing?
+       env["Cen"] = record.value("censored").toBool() ? "1" : "0";
+       env["Ver"] = record.value("version").toString();
+       env["Source"] = record.value("type").toString(); // TODO This is called source on AniDB
+       env["Quality"] = record.value("quality").toString();
+       env["CurrentFN"] = QFileInfo(record.value("path").toString()).fileName();
+       env["FCrc"] = record.value("crc").toString();
+       env["FVideoRes"] = record.value("resolution").toString();
+       env["FALng"] = record.value("audio_language").toString();
+       env["FSLng"] = record.value("subtitle_language").toString();
+       env["FACodec"] = record.value("audio_codec").toString();
+       env["FVCodec"] = record.value("video_codec").toString();
+       env["Watched"] = record.value("my_watched").toDateTime().isValid() ? "1" : "0";
+
+       // These variables are not in the original applet
+       env["CurrentFilePath"] = QFileInfo(record.value("path").toString()).path();
+       env["HostId"] = record.value("host_id").toString();
+       env["CurrentHostId"] = QString::number(MyList::instance()->hostId());
+}
+
+} // namespace LocalMyList

diff --git a/localmylist/renameutils.h b/localmylist/renameutils.h
new file mode 100644 (file)
index 0000000..c5628a0
--- /dev/null
+++ b/localmylist/renameutils.h
@@ -0,0 +1,14 @@
+#ifndef RENAMEUTILS_H
+#define RENAMEUTILS_H
+
+#include "localmylist_global.h"
+#include <RenameParser/RenameEngine>
+#include <QSqlRecord>
+
+namespace LocalMyList {
+
+void LOCALMYLISTSHARED_EXPORT setupRenameEnv(const QSqlRecord &record, RenameParser::Environment &env);
+
+} // namespace LocalMyList
+
+#endif // RENAMEUTILS_H