:>j  dZddlZddlZddlZddlZddlmZ ddlmZmZn&#e $re dej dYnwxYwej ej eZej edZgdZd ed efd Zd ed ed efd ZdZdZdefdZded efdZd2ded edededef dZdZdefdZdedefdZe dkre!ej"dkrRe de d e d!e d"e d#e d$ej dej"dZ#e#d%kr edSe#d&kre!ej"d'kre d(ej dej"dZ$ej"d)Z%ej"d*Z&e!ej"d'kr ej"d'ndZ'e!ej"d+kr ej"d+ndZ(ee$e%e&e'e(dSe#d,kr edSe#d-krGe!ej"d)kre d.ej deej"ddSe#d/krSe!ej"d*kre d0ej deej"dej"d)dSe d1e#ej ddSdS)3uv 用户管理工具 - 用户数据存储在 users.xlsx (Excel) - 密码使用 SHA256 哈希存储 - 支持命令行操作：add / list / delete / reset-pwd 用法： python user_manager.py add [role] python user_manager.py list python user_manager.py delete python user_manager.py reset-pwd N)datetime)Workbook load_workbooku+请先安装 openpyxl: pip install openpyxlz users.xlsx)idusernamepassword display_namephonerole created_atr returnc|d}tj||zS)uSHA256 密码哈希zhengfangyuan2026)hashlibsha256encode hexdigest)r salts %D:\AI-Ad-Agent\portal\user_manager.py hash_passwordr s3 D >8d?2244 5 5 ? ? A AAhashedc(t||kS)u 验证密码)r)r rs rverify_passwordr&s  " "f ,,rc $tjtst }|j}d|_|t|ttj dddtddddtjdg|tt#d tt#d d Sd S) u初始化 Excel 数据库usersNadminadmin123u 管理员%Y-%m-%d %H:%M:%Su已创建用户数据库: u'默认管理员账号: admin / admin123TF)ospathexistsDB_FILEractivetitleappendCOLUMNSstruuiduuid4rrnowstrftimesaveprint)wbwss rinit_dbr4+s 7>>' " " ZZ Y '   bqb !  * % %    LNN # #$7 8 8      4744555 7888t 5rctjtst t t}|j}g}|ddD]J}|dr@ttt|}d|vrd|d<| |K| |S)u获取所有用户T)min_row values_onlyrr r!) r#r$r%r&r4rr' iter_rowsdictzipr*r)close)r2r3rrowus r get_all_usersr?Cs 7>>' " " w  B B E||A4|88 q6 S#&&''Aa' LLOOOHHJJJ LrrcLt}|D]}|d|kr|cSdS)u 查找用户rN)r?)rrr>s r find_userrAUs< OOE  Z=H $ $HHH % 4rct|}|rCt||dr-|d|d|dd|ddSdS) uA验证用户登录，成功返回用户信息，失败返回 Noner rr r r!r )rr r r N)rArget)rr users r verify_userrE^sk X  D  $z*:;; Z( 0XXgr**L    4rr!rDr r r c Zt|rtd|ddStjt st tt }|j}| ttj dd|t||||tjdg|t |td|d|d |d S) u 添加用户错误: 用户 'u ' 已存在FNrr"用户 'z' (u) 添加成功，角色: T)rAr1r#r$r%r&r4rr'r)r+r,r-rrr.r/r0r<)rr r r r r2r3s radd_userrIks 6666777u 7>>' " " w  B BII DJLL"1"h    344GGGHHJJJ OX O O, O O O OPPP 4rc t}|stddStdddddddd d dd d dd td|D]O}t|ddd|ddd|ddd d|dd d|d Ptdt|ddS)u列出所有用户u 暂无用户N u 用户名z<16 u 显示名称z<12u 手机号z<14u角色z<10u 创建时间zN------------------------------------------------------------------------------rr r r!r r u 共 u 个用户)r?r1rClen)rr>s r list_usersrNs OOE  n g{ g g g g g gk g g g g g gWe g ghhh (OOO xx :vvvQ~%6vvvQUU72=N=NvvvUVW]U^vvvefgsetvvwwww )3u:: ) ) )*****rc|dkrtddSt|std|ddStt}|j}|dD]5}|dj|kr"||d jn6| t| tj td |d }tj |r&tj|td |td |ddS)u 删除用户ru#错误: 不能删除管理员账号FrG ' 不存在r6r7rrdataz _records.jsonu已删除用户数据文件: rHu ' 已删除T)r1rArr&r'r9value delete_rowsr=r0r<r#r$joinBASE_DIRr%remove)rr2r3r= records_files r delete_userrYsS7 3444u X   6666777u w  B B||A|&& q6<8 # # NN3q6: & & & E $GGGHHJJJ7<<&X2L2L2LMML w~~l##> ,  <密码> <显示名> [手机号] [角色]uI python user_manager.py list # 列出所有用户uG python user_manager.py delete <用户名> # 删除用户uJ python user_manager.py reset-pwd <用户名> <新密码> # 重置密码initadduX用法: python user_manager.py add <用户名> <密码> <显示名> [手机号] [角色]listdeleteu1用法: python user_manager.py delete <用户名>z reset-pwdu@用法: python user_manager.py reset-pwd <用户名> <新密码>u未知命令: )r!rD))__doc__rr#sysr,ropenpyxlrr ImportErrorr1exitr$dirnameabspath__file__rVrUr&r*r+rboolrr4r?rArErIrNrYr\__name__rMargvcmdrr r r r rrrss   000000000 E 7888 CHQKKKKK 7??27??844 5 5 ',,x . . W W WBCBCBBBB -c-3-4---- 0$ #     scSUX4 + + +#:S& z s38}}q i YZZZ bccc YZZZ WXXX Z[[[ (1+C f}}  3sx==1   El m m m CHQKKK8A;8A;x{ "s38}}q00 b!c#(mma//sx{{V8\5$?????   3sx==1   EE F F F CHQKKK CHQK       3sx==1   ET U U U CHQKKKsx{CHQK00000 $s$$%%% Ms# AA