B 3\`+@sddlZddlZddlmZddlmZddlmZddlm Z ddl m Z ddl mZddlmZejjZdd d ZGd ddeZGdddZdddZedddZddZddZddZdS)N) ActionChains)By)Keys) WebDriverWait)expected_conditions) WebElement)contextmanager FcCsHt||}|r"|rtj}q4tj}n|r.tj}ntj}||tj|fS)N) rECZvisibility_of_element_locatedZpresence_of_element_locatedZ"visibility_of_all_elements_locatedZ presence_of_all_elements_locateduntilrZ CSS_SELECTOR)browserselectorZtimeoutZvisiblesinglewaitZ conditionalrd?Zd@dAZdBdCZ e!dPdEdFZ"d-S)QNotebookcCs||_|dS)N)r #disable_autosave_and_onbeforeunload)rr rrrr&szNotebook.__init__cCs t|jS)N)lencells)rrrr__len__*szNotebook.__len__cCs |j|S)N)r)rkeyrrr __getitem__-szNotebook.__getitem__cCst|tr|j||dddS)NF)indexcontentrender) isinstanceint edit_cell)rr itemrrr __setitem__0s zNotebook.__setitem__cCsdd|jDS)Ncss|] }|VqdS)Nr).0cellrrr <sz$Notebook.__iter__..)r)rrrr__iter__;szNotebook.__iter__cCs |jdS)Nbody)r Zfind_element_by_tag_name)rrrrr.>sz Notebook.bodycCs |jdS)z7Gets all cells once they are visible. r+)r Zfind_elements_by_class_name)rrrrrBszNotebook.cellscCs ||jS)N)r" current_cell)rrrr current_indexIszNotebook.current_indexcCs |j|S)N)rr")rr+rrrr"MszNotebook.indexcCs|jd|jddS)zDisable request to save before closing window and autosave. This is most easily done by using js directly. zwindow.onbeforeunload = null;z)Jupyter.notebook.set_autosave_interval(0)N)r execute_script)rrrrrPs z,Notebook.disable_autosave_and_onbeforeunloadcCs|jtj|jddS)zHChanges us into command mode on currently focused cell zireturn Jupyter.notebook.handle_command_mode(Jupyter.notebook.get_cell(Jupyter.notebook.get_edit_index()))N)r. send_keysrZESCAPEr r1)rrrrto_command_modeXszNotebook.to_command_modercCs$|j|}||||_dS)N)rclickr3r/)rr"r+rrr focus_cellas zNotebook.focus_cellrcCsv||||jdt|jddd|jd|jd||jd||jddS) Nfz#find-and-replaceT)rZfindreplace_allcells_btnZfindreplace_find_inpZfindreplace_replace_inpZfindreplace_replaceall_btn)r5r3r.r2rr Zfind_element_by_idr4)rr"Zfind_txtZ replace_txtrrrfind_and_replacegs  zNotebook.find_and_replacecodecCs~|||j|}|dkr*|jdn:|dkr@|jdn$|dkrV|jdntd||||||jS)Nmarkdownmrawrr8yz<{} is not a valid cell type,use 'code', 'markdown', or 'raw')r5rr/r2rformatwait_for_stale_cell)rr" cell_typer+rrrconvert_cell_typeqs    zNotebook.convert_cell_typecCs t|jd}|t|}dS)z This is needed to switch a cell's mode and refocus it, or to render it. Warning: there is currently no way to do this when changing between markdown and raw cells. r N)rr r r Z staleness_of)rr+relementrrrr?s zNotebook.wait_for_stale_cellcCsd}|j|S)NzKreturn Jupyter.notebook.get_cells().map(function(c) {return c.get_text();}))r r1)rJSrrrget_cells_contentsszNotebook.get_cells_contentsdiv .CodeMirror-codecCs|j||jS)N)rZfind_element_by_css_selectortext)rr"r rrrget_cell_contentsszNotebook.get_cell_contentscCsd|||}|j|S)Nz.Jupyter.notebook.get_cell({}).metadata.{} = {})r>r r1)rr"r valuerCrrrset_cell_metadataszNotebook.set_cell_metadatacCsd|}|j|S)Nz.return Jupyter.notebook.get_cell({}).cell_type)r>r r1)rr"rCrrr get_cell_types zNotebook.get_cell_typecCsd||}|j|dS)Nz2Jupyter.notebook.get_cell({}).set_input_prompt({}))r>r r1)rr"Z prmpt_valrCrrrset_cell_input_prompts zNotebook.set_cell_input_promptNFcCs|dk r||}|||jtjt|jd|jtjx@t | D]0\}}|dkrr|jtjd|jtj|qRW|r| |j dS)zLSet the contents of a cell to *content*, by cell object or by index Nar ) r"r5r/r2rENTERctrlr ZDELETE enumerate splitlines execute_cellr0)rr+r"r#r$Zline_nolinerrrr's   zNotebook.edit_cellcCsNt|tr|}nt|tr&||}ntd|||jtj tj dS)Nz0execute_cell only accepts a WebElement or an int) r%r&rr" TypeErrorr5r/r2rZCONTROLrN)rZ cell_or_indexr"rrrrRs    zNotebook.execute_cellcCsV|||jd|dkr&|dn|}|r<|j||d|dkrR|j||ddS)Nbr)r"r#r8)r"r@)r5r/r2r'rA)rr"r@r#Z new_indexrrradd_cells  zNotebook.add_cellcCs"||||jddS)NZdd)r5r3r/r2)rr"rrr delete_cells zNotebook.delete_cellTcCs"|j|dd|j|||ddS)Nr9)r@)r"r#r$)rXr')rr"r#r$rrradd_markdown_cellszNotebook.add_markdown_cell)r@cGs@x:t|D].\}}t|tr,|j||dq td|q WdS)N)r@r#z"Don't know how to add cell from %r)rPr%strrXrT)rr@valuesirHrrrappends   zNotebook.appendcCs|j|dS)N)r^)rr\rrrextendszNotebook.extendcCsx|D]}||qWdS)N)rR)rr+rrrrun_alls zNotebook.run_allcCst|j|dS)N)trigger_keystrokesr.)rkeysrrrtrigger_keydownszNotebook.trigger_keydownkernel-python3c Cs,t|ddt||dWdQRX||S)Nz.cell)r ) kernel_name) new_window select_kernel)clsr rerrr new_notebookszNotebook.new_notebook)r)rrr)rr8)rrE)r)NrrF)N)rUr8r)rUrT)rd)#rrrrrr!r)r-propertyr.rr0r"rr3r5r7rAr?rDrGrIrJrKr'rRrXrYrZr^r_r`rc classmethodrirrrrr$s>            rkernel-python3cCsLt|d}|ttjdf}|d|}t||dd}|dS)zCClicks the "new" button and selects a kernel from the options. r znew-dropdown-buttonz#{} aT)rN) rr r Zelement_to_be_clickablerZIDr4r>r)r rerZ new_buttonZkernel_selectorZkernelrrrrgs   rgc#sD|jdVtfdd|jD}|||dk r@t||dS)uContextmanager for switching to & waiting for a window created. This context manager gives you the ability to create a new window inside the created context and it will switch you to that new window. If you know a CSS selector that can be expected to appear on the window, then this utility can wait on that selector appearing on the page before releasing the context. Usage example: from notebook.tests.selenium.utils import new_window, Notebook ⋮ # something that creates a browser object with new_window(browser, selector=".cell"): select_kernel(browser, kernel_name=kernel_name) nb = Notebook(browser) Nc3s|]}|kr|VqdS)Nr)r*Zwindow)initial_window_handlesrrr, sznew_window..)Zwindow_handlesnextZswitch_to_windowr)r r Znew_window_handler)rmrrfs  rfcCst|d|dS)zSend key combination Shift+(k)zshift-%sN)ra)r krrrshiftsrpcCst|d|dS)zSend key combination Ctrl+(k)z control-%sN)ra)r rorrrrOsrOcGsx|D]}|d}t|dkrdd|ddD}t|}x|D]}||}qDW||dx |dddD]}||}qtW|q|tt|d |dqWdS)a@ Send the keys in sequence to the browser. Handles following key combinations 1. with modifiers eg. 'control-alt-a', 'shift-c' 2. just modifiers eg. 'alt', 'esc' 3. non-modifiers eg. 'abc' Modifiers : http://seleniumhq.github.io/selenium/docs/api/py/webdriver/selenium.webdriver.common.keys.html -rWcSsg|]}tt|qSr)getattrrupper)r*xrrr "sz&trigger_keystrokes..NrUr) splitrrZkey_downr2Zkey_upZperformrrrrs)r rbZeach_key_combinationZmodifiers_keysZacr]rrrras     ra)r FF)rl)N)osZtimeZselenium.webdriverrZselenium.webdriver.common.byrZselenium.webdriver.common.keysrZselenium.webdriver.support.uirZselenium.webdriver.supportrr Z$selenium.webdriver.remote.webelementr contextlibrpathjoinZpjoinr ValueErrorrrrgrfrprOrarrrrs&        D