mirror of
https://github.com/explosion/spaCy.git
synced 2024-11-14 05:37:03 +03:00
3141e04822
* Add spacy.errors module * Update deprecation and user warnings * Replace errors and asserts with new error message system * Remove redundant asserts * Fix whitespace * Add messages for print/util.prints statements * Fix typo * Fix typos * Move CLI messages to spacy.cli._messages * Add decorator to display error code with message An implementation like this is nice because it only modifies the string when it's retrieved from the containing class – so we don't have to worry about manipulating tracebacks etc. * Remove unused link in spacy.about * Update errors for invalid pipeline components * Improve error for unknown factories * Add displaCy warnings * Update formatting consistency * Move error message to spacy.errors * Update errors and check if doc returned by component is None
57 lines
1.9 KiB
Python
57 lines
1.9 KiB
Python
# coding: utf8
|
|
from __future__ import unicode_literals
|
|
|
|
import functools
|
|
|
|
from ..errors import Errors
|
|
|
|
|
|
class Underscore(object):
|
|
doc_extensions = {}
|
|
span_extensions = {}
|
|
token_extensions = {}
|
|
|
|
def __init__(self, extensions, obj, start=None, end=None):
|
|
object.__setattr__(self, '_extensions', extensions)
|
|
object.__setattr__(self, '_obj', obj)
|
|
# Assumption is that for doc values, _start and _end will both be None
|
|
# Span will set non-None values for _start and _end
|
|
# Token will have _start be non-None, _end be None
|
|
# This lets us key everything into the doc.user_data dictionary,
|
|
# (see _get_key), and lets us use a single Underscore class.
|
|
object.__setattr__(self, '_doc', obj.doc)
|
|
object.__setattr__(self, '_start', start)
|
|
object.__setattr__(self, '_end', end)
|
|
|
|
def __getattr__(self, name):
|
|
if name not in self._extensions:
|
|
raise AttributeError(Errors.E046.format(name=name))
|
|
default, method, getter, setter = self._extensions[name]
|
|
if getter is not None:
|
|
return getter(self._obj)
|
|
elif method is not None:
|
|
return functools.partial(method, self._obj)
|
|
else:
|
|
return self._doc.user_data.get(self._get_key(name), default)
|
|
|
|
def __setattr__(self, name, value):
|
|
if name not in self._extensions:
|
|
raise AttributeError(Errors.E047.format(name=name))
|
|
default, method, getter, setter = self._extensions[name]
|
|
if setter is not None:
|
|
return setter(self._obj, value)
|
|
else:
|
|
self._doc.user_data[self._get_key(name)] = value
|
|
|
|
def set(self, name, value):
|
|
return self.__setattr__(name, value)
|
|
|
|
def get(self, name):
|
|
return self.__getattr__(name)
|
|
|
|
def has(self, name):
|
|
return name in self._extensions
|
|
|
|
def _get_key(self, name):
|
|
return ('._.', name, self._start, self._end)
|