import os

if __name__ == '__main__':
    in_dir = "/mnt/database2/jiangqian/0808/online_orig_train_datas_2"
    out_dir = "/mnt/database2/jiangqian/0808/online_train_datas_2"

    dirs = os.listdir(in_dir)

    for single_dir in dirs:
        if "traindata" in single_dir:
            in_single_dir = os.path.join(in_dir, single_dir)
            out_single_dir = os.path.join(out_dir, single_dir)

            if not os.path.exists(out_single_dir):
                os.makedirs(out_single_dir)

            images_dir = os.path.join(out_single_dir, "images")
            if not os.path.exists(images_dir):
                os.makedirs(images_dir)

            hairstyle_dir = os.path.join(images_dir, "1_hairstyle")
            if not os.path.exists(hairstyle_dir):
                os.makedirs(hairstyle_dir)

            for root, dirs, files in os.walk(in_single_dir):
                for file in files:
                    in_file = os.path.join(root, file)
                    out_file = os.path.join(hairstyle_dir, file)
                    os.system("cp %s %s" % (in_file, out_file))
                    print("copy %s to %s" % (in_file, out_file))