lang_detect_exception.py 519 Bytes
Newer Older
chenzk's avatar
v1.0  
chenzk committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
_error_codes = {
    'NoTextError': 0,
    'FormatError': 1,
    'FileLoadError': 2,
    'DuplicateLangError': 3,
    'NeedLoadProfileError': 4,
    'CantDetectError': 5,
    'CantOpenTrainData': 6,
    'TrainDataFormatError': 7,
    'InitParamError': 8,
}

ErrorCode = type('ErrorCode', (), _error_codes)


class LangDetectException(Exception):
    def __init__(self, code, message):
        super(LangDetectException, self).__init__(message)
        self.code = code

    def get_code(self):
        return self.code