a
    Xä%bx  ã                   @   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
   úp/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/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
   