diff --git a/main.py b/main.py index f122a93..deffedd 100644 --- a/main.py +++ b/main.py @@ -22,11 +22,18 @@ def pseudo_zsh(): while True: try: + dir = os.getcwd() + if dir == '/': + CmdDir = '/' + elif re.match(dir, "/home/*"): # 用户目录下的任意文件夹 + CmdDir = '~' + else: + CmdDir = dir.split('/')[-1] # 显示提示符并获取用户输入 - cmd = input("20240915786@\u9648\u5764\u9633 ~ % ") + cmd = input(f"20250910553@何相龙 {CmdDir} % ") - # 如果输入特定命令 "hexianglong",退出程序 - if cmd.strip() == "hexianglong": + # 如果输入特定命令 "hxl",退出程序 + if cmd.strip() == "hxl": print("Exiting secret mode...") break