"""Read and write notebook files as XML."""

from xml.etree import ElementTree as ET

from .base import NotebookReader, NotebookWriter
from .nbdict import new_code_cell, new_worksheet, new_notebook


class XMLReader(NotebookReader):

    def reads(s, **kwargs):
        pass


class XMLWriter(NotebookWriter):

    def writes(nb, **kwargs):
        nb_e = ET.Element('notebook')
        name_e = ET.SubElement(nb_e, 'name')
        name_e.text = nb.get('name','')
        id_e = ET.SubElement(nb_e, 'id')
        id_e.text = nb.get('id','')
        for ws in nb['worksheets']:
            ws_e = ET.SubElement(nb_e, 'worksheet')
            ws_name_e = ET.SubElement(ws_e, 'name')
            ws_name_e.text = ws.get('name','')
            for cell in ws['cells']:
                cell_type = cell['cell_type']
                if cell_type == 'code':
                    output = cell['output']
                    cell_e = ET.SubElement(ws_e, 'cell')
                    input_e = ET.SubElement(cell_e, 'input')
                    input_e.text = cell.get('input','')
                    output_e = ET.SubElement(cell_e, 'output')
                    text_e = ET.SubElement(output_e, 'text')
                    text_e.text = cell.output
                elif cell_type == 'text':
                    pass
                    


_reader = XMLReader()
_writer = XMLWriter()

reads = _reader.reads
read = _reader.read
write = _writer.write
writes = _writer.writes