Skip to content

๐ŸŽž๏ธโ“ A minimal asynchronous API wrapper for trace.moe

License

Notifications You must be signed in to change notification settings

IchBinLeoon/tracemoe-py

Folders and files

NameName
Last commit message
Last commit date

Latest commit

ย 

History

4 Commits
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 

Repository files navigation

tracemoe-py

PyPI - Python Version PyPI - Downloads PyPI License

A minimal asynchronous API wrapper for trace.moe.

Installation

Python 3.6 or higher is required.

pip install tracemoe-py

Usage

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())

Advanced Usage

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())

Contribute

Contributions are welcome! Feel free to open issues or submit pull requests!

License

MIT ยฉ IchBinLeoon