o ^iJ'@s>dZddlZddlZddlZddlmmZddlm Z ddl m Z m Z m Z ddlmZmZddlmZmZddlmZddlmZdd lmZdd lmZmZdd lmZd Zd ZdZ de!fddZ"e ddGdddZ#Gdddej$Z%Gddde%Z&Gddde%Z'GdddeZ(GdddeZ)d Z*d!Z+dS)"z%Announcements handler for JupyterLab.N) Awaitable)asdict dataclassfield)datetimetimezone)OptionalUnion) APIHandler) translator)parse) httpclientweb) __version__z%Y-%m-%dT%H:%M:%S%zz%https://pypi.org/pypi/jupyterlab/jsonz7https://github.com/jupyterlab/jupyterlab/releases/tag/vdt_strcCst|dS)Ni)r fromisoformat timestamp)rr\/home/ubuntu/hpml_nyu/venv/lib/python3.10/site-packages/jupyterlab/handlers/announcements.pyformat_datetimesrT)frozenc@sbeZdZUdZeed<eed<eed<dZeed<ee dZ e eefed<ee dZ e ed <d S) NotificationuKNotification Attributes: createdAt: Creation date message: Notification message modifiedAt: Modification date type: Notification type — ["default", "error", "info", "success", "warning"] link: Notification link button as a tuple (label, URL) options: Notification options createdAtmessage modifiedAtdefaulttype)default_factorylinkoptionsN) __name__ __module__ __qualname____doc__float__annotations__strrrtuplerdictrrrrrrs   rc @sTeZdZdZdeddfddZejdee dee ee eeffffddZ dS) CheckForUpdateABCzAbstract class to check for update. Args: version: Current JupyterLab version Attributes: version - str: Current JupyterLab version logger - logging.Logger: Server logger versionreturnNcCs ||_dSN)r*)selfr*rrr__init__?s zCheckForUpdateABC.__init__csd}t|)aGet the notification message if a new version is available. Returns: None if there is not update. or the notification message or the notification message and a tuple(label, URL link) for the user to get more information z-CheckForUpdateABC.__call__ is not implemented)NotImplementedError)r-msgrrr__call__Bs zCheckForUpdateABC.__call__) r r!r"r#r&r.abcabstractmethodrr r'r1rrrrr)4s  2r)c@s2eZdZdZdeeeeeefffddZdS)CheckForUpdatezDefault class to check for update. Args: version: Current JupyterLab version Attributes: version - str: Current JupyterLab version logger - logging.Logger: Server logger r+c st}z|jtddidIdH}t|jd}|d}Wnty;}z|j j d|dWYd}~dSd}~wwt |j t |kr^t d }|d |d |d t|ffSdS) Get the notification message if a new version is available. Returns: None if there is no update. or the notification message or the notification message and a tuple(label, URL link) for the user to get more information Content-Typezapplication/jsonheadersNinfor*zFailed to get latest versionexc_info jupyterlabzA newer version (z) of JupyterLab is available.u Read more…)r AsyncHTTPClientfetchJUPYTERLAB_LAST_RELEASE_URLjsonloadsbodyget Exceptionloggerdebugr r*r load__JUPYTERLAB_RELEASE_URL)r- http_clientresponsedata last_versionetransrrrr1Zs(   zCheckForUpdate.__call__N)r r!r"r#rr'r&r1rrrrr4Os& r4c@s"eZdZdZdedfddZdS)NeverCheckForUpdatea7Check update version that does nothing. This is provided for administrators that want to turn off requesting external resources. Args: version: Current JupyterLab version Attributes: version - str: Current JupyterLab version logger - logging.Logger: Server logger r+NcsdS)r5Nr)r-rrrr1szNeverCheckForUpdate.__call__)r r!r"r#rr1rrrrrPxs rPc@eZdZdZ d deeddffdd ZejddZ Z S) CheckForUpdateHandlerziCheck for Updates API handler. Args: update_check: The class checking for a new version Nupdate_checkerr+cs.t|dur ttn||_|j|j_dSr,)super initializerPrrSlogrE)r-rS __class__rrrUs z CheckForUpdateHandler.initializec sd}|IdH}|r=t|tr|dfn|\}}tjtjdd}t |  }t |||d|d|dgdid }| d |td |durMdnt|idS) zCheck for updates. Response: { "notification": Optional[Notification] } Nr)tzg@@r9rLupdateidtagsrrrrrr notification)rS isinstancer&rnowrutcrhashlibsha1encode hexdigestr set_statusfinishr@dumpsr)r-r`outrrrbhash_rrrrCs& zCheckForUpdateHandler.getr,) r r!r"r#rr4rUr authenticatedrC __classcell__rrrWrrRs rRcrQ) NewsHandlerzSNews API handler. Args: news_url: The Atom feed to fetch for news Nnews_urlr+cst||_dSr,)rTrUrp)r-rprWrrrUs  zNewsHandler.initializec sg}t}|jdurmtdddiD] \}}t||qz*|j|jddidIdH}t |j }fdd }t || d }| |Wntyl} z|jjd |j| d WYd} ~ nd} ~ ww|d |tdtt t|idS)zpGet the news. Response: { "news": List[Notification] } Nr<atomzhttp://www.w3.org/2005/Atomr6zapplication/atom+xmlr7c sddtdttdtffdd }|d}|d}|d}|d |}|d d d }d }t|dkrHttdd|}|rC|dn|d} n t|dkrR|dnd} | dur]| dnd} |rhd||gn|} t|} t|} t | | | d| dur}dn d| fd|dgdid}|S)Nattrrr+cs>d|}|dur|jS|dur|Sd|}t|)Nzatom:z7atom feed entry does not contain a required attribute: )findtextKeyError)rrr node_itemerror_m)nodexml_namespacesrr get_xml_textsz:NewsHandler.get..build_entry..get_xml_texttitler\updated publishedsummary)rz atom:linkcSs|ddkS)Nrel alternate)rC)elemrrrsz6NewsHandler.get..build_entry..rhref r9zOpen full postrLnewsr[r^r,) r&rfindalllenlistfilterrCjoinrrrH)rxrz entry_titleentry_id entry_updatedentry_published entry_summarylinksr link_node entry_linkr modified_at created_atr`rOry)rxr build_entrysB"     z$NewsHandler.get..build_entryz atom:entryz,Failed to get announcements from Atom feed: r:r_r)r r=rpr rGitemsETregister_namespacer> fromstringrBmaprextendrDrVrFrhrir@rjrr) r-rrJkeyspecrKtreerentriesrNrrrrCs4    3  "zNewsHandler.getr,) r r!r"r#rr&rUrrmrCrnrrrWrrosroz /lab/api/newsz/lab/api/update),r#r2rdr@xml.etree.ElementTreeetree ElementTreercollections.abcr dataclassesrrrrrtypingrr jupyter_server.base.handlersr #jupyterlab_server.translation_utilsr packaging.versionr tornador rjupyterlab._versionrISO8601_FORMATr?rIr&rrABCr)r4rPrRronews_handler_pathcheck_update_handler_pathrrrrs6     ).i