protected virtual List <string> GetScrapeArguments(IPythonInterpreterFactory factory)
{
var args = new List <string> {
"-B", "-E"
};
ModulePath mp = AstPythonInterpreterFactory.FindModule(factory, _filePath);
if (string.IsNullOrEmpty(mp.FullName))
{
return(null);
}
var sm = PythonToolsInstallPath.TryGetFile("scrape_module.py", GetType().Assembly);
if (!File.Exists(sm))
{
return(null);
}
args.Add(sm);
args.Add("-u8");
args.Add(mp.ModuleName);
args.Add(mp.LibraryPath);
return(args);
}
public void Imported(IModuleContext context)
{
if (_scraped)
{
return;
}
_scraped = true;
var interp = context as AstPythonInterpreter;
var fact = interp?.Factory;
if (fact == null || !File.Exists(fact.Configuration.InterpreterPath))
{
return;
}
ModulePath mp = AstPythonInterpreterFactory.FindModule(fact, _filePath);
if (string.IsNullOrEmpty(mp.FullName))
{
return;
}
var sm = PythonToolsInstallPath.TryGetFile("scrape_module.py", GetType().Assembly);
if (!File.Exists(sm))
{
return;
}
Stream code = null;
using (var p = ProcessOutput.RunHiddenAndCapture(
fact.Configuration.InterpreterPath, "-E", sm, mp.LibraryPath, mp.ModuleName
)) {
p.Wait();
if (p.ExitCode == 0)
{
var ms = new MemoryStream();
code = ms;
using (var sw = new StreamWriter(ms, Encoding.UTF8, 4096, true)) {
foreach (var line in p.StandardOutputLines)
{
sw.WriteLine(line);
}
}
}
}
if (code == null)
{
return;
}
PythonAst ast;
code.Seek(0, SeekOrigin.Begin);
using (var sr = new StreamReader(code, Encoding.UTF8))
using (var parser = Parser.CreateParser(sr, fact.GetLanguageVersion())) {
ast = parser.ParseFile();
}
lock (_members) {
var walker = new AstAnalysisWalker(interp, ast, this, _filePath, _members, false);
ast.Walk(walker);
}
}