53 lines
1.5 KiB
Python
53 lines
1.5 KiB
Python
import sys
|
|
from datetime import datetime
|
|
|
|
from pyrogram import Client
|
|
from utils import config
|
|
|
|
common_params = {
|
|
'api_id': config.api_id,
|
|
'api_hash': config.api_hash,
|
|
'hide_password': True,
|
|
'test_mode': config.test_server,
|
|
}
|
|
|
|
if __name__ == '__main__':
|
|
if config.STRINGSESSION:
|
|
common_params['session_string'] = config.STRINGSESSION
|
|
|
|
app = Client('my_account', **common_params)
|
|
|
|
if config.db_type in ['mongo', 'mongodb']:
|
|
from pymongo import MongoClient, errors
|
|
|
|
db = MongoClient(config.db_url)
|
|
try:
|
|
db.server_info()
|
|
except errors.ConnectionFailure as e:
|
|
raise RuntimeError(
|
|
"MongoDB server isn't available! "
|
|
f'Provided url: {config.db_url}. '
|
|
'Enter valid URL and restart installation'
|
|
) from e
|
|
|
|
install_type = sys.argv[1] if len(sys.argv) > 1 else '3'
|
|
if install_type == '1':
|
|
restart = 'pm2 restart Moon'
|
|
elif install_type == '2':
|
|
restart = 'sudo systemctl restart Moon'
|
|
else:
|
|
restart = 'cd Moon-Userbot/ && python main.py'
|
|
|
|
app.start()
|
|
try:
|
|
app.send_message(
|
|
'me',
|
|
f'<b>[{datetime.now()}] Userbot launched! \n'
|
|
'Custom modules: @moonub_modules\n'
|
|
f'For restart, enter:</b>\n'
|
|
f'<code>{restart}</code>',
|
|
)
|
|
except Exception as e:
|
|
print(f'[ERROR]: Sending Message to me failed! {e}')
|
|
app.stop()
|