import os
import json
import logging

from core import GsxObject, GsxError, GsxCache

MODIFIERS = (
    ("A", "Not Applicable"),
    ("B", "Continuous"),
    ("C", "Intermittent"),
    ("D", "Fails After Warm Up"),
    ("E", "Environmental"),
    ("F", "Configuration: Peripheral"),
    ("G", "Damaged"),
)

GROUPS = (
    ('0', "General"),
    ('1', "Visual"),
    ('2', "Displays"),
    ('3', "Mass Storage"),
    ('4', "Input Devices"),
    ('5', "Boards"),
    ('6', "Power"),
    ('7', "Printer"),
    ('8', "Multi-function Device"),
    ('9', "Communication Devices"),
    ('A', "Share"),
    ('B', "iPhone"),
    ('E', "iPod"),
    ('F', "iPad"),
)


class CompTIA(GsxObject):
    "Stores and accesses CompTIA codes."
    _namespace = "glob:"

    def __init__(self):
        """
        Initialize CompTIA symptoms from the local JSON file
        """
        self._cache = GsxCache("comptia")
        #df = open(os.path.join(os.path.dirname(__file__), 'comptia.json'))
        #self._data = json.load(df)

    def fetch(self):
        """
        Description:
        The CompTIA Codes Lookup API retrieves a list of CompTIA groups and modifiers.

        Context:
        The CompTIA Codes (Symptom Codes) are the current available selections based on
        the component group code for parts.
        The API can be executed only after valid Authentication.
        Users can use the API at any point to retrieve the CompTIA code and modifier details,
        in order to create or update repairs.

        >>> CompTIA().fetch() # doctest: +ELLIPSIS
        {'A': {'989': 'Remote Inoperable', ...
        """
        self._submit("ComptiaCodeLookupRequest", "ComptiaCodeLookup", "comptiaInfo", True)

        if self._cache.get():
            return self._cache.get()

        root = self._req.objects[0]

        for el in root.findall(".//comptiaGroup"):
            group = {}
            comp_id = el[0].text

            for ci in el.findall('comptiaCodeInfo'):
                group[ci[0].text] = ci[1].text

            self._data[comp_id] = group

        self._cache.set(self._data)
        return self._data

    def symptoms(self, component=None):
        """
        Returns all known CompTIA symptom codes or just the ones
        belonging to the given component code.
        """
        r = dict()

        for g, codes in self.data.items():
            r[g] = list()
            for k, v in codes.items():
                r[g].append((k, v,))

        return r[component] if component else r

if __name__ == '__main__':
    import sys
    import doctest
    from core import connect
    logging.basicConfig(level=logging.DEBUG)
    connect(*sys.argv[1:4])
    doctest.testmod()