Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- #!/usr/bin/python3
- # tpt = Temp PaTh，临时目录
- import os
- import random
- from datetime import datetime
- import tempfile
- import shutil
- def main():
- current_time = datetime.now().strftime("%m%d.%H%M%S")
- rand_num = f"{random.randint(0, 999999):06d}"
- dir_name = f"{current_time}.{rand_num}"
- # TMP = os.environ.get("TMP", tempfile.gettempdir())
- TMP = '/tmp/tpt'
- dir_path = os.path.join(TMP, dir_name)
- os.makedirs(dir_path, exist_ok=True)
- os.chdir(dir_path)
- print(f"Entered directory: {os.getcwd()}")
- os.system(os.environ.get("SHELL", "bash") + " --login")
- # Ctrl+D in SHELL
- os.chdir(TMP)
- c = ""
- try:
- c = input(f"Delete directory {dir_path}? (y(d)/n/r) ")
- except EOFError:
- c = "y"
- print()
- cmd = c.split()
- if len(cmd) == 0 or cmd[0] == "y" or cmd[0] == "d":
- # os.rmdir(dir_path)
- shutil.rmtree(dir_path, ignore_errors=True)
- print(f"Deleted directory {dir_path}")
- elif cmd[0] == "n":
- print(f"Directory {dir_path} is not deleted")
- elif cmd[0] == "r":
- # rename
- if len(cmd) < 2 or cmd[1].strip() == "":
- name = input("Input new path: ").strip()
- else:
- name = cmd[1].strip()
- new_dir_path = os.path.join(TMP, name)
- # os.rename(dir_path, new_dir_path)
- shutil.move(dir_path, new_dir_path)
- print(f"Renamed directory {dir_path} to {new_dir_path}")
- else:
- print(f"Invalid input {c}")
- if __name__ == "__main__":
- main()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement
