import os
from ConfigParser import ConfigParser


class GetFiles:
    query = None
    dir = None
    filenames = None
    base_url = None

    def __init__(self):
        self.base_url = '/Users/m.vasilyeva/PycharmProjects/news-autotests/agr/cladmin/smoke_selenium'

    def _get_files(self, dir, query):
        files = []
        for root, dirs, filenames in os.walk(dir):
            for f in filenames:
                if query in f:
                    files.append(f)
                    # log = open(os.path.join(root, f), 'r')
        return files

    def get_python_files(self):
        dir = self.base_url
        query = 'Tests.py'
        files = self._get_files(dir, query)
        files = [py_file for py_file in files if str(py_file).endswith('Tests.py')]
        return files

    def get_config_files(self):
        dir = '%s/config' % self.base_url
        query = 'selectors'
        files = self._get_files(dir, query)
        return files


class GetLocatorsFromIni(GetFiles):
    result = None

    def _parse_config(self, loc_files):
        result = dict()
        parser = ConfigParser()
        parser.read(loc_files)
        for section in parser.sections():
            result[section] = {'type': parser.get(section, 'type'), 'value': parser.get(section, 'value')}

        locators = result.keys()

        return locators

    def get_locators(self):
        loc_files = self.get_config_files()
        new_loc_files = []
        for lc in loc_files:
            lc = './config/' + str(lc)
            new_loc_files.append(lc)

        locators = self._parse_config(new_loc_files)
        return locators


class SearchLocatorsInFiles():
    gl = GetLocatorsFromIni()
    locators = gl.get_locators()
    py_files = gl.get_python_files()

    def search_in_files(self):
        locators = self.locators
        py_files = self.py_files
        failures = []
        for py_file in py_files:
            py_file = open(py_file, 'r')
            for locator in locators:
                sucess = locator in py_file
                if not sucess:
                    failures.append(locator)

        print failures

s = SearchLocatorsInFiles()
s.search_in_files()