main.py

import argparse
from reptile import fetch
from parse import parse, ParseError
from output import output_to_console, output_to_file, output_to_file_kotlin
from platform_ import Platform
import asyncio
from itertools import chain

parser = argparse.ArgumentParser(description='Y-API 2 JSON BEAN🚀🚀🚀')
parser.add_argument('ids', help='需要转换的Y-api的接口id, 可多传', nargs='+')
parser.add_argument('-i', '--ios', help='生成iOS平台 Swift Decodable Model', action='store_true', default=True)
parser.add_argument('-a', '--android', help='生成Android平台 Kotlin Json Bean', action='store_true')
parser.add_argument('-f', '--file', help='将结果保存到文件中', action='store_true')


async def main():
    args = parser.parse_args()
    tasks = []
    async with asyncio.TaskGroup() as group:
        for api_id in args.ids:
            try:
                task = group.create_task(
                    parse(fetch(api_id), platform=Platform.Swift if args.ios else Platform.Kotlin)
                )
            except ParseError as e:
                print(f'😈{e}')
            else:
                tasks.append(task)

    results = chain.from_iterable([task.result() for task in tasks])

    if args.file:
        if args.ios:
            output_to_file(results)
        if args.android:
            output_to_file_kotlin(results)

    else:
        output_to_console(results)


if __name__ == '__main__':
    """
    举个栗子: python main.py 5478 5595
    """
    print('🚀START')
    asyncio.run(main())
    print('🎆END')