2600b8098f
`subprocess.run` uses a default of `check=False`, which means that a nonzero exit code will be ignored by default, instead of raising an exception. You can ignore this issue if this behaviour is intended.
135 lines
4.2 KiB
Python
135 lines
4.2 KiB
Python
# Moon-Userbot - telegram userbot
|
|
# Copyright (C) 2020-present Moon Userbot Organization
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
# import asyncio
|
|
import os
|
|
import logging
|
|
|
|
import sqlite3
|
|
import platform
|
|
import subprocess
|
|
from pathlib import Path
|
|
|
|
from pyrogram import Client, idle, errors
|
|
from pyrogram.enums.parse_mode import ParseMode
|
|
from pyrogram.raw.functions.account import GetAuthorizations, DeleteAccount
|
|
|
|
from utils import config
|
|
from utils.db import db
|
|
from utils.misc import gitrepo, userbot_version
|
|
from utils.scripts import restart, load_module
|
|
|
|
script_path = os.path.dirname(os.path.realpath(__file__))
|
|
if script_path != os.getcwd():
|
|
os.chdir(script_path)
|
|
|
|
common_params = {
|
|
"api_id": config.api_id,
|
|
"api_hash": config.api_hash,
|
|
"hide_password": True,
|
|
"workdir": script_path,
|
|
"app_version": userbot_version,
|
|
"device_model": f"Moon-Userbot @ {gitrepo.head.commit.hexsha[:7]}",
|
|
"system_version": platform.version() + " " + platform.machine(),
|
|
"sleep_threshold": 30,
|
|
"test_mode": config.test_server,
|
|
"parse_mode": ParseMode.HTML,
|
|
}
|
|
|
|
if config.STRINGSESSION:
|
|
common_params["session_string"] = config.STRINGSESSION
|
|
|
|
app = Client("my_account", **common_params)
|
|
|
|
async def main():
|
|
logging.basicConfig(
|
|
format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
|
|
handlers=[logging.FileHandler("moonlogs.txt"), logging.StreamHandler()],
|
|
level=logging.INFO,
|
|
)
|
|
DeleteAccount.__new__ = None
|
|
|
|
try:
|
|
await app.start()
|
|
except sqlite3.OperationalError as e:
|
|
if str(e) == "database is locked" and os.name == "posix":
|
|
logging.warning(
|
|
"Session file is locked. Trying to kill blocking process..."
|
|
)
|
|
subprocess.run(["fuser", "-k", "my_account.session"], check=True)
|
|
restart()
|
|
raise
|
|
except (errors.NotAcceptable, errors.Unauthorized) as e:
|
|
logging.error(
|
|
f"{e.__class__.__name__}: {e}\n"
|
|
f"Moving session file to my_account.session-old..."
|
|
)
|
|
os.rename("./my_account.session", "./my_account.session-old")
|
|
restart()
|
|
|
|
success_modules = 0
|
|
failed_modules = 0
|
|
|
|
for path in Path("modules").rglob("*.py"):
|
|
try:
|
|
await load_module(
|
|
path.stem, app, core="custom_modules" not in path.parent.parts
|
|
)
|
|
except Exception:
|
|
logging.warning(f"Can't import module {path.stem}", exc_info=True)
|
|
failed_modules += 1
|
|
else:
|
|
success_modules += 1
|
|
|
|
logging.info(f"Imported {success_modules} modules")
|
|
if failed_modules:
|
|
logging.warning(f"Failed to import {failed_modules} modules")
|
|
|
|
if info := db.get("core.updater", "restart_info"):
|
|
text = {
|
|
"restart": "<b>Restart completed!</b>",
|
|
"update": "<b>Update process completed!</b>",
|
|
}[info["type"]]
|
|
try:
|
|
await app.edit_message_text(
|
|
info["chat_id"], info["message_id"], text
|
|
)
|
|
except errors.RPCError:
|
|
pass
|
|
db.remove("core.updater", "restart_info")
|
|
|
|
# required for sessionkiller module
|
|
if db.get("core.sessionkiller", "enabled", False):
|
|
db.set(
|
|
"core.sessionkiller",
|
|
"auths_hashes",
|
|
[
|
|
auth.hash
|
|
for auth in (
|
|
await app.invoke(GetAuthorizations())
|
|
).authorizations
|
|
],
|
|
)
|
|
|
|
logging.info("Moon-Userbot started!")
|
|
|
|
await idle()
|
|
|
|
await app.stop()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
app.run(main())
|