mirror of
https://github.com/PaiGramTeam/PaiGram.git
synced 2024-09-28 11:48:21 +00:00
68 lines
2.4 KiB
Python
68 lines
2.4 KiB
Python
import asyncio
|
|
import os
|
|
from sys import executable
|
|
|
|
from aiofiles import open as async_open
|
|
from telegram import Message, Update
|
|
from telegram.error import NetworkError
|
|
from telegram.ext import CallbackContext
|
|
|
|
from core.plugin import Plugin, handler
|
|
from utils.helpers import execute
|
|
from utils.log import logger
|
|
|
|
try:
|
|
import ujson as jsonlib
|
|
|
|
except ImportError:
|
|
import json as jsonlib
|
|
|
|
current_dir = os.getcwd()
|
|
|
|
UPDATE_DATA = os.path.join(current_dir, "data", "update.json")
|
|
|
|
|
|
class UpdatePlugin(Plugin):
|
|
def __init__(self):
|
|
self.lock = asyncio.Lock()
|
|
|
|
async def initialize(self) -> None:
|
|
if os.path.exists(UPDATE_DATA):
|
|
async with async_open(UPDATE_DATA) as file:
|
|
data = jsonlib.loads(await file.read())
|
|
try:
|
|
reply_text = Message.de_json(data, self.application.telegram.bot)
|
|
await reply_text.edit_text("重启成功")
|
|
except NetworkError as exc:
|
|
logger.error("编辑消息出现错误 %s", exc.message)
|
|
except jsonlib.JSONDecodeError:
|
|
logger.error("JSONDecodeError")
|
|
except KeyError as exc:
|
|
logger.error("编辑消息出现错误", exc_info=exc)
|
|
os.remove(UPDATE_DATA)
|
|
|
|
@handler.command("update", block=False, admin=True)
|
|
async def update(self, update: Update, context: CallbackContext):
|
|
user = update.effective_user
|
|
message = update.effective_message
|
|
args = self.get_args(context)
|
|
logger.info("用户 %s[%s] update命令请求", user.full_name, user.id)
|
|
if self.lock.locked():
|
|
await message.reply_text("程序正在更新 请勿重复操作")
|
|
return
|
|
async with self.lock:
|
|
reply_text = await message.reply_text("正在更新")
|
|
logger.info("正在更新代码")
|
|
await execute("git fetch --all")
|
|
if len(args) > 0:
|
|
await execute("git reset --hard origin/main")
|
|
await execute("git pull --all")
|
|
await execute("git submodule update")
|
|
if len(args) > 1:
|
|
await execute(f"{executable} -m poetry install --extras all")
|
|
logger.info("更新成功 正在重启")
|
|
await reply_text.edit_text("更新成功 正在重启")
|
|
async with async_open(UPDATE_DATA, mode="w", encoding="utf-8") as file:
|
|
await file.write(reply_text.to_json())
|
|
raise SystemExit
|