import json
import os


class Suite(object):
    def __init__(self, suitename):
        filename = "{0}.suite".format(suitename)
        filepath = os.path.join(os.path.dirname(__file__), filename)
        with open(filepath) as suitefile:
            _suite = json.load(suitefile)
        self.urls = _suite['urls']
        self.name = suitename
        self.attempts = 2
        self.max_attempts = 2

    @classmethod
    def get_available_suites(cls):
        suitesdir = os.path.dirname(__file__)
        suiteslist = [os.path.splitext(f)[0].lower() for f in os.listdir(suitesdir) if f.endswith('.suite')]
        if not suiteslist:
            raise Exception('Cant find any .suite files in directory %s' % suitesdir)
        return suiteslist