evanferrao
/
dump-mirrorbot
mirror of https://github.com/Box-boi/dump-mirrorbot.git
1
0
Fork 0
Code Issues Actions Packages Projects Releases Wiki Activity

change logs in update

This commit is contained in:
Juned KH 2021-08-22 11:07:59 +05:30 committed by GitHub
parent 1da86c8b98
commit c83d5768a9
Signed by untrusted user: GitHub
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 44 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

72
bot/modules/updates.py
View File

@ -13,19 +13,26 @@ from git.exc import GitCommandError, InvalidGitRepositoryError, NoSuchPathError
from pyrogram import filters
from bot import app, OWNER_ID, UPSTREAM_REPO, UPSTREAM_BRANCH, bot
from bot import app, OWNER_ID, UPSTREAM_REPO, UPSTREAM_BRANCH
from bot.helper import get_text, HEROKU_URL
from bot.helper.telegram_helper.bot_commands import BotCommands
REPO_ = UPSTREAM_REPO
BRANCH_ = UPSTREAM_BRANCH
def gen_chlog(repo, diff):
ch_log = ''
d_form = "%d/%m/%y"
for c in repo.iter_commits(diff):
ch_log += f'• [{c.committed_datetime.strftime(d_form)}]: {c.summary} **{c.author}**\n'
return ch_log
# Update Command
@app.on_message(filters.command([BotCommands.UpdateCommand, f'{BotCommands.UpdateCommand}@{bot.username}']) & filters.user(OWNER_ID))
async def update_it(client, message):
msg_ = await message.reply_text("`Updating Please Wait!`")
text = get_text(message)
try:
repo = Repo()
except GitCommandError:
@ -52,32 +59,41 @@ async def update_it(client, message):
pass
ups_rem = repo.remote("upstream")
ups_rem.fetch(UPSTREAM_BRANCH)
if not HEROKU_URL:
try:
ups_rem.pull(UPSTREAM_BRANCH)
except GitCommandError:
repo.git.reset("--hard", "FETCH_HEAD")
subprocess.run(["pip3", "install", "--no-cache-dir", "-r", "requirements.txt"])
await msg_.edit("`Updated Sucessfully! Give Me Some Time To Restart!`")
with open("./aria.sh", 'rb') as file:
script = file.read()
subprocess.call("./aria.sh", shell=True)
args = [sys.executable, "-m", "bot"]
execle(sys.executable, *args, environ)
exit()
clogs = gen_chlog(repo, f'HEAD..upstream/{UPSTREAM_BRANCH}')
if not clogs:
await msg_.edit(f"Bot up-to-date with **{UPSTREAM_BRANCH}**", parse_mode="Markdown")
return
else:
await msg_.edit("`Heroku Detected! Pushing, Please wait!`")
ups_rem.fetch(UPSTREAM_BRANCH)
repo.git.reset("--hard", "FETCH_HEAD")
if "heroku" in repo.remotes:
remote = repo.remote("heroku")
remote.set_url(HEROKU_URL)
if "now" == text:
await msg_.edit(f"`Bot Updating with` **{UPSTREAM_BRANCH}** `Branch Please Wait...`", parse_mode="Markdown")
if not HEROKU_URL:
try:
ups_rem.pull(UPSTREAM_BRANCH)
except GitCommandError:
repo.git.reset("--hard", "FETCH_HEAD")
await subprocess.run(["pip3","install","--no-cache-dir","-r","requirements.txt"])
await msg_.edit("`Updated Sucessfully! Give Me Some Time To Restart!`")
with open("./aria.sh", 'rb') as file:
script = file.read()
subprocess.call("./aria.sh", shell=True)
args = [sys.executable, "-m", "bot"]
execle(sys.executable, *args, environ)
else:
remote = repo.create_remote("heroku", HEROKU_URL)
try:
remote.push(refspec="HEAD:refs/heads/master", force=True)
except BaseException as error:
await msg_.edit(f"**Updater Error** \nTraceBack : `{error}`")
return repo.__del__()
await msg_.edit(f"`Updated Sucessfully! \n\nCheck your config with` `/{BotCommands.ConfigMenuCommand}`")
await msg_.edit("`Heroku Detected! Pushing, Please wait!`")
ups_rem.fetch(UPSTREAM_BRANCH)
repo.git.reset("--hard", "FETCH_HEAD")
if "heroku" in repo.remotes:
remote = repo.remote("heroku")
remote.set_url(HEROKU_URL)
else:
remote = repo.create_remote("heroku", HEROKU_URL)
try:
remote.push(refspec="HEAD:refs/heads/master", force=True)
except BaseException as error:
await msg_.edit(f"**Updater Error** \nTraceBack : `{error}`")
return repo.__del__()
await msg_.edit(f"`Updated Sucessfully! \n\nCheck your config with` `/{BotCommands.ConfigMenuCommand}`")
else:
await msg_.edit(f"**New Update Available**\n\nFrom [REPO]({REPO_})\nCHANGELOG:\n\n{clogs}\n\nDo `/update now` to Update BOT.", parse_mode="Markdown",disable_web_page_preview=True)
return