__init__.py 文件源码

def augAssignSpecialFunction(cv, orig):
    if (not isinstance(cv, DeleteVector)) and (not isStatement(cv.oldSubtree)) and \
        (childHasTag(cv.oldSubtree, "augAssignVal") or childHasTag(cv.oldSubtree, "augAssignBinOp")):
        # First, create the oldTree and newTree in full
        cvCopy = cv.deepcopy()
        cvCopy.start = deepcopy(cv.start)
        newTree = cvCopy.applyChange(caller="augAssignSpecialFunction")

        # This should be in an augassign, move up in the tree until we reach it.
        spot = cv.oldSubtree
        cvCopy = cv
        i = 0
        while type(spot) not in [ast.Assign, ast.AugAssign] and len(cvCopy.path) > i:
            i += 1
            cvCopy = cv.deepcopy()
            cvCopy.path = cv.path[i:]
            spot = deepcopy(cvCopy.traverseTree(cv.start))

        # Double check to make sure this is actually still an augassign
        if type(spot) in [ast.Assign, ast.AugAssign] and hasattr(spot, "global_id"):
            newCv = cv.deepcopy()
            newCv.path = cv.path[i+1:]
            newCv.oldSubtree = spot
            # find the new spot
            cvCopy = cv.deepcopy()
            cvCopy.path = cv.path[i:]
            newSpot = cvCopy.traverseTree(newTree)
            if type(newSpot) == type(spot):
                # Don't do special things when they aren't needed
                if type(newSpot) == ast.Assign:
                    if compareASTs(newSpot.targets, spot.targets, checkEquality=True) == 0:
                        # If the two have the same targets and are both binary operations with the target as the left value... 
                        # just change the value
                        if type(newSpot.value) == type(spot.value) == ast.BinOp:
                            if compareASTs(spot.targets[0], spot.value.left, checkEquality=True) == 0 and \
                                compareASTs(newSpot.targets[0], newSpot.value.left, checkEquality=True) == 0:
                                # we just want to change the values
                                return ChangeVector([("right", "Binary Operation"), ("value", "Assign")] + newCv.path, 
                                                    spot.value.right, newSpot.value.right, newCv.start)
                    elif compareASTs(newSpot.value, spot.value, checkEquality=True) == 0:
                        return cv
                    else:
                        log("Assign", "bug")
                elif type(newSpot) == ast.AugAssign:
                    diffCount = 0
                    if compareASTs(newSpot.op, spot.op, checkEquality=True) != 0:
                        diffCount += 1
                    if compareASTs(newSpot.target, spot.target, checkEquality=True) != 0:
                        diffCount += 1
                    if compareASTs(newSpot.value, spot.value, checkEquality=True) != 0:
                        diffCount += 1
                    if diffCount == 1:
                        return cv
                    else:
                        log("AugAssign: " + str(diffCount), "bug")
            else:
                log("Mismatched types: " + str(type(newSpot)) + "," + str(type(spot)), "bug")
            return ChangeVector(newCv.path, spot, newSpot, start=newCv.start)
    return cv