diff options
author | Kyomotoi <[email protected]> | 2020-12-20 17:14:41 +0800 |
---|---|---|
committer | Kyomotoi <[email protected]> | 2020-12-20 17:14:41 +0800 |
commit | 9bf222471d34cb756a4878b103ec82c6c4bfb191 (patch) | |
tree | 1420d498c41f71858e74b4926b8d7206b5235fe6 /ATRI/utils/utils_history | |
parent | 9ac2c2eb7ef2eb267e926d599dc077f5afae98f9 (diff) | |
download | ATRI-9bf222471d34cb756a4878b103ec82c6c4bfb191.tar.gz ATRI-9bf222471d34cb756a4878b103ec82c6c4bfb191.tar.bz2 ATRI-9bf222471d34cb756a4878b103ec82c6c4bfb191.zip |
[Update]
- 创建项目结构
Diffstat (limited to 'ATRI/utils/utils_history')
-rw-r--r-- | ATRI/utils/utils_history/__init__.py | 105 |
1 files changed, 0 insertions, 105 deletions
diff --git a/ATRI/utils/utils_history/__init__.py b/ATRI/utils/utils_history/__init__.py deleted file mode 100644 index 3aae3fa..0000000 --- a/ATRI/utils/utils_history/__init__.py +++ /dev/null @@ -1,105 +0,0 @@ -#!/usr/bin/env python3 -# -*- encoding: utf-8 -*- -''' -@File : __init__.py -@Time : 2020/11/07 14:33:22 -@Author : Kyomotoi -@Contact : [email protected] -@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(): - os.makedirs(Path('.') / 'ATRI' / 'data' / 'data_Group', exist_ok=True) - os.makedirs(Path('.') / 'ATRI' / 'data' / 'data_Group' / f'{group}', - exist_ok=True) - - with open(GROUP_PATH, 'w') as f: - f.write(json.dumps({})) - - if not PRIVATE_PATH.is_file(): - os.makedirs(Path('.') / 'ATRI' / 'data' / 'data_Private_Message', - exist_ok=True) - - 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} |