Option to specify filename pattern

This commit is contained in:
zvon 2019-01-23 14:08:40 +01:00
parent b00e8a4305
commit c5328f4057
5 changed files with 115 additions and 17 deletions

View File

@ -156,6 +156,14 @@ void printHelp() {
std::cout << " --show-path show path, -p show path" << std::endl;
std::cout << "\t\t\tPath of the directory with episodes" << std::endl;
std::cout << " --correct-path, -c\tThis is the correct path, stop asking me!" << std::endl;
std::cout << " --name-pattern pattern" << std::endl;
std::cout << "\t\t\tPattern to which change the file name. Possible sequences are:" << std::endl;
std::cout << "\t\t\t\t\%filename - original filename (without filetype extension)" << std::endl;
std::cout << "\t\t\t\t\%show - show name from thetvdb" << std::endl;
std::cout << "\t\t\t\t\%epname - episode name from thetvdb" << std::endl;
std::cout << "\t\t\t\t\%season - season number, possible to specify leading 0 like this: \%2season" << std::endl;
std::cout << "\t\t\t\t\%episode - episode number, possible to specify leading 0 like this: \%2season" << std::endl;
std::cout << "\t\t\tDefault pattern is \"$filename - $epname\"" << std::endl;
std::cout << " --trust, -t\t\tDon't ask whether the names are correct" << std::endl;
std::cout << " --linux, -x\t\tDon't replace characters characters that are illegal in Windows" << std::endl;
std::cout << " --lang language, -l language" << std::endl;
@ -211,3 +219,85 @@ std::string encodeUrl( const std::string &url ) {
return encoded.str();
}
std::string compilePattern(const std::string &pattern, int season, int episode, const std::string &filename, const std::string &episodeName, const std::string &showName) {
std::string output;
for( size_t i = 0; i < pattern.size(); i++ ) {
if( pattern[i] == '%' ) {
// if current character is % check if a pattern follows, otherwise put %
// check for numbers right after % indicating size of zero padding for numbers
auto pos = pattern.find_first_not_of("0123456789", i+1);
if( pattern.find( "season", pos - 1 ) == pos ) {
// if season is AFTER numbers, put season number padded
// with zeros
// get number of leading zeros
auto leading = std::atoi( pattern.c_str() + i + 1 );
// move i to the last char of 'season'
i = pos + 5;
auto season_num = std::to_string(season);
// get number of zeros to be put before the season number
leading -= season_num.size();
if( leading < 0 )
leading = 0;
// add padded season to output
output += std::string(leading, '0') + season_num;
} else if ( pattern.find( "season", i ) == i + 1 ) {
// if season isn't after numbers, just put season number to output
i += 6;
output += std::to_string(season);
} else if ( pattern.find( "episode", pos - 1 ) == pos ) {
// same principle as with season after number
auto leading = std::atoi( pattern.c_str() + i + 1 );
i = pos + 6;
auto ep_num = std::to_string(episode);
leading -= ep_num.size();
if( leading < 0 )
leading = 0;
output += std::string(leading, '0') + ep_num;
} else if ( pattern.find( "episode", i ) == i + 1 ) {
// if episode isn't after number, just put the episode number to output
i += 7;
output += std::to_string(episode);
} else if ( pattern.find( "epname", i ) == i + 1 ) {
// episode name from thetvdb
i += 6;
output += episodeName;
} else if ( pattern.find( "show", i ) == i + 1 ) {
// show name from thetvdb
i += 4;
output += showName;
} else if ( pattern.find( "filename", i ) == i + 1 ) {
// original file name
i += 8;
output += filename;
} else {
// output % if no escape sequence was found
output += '%';
}
} else if ( pattern[i] == '\\' ) {
// possibility to escape %
if( pattern[i+1] == '%' ) {
output += '%';
i++;
} else if ( pattern[i+1] == '\\' ) {
output += '\\';
i++;
} else {
output += '\\';
}
} else {
// if char isn't % or / just add it to the output string
output += pattern[i];
}
}
return output;
}

View File

@ -22,5 +22,6 @@ void printLangs();
bool findLanguage( const char *language );
std::string encodeUrl( const std::string &url );
std::string compilePattern(const std::string &pattern, int season, int episode, const std::string &filename, const std::string &episodeName, const std::string &showName);
#endif

View File

@ -4,7 +4,7 @@
#include "functions.hpp"
#include "tv_rename.hpp"
int parseCommandLine(std::string &show, std::set<int> &seasons_num, std::string &path, bool &change_dir, std::string &language, bool &linux, bool &trust, int argc, char **argv) {
int parseCommandLine(std::string &show, std::set<int> &seasons_num, std::string &path, bool &change_dir, std::string &language, std::string &pattern, bool &linux, bool &trust, int argc, char **argv) {
static struct option long_options[] = {
{"show", required_argument, 0, 's' },
{"season", required_argument, 0, 'n' },
@ -14,12 +14,13 @@ int parseCommandLine(std::string &show, std::set<int> &seasons_num, std::string
{"linux", no_argument, 0, 'x' },
{"lang", required_argument, 0, 'l' },
{"print-langs", no_argument, 0, '0' },
{"name-pattern", required_argument, 0, '1' },
{"help", no_argument, 0, 'h' }
};
while(1) {
int option_index{0};
auto c = getopt_long(argc, argv, "s:n:cp:txl:0h", long_options, &option_index);
auto c = getopt_long(argc, argv, "s:n:cp:txl:01:h", long_options, &option_index);
if( c == -1 )
break;
switch(c) {
@ -58,6 +59,9 @@ int parseCommandLine(std::string &show, std::set<int> &seasons_num, std::string
case 'h':
printHelp();
return 1;
case '1':
pattern = optarg;
break;
default:
return -1;
}
@ -74,10 +78,11 @@ int main(int argc, char** argv) {
std::string language{"en"};
bool linux{false};
bool trust{false};
std::string pattern{"%file - %epname"};
Curl c;
{
auto tmp = parseCommandLine(show, seasons_num, path, change_dir, language, linux, trust, argc, argv);
auto tmp = parseCommandLine(show, seasons_num, path, change_dir, language, pattern, linux, trust, argc, argv);
if( tmp == -1 )
return 1;
else if ( tmp == 1 )
@ -122,11 +127,11 @@ int main(int argc, char** argv) {
}
if( seasons_num.size() == 1 ) {
singleSeason(path, show, *seasons_num.begin(), "", language, linux, trust, c);
singleSeason(path, show, *seasons_num.begin(), "", language, pattern, linux, trust, c);
} else if ( seasons_num.size() != 0 ) {
multipleSeasons(path, show, seasons_num, language, linux, trust, c);
multipleSeasons(path, show, seasons_num, language, pattern, linux, trust, c);
} else {
allSeasons(path, show, language, linux, trust, c);
allSeasons(path, show, language, pattern, linux, trust, c);
}
}

View File

@ -31,7 +31,7 @@ std::vector<std::string> parseEpisodeNames( const std::string &season_code, cons
return episodes;
}
void singleSeason( const std::string &path, const std::string &show, int season, std::string url, const std::string &language, const bool &linux, const bool &trust, Curl &c, std::set<std::string> const *files) {
void singleSeason( const std::string &path, const std::string &show, int season, std::string url, const std::string &language, const std::string &pattern, const bool &linux, const bool &trust, Curl &c, std::set<std::string> const *files) {
if( url.empty() )
url = getDefUrl(show, language, c);
url += "/seasons/" + std::to_string(season);
@ -93,7 +93,9 @@ void singleSeason( const std::string &path, const std::string &show, int season,
num -= 1;
if( num < episodes.size() ) {
auto pos = name.find_last_of('.');
name.insert(pos, " - " + episodes[num]);
// get desired filename
name = compilePattern(pattern, season, num+1, name.substr(0, pos), episodes[num], show) + name.substr(pos);
// replace characters illegal in windows if desired
if( !linux ) {
season_code.erase(std::remove_if(name.begin(), name.end(), [](char x) {return x == '?' || x == '"' || x == '\\' || x == '|' || x == '*';}), season_code.end());
size_t max{name.size()};
@ -140,23 +142,23 @@ void singleSeason( const std::string &path, const std::string &show, int season,
}
}
void multipleSeasons( const std::string &path, const std::string &show, const std::map<int, std::set<std::string>> &seasons, const std::string &language, const bool &linux, const bool &trust, Curl &c) {
void multipleSeasons( const std::string &path, const std::string &show, const std::map<int, std::set<std::string>> &seasons, const std::string &language, const std::string &pattern, const bool &linux, const bool &trust, Curl &c) {
auto url = getDefUrl(show, language, c);
for( const auto &x : seasons ) {
singleSeason( path, show, x.first, url, language, linux, trust, c, &x.second);
singleSeason( path, show, x.first, url, language, pattern, linux, trust, c, &x.second);
}
}
void multipleSeasons( const std::string &path, const std::string &show, const std::set<int> seasons, const std::string &language, const bool &linux, const bool &trust, Curl &c) {
void multipleSeasons( const std::string &path, const std::string &show, const std::set<int> seasons, const std::string &language, const std::string &pattern, const bool &linux, const bool &trust, Curl &c) {
std::map<int, std::set<std::string>> season_map;
findSeasons(season_map, path, seasons);
multipleSeasons(path, show, season_map, language, linux, trust, c);
multipleSeasons(path, show, season_map, language, pattern, linux, trust, c);
}
void allSeasons( const std::string &path, const std::string &show, const std::string &language, const bool &linux, const bool &trust, Curl &c) {
void allSeasons( const std::string &path, const std::string &show, const std::string &language, const std::string &pattern, const bool &linux, const bool &trust, Curl &c) {
std::map<int, std::set<std::string>> seasons;
//get all season number from this directory and subdirectories
iterateFS(seasons, path);
multipleSeasons( path, show, seasons, language, linux, trust, c);
multipleSeasons( path, show, seasons, language, pattern, linux, trust, c);
}

View File

@ -4,8 +4,8 @@
#include <set>
#include "network.hpp"
void singleSeason( const std::string &path, const std::string &show, int season, std::string url, const std::string &language, const bool &linux, const bool &trust, Curl &c, std::set<std::string> const *files=nullptr);
void multipleSeasons( const std::string &path, const std::string &show, const std::set<int> seasons, const std::string &language, const bool &linux, const bool &trust, Curl &c);
void allSeasons( const std::string &path, const std::string &show, const std::string &language, const bool &linux, const bool &trust, Curl &c);
void singleSeason( const std::string &path, const std::string &show, int season, std::string url, const std::string &language, const std::string &pattern, const bool &linux, const bool &trust, Curl &c, std::set<std::string> const *files=nullptr);
void multipleSeasons( const std::string &path, const std::string &show, const std::set<int> seasons, const std::string &language, const std::string &pattern, const bool &linux, const bool &trust, Curl &c);
void allSeasons( const std::string &path, const std::string &show, const std::string &language, const std::string &pattern, const bool &linux, const bool &trust, Curl &c);
#endif