##############################################################################
# Python From Scratch
# Autor: Nilo Ney Coutinho Menezes
# Editora Novatec (c) 2010-2024
# Site: https://pythonfromscratch.com
#
# File: listing\chapter 13\13.09 - Site Manager.py
# Description: Site Manager
##############################################################################

import json
from site_register import Site

class SiteManager:
    def __init__(self):
        self.sites = {}

    def load(self, filename):
        with open(filename) as file:
            data = json.load(file)
        self.sites.clear()
        for record in data:
            site = Site(
                id = record.get("id"),
                category = record.get("category"),
                date_added = record["date"],
                url = record["url"],
                notes = record.get("notes"))
            self.sites[site.id] = site