Python: Store step for generators

This commit is contained in:
Rasmus Lerchedahl Petersen
2020-08-25 15:36:26 +02:00
parent ecf3928ed1
commit 56b78a664e
3 changed files with 25 additions and 1 deletions

View File

@@ -358,6 +358,10 @@ predicate comprehensionStoreStep(CfgNode nodeFrom, Content c, CfgNode nodeTo) {
// Dictionary
nodeTo.getNode().getNode().(DictComp).getElt() = nodeFrom.getNode().getNode() and
c instanceof DictionaryElementAnyContent
or
// Generator
nodeTo.getNode().getNode().(GeneratorExp).getElt() = nodeFrom.getNode().getNode() and
c instanceof ListElementContent
}
/**