Staging
v0.5.1
https://github.com/python/cpython
Raw File
Tip revision: dcd4c4f9c9f9fb75ad4c1f138a285fbe19a9cb61 authored by Ɓukasz Langa on 23 March 2020, 16:19:13 UTC
Python 3.9.0a5
Tip revision: dcd4c4f
fix_buffer.py
# Copyright 2007 Google, Inc. All Rights Reserved.
# Licensed to PSF under a Contributor Agreement.

"""Fixer that changes buffer(...) into memoryview(...)."""

# Local imports
from .. import fixer_base
from ..fixer_util import Name


class FixBuffer(fixer_base.BaseFix):
    BM_compatible = True

    explicit = True # The user must ask for this fixer

    PATTERN = """
              power< name='buffer' trailer< '(' [any] ')' > any* >
              """

    def transform(self, node, results):
        name = results["name"]
        name.replace(Name("memoryview", prefix=name.prefix))
back to top