SeanOMik
/
dropout-dl
mirror of https://github.com/SeanOMik/dropout-dl.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Options: Added '--output-directory' Option 

This changes where the show files are placed.
This commit is contained in:
Moss 2022-09-29 18:04:04 -04:00
parent 05eec0da88
commit eb39b10284
4 changed files with 30 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
src/main.cpp
View File

@ -302,7 +302,7 @@ public:
bool season = false;
std::string quality;
std::string filename;
std::string series_dir;
std::string output_directory;
std::string episode;
std::vector<std::string> cookies;
@ -355,7 +355,7 @@ public:
std::cerr << "ARGUMENT PARSE ERROR: --output-directory used with too few following arguments\n";
exit(8);
}
series_dir = args[++i];
output_directory = args[++i];
}
else if (arg == "series") {
series = true;
@ -382,6 +382,9 @@ public:
}
}
if (output_directory.empty()) {
output_directory = ".";
}
if (season && series) {
std::cerr << "ARGUMENT PARSE ERROR: Season and Series arguments used\n";
@ -420,28 +423,18 @@ int main(int argc, char** argv) {
if (options.series) {
dropout_dl::series series(options.url, options.cookies);
if (options.series_dir.empty()) {
options.series_dir = series.name;
std::replace(options.series_dir.begin(), options.series_dir.end(), ' ', '_');
std::replace(options.series_dir.begin(), options.series_dir.end(), ',', '_');
}
series.download(options.quality, options.series_dir);
series.download(options.quality, options.output_directory);
}
else if (options.season) {
dropout_dl::season season = dropout_dl::series::get_season(options.url, options.cookies);
if (options.series_dir.empty()) {
options.series_dir = season.series_name;
std::string series_directory = season.series_name;
std::replace(options.series_dir.begin(), options.series_dir.end(), ' ', '_');
std::replace(series_directory.begin(), series_directory.end(), ' ', '_');
std::replace(options.series_dir.begin(), options.series_dir.end(), ',', '_');
}
std::replace(series_directory.begin(), series_directory.end(), ',', '_');
season.download(options.quality, options.series_dir);
season.download(options.quality, options.output_directory + "/" + series_directory);
}
else {
dropout_dl::episode ep(options.url, options.cookies, options.verbose);
@ -460,10 +453,6 @@ int main(int argc, char** argv) {
std::cout << "filename: " << options.filename << '\n';
}
if (options.series_dir.empty()) {
options.series_dir = ep.series;
}
if (!std::filesystem::is_directory(ep.series)) {
std::filesystem::create_directories(ep.series);
if (options.verbose) {
@ -471,7 +460,7 @@ int main(int argc, char** argv) {
}
}
ep.download(options.quality, options.series_dir, options.filename);
ep.download(options.quality, options.output_directory + "/" + ep.series, options.filename);
}

5
src/season.cpp
View File

@ -54,6 +54,11 @@ namespace dropout_dl {
}
void season::download(const std::string &quality, const std::string &series_directory) {
if (!std::filesystem::is_directory(series_directory)) {
std::filesystem::create_directories(series_directory);
std::cout << "Creating series directory" << '\n';
}
std::string dir = series_directory + "/" + this->name;
std::replace(dir.begin(), dir.end(), ' ', '_');

9
src/series.cpp
View File

@ -219,9 +219,14 @@ namespace dropout_dl {
exit(9);
}
void series::download(const std::string &quality, const std::string &series_directory) {
void series::download(const std::string &quality, const std::string& base) {
if (!std::filesystem::is_directory(base + "/" + series_directory)) {
std::filesystem::create_directories(base + "/" + series_directory);
std::cout << "Creating series directory" << '\n';
}
for (auto& season : seasons) {
season.download(quality, series_directory);
season.download(quality, base + "/" + series_directory);
}
}

8
src/series.h
View File

@ -15,6 +15,7 @@ namespace dropout_dl {
std::string name;
std::string url;
std::string page_data;
std::string series_directory;
std::vector<season> seasons;
static std::string get_series_name(const std::string& html_data);
@ -23,7 +24,7 @@ namespace dropout_dl {
static season get_season(const std::string& url, const std::vector<std::string>& cookies);
void download(const std::string& quality, const std::string& series_directory);
void download(const std::string& quality, const std::string& base);
explicit series(const std::string& url, const std::vector<std::string>& cookies) {
this->url = url;
@ -33,6 +34,11 @@ namespace dropout_dl {
std::cerr << "SERIES PARSE ERROR: Could not parse series name\n";
exit(10);
}
this->series_directory = name;
std::replace(this->series_directory.begin(), this->series_directory.end(), ' ', '_');
std::replace(this->series_directory.begin(), this->series_directory.end(), ',', '_');
this->seasons = get_seasons(page_data, cookies);
}
};