updated code
This commit is contained in:
130
app/app.py
130
app/app.py
@@ -6,9 +6,20 @@ import requests
|
|||||||
import logging
|
import logging
|
||||||
import threading
|
import threading
|
||||||
import time
|
import time
|
||||||
|
import datetime
|
||||||
|
import hashlib
|
||||||
|
|
||||||
# Configure logging
|
# Configure logging
|
||||||
logging.basicConfig(level=logging.INFO)
|
LOG_FOLDER = './logs'
|
||||||
|
os.makedirs(LOG_FOLDER, exist_ok=True)
|
||||||
|
logging.basicConfig(
|
||||||
|
level=logging.INFO,
|
||||||
|
format='%(asctime)s - %(levelname)s - %(message)s',
|
||||||
|
handlers=[
|
||||||
|
logging.FileHandler(os.path.join(LOG_FOLDER, 'app.log')),
|
||||||
|
logging.StreamHandler()
|
||||||
|
]
|
||||||
|
)
|
||||||
Logger = logging.getLogger(__name__)
|
Logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
app = Flask(__name__, static_folder='static')
|
app = Flask(__name__, static_folder='static')
|
||||||
@@ -63,22 +74,38 @@ def download_media_files(playlist):
|
|||||||
file_url = media.get('url', '')
|
file_url = media.get('url', '')
|
||||||
file_path = os.path.join(RESOURCES_FOLDER, file_name)
|
file_path = os.path.join(RESOURCES_FOLDER, file_name)
|
||||||
|
|
||||||
|
if not file_name or not file_url:
|
||||||
|
Logger.error(f"Invalid media entry: {media}")
|
||||||
|
continue
|
||||||
|
|
||||||
|
Logger.debug(f"Downloading file: {file_name} from {file_url}")
|
||||||
|
|
||||||
try:
|
try:
|
||||||
response = requests.get(file_url)
|
response = requests.get(file_url, stream=True)
|
||||||
if response.status_code == 200:
|
if response.status_code == 200:
|
||||||
with open(file_path, 'wb') as file:
|
with open(file_path, 'wb') as file:
|
||||||
file.write(response.content)
|
for chunk in response.iter_content(chunk_size=8192):
|
||||||
|
file.write(chunk)
|
||||||
Logger.info(f"Downloaded {file_name} to {file_path}")
|
Logger.info(f"Downloaded {file_name} to {file_path}")
|
||||||
else:
|
else:
|
||||||
Logger.error(f"Failed to download {file_name}: {response.status_code}")
|
Logger.error(f"Failed to download {file_name}: {response.status_code}")
|
||||||
except requests.exceptions.RequestException as e:
|
except requests.exceptions.RequestException as e:
|
||||||
Logger.error(f"Failed to download {file_name}: {e}")
|
Logger.error(f"Failed to download {file_name}: {e}")
|
||||||
|
|
||||||
|
def calculate_file_hash(file_path):
|
||||||
|
"""Calculate the SHA256 hash of a file."""
|
||||||
|
if not os.path.exists(file_path):
|
||||||
|
return None
|
||||||
|
sha256 = hashlib.sha256()
|
||||||
|
with open(file_path, 'rb') as file:
|
||||||
|
while chunk := file.read(8192):
|
||||||
|
sha256.update(chunk)
|
||||||
|
return sha256.hexdigest()
|
||||||
|
|
||||||
# Download playlist files from server
|
# Download playlist files from server
|
||||||
def download_playlist_files_from_server():
|
def download_playlist_files_from_server():
|
||||||
Logger.info("Starting playlist file download using app configuration...")
|
Logger.info("Starting playlist file download using app configuration...")
|
||||||
|
|
||||||
# Load app configuration
|
|
||||||
app_config = load_app_config()
|
app_config = load_app_config()
|
||||||
server_address = app_config.get('server_address', '')
|
server_address = app_config.get('server_address', '')
|
||||||
port = app_config.get('port', 1025)
|
port = app_config.get('port', 1025)
|
||||||
@@ -89,7 +116,6 @@ def download_playlist_files_from_server():
|
|||||||
Logger.error("Missing required configuration values.")
|
Logger.error("Missing required configuration values.")
|
||||||
return
|
return
|
||||||
|
|
||||||
# Construct the request URL and parameters
|
|
||||||
server_ip = f'{server_address}:{port}'
|
server_ip = f'{server_address}:{port}'
|
||||||
url = f'http://{server_ip}/api/playlists'
|
url = f'http://{server_ip}/api/playlists'
|
||||||
params = {
|
params = {
|
||||||
@@ -98,21 +124,64 @@ def download_playlist_files_from_server():
|
|||||||
}
|
}
|
||||||
|
|
||||||
try:
|
try:
|
||||||
# Send request to fetch the playlist
|
|
||||||
response = requests.get(url, params=params)
|
response = requests.get(url, params=params)
|
||||||
Logger.debug(f"Status Code: {response.status_code}")
|
Logger.debug(f"Status Code: {response.status_code}")
|
||||||
Logger.debug(f"Response Content: {response.text}")
|
Logger.debug(f"Response Content: {response.text}")
|
||||||
|
|
||||||
if response.status_code == 200:
|
if response.status_code == 200:
|
||||||
try:
|
try:
|
||||||
playlist = response.json().get('playlist', [])
|
server_playlist = response.json().get('playlist', [])
|
||||||
Logger.info("Playlist retrieved successfully.")
|
if not server_playlist:
|
||||||
save_playlist(playlist) # Save the playlist locally
|
Logger.error("Playlist is empty or missing in the server response.")
|
||||||
download_media_files(playlist) # Download media files
|
return
|
||||||
|
|
||||||
|
Logger.info("Server playlist retrieved successfully.")
|
||||||
|
|
||||||
|
# Calculate the hash of the current local playlist
|
||||||
|
local_playlist_hash = calculate_file_hash(PLAYLIST_FILE)
|
||||||
|
|
||||||
|
# Calculate the hash of the server playlist
|
||||||
|
server_playlist_hash = hashlib.sha256(json.dumps(server_playlist, sort_keys=True).encode()).hexdigest()
|
||||||
|
|
||||||
|
# Compare hashes to determine if the playlist has changed
|
||||||
|
if local_playlist_hash == server_playlist_hash:
|
||||||
|
Logger.info("No changes detected in the server playlist. Skipping download.")
|
||||||
|
return
|
||||||
|
|
||||||
|
Logger.info("Changes detected in the server playlist. Downloading updated files...")
|
||||||
|
|
||||||
|
# Compare and download missing or updated files
|
||||||
|
for media in server_playlist:
|
||||||
|
file_name = media.get('file_name', '')
|
||||||
|
file_url = media.get('url', '')
|
||||||
|
file_path = os.path.join(RESOURCES_FOLDER, file_name)
|
||||||
|
|
||||||
|
if not file_name or not file_url:
|
||||||
|
Logger.error(f"Invalid media entry: {media}")
|
||||||
|
continue
|
||||||
|
|
||||||
|
# Check if the file exists locally and matches the server version
|
||||||
|
if not os.path.exists(file_path) or os.path.getsize(file_path) != media.get('size', 0):
|
||||||
|
Logger.info(f"Downloading file: {file_name}")
|
||||||
|
response = requests.get(file_url, stream=True)
|
||||||
|
if response.status_code == 200:
|
||||||
|
with open(file_path, 'wb') as file:
|
||||||
|
for chunk in response.iter_content(chunk_size=8192):
|
||||||
|
file.write(chunk)
|
||||||
|
Logger.info(f"Downloaded {file_name} to {file_path}")
|
||||||
|
else:
|
||||||
|
Logger.error(f"Failed to download {file_name}: {response.status_code}")
|
||||||
|
|
||||||
|
# Save the server playlist locally
|
||||||
|
save_playlist(server_playlist)
|
||||||
|
|
||||||
|
# Update the updated_playlist.json
|
||||||
|
create_updated_playlist()
|
||||||
|
|
||||||
except json.JSONDecodeError as e:
|
except json.JSONDecodeError as e:
|
||||||
Logger.error(f"Failed to parse playlist JSON: {e}")
|
Logger.error(f"Failed to parse server playlist JSON: {e}")
|
||||||
else:
|
else:
|
||||||
Logger.error(f"Failed to retrieve playlist: {response.text}")
|
Logger.error(f"Failed to retrieve server playlist: {response.text}")
|
||||||
except requests.exceptions.RequestException as e:
|
except requests.exceptions.RequestException as e:
|
||||||
Logger.error(f"Failed to connect to server: {e}")
|
Logger.error(f"Failed to connect to server: {e}")
|
||||||
|
|
||||||
@@ -183,9 +252,9 @@ def create_updated_playlist():
|
|||||||
"""Create a new playlist file with local file paths."""
|
"""Create a new playlist file with local file paths."""
|
||||||
Logger.info("Creating updated playlist with local file paths...")
|
Logger.info("Creating updated playlist with local file paths...")
|
||||||
|
|
||||||
# Load the existing playlist
|
# Load the existing server playlist
|
||||||
if not os.path.exists(PLAYLIST_FILE):
|
if not os.path.exists(PLAYLIST_FILE):
|
||||||
Logger.error(f"Playlist file not found: {PLAYLIST_FILE}")
|
Logger.error(f"Server playlist file not found: {PLAYLIST_FILE}")
|
||||||
return
|
return
|
||||||
|
|
||||||
with open(PLAYLIST_FILE, 'r') as file:
|
with open(PLAYLIST_FILE, 'r') as file:
|
||||||
@@ -216,7 +285,8 @@ def create_updated_playlist():
|
|||||||
# Check and download playlist on app startup
|
# Check and download playlist on app startup
|
||||||
def initialize_playlist():
|
def initialize_playlist():
|
||||||
Logger.info("Initializing playlist...")
|
Logger.info("Initializing playlist...")
|
||||||
download_playlist_files_from_server()
|
download_playlist_files_from_server() # Download playlist from the server
|
||||||
|
create_updated_playlist() # Create the updated playlist with local file paths
|
||||||
Logger.info("Playlist initialization complete.")
|
Logger.info("Playlist initialization complete.")
|
||||||
|
|
||||||
# Function to check for playlist updates every 5 minutes
|
# Function to check for playlist updates every 5 minutes
|
||||||
@@ -224,10 +294,8 @@ def periodic_playlist_check():
|
|||||||
while True:
|
while True:
|
||||||
try:
|
try:
|
||||||
Logger.info("Checking for playlist updates...")
|
Logger.info("Checking for playlist updates...")
|
||||||
# Download the playlist files from the server
|
download_playlist_files_from_server() # Download playlist from the server
|
||||||
download_playlist_files_from_server()
|
create_updated_playlist() # Create the updated playlist with local file paths
|
||||||
# Create the updated playlist with local file paths
|
|
||||||
create_updated_playlist()
|
|
||||||
Logger.info("Playlist check complete.")
|
Logger.info("Playlist check complete.")
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
Logger.error(f"Error during playlist check: {e}")
|
Logger.error(f"Error during playlist check: {e}")
|
||||||
@@ -238,8 +306,30 @@ def start_playlist_check_thread():
|
|||||||
thread = threading.Thread(target=periodic_playlist_check, daemon=True)
|
thread = threading.Thread(target=periodic_playlist_check, daemon=True)
|
||||||
thread.start()
|
thread.start()
|
||||||
|
|
||||||
|
# Function to delete log files older than 2 days
|
||||||
|
def delete_old_logs(log_folder='./logs', days=2):
|
||||||
|
Logger.info("Checking for old log files to delete...")
|
||||||
|
if not os.path.exists(log_folder):
|
||||||
|
Logger.warning(f"Log folder does not exist: {log_folder}")
|
||||||
|
return
|
||||||
|
|
||||||
|
now = time.time()
|
||||||
|
cutoff = now - (days * 86400) # Convert days to seconds
|
||||||
|
|
||||||
|
for file_name in os.listdir(log_folder):
|
||||||
|
file_path = os.path.join(log_folder, file_name)
|
||||||
|
if os.path.isfile(file_path):
|
||||||
|
file_modified_time = os.path.getmtime(file_path)
|
||||||
|
if file_modified_time < cutoff:
|
||||||
|
try:
|
||||||
|
os.remove(file_path)
|
||||||
|
Logger.info(f"Deleted old log file: {file_path}")
|
||||||
|
except Exception as e:
|
||||||
|
Logger.error(f"Failed to delete log file {file_path}: {e}")
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
initialize_playlist() # Check and download playlist on startup
|
initialize_playlist() # Check and download playlist on startup
|
||||||
create_updated_playlist() # Create the updated playlist
|
create_updated_playlist() # Create the updated playlist
|
||||||
start_playlist_check_thread() # Start the background thread for periodic checks
|
start_playlist_check_thread() # Start the background thread for periodic checks
|
||||||
app.run(host='0.0.0.0', port=1025)
|
app.run(host='0.0.0.0', port=1025)
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user