tracemoe-py

trace.moe的最小异步 API 包装器。

安装

需要 Python 3.6 或更高版本。

pip install tracemoe-py

用法

import asyncio

from tracemoe import TraceMoe


async def main():
    async with TraceMoe() as tracemoe:

        # Search by image URL
        results: list = await tracemoe.search('https://XXX/XXX.jpg')
        print(results)

        # Search by image upload
        results: list = await tracemoe.search(open('/home/ichbinleoon/XXX.jpg', 'rb'))
        print(results)

        # Get account info
        info: dict = await tracemoe.me()
        print(info)

asyncio.run(main())

高级用法

import asyncio

import aiohttp

from tracemoe import TraceMoe


async def main():

    # Use an API key
    tracemoe = TraceMoe(api_key='Your API key')

    # Cut black borders
    results: list = await tracemoe.search('https://XXX/XXX.jpg', cut_borders=True)
    print(results)

    # Filter by AniList ID
    results: list = await tracemoe.search('https://XXX/XXX.jpg', anilist_id=11617)
    print(results)

    # Include AniList info
    results: list = await tracemoe.search('https://XXX/XXX.jpg', anilist_info=True)
    print(results)
    
    await tracemoe.close()

    # Use your own aiohttp session
    session = aiohttp.ClientSession()
    tracemoe = TraceMoe(session=session)
    
    # ...
    
    await tracemoe.close()

asyncio.run(main())

贡献

欢迎投稿！随意打开问题或提交拉取请求！

执照

麻省理工学院 © IchBinLeoon