用户可以从控制台输入3个选项:1、2、q
控制台传入”1”,则生成1个目录A,目录A下面有 2个文件a.txt,b.txt
控制台传入”2”,则生成2个目录(目录A和目录B),目录A下面有2个文件a.txt和b.txt,目录B下面有2个文件b.txt和c.txt
控制台q,则退出
控制台输入其他数据,提示:输入有误!
while 1:
import os
try:
os.mkdir(r"E:\1003")
except FileExistsError:
print("'E:\\1003'已存在")
os.chdir(r"E:\1003")
print("输入3个选项:1、2、q")
command = input("please input a command:")
if command == "1":
try:
os.mkdir(r"E:\1003\A")
except FileExistsError:
print("'E:\\1003\A'已存在")
os.chdir(r"E:\1003\A")
with open(r"e:\1003\A\a.txt","w",encoding="utf-8")as fp:
pass
with open(r"e:\1003\A\b.txt","w",encoding="utf-8")as fp1:
pass
elif command == "2":
try:
os.mkdir(r"E:\1003\A")
except FileExistsError:
print("'E:\\1003\A'已存在")
try:
os.mkdir(r"E:\1003\B")
except FileExistsError:
print("'E:\\1003\B'已存在")
with open(r"e:\1003\A\a.txt","w",encoding="utf-8")as fp:
pass
with open(r"e:\1003\A\b.txt","w",encoding="utf-8")as fp1:
pass
with open(r"e:\1003\B\a.txt","w",encoding="utf-8")as fp:
pass
with open(r"e:\1003\B\b.txt","w",encoding="utf-8")as fp1:
pass
elif command == "q":
break
else:
print("输入有误!")