写一个行数,能通过传入的参数不同自动构造目录和目录下文件,规则如下:

用户可以从控制台输入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("输入有误!") 

你可能感兴趣的:(写一个行数,能通过传入的参数不同自动构造目录和目录下文件,规则如下:)