diff options
Diffstat (limited to 'utils/utils_history')
| -rw-r--r-- | utils/utils_history/__init__.py | 108 | 
1 files changed, 0 insertions, 108 deletions
| diff --git a/utils/utils_history/__init__.py b/utils/utils_history/__init__.py deleted file mode 100644 index 4453e11..0000000 --- a/utils/utils_history/__init__.py +++ /dev/null @@ -1,108 +0,0 @@ -#!/usr/bin/env python3 -# -*- encoding: utf-8 -*- -''' -@File    :   __init__.py -@Time    :   2020/11/07 14:33:22 -@Author  :   Kyomotoi -@Contact :   kyomotoiowo@gmail.com -@Github  :   https://github.com/Kyomotoi -@License :   Copyright © 2018-2020 Kyomotoi, All Rights Reserved. -''' -__author__ = 'kyomotoi' - -import os -import json -import datetime -from pathlib import Path -from typing import Optional - - -def saveMessage(message_id: str, -                message: str, -                user: str, -                group: Optional[str] = None) -> None: -    """储存消息""" -    GROUP_PATH = Path( -        '.' -    ) / 'ATRI' / 'data' / 'data_Group' / f'{group}' / f"{datetime.datetime.now().strftime('%Y-%m-%d')}-message.json" -    PRIVATE_PATH = Path( -        '.' -    ) / 'ATRI' / 'data' / 'data_Private_Message' / f"{datetime.datetime.now().strftime('%Y-%m-%d')}-private-message.json" - -    # 检查目标文件目录 -    if not GROUP_PATH.is_file(): -        try: -            os.mkdir(Path('.') / 'ATRI' / 'data' / 'data_Group' / f'{group}') -        except: -            pass - -        with open(GROUP_PATH, 'w') as f: -            f.write(json.dumps({})) - -    if not PRIVATE_PATH.is_file(): -        try: -            os.mkdir(Path('.') / 'ATRI' / 'data' / 'data_Private_Message') -        except: -            pass - -        with open(PRIVATE_PATH, 'w') as f: -            f.write(json.dumps({})) - -    # 加载目标文件 -    with open(GROUP_PATH, 'r') as f: -        DATA_GROUP = json.load(f) - -    with open(PRIVATE_PATH, 'r') as f: -        DATA_PRIVATE = json.load(f) - -    # 写入 -    if group: -        DATA_GROUP[f"{message_id}"] = { -            "message": f"{message}", -            "user_id": f"{user}", -            "group_id": f"{group}" -        } - -        with open(GROUP_PATH, 'w') as f: -            f.write(json.dumps(DATA_GROUP)) - -    else: -        DATA_PRIVATE[f"{message_id}"] = { -            "message": f"{message}", -            "user_id": f"{user}" -        } - -        with open(PRIVATE_PATH, 'w') as f: -            f.write(json.dumps(DATA_PRIVATE)) - - -def getMessage(message_id: str, group: Optional[str] = None) -> dict: -    ''' -    传入消息id以获取对应信息 -     -    :return: dict -    ''' -    GROUP_PATH = Path( -        '.' -    ) / 'ATRI' / 'data' / 'data_Group' / f'{group}' / f"{datetime.datetime.now().strftime('%Y-%m-%d')}-message.json" -    PRIVATE_PATH = Path( -        '.' -    ) / 'ATRI' / 'data' / 'data_Private_Message' / f"{datetime.datetime.now().strftime('%Y-%m-%d')}-private-message.json" - -    if group: -        try: -            with open(GROUP_PATH, 'r') as f: -                data_group = json.load(f) -            return data_group[message_id] - -        except: -            return {"status": 0} - -    else: -        try: -            with open(PRIVATE_PATH, 'r') as f: -                data_private = json.load(f) -            return data_private[message_id] - -        except: -            return {"status": 0} | 
