:vjdZddlmZddlZddlZddlZddlmZddlm Z ddl m Z ddl m Z mZdd lmZdd lmZdd lmZd ZGd de eZGdde eZGdde eZy)zA workspace management CLI) annotationsN)Path)Any) JupyterApp)BoolUnicode) __version__) LabConfig)WorkspacesManagerdefaultceZdZdZeZdZedddiidfdddiidfZe d dd Z e d dd Z dfd Z ddZ xZS)WorkspaceListAppzAn app to list workspaces.a Print all the workspaces available If '--json' flag is passed in, a single 'json' object is printed. If '--jsonlines' flag is passed in, 'json' object of each workspace separated by a new line is printed. If nothing is passed in, workspace ids list is printed. jsonlinesTz+Produce machine-readable JSON Lines output.jsonz,Produce machine-readable JSON object output.)rrFzIf True, the output will be a newline-delimited JSON (see https://jsonlines.org/) of objects, one per JupyterLab workspace, each with the details of the relevant workspace)confighelpzUIf True, each line of output will be a JSON object with the details of the workspace.cXt||i|t|j|_yzInitialize the app.Nsuper initializer workspaces_dirmanagerselfargskwargs __class__s q/workspace/moltex-experiments/.pixi/envs/default/lib/python3.12/site-packages/jupyterlab_server/workspaces_app.pyrzWorkspaceListApp.initialize@' D+F+()<)<= c"|jj}|jr&|D] }tt j |"y|jrtt j |y|D]}t|ddy)Start the app.metadataidN)rlist_workspacesrprintrdumps)r workspaces workspaces r startzWorkspaceListApp.startEsj\\113 >>' djj+,( YY $**Z( )' i +D12(r"rrrrreturnNoner.r/)__name__ __module__ __qualname____doc__r version descriptiondictflagsrrrrr, __classcell__rs@r rrs$GK  +t!4 5 :  &$ 0 ; E  \ I   (  D> 3r"rc4eZdZdZeZdZdfd ZddZxZ S)WorkspaceExportAppzA workspace export app.a Export a JupyterLab workspace If no arguments are passed in, this command will export the default workspace. If a workspace name is passed in, this command will export that workspace. If no workspace is found, this command will export an empty workspace. cXt||i|t|j|_yrrrs r rzWorkspaceExportApp.initialize_r!r"c t|jdkDr&tjd|j d|jst n|jd} |j j|}ttj|y#t$rN|jjtjttt|YywxYw)r$r z6Too many arguments were provided for workspace export.r)r&)datar%N)len extra_argswarningswarnexitDEFAULT_WORKSPACErloadr(rr) Exceptionlogerrorr7)rrawr+s r r,zWorkspaceExportApp.startds t ! # MMR S IIaL'+DOOA Qr"r<cdeZdZdZeZdZeddddZddiZ d fd Z dd Z dd Z dd Z xZS)WorkspaceImportAppzA workspace import app.z Import a JupyterLab workspace This command will import a workspace from a JSON file. The format of the file must be the same as what the export functionality emits. NTz Workspace name. If given, the workspace ID in the imported file will be replaced with a new ID pointing to this workspace name. )r allow_nonernamez!WorkspaceImportApp.workspace_namecXt||i|t|j|_yrrrs r rzWorkspaceImportApp.initializer!r"ct|jdk7r,|jjd|j d|j 5} |j |}ddd |jjddtj|}|jjdy#t$rB}|jjd|j||j dYd}~d}~wwxYw#1swYxYw#t$r7}|jjd||j dYd}~d}~wwxYw) r$r z.One argument is required for workspace import.z%s is not a valid workspace: %sNr%r&z#Workspace could not be exported: %szSaved workspace: %s) r@rArHinforD _smart_open _validaterGrNrsaverr))rfidr+eworkspace_paths r r,zWorkspaceImportApp.starts  t 1 $ HHMMJ K IIaL    3  NN3/   !\\..y/DT/JDJJW`LabN  +^<  @#((AN !     HHMM@! D IIaLL sHDC15D D 8D D DDD E&-EEc|jd}|dk(rtjSt|j }|j s-|j jd||jd|jdS)Nr-z%s does not exist.r zutf-8)encoding) rAsysstdinrresolveexistsrHrQrDopen)r file_name file_paths r rRzWorkspaceImportApp._smart_opensoOOA&  99 O++- ! HHMM. : IIaL~~w~//r"ctj|}d|vr d}t||j|jrd|ji|d<|Sd|dvr d}t||S)Nr?zThe `data` field is missing.r&r%z(The `id` field is missing in `metadata`.)rrFrGworkspace_name)rr?r+msgs r rSzWorkspaceImportApp._validates}IIdO  "0CC.     *t/B/B%)4+>+>$?Ij !  :. . =* 0r"rL)r4 __future__rrr[rBpathlibrtypingrjupyter_core.applicationr traitletsrr_versionr rr workspaces_handlerr rErr<rLr"r rnsc!" /#!163z963rQYQ@NYNr"