# -*- coding: utf-8 -*-
from __future__ import print_function
import shlex
from hashlib import sha1
from .basexml import BaseXML
from .fucking_string import ensure_utf8


def read_python_file(pypath):
    vars = {}
    with open(pypath, 'rb') as fp:
        exec(compile(fp.read(), pypath, 'exec'), {}, vars)
    return vars


def get_checksum(s):
    return sha1(ensure_utf8(s).encode('utf-8')).hexdigest()[:7]


def escape_string(s):
    '''keep char unchanged if char is number or letter or unicode'''
    chs = []
    for ch in s:
        ch = ch if ord(ch) > 128 or ch.isalnum() else ' '
        chs.append(ch)

    # remove multiple whitespaces and replace whitespace with '_'
    return '_'.join(''.join(chs).split())


def is_hex(s):
    try:
        _ = int(s, 16)
        return True
    except ValueError:
        return False


def is_list_or_tuple(obj):
    return isinstance(obj, (list, tuple))


def split_ignore_quote(s):
    return shlex.split(s)


def remove_all_space(s):
    return ''.join(s.split())


def is_xml_element_equal(node1, node2):
    if len(node1) != len(node2):
        return False
    if node1.tag != node2.tag:
        return False
    if node1.attrib != node2.attrib:
        return False

    text1 = '' if node1.text is None else remove_all_space(node1.text)
    text2 = '' if node2.text is None else remove_all_space(node2.text)
    return text1 == text2


def is_xml_tree_equal(tree1, tree2, ignore_tags=tuple()):
    if tree1.tag == tree2.tag and tree1.tag in ignore_tags:
        return True
    elif is_xml_element_equal(tree1, tree2):
        elems1 = list(tree1)
        elems2 = list(tree2)

        for i in range(len(elems1)):
            if not is_xml_tree_equal(elems1[i], elems2[i], ignore_tags=ignore_tags):
                return False
        return True
    else:
        return False


def assert_xml_equal(xml_tree1, xml_tree2, ignore_tags=tuple()):
    if isinstance(xml_tree1, BaseXML):
        xml_tree1 = xml_tree1.to_xml()
    else:
        xml_tree1 = BaseXML.parse_string(xml_tree1)

    if isinstance(xml_tree2, BaseXML):
        xml_tree2 = xml_tree2.to_xml()
    else:
        xml_tree2 = BaseXML.parse_string(xml_tree2)

    assert(is_xml_tree_equal(xml_tree1, xml_tree2, ignore_tags=ignore_tags))

# -*- coding: utf-8 -*-
from __future__ import print_function
import shlex
from hashlib import sha1
from .basexml import BaseXML
from .fucking_string import ensure_utf8


def read_python_file(pypath):
    vars = {}
    with open(pypath, 'rb') as fp:
        exec(compile(fp.read(), pypath, 'exec'), {}, vars)
    return vars


def get_checksum(s):
    return sha1(ensure_utf8(s).encode('utf-8')).hexdigest()[:7]


def escape_string(s):
    '''keep char unchanged if char is number or letter or unicode'''
    chs = []
    for ch in s:
        ch = ch if ord(ch) > 128 or ch.isalnum() else ' '
        chs.append(ch)

    # remove multiple whitespaces and replace whitespace with '_'
    return '_'.join(''.join(chs).split())


def is_hex(s):
    try:
        _ = int(s, 16)
        return True
    except ValueError:
        return False


def is_list_or_tuple(obj):
    return isinstance(obj, (list, tuple))


def split_ignore_quote(s):
    return shlex.split(s)


def remove_all_space(s):
    return ''.join(s.split())


def is_xml_element_equal(node1, node2):
    if len(node1) != len(node2):
        return False
    if node1.tag != node2.tag:
        return False
    if node1.attrib != node2.attrib:
        return False

    text1 = '' if node1.text is None else remove_all_space(node1.text)
    text2 = '' if node2.text is None else remove_all_space(node2.text)
    return text1 == text2


def is_xml_tree_equal(tree1, tree2, ignore_tags=tuple()):
    if tree1.tag == tree2.tag and tree1.tag in ignore_tags:
        return True
    elif is_xml_element_equal(tree1, tree2):
        elems1 = list(tree1)
        elems2 = list(tree2)

        for i in range(len(elems1)):
            if not is_xml_tree_equal(elems1[i], elems2[i], ignore_tags=ignore_tags):
                return False
        return True
    else:
        return False


def assert_xml_equal(xml_tree1, xml_tree2, ignore_tags=tuple()):
    if isinstance(xml_tree1, BaseXML):
        xml_tree1 = xml_tree1.__str__()
    if isinstance(xml_tree2, BaseXML):
        xml_tree2 = xml_tree2.__str__()

    nospaces1 = ''.join(xml_tree1.split())
    nospaces2 = ''.join(xml_tree2.split())

    if nospaces1 != nospaces2:
        assert(is_xml_tree_equal(xml_tree1, xml_tree2, ignore_tags=ignore_tags))

change the implementation of assert_xml_equal