Fix poster downoad name files

This commit is contained in:
Pierret Hugo
2024-12-22 00:27:31 +01:00
committed by Hugo Pierret
parent b633cd11c3
commit 258480ef9d
2 changed files with 33 additions and 22 deletions

View File

@@ -20,6 +20,7 @@ ALLOWED_EXTENSIONS = {'csv'}
user_csvs = {}
@lru_cache(maxsize=1000)
def get_poster_url(movie_name, movie_year):
movie = tmdb_service.search_movie(movie_name, movie_year)
@@ -29,6 +30,7 @@ def get_poster_url(movie_name, movie_year):
return posters[0]
return None
@api_bp.route('/movies', methods=['GET'])
def get_movies():
user_id = session.get('user_id') # Récupérer l'ID de l'utilisateur de la session
@@ -66,7 +68,7 @@ def get_movies():
'id': start_index + i + 1,
'Watched Date': movie['Watched Date'],
'Name': movie['Name'],
'Year': movie['Year'],
'Year': movie['Year'],
'Poster': get_poster_url(movie['Name'], movie['Year']),
}
for i, movie in enumerate(movies)
@@ -82,9 +84,9 @@ def get_movies():
print(f"Error in get_movies: {str(e)}")
return jsonify({'error': 'An error occurred while fetching movies'}), 500
@api_bp.route('/posters/<movie_name>/<movie_year>', methods=['GET'])
def get_movie_posters(movie_name, movie_year):
tmdb_service = TMDBService()
movie = tmdb_service.search_movie(movie_name, movie_year)
@@ -95,9 +97,11 @@ def get_movie_posters(movie_name, movie_year):
else:
return jsonify({'error': 'Movie not found'}), 404
def allowed_file(filename):
return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
@api_bp.route('/upload-csv', methods=['POST'])
def upload_csv():
user_id = session.get('user_id')
@@ -126,6 +130,7 @@ def upload_csv():
return jsonify({'error': f'Error processing file: {str(e)}'}), 500
return jsonify({'error': 'File type not allowed'}), 400
@api_bp.route('/check-csv', methods=['GET'])
def check_csv():
user_id = session.get('user_id')
@@ -139,6 +144,7 @@ def check_csv():
print(f"Error in check_csv: {str(e)}")
return jsonify({'error': 'An error occurred while checking the CSV file'}), 500
@api_bp.route('/delete-csv', methods=['DELETE'])
def delete_csv():
user_id = session.get('user_id')
@@ -160,6 +166,8 @@ def download_posters():
data = request.get_json()
selected_posters = data.get('posters', [])
print(data)
if not selected_posters:
return jsonify({'error': 'No posters selected'}), 400
@@ -169,12 +177,15 @@ def download_posters():
for poster in selected_posters:
movie_id = poster['movieId']
poster_id = poster['posterId']
poster_url = f"https://image.tmdb.org/t/p/w500{poster_id}"
watched_date = poster.get('watchedDate', 'unknown-date')
poster_url = f"https://image.tmdb.org/t/p/original{poster_id}"
try:
response = requests.get(poster_url)
if response.status_code == 200:
zip_file.writestr(f"{movie_id}.jpg", response.content)
file_name = f"{watched_date}_{movie_id}.jpg"
zip_file.writestr(file_name, response.content)
else:
print(f"Failed to download poster for movie ID {movie_id}")
except Exception as e: