ralsina/tartrazine
1
0
Fork 0
mirror of https://github.com/ralsina/tartrazine.git synced 2025-06-18 22:23:07 -03:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

py: expose highest-level enry.language()

Browse Source 
Signed-off-by: Alexander Bezzubov <bzz@apache.org>
This commit is contained in:
Alexander Bezzubov
2019-11-17 15:13:41 +01:00
parent 1d23012ae6
commit 35575d0a3e
3 changed files with 60 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files

72
python/enry.py
View File

@ -4,73 +4,85 @@ Python library calling enry Go implementation trough cFFI (API, out-of-line) and
from _c_enry import ffi, lib
## Helpers
## cgo -> ffi helpers
def py_bytes_to_go(py_bytes: bytes):
c_bytes = ffi.new("char[]", len(py_bytes))
go_slice = ffi.new("GoSlice *", [c_bytes, len(py_bytes), len(py_bytes)])
return (go_slice[0], c_bytes)
def py_str_to_go(py_str: str):
str_bytes = py_str.encode()
c_str = ffi.new("char[]", str_bytes)
go_str = ffi.new("_GoString_ *", [c_str, len(str_bytes)])
return (go_str[0], c_str)
def go_str_to_py(go_str):
def go_str_to_py(go_str: str):
str_len = go_str.n
if str_len > 0:
return ffi.unpack(go_str.p, go_str.n).decode()
return ""
def py_str_to_go(py_str):
str_bytes = py_str.encode()
c_str = ffi.new("char[]", str_bytes)
go_str = ffi.new("_GoString_ *", [c_str, len(str_bytes)])
return go_str[0]
def go_bool_to_py(go_bool):
def go_bool_to_py(go_bool: bool):
return go_bool == 1
## API
## API, TODO(bzz): add docstrings
def language(filename: str, content: bytes) -> str:
fName, c_str = py_str_to_go(filename)
fContent, c_bytes = py_bytes_to_go(content)
guess = lib.GetLanguage(fName, fContent)
lang = go_str_to_py(guess)
return lang
def language_by_extension(filename: str) -> str:
fName = py_str_to_go(filename)
fName, c_str = py_str_to_go(filename)
guess = lib.GetLanguageByExtension(fName)
lang = go_str_to_py(guess.r0)
return lang
def language_by_filename(filename: str) -> str:
fName = py_str_to_go(filename)
fName, c_str = py_str_to_go(filename)
guess = lib.GetLanguageByFilename(fName)
lang = go_str_to_py(guess.r0)
return lang
def is_vendor(filename: str) -> bool:
fName = py_str_to_go(filename)
fName, c_str = py_str_to_go(filename)
guess = lib.IsVendor(fName)
return go_bool_to_py(guess)
## Tests
from collections import namedtuple
def main():
TestFile = namedtuple("TestFile", "name, content, lang")
files = [
"Parse.hs", "some.cpp", "and.go", "type.h", ".bashrc", ".gitignore"
TestFile("Parse.hs", b"", "Haskell"), TestFile("some.cpp", b"", "C++"),
TestFile("orand.go", b"", "Go"), TestFile("type.h", b"", "C"),
TestFile(".bashrc", b"", "Shell"), TestFile(".gitignore", b"", "Ignore List")
]
print("strategy: extension")
for filename in files:
lang = language_by_extension(filename)
print("file: {:10s} language: '{}'".format(filename, lang))
print("\nstrategy: extension")
for f in files:
lang = language_by_extension(f.name)
print("\tfile: {:10s} language: '{}'".format(f.name, lang))
print("\nstrategy: filename")
for filename in files:
lang = language_by_filename(filename)
print("file: {:10s} language: '{}'".format(filename, lang))
for f in files:
lang = language_by_filename(f.name)
print("\tfile: {:10s} language: '{}'".format(f.name, lang))
print("\ncheck: is vendor?")
for filename in files:
vendor = is_vendor(filename)
print("file: {:10s} vendor: '{}'".format(filename, vendor))
for f in files:
vendor = is_vendor(f.name)
print("\tfile: {:10s} vendor: '{}'".format(f.name, vendor))
print("\nstrategy: all")
for f in files:
lang = language(f.name, f.content)
print("\tfile: {:10s} language: '{}'".format(f.name, lang))
assert lang == f.lang, "Expected '{}' but got '{}'".format(f.lang, lang)
if __name__ == "__main__":
main()