
c'oTx                 @   sk   d  Z  d d l m Z d d l m Z m Z m Z d d l m Z d d   Z	 Gd d   d e j
  Z d	 S)
z8
Fixer for "class Foo: ..." -> "class Foo(object): ..."
    )
fixer_base)LParenRParenName)touch_import_topc             C   s@   |  j  | t    |  j  | t d   |  j  | t    d  S)Nobject)Zinsert_childr   r   r   )nodeidx r
   B/usr/lib/python3/dist-packages/libpasteurize/fixes/fix_newstyle.pyinsert_object   s    r   c               @   s"   e  Z d  Z d Z d d   Z d S)FixNewstylez0classdef< 'class' NAME ['(' ')'] colon=':' any >c             C   s   | d } | j  j |  } | j  | d j d k rn | j  | d j d k rn | j  | d |  =| d 8} t | |  t d d |  d  S)Ncolon   (   )builtinsr   )Zchildrenindexvaluer   r   )selfr   Zresultsr   r	   r
   r
   r   	transform   s    

zFixNewstyle.transformN)__name__
__module____qualname__ZPATTERNr   r
   r
   r
   r   r      s   r   N)__doc__Zlib2to3r   Zlib2to3.fixer_utilr   r   r   Zlibfuturize.fixer_utilr   r   ZBaseFixr   r
   r
   r
   r   <module>   s
   