e<4dZddlZddlmZddlmZmZmZmZm Z m Z m Z ddl m Z mZejdZGdd eZGd d e ZGd d eZeZeZejZejZejZejZejZdS)zJRead and write notebooks as regular .py files. Authors: * Brian Granger N)List)nbformatnbformat_minor new_code_cellnew_heading_cell new_notebook new_text_cell new_worksheet)NotebookReaderNotebookWriterz^#.*coding[:=]\s*([-\w.]+)ceZdZdZdS) PyReaderErrorz%An error raised for a pyreader error.N)__name__ __module__ __qualname____doc__0lib/python3.11/site-packages/nbformat/v3/nbpy.pyrr(s//Drrc0eZdZdZdZdZdZdZdZdS)PyReaderzA python notebook reader.c |j|fi|S)Convert a string to a notebook) to_notebook)selfskwargss rreadszPyReader.reads1st,,V,,,rc |}g}g}i}d}|D]}|dst|r3|dr-|j||fi|}|||d}g}i}u|dr-|j||fi|}|||d}g}i}|dr-|j||fi|}|||d}g}i}|d s|d r.|j||fi|}|||d }g}i}Q|d rr|j||fi|}|||g}t jd |} | +d}i}t| d|d<d}i}g}|||r3|dkr-|||}|||t|} t| g} | S)rcodecellz # # N # htmlcell# markdowncell # z# rawcellz# \d)> headingcelllevel)cells) worksheets) splitlines startswith_encoding_declaration_rematchnew_cellappendreintgroupr r ) rrrlinesr+ cell_linesstatelinecellmwsnbs rrzPyReader.to_notebook5s  " / (/ (D~... (2J2P2PQU2V2V. (00, ($t}UJAA&AA#LL&&&" 00% ($t}UJAA&AA#LL&&&" !344 ($t}UJAA&AA#LL&&&& // (4??CV3W3W ($t}UJAA&AA#LL&&&! !122 ($t}UJAA&AA#LL&&&!#JHCTJJ=)EF&)!'''*:*:&;&;F7OO&EF!#JJ!!$''''  #%:--== 33D T"""  ' ' ' bT * * * rc :|dkr>d|}|d}|rt|SdS|dkr*||}|rt d|SdS|dkr*||}|rt d|SdS|d kr*||}|rt d |SdS|d kr>||}|d d }|rt ||SdSdS)zCreate a new cell.r! )inputr$html)sourcer&markdownr(rawr)r*r)rBr*N)joinstripr_remove_commentsr getr)rr8r6rinput_textr*s rr1zPyReader.new_cellts| J  YYu%%F\\$''F 3$62222 3 3 j ((//D :$VD9999 : : n $ $((//D >$Z==== > > i  ((//D 9$U48888 9 9 m # #((//DJJw**E B't5AAAA $ # B Brcg}|D]J}|dr||dd5||Kd|}|d}|S)N#r?)r.r2rErF)rr6 new_linesr9rJs rrGzPyReader._remove_commentss  ' 'Ds## '  abb****  &&&&yy##zz$ rc#Kt|dkr|dVtddl}d|}||}d|jD}t t|dz D]C}d|||||dzdVDd||dddVdS)zSplit lines into code blocks.rrNr?c"g|] }|jdz  S)r)lineno).0xs r z4PyReader.split_lines_into_blocks..s2221!(Q,222r)len StopIterationastrEparsebodyrangerF)rr6rXrBcodestartsis rsplit_lines_into_blocksz PyReader.split_lines_into_blockss u::??(NNN// ! 5!!yy  22 222s6{{Q'' J JA))E&)fQUm";<==CCDII I I I IiifRjll+,,2248888888rN) rrrrrrr1rGr_rrrrr.si##---===~BBB2    9 9 9 9 9rrceZdZdZdZdS)PyWriterzA Python notebook writer.c dg}|dttfzdg|jD]}|jD]}|jdkrk|d}|S|ddg|||dy|jdkru|d }|]|d dg|d |D|d|jd krv|d }|]|d dg|d|D|dz|jdkrv|d }|]|ddg|d|D|d|jdkr|d }|dd}|`|d|zdg|d|D|d|dd |S)zConvert a notebook to a string.z# -*- coding: utf-8 -*-z# %i.%ir\r@Nr"rArBr#cg|]}d|zSz# rrRr9s rrTz#PyWriter.writes..%R%R%RddTk%R%R%RrrCr%cg|]}d|zSrerrfs rrTz#PyWriter.writes..rgrrDr'cg|]}d|zSrerrfs rrTz#PyWriter.writes..rgrheadingr*rz# cg|]}d|zSrerrfs rrTz#PyWriter.writes..rgrr?) extendrrr,r+ cell_typerHr-r2rE)rr=rr6r<r:rIr*s rwriteszPyWriter.writess*+ .(N1KK    - ) )B ) )>V++!XXg..F) nb%9::: V%6%6%8%8999 R(((^v--!XXh//F) nb%9::: %R%Rf>O>O>Q>Q%R%R%RSSS R(((^z11!XXh//F) &8"%=>>> %R%Rf>O>O>Q>Q%R%R%RSSS R(((^u,,!XXh//F) mR%8999 %R%Rf>O>O>Q>Q%R%R%RSSS R(((^y00!XXh//F HHWa00E) &@5&H"%MNNN %R%Rf>O>O>Q>Q%R%R%RSSS R(((? )@  RyyrN)rrrrrnrrrraras)##+ + + + + rra)rr3typingrnbbaserrrrr r r rwbaser r compiler/ Exceptionrrra_reader_writerrreadrwriternrrrrxs}$ 32222222 &2:&CDD     I   v9v9v9v9v9~v9v9v9r. . . . . ~. . . b (** (** |!   r