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

Quality: Added 'all' Quality Option 

This allows for the user to download all qualities at once and have them
be automatically placed into quality folders.
This commit is contained in:
Moss 2022-09-28 22:22:20 -04:00
parent 900303d9d0
commit b5c3e1e3bc
2 changed files with 21 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/episode.h
View File

@ -113,7 +113,7 @@ namespace dropout_dl {
std::replace(this->series.begin(), this->series.end(), ',', '_'); std::replace(this->series.begin(), this->series.end(), ',', '_');
this->filename = this->series + "/S" + (this->season_number.size() < 2 ? "0" + this->season_number : this->season_number) + "E" + (this->episode_number.size() < 2 ? "0" + this->episode_number : this->episode_number) + this->name + ".mp4"; this->filename = "S" + (this->season_number.size() < 2 ? "0" + this->season_number : this->season_number) + "E" + (this->episode_number.size() < 2 ? "0" + this->episode_number : this->episode_number) + this->name + ".mp4";
std::replace(filename.begin(), filename.end(), ' ', '_'); std::replace(filename.begin(), filename.end(), ' ', '_');

31
src/main.cpp
View File

@ -11,7 +11,6 @@
#endif #endif
#endif #endif
static int sqlite_write_callback(void* data, int argc, char** argv, char** azColName) static int sqlite_write_callback(void* data, int argc, char** argv, char** azColName)
{ {
if (argc < 1) { if (argc < 1) {
@ -298,8 +297,8 @@ std::vector<std::string> get_cookies(bool verbose = false) {
int main(int argc, char** argv) { int main(int argc, char** argv) {
bool verbose = true; bool verbose = false;
std::string quality = "1080p"; std::string quality = "all";
std::string firefox_profile; std::string firefox_profile;
std::string chrome_profile; std::string chrome_profile;
@ -325,12 +324,6 @@ int main(int argc, char** argv) {
dropout_dl::episode ep(episode_url, cookies, verbose); dropout_dl::episode ep(episode_url, cookies, verbose);
std::string video_url = ep.get_video_url(quality);
if (verbose) {
std::cout << "Got video url: " << video_url << '\n';
}
if (!std::filesystem::is_directory(ep.series)) { if (!std::filesystem::is_directory(ep.series)) {
std::filesystem::create_directories(ep.series); std::filesystem::create_directories(ep.series);
if (verbose) { if (verbose) {
@ -338,9 +331,25 @@ int main(int argc, char** argv) {
} }
} }
std::fstream out(ep.filename, std::ios_base::in|std::ios_base::out|std::ios_base::trunc); if (quality == "all") {
for (const auto& possible_quality : ep.qualities) {
if (!std::filesystem::is_directory(ep.series + "/" + possible_quality)) {
std::filesystem::create_directories(ep.series + "/" + possible_quality);
if (verbose) {
std::cout << "Creating series directory" << '\n';
}
}
std::fstream out(ep.series + "/" + possible_quality + "/" + ep.filename,
std::ios_base::in | std::ios_base::out | std::ios_base::trunc);
out << ep.get_video_data(quality) << std::endl; out << ep.get_video_data(possible_quality) << std::endl;
}
}
else {
std::fstream out(ep.series + "/" + ep.filename, std::ios_base::in | std::ios_base::out | std::ios_base::trunc);
out << ep.get_video_data(quality) << std::endl;
}
return 0; return 0;
} }