Add a more API focused grammar.

This makes it so we don't have to expose all the details of a pgen grammar to the user.
This commit is contained in:
Dave Halter
2017-05-21 17:02:15 -04:00
parent 973f7c5f61
commit 23d1470618
8 changed files with 81 additions and 57 deletions

View File

@@ -1,11 +1,57 @@
import os
from parso._compatibility import FileNotFoundError
from parso.parser import ParserSyntaxError
from parso.pgen2.pgen import generate_grammar
from parso import python
from parso import grammar
from parso.tokenize import generate_tokens
from parso.python.parser import Parser
__version__ = '0.0.1'
_loaded_grammars = {}
def parse(grammar, code):
raise NotImplementedError
Parser(grammar, code)
def create_grammar(text, tokenizer=generate_tokens):
"""
:param text: A BNF representation of your grammar.
"""
return grammar.Grammar(text, tokenizer, parser=None)
def load_python_grammar(version=None):
"""
Loads a Python grammar. The default version is always the latest.
If you need support for a specific version, please use e.g.
`version='3.3'`.
"""
if version is None:
version = '3.6'
if version in ('3.2', '3.3'):
version = '3.4'
elif version == '2.6':
version = '2.7'
file = 'python/grammar' + version + '.txt'
global _loaded_grammars
path = os.path.join(os.path.dirname(__file__), file)
try:
return _loaded_grammars[path]
except KeyError:
try:
with open(path) as f:
bnf_text = f.read()
grammar = create_grammar(bnf_text)
return _loaded_grammars.setdefault(path, grammar)
except FileNotFoundError:
# Just load the default if the file does not exist.
return load_python_grammar()