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