import argparse
import os


def main():
    parser = argparse.ArgumentParser("生成label studio 需要的文件目录\n")
    parser.add_argument("input", help="")
    parser.add_argument("--wildcard")
    parser.add_argument("output", help="")

    args = parser.parse_args()

    try:
        with open(args.output, 'w') as outfile:

            # 遍历目录中的文件和文件夹
            for root, dirs, files in os.walk(args.input):
                # 只输出文件名，不包括文件夹
                for file in files:
                    if file.startswith("."):
                        continue
                    if args.wildcard and file.endswith(args.wildcard):
                        file_path = os.path.join(root, file)
                        outfile.write('http://192.168.10.18:10088/' + file_path + '\n')

    except Exception as e:
        print(e)


if __name__ == '__main__':
    main()