123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551 |
- # cython: infer_types=True, language_level=3, py2_import=True, auto_pickle=False
- #
- # Cython Scanner
- #
- from __future__ import absolute_import
- import cython
- cython.declare(make_lexicon=object, lexicon=object,
- print_function=object, error=object, warning=object,
- os=object, platform=object)
- import os
- import platform
- from .. import Utils
- from ..Plex.Scanners import Scanner
- from ..Plex.Errors import UnrecognizedInput
- from .Errors import error, warning
- from .Lexicon import any_string_prefix, make_lexicon, IDENT
- from .Future import print_function
- debug_scanner = 0
- trace_scanner = 0
- scanner_debug_flags = 0
- scanner_dump_file = None
- lexicon = None
- def get_lexicon():
- global lexicon
- if not lexicon:
- lexicon = make_lexicon()
- return lexicon
- #------------------------------------------------------------------
- py_reserved_words = [
- "global", "nonlocal", "def", "class", "print", "del", "pass", "break",
- "continue", "return", "raise", "import", "exec", "try",
- "except", "finally", "while", "if", "elif", "else", "for",
- "in", "assert", "and", "or", "not", "is", "lambda",
- "from", "yield", "with",
- ]
- pyx_reserved_words = py_reserved_words + [
- "include", "ctypedef", "cdef", "cpdef",
- "cimport", "DEF", "IF", "ELIF", "ELSE"
- ]
- class Method(object):
- def __init__(self, name, **kwargs):
- self.name = name
- self.kwargs = kwargs or None
- self.__name__ = name # for Plex tracing
- def __call__(self, stream, text):
- method = getattr(stream, self.name)
- # self.kwargs is almost always unused => avoid call overhead
- return method(text, **self.kwargs) if self.kwargs is not None else method(text)
- def __copy__(self):
- return self # immutable, no need to copy
- def __deepcopy__(self, memo):
- return self # immutable, no need to copy
- #------------------------------------------------------------------
- class CompileTimeScope(object):
- def __init__(self, outer=None):
- self.entries = {}
- self.outer = outer
- def declare(self, name, value):
- self.entries[name] = value
- def update(self, other):
- self.entries.update(other)
- def lookup_here(self, name):
- return self.entries[name]
- def __contains__(self, name):
- return name in self.entries
- def lookup(self, name):
- try:
- return self.lookup_here(name)
- except KeyError:
- outer = self.outer
- if outer:
- return outer.lookup(name)
- else:
- raise
- def initial_compile_time_env():
- benv = CompileTimeScope()
- names = ('UNAME_SYSNAME', 'UNAME_NODENAME', 'UNAME_RELEASE', 'UNAME_VERSION', 'UNAME_MACHINE')
- for name, value in zip(names, platform.uname()):
- benv.declare(name, value)
- try:
- import __builtin__ as builtins
- except ImportError:
- import builtins
- names = (
- 'False', 'True',
- 'abs', 'all', 'any', 'ascii', 'bin', 'bool', 'bytearray', 'bytes',
- 'chr', 'cmp', 'complex', 'dict', 'divmod', 'enumerate', 'filter',
- 'float', 'format', 'frozenset', 'hash', 'hex', 'int', 'len',
- 'list', 'map', 'max', 'min', 'oct', 'ord', 'pow', 'range',
- 'repr', 'reversed', 'round', 'set', 'slice', 'sorted', 'str',
- 'sum', 'tuple', 'zip',
- ### defined below in a platform independent way
- # 'long', 'unicode', 'reduce', 'xrange'
- )
- for name in names:
- try:
- benv.declare(name, getattr(builtins, name))
- except AttributeError:
- # ignore, likely Py3
- pass
- # Py2/3 adaptations
- from functools import reduce
- benv.declare('reduce', reduce)
- benv.declare('unicode', getattr(builtins, 'unicode', getattr(builtins, 'str')))
- benv.declare('long', getattr(builtins, 'long', getattr(builtins, 'int')))
- benv.declare('xrange', getattr(builtins, 'xrange', getattr(builtins, 'range')))
- denv = CompileTimeScope(benv)
- return denv
- #------------------------------------------------------------------
- class SourceDescriptor(object):
- """
- A SourceDescriptor should be considered immutable.
- """
- filename = None
- _file_type = 'pyx'
- _escaped_description = None
- _cmp_name = ''
- def __str__(self):
- assert False # To catch all places where a descriptor is used directly as a filename
- def set_file_type_from_name(self, filename):
- name, ext = os.path.splitext(filename)
- self._file_type = ext in ('.pyx', '.pxd', '.py') and ext[1:] or 'pyx'
- def is_cython_file(self):
- return self._file_type in ('pyx', 'pxd')
- def is_python_file(self):
- return self._file_type == 'py'
- def get_escaped_description(self):
- if self._escaped_description is None:
- esc_desc = \
- self.get_description().encode('ASCII', 'replace').decode("ASCII")
- # Use forward slashes on Windows since these paths
- # will be used in the #line directives in the C/C++ files.
- self._escaped_description = esc_desc.replace('\\', '/')
- return self._escaped_description
- def __gt__(self, other):
- # this is only used to provide some sort of order
- try:
- return self._cmp_name > other._cmp_name
- except AttributeError:
- return False
- def __lt__(self, other):
- # this is only used to provide some sort of order
- try:
- return self._cmp_name < other._cmp_name
- except AttributeError:
- return False
- def __le__(self, other):
- # this is only used to provide some sort of order
- try:
- return self._cmp_name <= other._cmp_name
- except AttributeError:
- return False
- def __copy__(self):
- return self # immutable, no need to copy
- def __deepcopy__(self, memo):
- return self # immutable, no need to copy
- class FileSourceDescriptor(SourceDescriptor):
- """
- Represents a code source. A code source is a more generic abstraction
- for a "filename" (as sometimes the code doesn't come from a file).
- Instances of code sources are passed to Scanner.__init__ as the
- optional name argument and will be passed back when asking for
- the position()-tuple.
- """
- def __init__(self, filename, path_description=None):
- filename = Utils.decode_filename(filename)
- self.path_description = path_description or filename
- self.filename = filename
- # Prefer relative paths to current directory (which is most likely the project root) over absolute paths.
- workdir = os.path.abspath('.') + os.sep
- self.file_path = filename[len(workdir):] if filename.startswith(workdir) else filename
- self.set_file_type_from_name(filename)
- self._cmp_name = filename
- self._lines = {}
- def get_lines(self, encoding=None, error_handling=None):
- # we cache the lines only the second time this is called, in
- # order to save memory when they are only used once
- key = (encoding, error_handling)
- try:
- lines = self._lines[key]
- if lines is not None:
- return lines
- except KeyError:
- pass
- with Utils.open_source_file(self.filename, encoding=encoding, error_handling=error_handling) as f:
- lines = list(f)
- if key in self._lines:
- self._lines[key] = lines
- else:
- # do not cache the first access, but remember that we
- # already read it once
- self._lines[key] = None
- return lines
- def get_description(self):
- try:
- return os.path.relpath(self.path_description)
- except ValueError:
- # path not under current directory => use complete file path
- return self.path_description
- def get_error_description(self):
- path = self.filename
- cwd = Utils.decode_filename(os.getcwd() + os.path.sep)
- if path.startswith(cwd):
- return path[len(cwd):]
- return path
- def get_filenametable_entry(self):
- return self.file_path
- def __eq__(self, other):
- return isinstance(other, FileSourceDescriptor) and self.filename == other.filename
- def __hash__(self):
- return hash(self.filename)
- def __repr__(self):
- return "<FileSourceDescriptor:%s>" % self.filename
- class StringSourceDescriptor(SourceDescriptor):
- """
- Instances of this class can be used instead of a filenames if the
- code originates from a string object.
- """
- def __init__(self, name, code):
- self.name = name
- #self.set_file_type_from_name(name)
- self.codelines = [x + "\n" for x in code.split("\n")]
- self._cmp_name = name
- def get_lines(self, encoding=None, error_handling=None):
- if not encoding:
- return self.codelines
- else:
- return [line.encode(encoding, error_handling).decode(encoding)
- for line in self.codelines]
- def get_description(self):
- return self.name
- get_error_description = get_description
- def get_filenametable_entry(self):
- return "stringsource"
- def __hash__(self):
- return id(self)
- # Do not hash on the name, an identical string source should be the
- # same object (name is often defaulted in other places)
- # return hash(self.name)
- def __eq__(self, other):
- return isinstance(other, StringSourceDescriptor) and self.name == other.name
- def __repr__(self):
- return "<StringSourceDescriptor:%s>" % self.name
- #------------------------------------------------------------------
- class PyrexScanner(Scanner):
- # context Context Compilation context
- # included_files [string] Files included with 'include' statement
- # compile_time_env dict Environment for conditional compilation
- # compile_time_eval boolean In a true conditional compilation context
- # compile_time_expr boolean In a compile-time expression context
- def __init__(self, file, filename, parent_scanner=None,
- scope=None, context=None, source_encoding=None, parse_comments=True, initial_pos=None):
- Scanner.__init__(self, get_lexicon(), file, filename, initial_pos)
- if filename.is_python_file():
- self.in_python_file = True
- self.keywords = set(py_reserved_words)
- else:
- self.in_python_file = False
- self.keywords = set(pyx_reserved_words)
- self.async_enabled = 0
- if parent_scanner:
- self.context = parent_scanner.context
- self.included_files = parent_scanner.included_files
- self.compile_time_env = parent_scanner.compile_time_env
- self.compile_time_eval = parent_scanner.compile_time_eval
- self.compile_time_expr = parent_scanner.compile_time_expr
- if parent_scanner.async_enabled:
- self.enter_async()
- else:
- self.context = context
- self.included_files = scope.included_files
- self.compile_time_env = initial_compile_time_env()
- self.compile_time_eval = 1
- self.compile_time_expr = 0
- if getattr(context.options, 'compile_time_env', None):
- self.compile_time_env.update(context.options.compile_time_env)
- self.parse_comments = parse_comments
- self.source_encoding = source_encoding
- self.trace = trace_scanner
- self.indentation_stack = [0]
- self.indentation_char = None
- self.bracket_nesting_level = 0
- self.begin('INDENT')
- self.sy = ''
- self.next()
- def commentline(self, text):
- if self.parse_comments:
- self.produce('commentline', text)
- def strip_underscores(self, text, symbol):
- self.produce(symbol, text.replace('_', ''))
- def current_level(self):
- return self.indentation_stack[-1]
- def open_bracket_action(self, text):
- self.bracket_nesting_level += 1
- return text
- def close_bracket_action(self, text):
- self.bracket_nesting_level -= 1
- return text
- def newline_action(self, text):
- if self.bracket_nesting_level == 0:
- self.begin('INDENT')
- self.produce('NEWLINE', '')
- string_states = {
- "'": 'SQ_STRING',
- '"': 'DQ_STRING',
- "'''": 'TSQ_STRING',
- '"""': 'TDQ_STRING'
- }
- def begin_string_action(self, text):
- while text[:1] in any_string_prefix:
- text = text[1:]
- self.begin(self.string_states[text])
- self.produce('BEGIN_STRING')
- def end_string_action(self, text):
- self.begin('')
- self.produce('END_STRING')
- def unclosed_string_action(self, text):
- self.end_string_action(text)
- self.error("Unclosed string literal")
- def indentation_action(self, text):
- self.begin('')
- # Indentation within brackets should be ignored.
- #if self.bracket_nesting_level > 0:
- # return
- # Check that tabs and spaces are being used consistently.
- if text:
- c = text[0]
- #print "Scanner.indentation_action: indent with", repr(c) ###
- if self.indentation_char is None:
- self.indentation_char = c
- #print "Scanner.indentation_action: setting indent_char to", repr(c)
- else:
- if self.indentation_char != c:
- self.error("Mixed use of tabs and spaces")
- if text.replace(c, "") != "":
- self.error("Mixed use of tabs and spaces")
- # Figure out how many indents/dedents to do
- current_level = self.current_level()
- new_level = len(text)
- #print "Changing indent level from", current_level, "to", new_level ###
- if new_level == current_level:
- return
- elif new_level > current_level:
- #print "...pushing level", new_level ###
- self.indentation_stack.append(new_level)
- self.produce('INDENT', '')
- else:
- while new_level < self.current_level():
- #print "...popping level", self.indentation_stack[-1] ###
- self.indentation_stack.pop()
- self.produce('DEDENT', '')
- #print "...current level now", self.current_level() ###
- if new_level != self.current_level():
- self.error("Inconsistent indentation")
- def eof_action(self, text):
- while len(self.indentation_stack) > 1:
- self.produce('DEDENT', '')
- self.indentation_stack.pop()
- self.produce('EOF', '')
- def next(self):
- try:
- sy, systring = self.read()
- except UnrecognizedInput:
- self.error("Unrecognized character")
- return # just a marker, error() always raises
- if sy == IDENT:
- if systring in self.keywords:
- if systring == u'print' and print_function in self.context.future_directives:
- self.keywords.discard('print')
- elif systring == u'exec' and self.context.language_level >= 3:
- self.keywords.discard('exec')
- else:
- sy = systring
- systring = self.context.intern_ustring(systring)
- self.sy = sy
- self.systring = systring
- if False: # debug_scanner:
- _, line, col = self.position()
- if not self.systring or self.sy == self.systring:
- t = self.sy
- else:
- t = "%s %s" % (self.sy, self.systring)
- print("--- %3d %2d %s" % (line, col, t))
- def peek(self):
- saved = self.sy, self.systring
- self.next()
- next = self.sy, self.systring
- self.unread(*next)
- self.sy, self.systring = saved
- return next
- def put_back(self, sy, systring):
- self.unread(self.sy, self.systring)
- self.sy = sy
- self.systring = systring
- def unread(self, token, value):
- # This method should be added to Plex
- self.queue.insert(0, (token, value))
- def error(self, message, pos=None, fatal=True):
- if pos is None:
- pos = self.position()
- if self.sy == 'INDENT':
- error(pos, "Possible inconsistent indentation")
- err = error(pos, message)
- if fatal: raise err
- def expect(self, what, message=None):
- if self.sy == what:
- self.next()
- else:
- self.expected(what, message)
- def expect_keyword(self, what, message=None):
- if self.sy == IDENT and self.systring == what:
- self.next()
- else:
- self.expected(what, message)
- def expected(self, what, message=None):
- if message:
- self.error(message)
- else:
- if self.sy == IDENT:
- found = self.systring
- else:
- found = self.sy
- self.error("Expected '%s', found '%s'" % (what, found))
- def expect_indent(self):
- self.expect('INDENT', "Expected an increase in indentation level")
- def expect_dedent(self):
- self.expect('DEDENT', "Expected a decrease in indentation level")
- def expect_newline(self, message="Expected a newline", ignore_semicolon=False):
- # Expect either a newline or end of file
- useless_trailing_semicolon = None
- if ignore_semicolon and self.sy == ';':
- useless_trailing_semicolon = self.position()
- self.next()
- if self.sy != 'EOF':
- self.expect('NEWLINE', message)
- if useless_trailing_semicolon is not None:
- warning(useless_trailing_semicolon, "useless trailing semicolon")
- def enter_async(self):
- self.async_enabled += 1
- if self.async_enabled == 1:
- self.keywords.add('async')
- self.keywords.add('await')
- def exit_async(self):
- assert self.async_enabled > 0
- self.async_enabled -= 1
- if not self.async_enabled:
- self.keywords.discard('await')
- self.keywords.discard('async')
- if self.sy in ('async', 'await'):
- self.sy, self.systring = IDENT, self.context.intern_ustring(self.sy)
|