mirror of
https://github.com/psycopg/psycopg2.git
synced 2024-11-29 20:23:45 +03:00
b075017ad9
Pulled the master branch from of Daniele's psycopg branch on github and merged the changes.
21 lines
594 B
Python
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']
|
|
|