+j/JddlZddlZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z ddl m Z ddlZddlmZddlmZdZGd d eZd ed ed efdZGddZdS)N)web)parse)args) AppSettings) TypedDictdefaultc8eZdZUeed<eed<eed<eed<dS)FileInfopathsizemodifiedcreatedN)__name__ __module__ __qualname__str__annotations__int0/home/wildlama/comfy/ComfyUI/app/user_manager.pyr r s4 III IIIMMM LLLLLrr r relative_toreturncvtj||tjdtj|t tj|dzt tj|dzdS)N/i)r r rr) osr relpathreplacesepgetsizergetmtimegetctime)r rs r get_file_infor$sk22::263GG%%((..566rw''--455   rc4eZdZdZdZdZd dZdZdZd S) UserManagercntj}t||_tj|sJt j|dtj s(tj dtj dtj rtj | rUt| 5}tj||_ddddS#1swxYwYdSi|_dSddi|_dS)NTexist_okzd****** User settings have been changed to be stored on the server instead of browser storage. ******zg****** For multi-user setups add the --multi-user CLI argument to enable multiple user profiles. ******r ) folder_pathsget_user_directoryrsettingsrr existsmakedirsr multi_userloggingwarningisfileget_users_fileopenjsonloadusers)selfuser_directoryfs r__init__zUserManager.__init__$sc%8::#D)) w~~n-- K K 6 6 6 6? K!GHHH!JKKK ? 0w~~d113344 $--//00.A!%1DJ..................  #Y/DJJJs/DDDcdtjtjdS)Nz users.json)rr joinr*r+)r8s rr3zUserManager.get_users_file7s!w||L;==|LLLrcd}tjrGd|jvr>|jd}|tjrt d|z||jvrt d|z|S)Nr z comfy-userzUnknown user: )rr/headers startswithr*SYSTEM_USER_PREFIXKeyErrorr7)r8requestusers rget_request_user_idzUserManager.get_request_user_id:sz ? 8|w>>?<0D|>?? 8/$6777 tz ! !+d233 3 ruserdataTc|dkrtj}ntd|z||}tj|}|dS|}t j||f|krdS||d|vrtj |}t j t j ||}t j||f|krdSt j |d} |r5t j | st j| d|S)NrFzUnknown filepath type:%rTr()r*r+rBrEget_public_user_directoryrr commonpathrunquoteabspathr=splitr-r.) r8rCfiletype create_dirroot_dirrD user_rootr parents rget_request_user_filepathz%UserManager.get_request_user_filepathGsB :  #688HH3d:;; ;''00 :4@@  4 7  x3 4 4 @ @4  d{{}T**7??27<< 4#@#@AADw!!9d"344 AAtt$$Q'  /bgnnV44 / K . . . . rc<|}|std|tjrtdt jdd|}|tjrtd|dzttj z}||j |<t| d5}tj|j |dddn #1swxYwY|S)Nzusername not providedzSystem User prefix not allowedz[^a-zA-Z0-9-_]+-_w)strip ValueErrorr@r*rAresubruuiduuid4r7r4r3r5dump)r8nameuser_idr:s radd_userzUserManager.add_userhs9zz|| 6455 5 ??<: ; ; ?=>> >&*C66   l= > > ?=>> >C-#djll"3"33" 7 $%%'' - - % Idj! $ $ $ % % % % % % % % % % % % % % %s*DDDcN j||dfd}|dfd}|dfd}|dfd}dfd |d  fd }|d  fd }|d  fd}|d fd} dS)Nz/userscKtjrtjdjdS|dd}tjdt j|dS)Nserver)storager7FrP)rfmigrated) rr/r json_responser7rTrr r-)rCuser_dirr8s r get_usersz)UserManager.add_routes..get_users}sz (X )S)STTT99'4TY9ZZ(' "x 8 8**rcK|d{V}|d}|jvrtjddidS |}n<#t $r/}tjdt|idcYd}~Sd}~wwxYwtj|S)NusernameerrorzDuplicate username.status)r5r7valuesrrirbrZr)rCbodyrmraer8s r post_usersz*UserManager.add_routes..post_userss ''''''DJ'H4:,,....('3H)IRUVVVV H--11 H H H('3q66):3GGGGGGGGG H$W-- -sA00 B):$B$B)$B)z /userdatac(K|jjdd}|stjddS ||stjddSt jstjdd S|jjd d d k}|jjd d d k|jjd d d k|r4t j tj dd}n2t j tj d}dtdttzttzffd fdtj ||D}tj|S)a List user data files in a specified directory. This endpoint allows listing files in a user's data directory, with options for recursion, full file information, and path splitting. Query Parameters: - dir (required): The directory to list files from. - recurse (optional): If "true", recursively list files in subdirectories. - full_info (optional): If "true", return detailed file information (path, size, modified time). - split (optional): If "true", split file paths into components (only applies when full_info is false). Returns: - 400: If 'dir' parameter is missing. - 403: If the requested path is not allowed. - 404: If the requested directory does not exist. - 200: JSON response with the list of files or file information. The response format depends on the query parameters: - Default: List of relative file paths. - full_info=true: List of dictionaries with file details. - split=true (and full_info=false): List of lists, each containing path components. dirrozDirectory not providedrqtextzInvalid directoryzDirectory not foundrecursetrue full_inforMz*** full_pathrcrt|Stj|tjd}r|g|dzS|S)Nr)r$rr rrr rM)rrel_pathrr split_paths rprocess_full_pathzGUserManager.add_routes..listuserdata..process_full_pathsh:(D9997??9d;;CCBFCPP<$:s(;(;;;rcdg|],}tj|!|-Sr)rr r2).0rrs r z@UserManager.add_routes..listuserdata..sH7>>),,!!),,r) recursive)rel_urlquerygetrResponserTrr r-lowerr=globescaperr listri) rC directoryr}patternresultsrr rrr8s @@@@r listuserdataz,UserManager.add_routes..listuserdatas2 -11%<>$'' L|35JKKKKo+// 2>>DDFF&PG-11+rBBHHJJfTI .227B??EEGG6QJ ?',,t{4'8'8$DD',,t{4'8'8#>> S X^d3i5O        !%7g!F!F!FG $W-- -rz /v2/userdatac K|jjdd} tj|}nG#t $r:}t jd|d|tj ddcYd}~Sd}~wwxYw  |dd }|r ||d }n|}nD#t$r7}t jd |tj d d cYd}~Sd}~wwxYw|stj ddStj |s0||krtjgStj ddStj |stj ddSg} tj|dD]\\}}}|D]y} tj || } tj | |tjd} || | ddz|D]} tj || } tj | |tjd} | | dd} tj| }|j|d<|j|d<n1#t4$r$}t jd| d|Yd}~nd}~wwxYw||א^nG#t4$r:}t jd|d|tj ddcYd}~Sd}~wwxYw|d tj|S)!a List files and directories in a user's data directory. This endpoint provides a structured listing of contents within a specified subdirectory of the user's data storage. Query Parameters: - path (optional): The relative path within the user's data directory to list. Defaults to the root (''). Returns: - 400: If the requested path is invalid, outside the user's data directory, or is not a directory. - 404: If the requested path does not exist. - 403: If the user is invalid. - 500: If there is an error reading the directory contents. - 200: JSON response containing a list of file and directory objects. Each object includes: - name: The name of the file or directory. - type: 'file' or 'directory'. - path: The relative path from the user's data root. - size (for files): The size in bytes. - modified (for files): The last modified timestamp (Unix epoch). r rxz!Failed to decode path parameter: z , Error: roz$Invalid characters in path parameterryNFrgzAccess denied for user: r{z!Invalid user specified in requestzInvalid path requestedr|zRequested path not foundz!Requested path is not a directoryT)topdownrr)r`r rOrNr rzCould not stat file z: zError listing directory iz Error reading directory contentscN|ddk|dfS)NrOrr`)r)xs rzBUserManager.add_routes..list_userdata_v2..;s"& [(@!F)//BSBS'Tr)key)rrrrrK Exceptionr0r1rrrTrBrr r-riisdirwalkr=rrr appendstatst_sizest_mtimeOSErrorrnsort)rCrequested_rel_pathrtbase_user_pathtarget_abs_pathrrootdirsfilesdir_namedir_pathr file_name file_path entry_infostats stat_errorr8s rlist_userdata_v2z0UserManager.add_routes..list_userdata_v2s52")!6!:!:62!F!F  ]%*]3E%F%F"" ] ] ] dDV d dab d deee|35[\\\\\\\\\ ] ["&"@"@$[`"@"a"a&6'+'E'EgOans'E't't__'5_ [ [ [!?A!?!?@@@ C6YZZZZZZZZZ [ # P C6NOOOO7>>/22 V"n44-b111 L:TUUUU7==11 [ C6YZZZZG Y)+$)O)O)O33%D$$(#%7<<h#?#?#%7??8^#L#L#T#TUWU[]`#a#a$,$,$/((&+33 $&GLLy$A$A #%7??9n#M#M#U#UVXV\^a#b#b$-$,$*&& !$&GI$6$6E16Jv.5:^Jz22&!!!#O,\9,\,\PZ,\,\]]] DDDD! z2222338 Y Y Y OOOAOOPPP|35WXXXXXXXXX Y LLTTL U U U$W-- -s: A>/A93A>9A>5B88 C9,C4.C94C9DL(KL K4K/*L/K44L M/M MMFrNc2|j|d}|stjdS||}|stjdS|r4t j|stjdS|S)Nrorpr{r|) match_inforrrrTrr r-)rC check_existsparamrNr r8s rget_user_data_pathz2UserManager.add_routes..get_user_data_path?s%))%66D 0|3////11'4@@D 0|3//// 0BGNN4$8$8 0|3////Krz/userdata/{file}cxK|d}t|ts|Stj|S)NTr) isinstancerr FileResponserCr rs r getuserdataz+UserManager.add_routes..getuserdataMsD%%gDAAADdC((  #D)) )rc.K |}t|ts|S|jdddk}|jdddk}|s5t j|rtj ddS | d{V}t j |}tj | \}} t j|d 5}||dddn #1swxYwYt j||n#t j|xYwnG#t$$r:} t'jd |d | tj d dcYd} ~ Sd} ~ wwxYw |d} |rt-|| } n t j|| } tj| S)aC Upload or update a user data file. This endpoint handles file uploads to a user's data directory, with options for controlling overwrite behavior and response format. Query Parameters: - overwrite (optional): If "false", prevents overwriting existing files. Defaults to "true". - full_info (optional): If "true", returns detailed file information (path, size, modified time). If "false", returns only the relative file path. Path Parameters: - file: The target file path (URL encoded if necessary). Returns: - 400: If 'file' parameter is missing. - 403: If the requested path is not allowed. - 409: If overwrite=false and the file already exists. - 200: JSON response with either: - Full file information (if full_info=true) - Relative file path (if full_info=false) The request body should contain the raw file content to be written. overwriter~falserFile already existsryN)rwwbzError saving file 'z': roz@Invalid filename. Please avoid special characters like :\/*?"<>|)rqreason)rrrrrrr r-rrreaddirnametempfilemkstempfdopenwriterunlinkrr0r1rTr$rri)rCr rrrsrfdtmp_pathr:rt user_pathresprr8s r post_userdataz-UserManager.add_routes..post_userdataUsu4&%g..DdC((   ))+v>>'II ))+w??EEGG6QI L!5!5 L|35JKKKK $\\^^++++++7??400'/H=== H2t,,& &&&&&&&&&&&&&&&Jx....Ih'''/    Bd B Bq B BCCC|_ 66wEEI 8$T955wtY77$T** *s[.AE3ED7+ E7D;;E>D;?EE3E//E33 F7=/F2,F72F7cK|d}t|ts|Stj|t jdS)NTrrp)rrrremoverrrs rdelete_userdataz/UserManager.add_routes..delete_userdatasS%%gDAAADdC((  IdOOO ''N)rrrrrrr r-rrr0infoshutilmoverTr$rri) rCsourcerrrrrrr8s r move_userdataz-UserManager.add_routes..move_userdatasf4('dCCCFfc**  %%gEPPPDdC((   ))+v>>'II ))+w??EEGG6QI L!5!5 L|35JKKKK L9F99$999 : : : K % % %66wEEI 8$T955wtY77$T** *r)FrN)r, add_routesrpostdelete) r8routesrkrurrrrrrrs ` @rrzUserManager.add_routeszs   ((( H          X   . . . .   . K = .= .= .= . ! = .~ N # #g .g .g .g . $ #g .R       & ' ' * * * * ( ' * ' ( (< +< +< +< +< + ) (< +| ) * * , , , , + * , 3 4 40 +0 +0 +0 +0 + 5 40 +0 +0 +rN)rFT) rrrr;r3rErTrbrrrrr&r&#s|000&MMM   B$U+U+U+U+U+rr&)r5rr[r]rrr0raiohttprurllibrcomfy.cli_argsrr* app_settingsrtypingr default_userr rr$r&rrrrsG %%%%%% y#(l+l+l+l+l+l+l+l+l+l+r