mirror of
https://github.com/github/codeql.git
synced 2026-01-03 09:40:17 +01:00
35 lines
1013 B
Python
35 lines
1013 B
Python
import os.path
|
|
|
|
from .parse import Parser
|
|
from .emit import Emitter
|
|
|
|
class Extractor(object):
|
|
|
|
def __init__(self, trap_folder, src_archive=None):
|
|
self.parser = Parser()
|
|
self.emitter = Emitter(trap_folder)
|
|
self.src_archive = src_archive
|
|
|
|
def _walk(self, path):
|
|
for dirpath, _, filenames in os.walk(path):
|
|
for filename in filenames:
|
|
if filename.endswith(".thrift"):
|
|
yield os.path.join(dirpath, filename)
|
|
|
|
def extract_files(self, files):
|
|
for file in files:
|
|
self.extract_file(file)
|
|
|
|
def extract_folder(self, path):
|
|
for file in self._walk(path):
|
|
self.extract_file(file)
|
|
|
|
def extract_file(self, file):
|
|
with open(file, "rb") as fd:
|
|
bytes_source = fd.read()
|
|
src = bytes_source.decode('utf-8')
|
|
tree = self.parser.parse(src)
|
|
self.emitter.emit(file, tree)
|
|
if self.src_archive:
|
|
self.src_archive.write(file, bytes_source)
|