summaryrefslogtreecommitdiffstats
path: root/Lib/lib2to3/fixes/fix_buffer.py
blob: 13168d6b2e4b17706bfbc772d9846dc3efe8029d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
# Copyright 2007 Google, Inc. All Rights Reserved.
# Licensed to PSF under a Contributor Agreement.

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

# Local imports
from . import basefix
from .util import Name


class FixBuffer(basefix.BaseFix):

    explicit = True # The user must ask for this fixer

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

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