manager.py

import json
import os

cwd = os.path.dirname(os.path.abspath(__file__))

if not os.path.exists("json_data"):
    os.mkdir(os.path.join(cwd, "json_data"))

DATABASE_FILE = os.path.join(cwd, "json_data/animerecord.json")


def ensure_file_exists():
    """
    Ensure the database file exists. If not, create an empty JSON file.
    """
    if not os.path.isfile(DATABASE_FILE):
        with open(DATABASE_FILE, 'w') as f:
            json.dump({}, f)

def load_database():
    """
    Load the database from the JSON file.
    """
    ensure_file_exists()  # Ensure the file exists before loading
    with open(DATABASE_FILE, 'r') as f:
        return json.load(f)

def save_database(data):
    """
    Save the provided data to the JSON database file.
    """
    with open(DATABASE_FILE, 'w') as f:
        json.dump(data, f, indent=4)  # Pretty-print the JSON with an indent of 4 spaces

def get_next_index(database):
    """
    Get the next available index for a new record.
    """
    # Find the maximum index from existing keys or default to 0 if the database is empty
    return max([int(key) for key in database.keys()] or [0]) + 1

def update_entry(record, database=None):
    """
    Update an existing record in the database.
    """

    if database is None:
        database = load_database()  # Load the current database only if not passed as an argument
    
    title = record[1].get('title')
    
    # Find the index of the existing record by matching the title
    existing_index = next((index for index, data in database.items() if data["title"] == title), None)
    
    if existing_index is None:
        print(f"No existing record found for title '{title}'. Adding as new record.")
        add_new_record(record, database)  # If no existing record, add it as a new record
        return
    
    # Extract record details
    keyword = record[0]
    anime_type = record[1].get('type')
    max_episode = record[1].get('episodes')
    year = record[1].get('year')
    cover = record[1].get('poster')
    
    changes = []
    for k,v in database.items():
        if database[k]['title'] == title :
            changes.append(database[k]['about'])
            changes.append(database[k]['current_episode'])
    
    about = changes[0]
    current_episode = changes[1]

    if isinstance(current_episode, str):
        current_episode = 0

    if len(record) == 3:
        if type(record[2]) == int:
            current_episode = record[2] if current_episode < record[2] else current_episode
        else:
            if ',' in record[2] and len(record[2]) < 30:
                latest_episode = max(list(int(s) for s in record[2].split(',')))
                current_episode = latest_episode if current_episode < latest_episode else current_episode
            else:
                about = record[2]

    elif len(record) == 4:
        about = record[2]
        current_episode_info = record[3]
        latest_episode = max(list(int(s) for s in current_episode_info.split(',')))
        current_episode = latest_episode if current_episode < latest_episode else current_episode

    # Determine the status based on the current episode
    status = "Not Started Watching"
    try:
        if current_episode > 0:
            if current_episode < max_episode:
                status = f"Watching Episode {current_episode}"
            else:
                status = "Completed"
    except TypeError:
        print(f"\n{current_episode} is currently str,'>' not supported between instances of 'str' and 'int'")
        pass

    # Update the existing record with new details
    database[existing_index] = {
        "title": title,
        "keyword": keyword,
        "type": anime_type,
        "cover_photo": cover,
        "rating": 0,
        "status": status,
        "current_episode": current_episode,
        "max_episode": max_episode,
        "year_aired": year,
        "about": about
    }
    save_database(database)  # Save the updated database

def add_new_record(record, database):
    """
    Add a new record to the database.
    """
    next_index = get_next_index(database)
    
    # Extract record details
    keyword = record[0]
    title = record[1].get('title')
    anime_type = record[1].get('type')
    max_episode = record[1].get('episodes')
    year = record[1].get('year')
    cover = record[1].get('poster')

    about = ""
    current_episode = 0

    if len(record) == 3:
        if ',' in record[2] and all(part.isdigit() for part in record[2].split(',')):
            current_episode = int(record[2].split(',')[-1])
        else:
            about = record[2]
    elif len(record) == 4:
        about = record[2]
        current_episode_info = record[3]
        current_episode = int(max(list(record[3])))

    status = "Not Started Watching"
    if current_episode > 0:
        if current_episode < max_episode:
            status = f"Watching Episode {current_episode}"
        else:
            status = "Completed"

    database[next_index] = {
        "title": title,
        "keyword": keyword,
        "type": anime_type,
        "cover_photo": cover,
        "rating": 0,
        "status": status,
        "current_episode": current_episode,
        "max_episode": max_episode,
        "year_aired": year,
        "about": about
    }
    save_database(database)

def process_record(record, update=False):
    """
    Process and add a new record to the database. If the record exists, update it if `update` is True.
    """
    database = load_database()
    
    title = record[1].get('title')
    
    # Check if the record already exists in the database by title
    existing_index = next((index for index, data in database.items() if data["title"] == title), None)
    
    if existing_index is not None:
        if update:
            print(f"Record with title '{title}' already exists. Updating it.")
            update_entry(record, database)
        else:
            print(f"Record with title '{title}' already exists. No action taken.")
    else:
        print(f"Adding new record with title '{title}'.")
        add_new_record(record, database)

def search_record(query):
    """
    Search for records in the database that match the query.
    """
    database = load_database()
    
    results = {}
    
    for key, value in database.items():
        lower_query = query.lower()
        if lower_query in value["title"].lower() or lower_query in value["keyword"].lower():
            results[key] = value
    
    return results

def print_all_records():
    """
    Print all records from the database in a formatted JSON.
    """
    database = load_database()
    print(json.dumps(database, indent=4))

# Example usage
sample = [
    'kabaneri',
    {
        'id': 875,
        'title': 'Koutetsujou no Kabaneri',
        'type': 'TV',
        'episodes': 12,
        'status': 'Finished Airing',
        'season': 'Spring',
        'year': 2016,
        'score': 7.27,
        'poster': 'https://i.animepahe.ru/posters/3c01c83a35626201293b677d166226fcef7e13b00b875991907f1a54aebad626.jpg',
        'session': 'fccced41-eb03-ea7c-ceaf-13b40bad9cd3'
    },
    'The world is in the midst of the industrial revolution when horrific creatures emerge from a mysterious virus...'
]

if __name__ == '__main__':
    process_record(sample, update=True)