psycopg2/scripts/fix_b.py
Daniele Varrazzo 332acccc6e Replace b('str') with b'str' in Python 3
This avoids an encode() call for each of these constants.

Use a custom 2to3 fixer in setup.py to perform the conversion.
2011-01-13 11:37:40 +00:00

21 lines
594 B
Python

"""Fixer to change b('string') into b'string'."""
# Author: Daniele Varrazzo
import token
from lib2to3 import fixer_base
from lib2to3.pytree import Leaf
class FixB(fixer_base.BaseFix):
PATTERN = """
power< wrapper='b' trailer< '(' arg=[any] ')' > rest=any* >
"""
def transform(self, node, results):
arg = results['arg']
wrapper = results["wrapper"]
if len(arg) == 1 and arg[0].type == token.STRING:
b = Leaf(token.STRING, 'b' + arg[0].value, prefix=wrapper.prefix)
node.children = [ b ] + results['rest']