a _=ics@sdZddlZddlZddlZddlZddlZddlZddlZedddgZ ddZ dd Z d d Z d d Z ddZddZdS)z-A module to support operations on ipynb filesNCodeLine cell_numbercodecCs.|ddko,|do,t|ddd S)z+Checks if the cell consists of Python code.Z cell_typersourcerz%%six ensure_str startswith)cellr u/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/tensorflow/tools/compatibility/ipynb.py is_pythons  r cCstdt|\}}dd|D}tjddd}|d||\}}} } |rz|rzt||t| d} t | |nt dd | |} | | ||} | }Wd n1s0Yt|j|| | |fS) z;The function where we inject the support for ipynb upgrade.z,Extracting code lines from original notebookcSsg|] }|jqSr )r).0clr r r (z process_file..wF)delete z&Was not able to process the file: %s N)print _get_codetempfileNamedTemporaryFileZupdate_string_pastajoin_update_notebookrrsplitjsondump SyntaxErrorZ _format_logshutilmovename)Z in_filenameZ out_filenameZupgraderraw_codenotebookZ raw_lines temp_fileZprocessed_fileZnew_file_contentlogZprocess_errors new_notebookZfiles_processedZ report_texterrorsr r r process_file$s(   "r)cCs$|D]}t||rdSqdS)aChecks if the cell has magic, that is not Python-based. Args: code_line: A line of Python code magic_list: A list of jupyter "magic" exceptions Returns: If the line jupyter "magic" line, not Python line >>> skip_magic('!ls -laF', ['%', '!', '?']) True TFr) code_lineZ magic_listmagicr r r skip_magicBsr,cCs td|S)zChecks if a line was split with `\`. Args: code_line: A line of Python code Returns: If the line was split with `\` >>> skip_magic("!gcloud ml-engine models create ${MODEL} \\\n") True z\\\s*\n$)research)r*r r r check_line_splitWs r/c Csg}t|}t|}Wdn1s,0Yd}|dD]}t|rB|d}d}t|D]\}} t| gds||rdt| } t| }|rt| }|t |dkrt|  d rt|  d d } | t |t|  d d qb|d7}qB||fS) z5Loads the ipynb file and returns a list of CodeLines.NrcellsrF)%!?###!!!r###===)openrloadr enumerater,rrr/lenendswithreplaceappendrrstrip) Z input_filer#in_filer$Z cell_indexr Z cell_linesZ is_line_splitZline_idxr*r r r rgs: (  rcst|}t|tks2Jdt|td|dD]\}t|sLq>fddt|D}fdd|D}d|dd d d|d <d 7q>|S) z)Updates notebook, once migration is done.zCThe lengths of input and converted files are not the same: {} vs {}rr0csg|]\}}|jkr|qSr )r)ridxr*) code_cell_idxr r rs z$_update_notebook..csg|] }|qSr r )rr@)updated_code_linesr r rrrr4rr6rr5)copydeepcopyr:formatr r9rr<)Zoriginal_notebookZoriginal_raw_linesrBr'r Zapplicable_linesZnew_coder )rArBr rs&     r)__doc__ collectionsrCrr-r rr namedtuplerr r)r,r/rrr r r r s/