U
    hx                     @   sN   d Z ddlmZ ddlmZmZmZ ddlmZ dd Z	G dd dej
Zd	S )
z8
Fixer for "class Foo: ..." -> "class Foo(object): ..."
    )
fixer_base)LParenRParenName)touch_import_topc                 C   s0   |  |t  |  |td |  |t  d S )Nobject)Zinsert_childr   r   r   )nodeidx r
   R/var/www/html/venv/lib/python3.8/site-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   sn   |d }|j |}|j |d  jdkrT|j |d  jdkrT|j |d |= |d8 }t|| tdd| d S )Ncolon   (   )builtinsr   )childrenindexvaluer   r   )selfr   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
   