KTVManager_Backend/routes/extra_urls.py

from flask import Blueprint, request, jsonify
import os

extra_urls_bp = Blueprint('extra_urls', __name__)

EXTRA_URLS_FILE = os.path.join(os.path.dirname(__file__), '..', 'ktvmanager', 'lib', 'extra_urls.txt')

def read_extra_urls_list():
    if not os.path.exists(EXTRA_URLS_FILE):
        return []
    with open(EXTRA_URLS_FILE, 'r') as f:
        return [line.strip() for line in f.readlines() if line.strip()]

def write_extra_urls_list(extra_urls_list):
    with open(EXTRA_URLS_FILE, 'w') as f:
        for item in extra_urls_list:
            f.write(f"{item}\n")

@extra_urls_bp.route('/extra_urls', methods=['GET'])
def get_extra_urls_list():
    """Gets the list of extra URLs."""
    return jsonify(read_extra_urls_list())

@extra_urls_bp.route('/extra_urls', methods=['POST'])
def add_extra_url():
    """Adds a new extra URL."""
    data = request.get_json()
    if not data or 'extra_url' not in data:
        return jsonify({'error': 'Missing extra_url in request body'}), 400
    
    extra_url = data.get('extra_url')
    if not extra_url:
        return jsonify({'error': 'Extra URL cannot be empty.'}), 400

    extra_urls_list = read_extra_urls_list()
    if extra_url in extra_urls_list:
        return jsonify({'message': 'Extra URL already exists.'}), 200

    extra_urls_list.append(extra_url)
    write_extra_urls_list(extra_urls_list)
    return jsonify({'message': 'Extra URL added successfully.'}), 201

@extra_urls_bp.route('/extra_urls', methods=['DELETE'])
def remove_extra_url():
    """Removes an extra URL."""
    data = request.get_json()
    if not data or 'extra_url' not in data:
        return jsonify({'error': 'Missing extra_url in request body'}), 400

    extra_url = data.get('extra_url')
    if not extra_url:
        return jsonify({'error': 'Extra URL cannot be empty.'}), 400

    extra_urls_list = read_extra_urls_list()
    if extra_url not in extra_urls_list:
        return jsonify({'error': 'Extra URL not found.'}), 404

    extra_urls_list.remove(extra_url)
    write_extra_urls_list(extra_urls_list)
    return jsonify({'message': 'Extra URL removed successfully.'}), 200