mirror of
https://github.com/xuthus83/LittlePaimon.git
synced 2024-10-21 16:27:15 +08:00
🎉 Init
This commit is contained in:
commit
49ea404eb8
7
.env.prod
Normal file
7
.env.prod
Normal file
@ -0,0 +1,7 @@
|
|||||||
|
HOST=0.0.0.0
|
||||||
|
PORT=13579
|
||||||
|
LOG_LEVEL=INFO
|
||||||
|
SUPERUSERS=["123456"]
|
||||||
|
NICKNAME=["派蒙"]
|
||||||
|
COMMAND_START=[""]
|
||||||
|
COMMAND_SEP=[""]
|
142
.gitignore
vendored
Normal file
142
.gitignore
vendored
Normal file
@ -0,0 +1,142 @@
|
|||||||
|
|
||||||
|
# Created by https://www.toptal.com/developers/gitignore/api/python
|
||||||
|
# Edit at https://www.toptal.com/developers/gitignore?templates=python
|
||||||
|
|
||||||
|
### Python ###
|
||||||
|
# Byte-compiled / optimized / DLL files
|
||||||
|
__pycache__/
|
||||||
|
*.py[cod]
|
||||||
|
*$py.class
|
||||||
|
|
||||||
|
# C extensions
|
||||||
|
*.so
|
||||||
|
|
||||||
|
# Distribution / packaging
|
||||||
|
.Python
|
||||||
|
build/
|
||||||
|
develop-eggs/
|
||||||
|
dist/
|
||||||
|
downloads/
|
||||||
|
eggs/
|
||||||
|
.eggs/
|
||||||
|
lib/
|
||||||
|
lib64/
|
||||||
|
parts/
|
||||||
|
sdist/
|
||||||
|
var/
|
||||||
|
wheels/
|
||||||
|
pip-wheel-metadata/
|
||||||
|
share/python-wheels/
|
||||||
|
*.egg-info/
|
||||||
|
.installed.cfg
|
||||||
|
*.egg
|
||||||
|
MANIFEST
|
||||||
|
|
||||||
|
# PyInstaller
|
||||||
|
# Usually these files are written by a python script from a template
|
||||||
|
# before PyInstaller builds the exe, so as to inject date/other infos into it.
|
||||||
|
*.manifest
|
||||||
|
*.spec
|
||||||
|
|
||||||
|
# Installer logs
|
||||||
|
pip-log.txt
|
||||||
|
pip-delete-this-directory.txt
|
||||||
|
|
||||||
|
# Unit test / coverage reports
|
||||||
|
htmlcov/
|
||||||
|
.tox/
|
||||||
|
.nox/
|
||||||
|
.coverage
|
||||||
|
.coverage.*
|
||||||
|
.cache
|
||||||
|
nosetests.xml
|
||||||
|
coverage.xml
|
||||||
|
*.cover
|
||||||
|
*.py,cover
|
||||||
|
.hypothesis/
|
||||||
|
.pytest_cache/
|
||||||
|
pytestdebug.log
|
||||||
|
|
||||||
|
# Translations
|
||||||
|
*.mo
|
||||||
|
*.pot
|
||||||
|
|
||||||
|
# Django stuff:
|
||||||
|
*.log
|
||||||
|
local_settings.py
|
||||||
|
db.sqlite3
|
||||||
|
db.sqlite3-journal
|
||||||
|
|
||||||
|
# Flask stuff:
|
||||||
|
instance/
|
||||||
|
.webassets-cache
|
||||||
|
|
||||||
|
# Scrapy stuff:
|
||||||
|
.scrapy
|
||||||
|
|
||||||
|
# Sphinx documentation
|
||||||
|
docs/_build/
|
||||||
|
doc/_build/
|
||||||
|
|
||||||
|
# PyBuilder
|
||||||
|
target/
|
||||||
|
|
||||||
|
# Jupyter Notebook
|
||||||
|
.ipynb_checkpoints
|
||||||
|
|
||||||
|
# IPython
|
||||||
|
profile_default/
|
||||||
|
ipython_config.py
|
||||||
|
|
||||||
|
# pyenv
|
||||||
|
.python-version
|
||||||
|
|
||||||
|
# pipenv
|
||||||
|
# According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
|
||||||
|
# However, in case of collaboration, if having platform-specific dependencies or dependencies
|
||||||
|
# having no cross-platform support, pipenv may install dependencies that don't work, or not
|
||||||
|
# install all needed dependencies.
|
||||||
|
#Pipfile.lock
|
||||||
|
|
||||||
|
# PEP 582; used by e.g. github.com/David-OConnor/pyflow
|
||||||
|
__pypackages__/
|
||||||
|
|
||||||
|
# Celery stuff
|
||||||
|
celerybeat-schedule
|
||||||
|
celerybeat.pid
|
||||||
|
|
||||||
|
# SageMath parsed files
|
||||||
|
*.sage.py
|
||||||
|
|
||||||
|
# Environments
|
||||||
|
.venv
|
||||||
|
env/
|
||||||
|
venv/
|
||||||
|
ENV/
|
||||||
|
env.bak/
|
||||||
|
venv.bak/
|
||||||
|
|
||||||
|
# Spyder project settings
|
||||||
|
.spyderproject
|
||||||
|
.spyproject
|
||||||
|
|
||||||
|
# Rope project settings
|
||||||
|
.ropeproject
|
||||||
|
|
||||||
|
# mkdocs documentation
|
||||||
|
/site
|
||||||
|
|
||||||
|
# mypy
|
||||||
|
.mypy_cache/
|
||||||
|
.dmypy.json
|
||||||
|
dmypy.json
|
||||||
|
|
||||||
|
# Pyre type checker
|
||||||
|
.pyre/
|
||||||
|
|
||||||
|
# pytype static type analyzer
|
||||||
|
.pytype/
|
||||||
|
|
||||||
|
.idea
|
||||||
|
|
||||||
|
# End of https://www.toptal.com/developers/gitignore/api/python
|
30
.pre-commit-config.yaml
Normal file
30
.pre-commit-config.yaml
Normal file
@ -0,0 +1,30 @@
|
|||||||
|
ci:
|
||||||
|
autofix_commit_msg: ":rotating_light: auto fix by pre-commit hooks"
|
||||||
|
autofix_prs: true
|
||||||
|
autoupdate_branch: dev
|
||||||
|
autoupdate_schedule: monthly
|
||||||
|
autoupdate_commit_msg: ":arrow_up: auto update by pre-commit hooks"
|
||||||
|
|
||||||
|
repos:
|
||||||
|
- repo: https://github.com/psf/black
|
||||||
|
rev: 23.1a1
|
||||||
|
hooks:
|
||||||
|
- id: black
|
||||||
|
args: [-S, --verbose]
|
||||||
|
verbose: true
|
||||||
|
|
||||||
|
|
||||||
|
- repo: https://github.com/python-poetry/poetry
|
||||||
|
rev: 1.3.1
|
||||||
|
hooks:
|
||||||
|
- id: poetry-check
|
||||||
|
- id: poetry-export
|
||||||
|
args:
|
||||||
|
[
|
||||||
|
"-f",
|
||||||
|
"requirements.txt",
|
||||||
|
"--without-hashes",
|
||||||
|
"--output",
|
||||||
|
"requirements.txt",
|
||||||
|
]
|
||||||
|
verbose: true
|
19
LittlePaimon/__init__.py
Normal file
19
LittlePaimon/__init__.py
Normal file
@ -0,0 +1,19 @@
|
|||||||
|
from pathlib import Path
|
||||||
|
from typing import Union
|
||||||
|
|
||||||
|
import nonebot
|
||||||
|
from .log import init_logger
|
||||||
|
from nonebot.adapters.onebot.v11 import Adapter as ONEBOT_V11Adapter
|
||||||
|
|
||||||
|
|
||||||
|
def run(*args, **kwargs):
|
||||||
|
init_logger()
|
||||||
|
nonebot.init()
|
||||||
|
app = nonebot.get_asgi()
|
||||||
|
driver = nonebot.get_driver()
|
||||||
|
driver.register_adapter(ONEBOT_V11Adapter)
|
||||||
|
nonebot.run(app=app, *args, **kwargs)
|
||||||
|
|
||||||
|
|
||||||
|
def load_plugin(name: Union[str, Path]):
|
||||||
|
nonebot.load_plugin(name)
|
85
LittlePaimon/hook.py
Normal file
85
LittlePaimon/hook.py
Normal file
@ -0,0 +1,85 @@
|
|||||||
|
import asyncio
|
||||||
|
import inspect
|
||||||
|
|
||||||
|
from nonebot import get_driver, get_bot
|
||||||
|
from nonebot.adapters import Bot
|
||||||
|
from nonebot.utils import is_coroutine_callable, run_sync
|
||||||
|
from typing import Callable
|
||||||
|
|
||||||
|
_start_up_func = []
|
||||||
|
_start_up_func_after_db = []
|
||||||
|
_shutdown_func = []
|
||||||
|
_shutdown_func_before_db = []
|
||||||
|
|
||||||
|
driver = get_driver()
|
||||||
|
|
||||||
|
|
||||||
|
@driver.on_startup
|
||||||
|
async def _run_start_up():
|
||||||
|
await asyncio.gather(*_start_up_func)
|
||||||
|
...
|
||||||
|
await asyncio.gather(*_start_up_func_after_db)
|
||||||
|
|
||||||
|
|
||||||
|
@driver.on_shutdown
|
||||||
|
async def _run_shutdown():
|
||||||
|
await asyncio.gather(*_shutdown_func)
|
||||||
|
...
|
||||||
|
await asyncio.gather(*_shutdown_func_before_db)
|
||||||
|
|
||||||
|
|
||||||
|
def on_startup(database: bool = False) -> Callable:
|
||||||
|
"""
|
||||||
|
包裹一个函数,使其在bot启动完成时运行,如果该函数有数据库相关处理,参数database需为True
|
||||||
|
|
||||||
|
:param database: 是否有数据库相关处理
|
||||||
|
"""
|
||||||
|
|
||||||
|
def return_func(func: Callable) -> Callable:
|
||||||
|
if database:
|
||||||
|
_start_up_func_after_db.append(func)
|
||||||
|
else:
|
||||||
|
_start_up_func.append(func)
|
||||||
|
return func
|
||||||
|
|
||||||
|
return return_func
|
||||||
|
|
||||||
|
|
||||||
|
def on_shutdown(database: bool = False) -> Callable:
|
||||||
|
"""
|
||||||
|
包裹一个函数,使其在bot停止前运行,如果该函数有数据库相关处理,参数database需为True
|
||||||
|
|
||||||
|
:param database: 是否有数据库相关处理
|
||||||
|
"""
|
||||||
|
|
||||||
|
def return_func(func: Callable) -> Callable:
|
||||||
|
if database:
|
||||||
|
_shutdown_func_before_db.append(func)
|
||||||
|
else:
|
||||||
|
_shutdown_func.append(func)
|
||||||
|
return func
|
||||||
|
|
||||||
|
return return_func
|
||||||
|
|
||||||
|
|
||||||
|
async def handle_func_params(func: Callable):
|
||||||
|
"""
|
||||||
|
处理函数依赖注入,已支持的注入参数有:
|
||||||
|
|
||||||
|
- bot: nonebot.adapters.Bot及其子类,默认值可为bot_id来指定bot,例如:bot: Bot = 123456789
|
||||||
|
|
||||||
|
:param func:
|
||||||
|
:return:
|
||||||
|
"""
|
||||||
|
func = func if is_coroutine_callable(func) else run_sync(func)
|
||||||
|
param = inspect.signature(func).parameters
|
||||||
|
if 'bot' in param and issubclass(param['bot'].annotation, Bot):
|
||||||
|
default = param['bot'].default
|
||||||
|
if default == inspect.Parameter.empty:
|
||||||
|
await func(bot=get_bot())
|
||||||
|
elif isinstance(default, (int, str)):
|
||||||
|
await func(bot=get_bot(default))
|
||||||
|
else:
|
||||||
|
await func()
|
||||||
|
else:
|
||||||
|
await func()
|
192
LittlePaimon/log.py
Normal file
192
LittlePaimon/log.py
Normal file
@ -0,0 +1,192 @@
|
|||||||
|
import sys
|
||||||
|
from typing import Union, TYPE_CHECKING
|
||||||
|
|
||||||
|
from loguru import logger as loguru_logger
|
||||||
|
from nonebot.log import logger_id, logger as nb_logger
|
||||||
|
from rich.console import Console
|
||||||
|
|
||||||
|
if TYPE_CHECKING:
|
||||||
|
from loguru import Record
|
||||||
|
|
||||||
|
|
||||||
|
def log_format(record: "Record") -> str:
|
||||||
|
# if record['message'] == 'NoneBot is initializing...':
|
||||||
|
# record['message'] = '正在初始化Nonebot...'
|
||||||
|
return ("<g>{time:MM-DD HH:mm:ss:SSS}</g> "
|
||||||
|
"[<lvl>{level}</lvl>] "
|
||||||
|
"<c><u>{name}</u></c> | "
|
||||||
|
"{message}\n")
|
||||||
|
|
||||||
|
|
||||||
|
def log_filter(record: "Record") -> bool:
|
||||||
|
"""默认的日志过滤器,根据 `config.log_level` 配置改变日志等级。"""
|
||||||
|
# if record['message']:
|
||||||
|
# return False
|
||||||
|
log_level = record['extra'].get('nonebot_log_level', 'INFO')
|
||||||
|
level_no = nb_logger.level(log_level).no if isinstance(log_level, str) else log_level
|
||||||
|
return record['level'].no >= level_no
|
||||||
|
|
||||||
|
|
||||||
|
def init_logger():
|
||||||
|
nb_logger.remove(logger_id)
|
||||||
|
nb_logger.add(
|
||||||
|
sys.stdout,
|
||||||
|
level=0,
|
||||||
|
format=log_format,
|
||||||
|
filter=log_filter,
|
||||||
|
backtrace=False,
|
||||||
|
diagnose=True,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
class logger:
|
||||||
|
"""
|
||||||
|
自定义格式、色彩logger
|
||||||
|
"""
|
||||||
|
_logger: loguru_logger = nb_logger
|
||||||
|
_new_level = _logger.level('data', no=38, color='<yellow>', icon='📂')
|
||||||
|
_rich_console = Console()
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def info(cls, command: str, info: str = '', **kwargs):
|
||||||
|
"""
|
||||||
|
info日志
|
||||||
|
|
||||||
|
:param command: 命令名
|
||||||
|
:param info: 信息
|
||||||
|
"""
|
||||||
|
cls._logger.opt(colors=True, **kwargs).info(f'{CommandColor(command)}{info}')
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def success(cls, command: str, info: str = '', **kwargs):
|
||||||
|
"""
|
||||||
|
success日志
|
||||||
|
|
||||||
|
:param command: 命令名
|
||||||
|
:param info: 信息
|
||||||
|
"""
|
||||||
|
cls._logger.opt(colors=True, **kwargs).success(f'{CommandColor(command)}{info}')
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def warning(cls, command: str, info: str = '', **kwargs):
|
||||||
|
"""
|
||||||
|
warning日志
|
||||||
|
|
||||||
|
:param command: 命令名
|
||||||
|
:param info: 信息
|
||||||
|
"""
|
||||||
|
cls._logger.opt(colors=True, **kwargs).warning(f'{CommandColor(command)}{info}')
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def debug(cls, command: str, info: str = '', **kwargs):
|
||||||
|
"""
|
||||||
|
debug日志
|
||||||
|
|
||||||
|
:param command: 命令名
|
||||||
|
:param info: 信息
|
||||||
|
"""
|
||||||
|
cls._logger.opt(colors=True, **kwargs).debug(f'{CommandColor(command)}{info}')
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def error(cls, command: str, info: str = '', **kwargs):
|
||||||
|
"""
|
||||||
|
error日志
|
||||||
|
|
||||||
|
:param command: 命令名
|
||||||
|
:param info: 信息
|
||||||
|
"""
|
||||||
|
cls._logger.opt(colors=True, **kwargs).error(f'{CommandColor(command)}{info}')
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def data(cls, command: str, info: str = '', **kwargs):
|
||||||
|
"""
|
||||||
|
data日志
|
||||||
|
|
||||||
|
:param command: 命令名
|
||||||
|
:param info: 信息
|
||||||
|
"""
|
||||||
|
cls._logger.opt(colors=True, **kwargs).log('data', f'{CommandColor(command)}{info}')
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def exception(cls, command: str, info: str = '', rich: bool = False, **kwargs):
|
||||||
|
"""
|
||||||
|
exception日志
|
||||||
|
|
||||||
|
:param command: 命令名
|
||||||
|
:param info: 信息
|
||||||
|
:param rich: 使用rich库打印异常
|
||||||
|
"""
|
||||||
|
if rich:
|
||||||
|
cls._logger.opt(colors=True, **kwargs).error(f'{CommandColor(command)}{info}')
|
||||||
|
cls._rich_console.print_exception(show_locals=True)
|
||||||
|
else:
|
||||||
|
cls._logger.opt(colors=True, **kwargs).exception(f'{CommandColor(command)}{info}')
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def status(cls, command: str, message: str, color: str = 'green'):
|
||||||
|
"""
|
||||||
|
状态动画
|
||||||
|
|
||||||
|
:param command: 命令名
|
||||||
|
:param message: 消息
|
||||||
|
:param color: 颜色
|
||||||
|
:return:
|
||||||
|
"""
|
||||||
|
return cls._rich_console.status(
|
||||||
|
f'[bold underline yellow][{command}][/bold underline yellow][{color}]{message}[/{color}]',
|
||||||
|
spinner='earth')
|
||||||
|
|
||||||
|
|
||||||
|
# def ParamColor(**kwargs) -> str:
|
||||||
|
# """参数颜色"""
|
||||||
|
# text = ''.join(f'{key}=<m>{value}</m> ' for key, value in kwargs.items())
|
||||||
|
# return text.strip()
|
||||||
|
|
||||||
|
|
||||||
|
def CommandColor(command: Union[str, int]):
|
||||||
|
"""命令颜色"""
|
||||||
|
return f'<u><y>[{command}]</y></u>'
|
||||||
|
|
||||||
|
|
||||||
|
def Underline(text: Union[str, int]) -> str:
|
||||||
|
"""下划线"""
|
||||||
|
return f'<u>{text}</u>'
|
||||||
|
|
||||||
|
|
||||||
|
def Yellow(text: Union[str, int]) -> str:
|
||||||
|
"""黄色"""
|
||||||
|
return f'<y>{text}</y>'
|
||||||
|
|
||||||
|
|
||||||
|
def Green(text: Union[str, int]) -> str:
|
||||||
|
"""绿色"""
|
||||||
|
return f'<g>{text}</g>'
|
||||||
|
|
||||||
|
|
||||||
|
def Red(text: Union[str, int]) -> str:
|
||||||
|
"""红色"""
|
||||||
|
return f'<r>{text}</r>'
|
||||||
|
|
||||||
|
|
||||||
|
def Magenta(text: Union[str, int]) -> str:
|
||||||
|
"""品红色"""
|
||||||
|
return f'<m>{text}</m>'
|
||||||
|
|
||||||
|
|
||||||
|
def Bold(text: Union[str, int]) -> str:
|
||||||
|
"""粗体"""
|
||||||
|
return f'<b>{text}</b>'
|
||||||
|
|
||||||
|
|
||||||
|
__all__ = [
|
||||||
|
'logger',
|
||||||
|
# 'ParamColor',
|
||||||
|
'CommandColor',
|
||||||
|
'Underline',
|
||||||
|
'Yellow',
|
||||||
|
'Green',
|
||||||
|
'Red',
|
||||||
|
'Magenta',
|
||||||
|
'Bold',
|
||||||
|
'init_logger'
|
||||||
|
]
|
349
LittlePaimon/scheduler.py
Normal file
349
LittlePaimon/scheduler.py
Normal file
@ -0,0 +1,349 @@
|
|||||||
|
import logging
|
||||||
|
from datetime import datetime, date, timedelta
|
||||||
|
from typing import Optional, Callable, Union
|
||||||
|
from apscheduler.schedulers.asyncio import AsyncIOScheduler
|
||||||
|
from apscheduler.triggers.cron import CronTrigger
|
||||||
|
from nonebot.log import LoguruHandler
|
||||||
|
from .hook import on_startup, on_shutdown
|
||||||
|
from .log import logger, Yellow
|
||||||
|
|
||||||
|
scheduler = AsyncIOScheduler()
|
||||||
|
scheduler.configure({'apscheduler.timezone': 'Asia/Shanghai'})
|
||||||
|
|
||||||
|
|
||||||
|
@on_startup()
|
||||||
|
async def _start_scheduler():
|
||||||
|
if not scheduler.running:
|
||||||
|
scheduler.start()
|
||||||
|
logger.info('定时任务', Yellow('启动定时任务'))
|
||||||
|
|
||||||
|
|
||||||
|
@on_shutdown()
|
||||||
|
async def _shutdown_scheduler():
|
||||||
|
if scheduler.running:
|
||||||
|
scheduler.shutdown()
|
||||||
|
logger.info('定时任务', Yellow('关闭定时任务'))
|
||||||
|
|
||||||
|
|
||||||
|
aps_logger = logging.getLogger('apscheduler')
|
||||||
|
aps_logger.setLevel(30)
|
||||||
|
aps_logger.handlers.clear()
|
||||||
|
aps_logger.addHandler(LoguruHandler())
|
||||||
|
|
||||||
|
|
||||||
|
def add_interval_job(
|
||||||
|
func: Callable,
|
||||||
|
job_id: Optional[str] = None,
|
||||||
|
weeks: int = 0,
|
||||||
|
days: int = 0,
|
||||||
|
hours: int = 0,
|
||||||
|
minutes: int = 0,
|
||||||
|
seconds: int = 0,
|
||||||
|
start_date: Union[date, datetime, str] = None,
|
||||||
|
end_date: Union[date, datetime, str] = None,
|
||||||
|
misfire_grace_time: Optional[int] = None,
|
||||||
|
*args,
|
||||||
|
**kwargs) -> str:
|
||||||
|
"""
|
||||||
|
使指定函数每隔一段时间执行一次
|
||||||
|
|
||||||
|
:param func: 要执行的函数
|
||||||
|
:param job_id: 任务ID,不设置则自动生成
|
||||||
|
:param weeks: 周数
|
||||||
|
:param days: 天数
|
||||||
|
:param hours: 小时
|
||||||
|
:param minutes: 分钟
|
||||||
|
:param seconds: 秒
|
||||||
|
:param start_date: 允许执行的开始日期,不设置则无限制
|
||||||
|
:param end_date: 允许执行的结束日期,不设置则无限制
|
||||||
|
:param misfire_grace_time: 容错时间
|
||||||
|
:param args: 其它传给函数的位置参数
|
||||||
|
:param kwargs: 其它传给函数的关键字参数
|
||||||
|
:return: 任务ID,可以用于取消任务
|
||||||
|
"""
|
||||||
|
job = scheduler.add_job(func,
|
||||||
|
'interval',
|
||||||
|
weeks=weeks,
|
||||||
|
days=days,
|
||||||
|
hours=hours,
|
||||||
|
minutes=minutes,
|
||||||
|
seconds=seconds,
|
||||||
|
start_date=start_date,
|
||||||
|
end_date=end_date,
|
||||||
|
misfire_grace_time=misfire_grace_time,
|
||||||
|
args=args,
|
||||||
|
kwargs=kwargs,
|
||||||
|
id=job_id)
|
||||||
|
return job.id
|
||||||
|
|
||||||
|
|
||||||
|
def on_interval_time(job_id: Optional[str] = None,
|
||||||
|
weeks: int = 0,
|
||||||
|
days: int = 0,
|
||||||
|
hours: int = 0,
|
||||||
|
minutes: int = 0,
|
||||||
|
seconds: int = 0,
|
||||||
|
start_date: Union[date, datetime, str] = None,
|
||||||
|
end_date: Union[date, datetime, str] = None,
|
||||||
|
misfire_grace_time: Optional[int] = None,
|
||||||
|
*args,
|
||||||
|
**kwargs) -> Callable:
|
||||||
|
"""
|
||||||
|
装饰器
|
||||||
|
装饰一个函数,使其在指定间隔时间执行
|
||||||
|
|
||||||
|
示例:
|
||||||
|
每1分钟执行一次打印test
|
||||||
|
```
|
||||||
|
@on_interval_time(minutes=1)
|
||||||
|
async def test():
|
||||||
|
print('test')
|
||||||
|
```
|
||||||
|
|
||||||
|
:param job_id: 任务ID,不设置则自动生成
|
||||||
|
:param weeks: 周数
|
||||||
|
:param days: 天数
|
||||||
|
:param hours: 小时
|
||||||
|
:param minutes: 分钟
|
||||||
|
:param seconds: 秒
|
||||||
|
:param start_date: 允许执行的开始日期,不设置则无限制
|
||||||
|
:param end_date: 允许执行的结束日期,不设置则无限制
|
||||||
|
:param misfire_grace_time: 容错时间
|
||||||
|
:param args: 其它传给函数的位置参数
|
||||||
|
:param kwargs: 其它传给函数的关键字参数
|
||||||
|
"""
|
||||||
|
|
||||||
|
def inner(func: Callable) -> Callable:
|
||||||
|
add_interval_job(func,
|
||||||
|
job_id=job_id,
|
||||||
|
weeks=weeks,
|
||||||
|
days=days,
|
||||||
|
hours=hours,
|
||||||
|
minutes=minutes,
|
||||||
|
seconds=seconds,
|
||||||
|
start_date=start_date,
|
||||||
|
end_date=end_date,
|
||||||
|
misfire_grace_time=misfire_grace_time,
|
||||||
|
args=args,
|
||||||
|
kwargs=kwargs)
|
||||||
|
return func
|
||||||
|
|
||||||
|
return inner
|
||||||
|
|
||||||
|
|
||||||
|
def add_cron_job(
|
||||||
|
func: Callable,
|
||||||
|
job_id: Optional[str] = None,
|
||||||
|
crontab: Optional[str] = None,
|
||||||
|
year: Union[int, str] = '*',
|
||||||
|
month: Union[int, str] = '*',
|
||||||
|
week: Union[int, str] = '*',
|
||||||
|
day: Union[int, str] = '*',
|
||||||
|
day_of_week: Union[int, str] = '*',
|
||||||
|
hour: Union[int, str] = '*',
|
||||||
|
minute: Union[int, str] = '*',
|
||||||
|
second: Union[int, str] = '*',
|
||||||
|
start_date: Union[date, datetime, str] = None,
|
||||||
|
end_date: Union[date, datetime, str] = None,
|
||||||
|
misfire_grace_time: Optional[int] = None,
|
||||||
|
*args,
|
||||||
|
**kwargs) -> str:
|
||||||
|
"""
|
||||||
|
使指定函数,按cron表达式来定期执行
|
||||||
|
|
||||||
|
:param func: 要执行的函数
|
||||||
|
:param job_id: 任务ID,不设置则自动生成
|
||||||
|
:param crontab: crontab表达式,如有,则后续year等参数将忽略
|
||||||
|
:param year: 年
|
||||||
|
:param month: 月
|
||||||
|
:param week: 周
|
||||||
|
:param day_of_week: 周内第几天或者星期几
|
||||||
|
:param day: 天
|
||||||
|
:param hour: 小时
|
||||||
|
:param minute: 分钟
|
||||||
|
:param second: 秒
|
||||||
|
:param start_date: 允许执行的开始日期,不设置则无限制
|
||||||
|
:param end_date: 允许执行的结束日期,不设置则无限制
|
||||||
|
:param misfire_grace_time: 容错时间
|
||||||
|
:param args: 其它传给函数的位置参数
|
||||||
|
:param kwargs: 其它传给函数的关键字参数
|
||||||
|
:return: 任务ID,可用于取消任务
|
||||||
|
"""
|
||||||
|
if crontab:
|
||||||
|
job = scheduler.add_job(func,
|
||||||
|
CronTrigger.from_crontab(crontab),
|
||||||
|
misfire_grace_time=misfire_grace_time,
|
||||||
|
args=args,
|
||||||
|
kwargs=kwargs,
|
||||||
|
id=job_id)
|
||||||
|
else:
|
||||||
|
job = scheduler.add_job(func,
|
||||||
|
'cron',
|
||||||
|
year=year,
|
||||||
|
month=month,
|
||||||
|
day=day,
|
||||||
|
week=week,
|
||||||
|
day_of_week=day_of_week,
|
||||||
|
hour=hour,
|
||||||
|
minute=minute,
|
||||||
|
second=second,
|
||||||
|
start_date=start_date,
|
||||||
|
end_date=end_date,
|
||||||
|
misfire_grace_time=misfire_grace_time,
|
||||||
|
args=args,
|
||||||
|
kwargs=kwargs,
|
||||||
|
id=job_id)
|
||||||
|
return job.id
|
||||||
|
|
||||||
|
|
||||||
|
def on_cron_time(
|
||||||
|
job_id: Optional[str] = None,
|
||||||
|
crontab: Optional[str] = None,
|
||||||
|
year: Union[int, str] = '*',
|
||||||
|
month: Union[int, str] = '*',
|
||||||
|
week: Union[int, str] = '*',
|
||||||
|
day: Union[int, str] = '*',
|
||||||
|
day_of_week: Union[int, str] = '*',
|
||||||
|
hour: Union[int, str] = '*',
|
||||||
|
minute: Union[int, str] = '*',
|
||||||
|
second: Union[int, str] = '*',
|
||||||
|
start_date: Union[date, datetime, str] = None,
|
||||||
|
end_date: Union[date, datetime, str] = None,
|
||||||
|
misfire_grace_time: Optional[int] = None,
|
||||||
|
*args,
|
||||||
|
**kwargs):
|
||||||
|
"""
|
||||||
|
装饰器
|
||||||
|
装饰一个函数,使其按cron表达式时间执行
|
||||||
|
|
||||||
|
示例:
|
||||||
|
在6,7,8,11,12月的第3个周五的1,2,3点打印test
|
||||||
|
```
|
||||||
|
@on_cron_time(month='6-8,11-12', day='3rd fri', hour='0-3')
|
||||||
|
async def test():
|
||||||
|
print('test')
|
||||||
|
```
|
||||||
|
|
||||||
|
:param job_id: 任务ID,不设置则自动生成
|
||||||
|
:param crontab: crontab表达式,如有,则后续year等参数将忽略
|
||||||
|
:param year: 年
|
||||||
|
:param month: 月
|
||||||
|
:param week: 周
|
||||||
|
:param day_of_week: 周内第几天或者星期几
|
||||||
|
:param day: 天
|
||||||
|
:param hour: 小时
|
||||||
|
:param minute: 分钟
|
||||||
|
:param second: 秒
|
||||||
|
:param start_date: 允许执行的开始日期,不设置则无限制
|
||||||
|
:param end_date: 允许执行的结束日期,不设置则无限制
|
||||||
|
:param misfire_grace_time: 容错时间
|
||||||
|
:param args: 其它传给函数的位置参数
|
||||||
|
:param kwargs: 其它传给函数的关键字参数
|
||||||
|
"""
|
||||||
|
|
||||||
|
def inner(func: Callable) -> Callable:
|
||||||
|
add_cron_job(func,
|
||||||
|
job_id=job_id,
|
||||||
|
crontab=crontab,
|
||||||
|
year=year,
|
||||||
|
month=month,
|
||||||
|
day=day,
|
||||||
|
week=week,
|
||||||
|
day_of_week=day_of_week,
|
||||||
|
hour=hour,
|
||||||
|
minute=minute,
|
||||||
|
second=second,
|
||||||
|
start_date=start_date,
|
||||||
|
end_date=end_date,
|
||||||
|
misfire_grace_time=misfire_grace_time,
|
||||||
|
args=args,
|
||||||
|
kwargs=kwargs)
|
||||||
|
return func
|
||||||
|
|
||||||
|
return inner
|
||||||
|
|
||||||
|
|
||||||
|
def add_date_job(
|
||||||
|
func: Callable,
|
||||||
|
run_date: Union[str, datetime],
|
||||||
|
job_id: Optional[str] = None,
|
||||||
|
misfire_grace_time: Optional[int] = None,
|
||||||
|
*args,
|
||||||
|
**kwargs) -> str:
|
||||||
|
"""
|
||||||
|
使一个函数在指定时间执行
|
||||||
|
|
||||||
|
:param func: 要执行的函数
|
||||||
|
:param run_date: 执行时间
|
||||||
|
:param job_id: 任务ID,不设置则自动生成
|
||||||
|
:param misfire_grace_time: 容错时间
|
||||||
|
:param args: 其它传给函数的位置参数
|
||||||
|
:param kwargs: 其它传给函数的关键字参数
|
||||||
|
:return: 任务ID,可用于取消任务
|
||||||
|
"""
|
||||||
|
job = scheduler.add_job(func,
|
||||||
|
'date',
|
||||||
|
run_date=run_date,
|
||||||
|
misfire_grace_time=misfire_grace_time,
|
||||||
|
args=args,
|
||||||
|
kwargs=kwargs,
|
||||||
|
id=job_id)
|
||||||
|
return job.id
|
||||||
|
|
||||||
|
|
||||||
|
def on_date_time(
|
||||||
|
run_date: Union[str, datetime],
|
||||||
|
job_id: Optional[str] = None,
|
||||||
|
misfire_grace_time: Optional[int] = None,
|
||||||
|
*args,
|
||||||
|
**kwargs):
|
||||||
|
"""
|
||||||
|
装饰器
|
||||||
|
装饰一个函数,使其在指定时间执行
|
||||||
|
|
||||||
|
:param run_date: 执行时间
|
||||||
|
:param job_id: 任务ID,不设置则自动生成
|
||||||
|
:param misfire_grace_time: 容错时间
|
||||||
|
:param args: 其它传给函数的位置参数
|
||||||
|
:param kwargs: 其它传给函数的关键字参数
|
||||||
|
"""
|
||||||
|
|
||||||
|
def inner(func: Callable) -> Callable:
|
||||||
|
add_date_job(func,
|
||||||
|
run_date=run_date,
|
||||||
|
misfire_grace_time=misfire_grace_time,
|
||||||
|
args=args,
|
||||||
|
kwargs=kwargs,
|
||||||
|
job_id=job_id)
|
||||||
|
return func
|
||||||
|
|
||||||
|
return inner
|
||||||
|
|
||||||
|
|
||||||
|
def add_run_after_job(seconds: int,
|
||||||
|
func: Callable,
|
||||||
|
*args,
|
||||||
|
**kwargs) -> str:
|
||||||
|
"""
|
||||||
|
使一个函数在指定秒数后执行
|
||||||
|
|
||||||
|
:param seconds: 秒数
|
||||||
|
:param func: 函数
|
||||||
|
:param args: 其它传给函数的位置参数
|
||||||
|
:param kwargs: 其它传给函数的关键字参数
|
||||||
|
:return: 任务ID,可用于取消任务
|
||||||
|
"""
|
||||||
|
a = scheduler.add_job(func, 'date', run_date=datetime.now() + timedelta(seconds=seconds), args=args, kwargs=kwargs)
|
||||||
|
return a.id
|
||||||
|
|
||||||
|
|
||||||
|
__all__ = [
|
||||||
|
'add_interval_job',
|
||||||
|
'add_cron_job',
|
||||||
|
'add_date_job',
|
||||||
|
'add_run_after_job',
|
||||||
|
'on_interval_time',
|
||||||
|
'on_cron_time',
|
||||||
|
'on_date_time',
|
||||||
|
'scheduler'
|
||||||
|
]
|
16
bot.py
Normal file
16
bot.py
Normal file
@ -0,0 +1,16 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
import nonebot
|
||||||
|
from nonebot.adapters.onebot.v11 import Adapter as ONEBOT_V11Adapter
|
||||||
|
from LittlePaimon.log import init_logger
|
||||||
|
|
||||||
|
|
||||||
|
init_logger()
|
||||||
|
nonebot.init()
|
||||||
|
app = nonebot.get_asgi()
|
||||||
|
|
||||||
|
driver = nonebot.get_driver()
|
||||||
|
driver.register_adapter(ONEBOT_V11Adapter)
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
nonebot.run(app='__mp_main__:app')
|
2513
poetry.lock
generated
Normal file
2513
poetry.lock
generated
Normal file
File diff suppressed because it is too large
Load Diff
52
pyproject.toml
Normal file
52
pyproject.toml
Normal file
@ -0,0 +1,52 @@
|
|||||||
|
[tool.poetry]
|
||||||
|
name = "LittlePaimon"
|
||||||
|
version = "4.0"
|
||||||
|
description = "小派蒙!原神qq群机器人,基于NoneBot2的UID查询、抽卡导出分析、模拟抽卡、实时便签、札记等多功能小助手。"
|
||||||
|
authors = ["惜月 <277073121@qq.com>"]
|
||||||
|
license = "AGPL"
|
||||||
|
|
||||||
|
[[tool.poetry.source]]
|
||||||
|
name = "aliyun"
|
||||||
|
default = true
|
||||||
|
url = "https://mirrors.aliyun.com/pypi/simple/"
|
||||||
|
|
||||||
|
[tool.poetry.dependencies]
|
||||||
|
python = "^3.8.1"
|
||||||
|
nonebot2 = {extras = ["fastapi"], version = "^2.0.0-rc.2"}
|
||||||
|
nonebot-adapter-onebot = "^2.1"
|
||||||
|
beautifulsoup4 = "^4.10.0"
|
||||||
|
httpx = "^0.23.0"
|
||||||
|
lxml = "^4.8.0"
|
||||||
|
Pillow = "^9.1.0"
|
||||||
|
tortoise-orm = "^0.19.2"
|
||||||
|
"ruamel.yaml" = "^0.17.21"
|
||||||
|
ujson = "^5.4.0"
|
||||||
|
jieba = "^0.42.1"
|
||||||
|
gitpython = "^3.1.27"
|
||||||
|
python-jose = "^3.3.0"
|
||||||
|
amis-python = "^1.0.5"
|
||||||
|
psutil = "^5.9.3"
|
||||||
|
playwright = "^1.27.1"
|
||||||
|
qrcode = "^7.3.1"
|
||||||
|
rich = "^13.3.1"
|
||||||
|
apscheduler = "^3.9.1"
|
||||||
|
|
||||||
|
[tool.poetry.dev-dependencies]
|
||||||
|
pre-commit = "^3.0.2"
|
||||||
|
black = "^22.12.0"
|
||||||
|
|
||||||
|
[tool.black]
|
||||||
|
line-length = 79
|
||||||
|
target-version = ["py38", "py39", "py310"]
|
||||||
|
include = '\.pyi?$'
|
||||||
|
skip-string-normalization = true
|
||||||
|
extend-exclude = '''
|
||||||
|
'''
|
||||||
|
|
||||||
|
[tool.nonebot]
|
||||||
|
plugins = []
|
||||||
|
plugin_dirs = []
|
||||||
|
|
||||||
|
[build-system]
|
||||||
|
requires = ["poetry_core>=1.0.0"]
|
||||||
|
build-backend = "poetry.core.masonry.api"
|
Loading…
Reference in New Issue
Block a user