zv0n/RenameServer
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

TheMovieDB: formatting

Browse Source
This commit is contained in:
zvon 2021-07-27 19:40:44 +02:00
parent b88d4c0e65
commit 1565aabf70
1 changed files with 55 additions and 39 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

48
themoviedb/moviedb.cpp
View File

@ -1,5 +1,6 @@
// API - 2ebc8e784a4072da457fae5c0d291e48 // API - 2ebc8e784a4072da457fae5c0d291e48
// API READ ONLY - eyJhbGciOiJIUzI1NiJ9.eyJhdWQiOiIyZWJjOGU3ODRhNDA3MmRhNDU3ZmFlNWMwZDI5MWU0OCIsInN1YiI6IjYwZTJlNGI5MjJlNDgwMDA2MDJmZDMzMyIsInNjb3BlcyI6WyJhcGlfcmVhZCJdLCJ2ZXJzaW9uIjoxfQ.c0y7bTCI5KSsfQRw7igPx1FR40mbMF6hGTJTHn0HXH8 // API READ ONLY -
// eyJhbGciOiJIUzI1NiJ9.eyJhdWQiOiIyZWJjOGU3ODRhNDA3MmRhNDU3ZmFlNWMwZDI5MWU0OCIsInN1YiI6IjYwZTJlNGI5MjJlNDgwMDA2MDJmZDMzMyIsInNjb3BlcyI6WyJhcGlfcmVhZCJdLCJ2ZXJzaW9uIjoxfQ.c0y7bTCI5KSsfQRw7igPx1FR40mbMF6hGTJTHn0HXH8
#include <algorithm> #include <algorithm>
#include <iostream> #include <iostream>
#include <map> #include <map>
@ -64,11 +65,16 @@ bool init(const string &config_path) {
#else #else
request.setServer( "https://api.themoviedb.org/3" ); request.setServer( "https://api.themoviedb.org/3" );
#endif #endif
_moviedb_api_token = "eyJhbGciOiJIUzI1NiJ9.eyJhdWQiOiIyZWJjOGU3ODRhNDA3MmRhNDU3ZmFlNWMwZDI5MWU0OCIsInN1YiI6IjYwZTJlNGI5MjJlNDgwMDA2MDJmZDMzMyIsInNjb3BlcyI6WyJhcGlfcmVhZCJdLCJ2ZXJzaW9uIjoxfQ.c0y7bTCI5KSsfQRw7igPx1FR40mbMF6hGTJTHn0HXH8"; _moviedb_api_token =
"eyJhbGciOiJIUzI1NiJ9."
"eyJhdWQiOiIyZWJjOGU3ODRhNDA3MmRhNDU3ZmFlNWMwZDI5MWU0OCIsInN1YiI6IjYwZT"
"JlNGI5MjJlNDgwMDA2MDJmZDMzMyIsInNjb3BlcyI6WyJhcGlfcmVhZCJdLCJ2ZXJzaW9u"
"IjoxfQ.c0y7bTCI5KSsfQRw7igPx1FR40mbMF6hGTJTHn0HXH8";
return true; return true;
} }
bool hasKey(const rapidjson::GenericValue<rapidjson::UTF8<>> &object, const std::string &key) { bool hasKey( const rapidjson::GenericValue< rapidjson::UTF8<> > &object,
const std::string &key ) {
return object.FindMember( key.c_str() ) != object.MemberEnd(); return object.FindMember( key.c_str() ) != object.MemberEnd();
} }
@ -87,12 +93,13 @@ searchMovie( const string &movie, const string &language, const string &year ) {
do { do {
cur_page++; cur_page++;
rapidjson::Document json; rapidjson::Document json;
auto request_uri = TEXT( "/search/movie?query=" ) + encoded_show + TEXT("&language=" ) + language + TEXT("&page=") + toString(std::to_string(cur_page)); auto request_uri = TEXT( "/search/movie?query=" ) + encoded_show +
TEXT( "&language=" ) + language + TEXT( "&page=" ) +
toString( std::to_string( cur_page ) );
if ( !year.empty() ) { if ( !year.empty() ) {
request_uri += TEXT( "&year=" ) + year; request_uri += TEXT( "&year=" ) + year;
} }
json.Parse( json.Parse( request.get( request_uri ).c_str() );
request.get( request_uri ).c_str() );
if ( json.HasParseError() ) if ( json.HasParseError() )
return {}; return {};
@ -105,14 +112,17 @@ searchMovie( const string &movie, const string &language, const string &year ) {
// find all possible movies // find all possible movies
for ( size_t i = 0; i < results.Size(); i++ ) { for ( size_t i = 0; i < results.Size(); i++ ) {
if(!hasKey(results[i], "title") || !hasKey(results[i], "id") || if ( !hasKey( results[i], "title" ) ||
!hasKey(results[i], "release_date") || !hasKey(results[i], "original_title")) { !hasKey( results[i], "id" ) ||
!hasKey( results[i], "release_date" ) ||
!hasKey( results[i], "original_title" ) ) {
continue; continue;
} }
auto movie = toString( results[i]["title"].GetString() ); auto movie = toString( results[i]["title"].GetString() );
auto id = toString( std::to_string( results[i]["id"].GetInt() ) ); auto id = toString( std::to_string( results[i]["id"].GetInt() ) );
string year = toString( results[i]["release_date"].GetString() ); string year = toString( results[i]["release_date"].GetString() );
string original = toString( results[i]["original_title"].GetString() ); string original =
toString( results[i]["original_title"].GetString() );
if ( year.empty() ) { if ( year.empty() ) {
year = "0000"; year = "0000";
} else { } else {
@ -173,8 +183,8 @@ std::string removeIllegalCharacters(const std::string &input) {
// replace characters illegal in windows // replace characters illegal in windows
ret.erase( std::remove_if( ret.begin(), ret.end(), ret.erase( std::remove_if( ret.begin(), ret.end(),
[]( char_t x ) { []( char_t x ) {
return x == '?' || x == '"' || return x == '?' || x == '"' || x == '\\' ||
x == '\\' || x == '*'; x == '*';
} ), } ),
ret.end() ); ret.end() );
for ( size_t i = 0; i < ret.size(); i++ ) { for ( size_t i = 0; i < ret.size(); i++ ) {
@ -211,11 +221,14 @@ MovieNames movieFromId( const string &id, const string &language ) {
if ( json.HasParseError() ) { if ( json.HasParseError() ) {
return { "", "" }; return { "", "" };
} }
return {removeIllegalCharacters( json["title"].GetString() ), removeIllegalCharacters( json["original_title"].GetString() )}; return { removeIllegalCharacters( json["title"].GetString() ),
removeIllegalCharacters( json["original_title"].GetString() ) };
} }
bool renameMovie( const string &path, const string &name, const string &year ) { bool renameMovie( const string &path, const string &name, const string &year ) {
return FSLib::rename(path, FSLib::canonical( FSLib::getContainingDirectory(path) ) + "/" + name + " (" + year + ")." + FSLib::getFileExtension(path) ); return FSLib::rename(
path, FSLib::canonical( FSLib::getContainingDirectory( path ) ) + "/" +
name + " (" + year + ")." + FSLib::getFileExtension( path ) );
} }
bool renamePath( const string &path, const RenameObject &renamer ) { bool renamePath( const string &path, const RenameObject &renamer ) {
@ -248,17 +261,20 @@ bool renamePath( const string &path, const RenameObject &renamer ) {
if ( results.empty() ) if ( results.empty() )
return false; return false;
id = results[0].id; id = results[0].id;
movie = { removeIllegalCharacters( results[0].name ), removeIllegalCharacters( results[0].original_name ) }; movie = { removeIllegalCharacters( results[0].name ),
removeIllegalCharacters( results[0].original_name ) };
year = results[0].year; year = results[0].year;
} else { } else {
id = renamer.getCustomFields().at( "id" ); id = renamer.getCustomFields().at( "id" );
movie = movieFromId( id, lang ); movie = movieFromId( id, lang );
} }
if((!use_original && movie.name == "") || (use_original && movie.original_name == "")) if ( ( !use_original && movie.name == "" ) ||
( use_original && movie.original_name == "" ) )
return false; return false;
return renameMovie( path, use_original ? movie.original_name : movie.name, year ); return renameMovie( path, use_original ? movie.original_name : movie.name,
year );
} }
std::vector< string > getCustomKeys() { std::vector< string > getCustomKeys() {