OpenCompass/opencompass/utils/lark.py

import argparse
import json
from typing import Dict, List, Optional, Union

import requests


class LarkReporter:

    def __init__(self, url: str):
        self.url = url

    def post(self,
             content: Union[str, List[List[Dict]]],
             title: Optional[str] = None):
        """Post a message to Lark.

        When title is None, message must be a str.
        otherwise msg can be in rich text format (see
        https://open.feishu.cn/document/uAjLw4CM/ukTMukTMukTM/im-v1/message/create_json#45e0953e
        for details).
        """
        if title is None:
            assert isinstance(content, str)
            msg = {'msg_type': 'text', 'content': {'text': content}}
        else:
            if isinstance(content, str):
                content = [[{'tag': 'text', 'text': content}]]
            msg = {
                'msg_type': 'post',
                'content': {
                    'post': {
                        'zh_cn': {
                            'title': title,
                            'content': content
                        }
                    }
                }
            }
        requests.post(self.url, data=json.dumps(msg))


def parse_args():
    parser = argparse.ArgumentParser(description='Lark bot reporter')
    parser.add_argument('url', help='Lark bot url')
    parser.add_argument('content', type=str, help='Content')
    parser.add_argument('--title', type=str, help='Title', default=None)
    args = parser.parse_args()
    return args


if __name__ == '__main__':
    args = parse_args()
    lark = LarkReporter(args.url)
    if args.title:
        lark.post(args.content, args.title)
    else:
        lark.post(args.content)