Improved imports
This commit is contained in:
parent
19b83c0631
commit
4fd4f0a6a4
52
bwtbot.py
52
bwtbot.py
@ -1,14 +1,15 @@
|
|||||||
#-*- coding: utf-8 -*-
|
#-*- coding: utf-8 -*-
|
||||||
|
|
||||||
import traceback
|
from os import getpid, system
|
||||||
from pyrogram import Client, filters, idle
|
from subprocess import call
|
||||||
from pyrogram.types import ForceReply, BotCommand, BotCommandScopeChat
|
from pyrogram import filters
|
||||||
|
from pyrogram.client import Client
|
||||||
|
from pyrogram.sync import idle
|
||||||
|
from pyrogram.types import ForceReply, BotCommand, BotCommandScopeChat, Message
|
||||||
from pyrogram.enums.chat_action import ChatAction
|
from pyrogram.enums.chat_action import ChatAction
|
||||||
from functions import *
|
from functions import *
|
||||||
from modules.colors import *
|
from modules.colors import *
|
||||||
from modules.bwt import *
|
from modules.bwt import *
|
||||||
import subprocess
|
|
||||||
import os
|
|
||||||
|
|
||||||
config = jsonLoad("config.json")
|
config = jsonLoad("config.json")
|
||||||
|
|
||||||
@ -18,7 +19,7 @@ app = Client(config["bot_name"], api_id=config["api_id"], api_hash=config["api_h
|
|||||||
|
|
||||||
|
|
||||||
@app.on_message(~ filters.scheduled & filters.command(["setcard", "задать карту"], prefixes=["/", ""]))
|
@app.on_message(~ filters.scheduled & filters.command(["setcard", "задать карту"], prefixes=["/", ""]))
|
||||||
async def setcard(_, msg):
|
async def setcard(_: Client, msg: Message):
|
||||||
if userGet(msg.from_user.id, "context") is None:
|
if userGet(msg.from_user.id, "context") is None:
|
||||||
userSet(msg.from_user.id, "context", "set")
|
userSet(msg.from_user.id, "context", "set")
|
||||||
await msg.reply_text(string("send_number"), reply_markup=ForceReply(placeholder=string("enter_number")))
|
await msg.reply_text(string("send_number"), reply_markup=ForceReply(placeholder=string("enter_number")))
|
||||||
@ -27,7 +28,7 @@ async def setcard(_, msg):
|
|||||||
|
|
||||||
|
|
||||||
@app.on_message(~ filters.scheduled & filters.command(["cancel", "відміна"], prefixes=["/", ""]))
|
@app.on_message(~ filters.scheduled & filters.command(["cancel", "відміна"], prefixes=["/", ""]))
|
||||||
async def cancel(_, msg):
|
async def cancel(_: Client, msg: Message):
|
||||||
if userGet(msg.from_user.id, "context") is not None:
|
if userGet(msg.from_user.id, "context") is not None:
|
||||||
userReset(msg.from_user.id, "context")
|
userReset(msg.from_user.id, "context")
|
||||||
await msg.reply_text(string("cancel"))
|
await msg.reply_text(string("cancel"))
|
||||||
@ -36,7 +37,7 @@ async def cancel(_, msg):
|
|||||||
|
|
||||||
|
|
||||||
@app.on_message(~ filters.scheduled & filters.command(["resetcard", "забути картку"], prefixes=["/", ""]))
|
@app.on_message(~ filters.scheduled & filters.command(["resetcard", "забути картку"], prefixes=["/", ""]))
|
||||||
async def resetcard(_, msg):
|
async def resetcard(_: Client, msg: Message):
|
||||||
if userGet(msg.from_user.id, "context") is None:
|
if userGet(msg.from_user.id, "context") is None:
|
||||||
if "card" in jsonLoad("data/database.json")[str(msg.from_user.id)]:
|
if "card" in jsonLoad("data/database.json")[str(msg.from_user.id)]:
|
||||||
userReset(msg.from_user.id, "card")
|
userReset(msg.from_user.id, "card")
|
||||||
@ -50,7 +51,7 @@ async def resetcard(_, msg):
|
|||||||
|
|
||||||
|
|
||||||
@app.on_message(~ filters.scheduled & filters.command(["balance", "баланс"], prefixes=["/", ""]))
|
@app.on_message(~ filters.scheduled & filters.command(["balance", "баланс"], prefixes=["/", ""]))
|
||||||
async def balance(_, msg):
|
async def balance(_: Client, msg: Message):
|
||||||
if userGet(msg.from_user.id, "context") is None:
|
if userGet(msg.from_user.id, "context") is None:
|
||||||
try:
|
try:
|
||||||
if "card" in jsonLoad("data/database.json")[str(msg.from_user.id)]:
|
if "card" in jsonLoad("data/database.json")[str(msg.from_user.id)]:
|
||||||
@ -68,15 +69,16 @@ async def balance(_, msg):
|
|||||||
await msg.reply_text(string("card_not_linked").format(string("get_number")))
|
await msg.reply_text(string("card_not_linked").format(string("get_number")))
|
||||||
appendLog(f"User {str(msg.from_user.id)} tried to get balance without card set")
|
appendLog(f"User {str(msg.from_user.id)} tried to get balance without card set")
|
||||||
except Exception as exp:
|
except Exception as exp:
|
||||||
await msg.reply_text(string("error_occured").format(string("get_number")))
|
if msg.from_user.id != config["owner_id"]:
|
||||||
await app.send_message(owner_id, f"Error occured by {str(msg.from_user.id)}:\nException: `{exp}`\nTraceback: `{traceback.format_exc()}`")
|
await msg.reply_text(string("error_occured").format(string("get_number")))
|
||||||
|
await app.send_message(owner_id, f"Error occured by {str(msg.from_user.id)}:\nException: `{exp}`\nTraceback: `{format_exc()}`")
|
||||||
appendLog(f"User {str(msg.from_user.id)} could not get left water amount")
|
appendLog(f"User {str(msg.from_user.id)} could not get left water amount")
|
||||||
else:
|
else:
|
||||||
await msg.reply_text(string("cancel_first"))
|
await msg.reply_text(string("cancel_first"))
|
||||||
|
|
||||||
|
|
||||||
@app.on_message(~ filters.scheduled & filters.command(["topup", "refill", "поповнити"], prefixes=["/", ""]))
|
@app.on_message(~ filters.scheduled & filters.command(["topup", "refill", "поповнити"], prefixes=["/", ""]))
|
||||||
async def topup_cmd(_, msg):
|
async def topup_cmd(_: Client, msg: Message):
|
||||||
if userGet(msg.from_user.id, "context") is None:
|
if userGet(msg.from_user.id, "context") is None:
|
||||||
try:
|
try:
|
||||||
if "card" in jsonLoad("data/database.json")[str(msg.from_user.id)]:
|
if "card" in jsonLoad("data/database.json")[str(msg.from_user.id)]:
|
||||||
@ -87,13 +89,13 @@ async def topup_cmd(_, msg):
|
|||||||
await msg.reply_text(string("card_not_linked").format(string("get_number")))
|
await msg.reply_text(string("card_not_linked").format(string("get_number")))
|
||||||
appendLog(f"User {str(msg.from_user.id)} tried to request top up without card set")
|
appendLog(f"User {str(msg.from_user.id)} tried to request top up without card set")
|
||||||
except Exception as exp:
|
except Exception as exp:
|
||||||
await msg.reply_text(exp)
|
await msg.reply_text(str(exp))
|
||||||
else:
|
else:
|
||||||
await msg.reply_text(string("cancel_first"))
|
await msg.reply_text(string("cancel_first"))
|
||||||
|
|
||||||
|
|
||||||
@app.on_message(~ filters.scheduled & filters.command(["start", "help", "допомога"], prefixes=["/", ""]))
|
@app.on_message(~ filters.scheduled & filters.command(["start", "help", "допомога"], prefixes=["/", ""]))
|
||||||
async def help(_, msg):
|
async def help(_: Client, msg: Message):
|
||||||
if userGet(msg.from_user.id, "context") is None:
|
if userGet(msg.from_user.id, "context") is None:
|
||||||
await msg.reply_text(string("welcome").format(string("get_number")))
|
await msg.reply_text(string("welcome").format(string("get_number")))
|
||||||
if msg.from_user.language_code in jsonLoad("strings.json"):
|
if msg.from_user.language_code in jsonLoad("strings.json"):
|
||||||
@ -103,13 +105,13 @@ async def help(_, msg):
|
|||||||
else:
|
else:
|
||||||
await msg.reply_text(string("cancel_first"))
|
await msg.reply_text(string("cancel_first"))
|
||||||
|
|
||||||
pid = os.getpid()
|
pid = getpid()
|
||||||
|
|
||||||
@app.on_message(~ filters.scheduled & filters.command(["kill", "die", "shutdown"], prefixes="/"))
|
@app.on_message(~ filters.scheduled & filters.command(["kill", "die", "shutdown"], prefixes="/"))
|
||||||
async def kill(_, msg):
|
async def kill(_: Client, msg: Message):
|
||||||
if msg.from_user.id == owner_id:
|
if msg.from_user.id == owner_id:
|
||||||
await msg.reply_text(f"Shutting down bot with pid **{pid}**")
|
await msg.reply_text(f"Shutting down bot with pid **{pid}**")
|
||||||
os.system(f"kill -9 {pid}")
|
system(f"kill -9 {pid}")
|
||||||
|
|
||||||
|
|
||||||
@app.on_message(~ filters.scheduled)
|
@app.on_message(~ filters.scheduled)
|
||||||
@ -122,8 +124,8 @@ async def any_message_handler(app, msg):
|
|||||||
|
|
||||||
print(f'{nowtime()} {WHITE}Starting with PID {YELLOW}{pid}{RESET}')
|
print(f'{nowtime()} {WHITE}Starting with PID {YELLOW}{pid}{RESET}')
|
||||||
|
|
||||||
app.start()
|
app.start() # type: ignore
|
||||||
app.send_message(owner_id, f"Starting bot with pid **{pid}**")
|
app.send_message(owner_id, f"Starting bot with pid **{pid}**") # type: ignore
|
||||||
|
|
||||||
app.set_bot_commands([
|
app.set_bot_commands([
|
||||||
BotCommand("help", "Меню допомоги"),
|
BotCommand("help", "Меню допомоги"),
|
||||||
@ -133,7 +135,7 @@ app.set_bot_commands([
|
|||||||
BotCommand("resetcard", "Відв'язати картку"),
|
BotCommand("resetcard", "Відв'язати картку"),
|
||||||
BotCommand("cancel", "Відмінити операцію"),
|
BotCommand("cancel", "Відмінити операцію"),
|
||||||
],
|
],
|
||||||
language_code="uk")
|
language_code="uk") # type: ignore
|
||||||
|
|
||||||
app.set_bot_commands([
|
app.set_bot_commands([
|
||||||
BotCommand("help", "Меню допомоги"),
|
BotCommand("help", "Меню допомоги"),
|
||||||
@ -143,7 +145,7 @@ app.set_bot_commands([
|
|||||||
BotCommand("resetcard", "Відв'язати картку"),
|
BotCommand("resetcard", "Відв'язати картку"),
|
||||||
BotCommand("cancel", "Відмінити операцію"),
|
BotCommand("cancel", "Відмінити операцію"),
|
||||||
],
|
],
|
||||||
language_code="ru")
|
language_code="ru") # type: ignore
|
||||||
|
|
||||||
app.set_bot_commands([
|
app.set_bot_commands([
|
||||||
BotCommand("help", "Help menu"),
|
BotCommand("help", "Help menu"),
|
||||||
@ -152,7 +154,7 @@ app.set_bot_commands([
|
|||||||
BotCommand("setcard", "Link card"),
|
BotCommand("setcard", "Link card"),
|
||||||
BotCommand("resetcard", "Unlink card"),
|
BotCommand("resetcard", "Unlink card"),
|
||||||
BotCommand("cancel", "Cancel operation"),
|
BotCommand("cancel", "Cancel operation"),
|
||||||
])
|
]) # type: ignore
|
||||||
|
|
||||||
app.set_bot_commands([
|
app.set_bot_commands([
|
||||||
BotCommand("help", "Help menu"),
|
BotCommand("help", "Help menu"),
|
||||||
@ -163,11 +165,11 @@ app.set_bot_commands([
|
|||||||
BotCommand("shutdown", "Turn off the bot"),
|
BotCommand("shutdown", "Turn off the bot"),
|
||||||
BotCommand("cancel", "Cancel operation"),
|
BotCommand("cancel", "Cancel operation"),
|
||||||
],
|
],
|
||||||
scope=BotCommandScopeChat(chat_id=owner_id))
|
scope=BotCommandScopeChat(chat_id=owner_id)) # type: ignore
|
||||||
|
|
||||||
idle()
|
idle()
|
||||||
|
|
||||||
app.send_message(owner_id, f"Shutting down bot with pid **{pid}**")
|
app.send_message(owner_id, f"Shutting down bot with pid **{pid}**") # type: ignore
|
||||||
print(f'\n{nowtime()} {WHITE}Shutting down with PID {YELLOW}{pid}{RESET}')
|
print(f'\n{nowtime()} {WHITE}Shutting down with PID {YELLOW}{pid}{RESET}')
|
||||||
|
|
||||||
subprocess.call(f'kill -9 {pid}', shell=True)
|
call(f'kill -9 {pid}', shell=True)
|
||||||
|
38
functions.py
38
functions.py
@ -1,34 +1,29 @@
|
|||||||
#-*- coding: utf-8 -*-
|
#-*- coding: utf-8 -*-
|
||||||
|
|
||||||
import json
|
from os import makedirs, stat
|
||||||
import os
|
from gzip import open as gzipopen
|
||||||
import shutil
|
from shutil import copyfileobj
|
||||||
import gzip
|
from time import sleep
|
||||||
import time
|
from ujson import loads, dumps
|
||||||
from modules.colors import *
|
from modules.colors import *
|
||||||
from datetime import datetime
|
from datetime import datetime
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
|
|
||||||
path = Path(__file__).resolve().parent
|
users_path = "users/"
|
||||||
|
logs_folder = "logs/"
|
||||||
days_path = str(path)+"/assets/days/"
|
|
||||||
users_path = str(path)+"/users/"
|
|
||||||
logs_folder = str(path)+"/logs/"
|
|
||||||
|
|
||||||
|
|
||||||
def jsonSave(filename, value):
|
def jsonSave(filename, value):
|
||||||
with open(filename, 'w', encoding="utf-8") as f:
|
with open(filename, 'w', encoding="utf-8") as f:
|
||||||
json.dump(value, f, indent=4, ensure_ascii=False)
|
f.write(dumps(value, indent=4, ensure_ascii=False))
|
||||||
f.close()
|
|
||||||
|
|
||||||
def jsonLoad(filename):
|
def jsonLoad(filename):
|
||||||
with open(filename, 'r', encoding="utf-8") as f:
|
with open(filename, 'r', encoding="utf-8") as f:
|
||||||
value = json.load(f)
|
value = loads(f.read())
|
||||||
f.close()
|
|
||||||
return value
|
return value
|
||||||
|
|
||||||
|
|
||||||
config = jsonLoad(f"{path}/config.json")
|
config = jsonLoad(f"config.json")
|
||||||
|
|
||||||
log_size = config["log_size"]
|
log_size = config["log_size"]
|
||||||
owner_id = config["owner_id"]
|
owner_id = config["owner_id"]
|
||||||
@ -44,12 +39,12 @@ def checkSize():
|
|||||||
|
|
||||||
while i < 2:
|
while i < 2:
|
||||||
try:
|
try:
|
||||||
log = os.stat(logs_folder + 'latest.log')
|
log = stat(logs_folder + 'latest.log')
|
||||||
|
|
||||||
if (log.st_size / 1024) > log_size:
|
if (log.st_size / 1024) > log_size:
|
||||||
with open(logs_folder + 'latest.log', 'rb') as f_in:
|
with open(logs_folder + 'latest.log', 'rb') as f_in:
|
||||||
with gzip.open(f'{logs_folder}{datetime.now().strftime("%d.%m.%Y_%H:%M:%S")}.zip', 'wb') as f_out:
|
with gzipopen(f'{logs_folder}{datetime.now().strftime("%d.%m.%Y_%H:%M:%S")}.zip', 'wb') as f_out:
|
||||||
shutil.copyfileobj(f_in, f_out)
|
copyfileobj(f_in, f_out)
|
||||||
|
|
||||||
open(logs_folder + 'latest.log', 'w').close()
|
open(logs_folder + 'latest.log', 'w').close()
|
||||||
|
|
||||||
@ -62,7 +57,7 @@ def checkSize():
|
|||||||
open(logs_folder + 'latest.log', 'a').close()
|
open(logs_folder + 'latest.log', 'a').close()
|
||||||
except:
|
except:
|
||||||
try:
|
try:
|
||||||
os.mkdir(logs_folder)
|
makedirs(logs_folder, exist_ok=True)
|
||||||
log = open(logs_folder + 'latest.log', 'a')
|
log = open(logs_folder + 'latest.log', 'a')
|
||||||
open(logs_folder + 'latest.log', 'a').close()
|
open(logs_folder + 'latest.log', 'a').close()
|
||||||
except:
|
except:
|
||||||
@ -80,14 +75,15 @@ def appendLog(message):
|
|||||||
open(logs_folder + 'latest.log', 'a').close()
|
open(logs_folder + 'latest.log', 'a').close()
|
||||||
except:
|
except:
|
||||||
try:
|
try:
|
||||||
os.mkdir(logs_folder)
|
makedirs(logs_folder, exist_ok=True)
|
||||||
log = open(logs_folder + 'latest.log', 'a')
|
log = open(logs_folder + 'latest.log', 'a')
|
||||||
open(logs_folder + 'latest.log', 'a').close()
|
open(logs_folder + 'latest.log', 'a').close()
|
||||||
except:
|
except:
|
||||||
time.sleep(2)
|
sleep(2)
|
||||||
print('Log file could not be created')
|
print('Log file could not be created')
|
||||||
return
|
return
|
||||||
|
|
||||||
|
print(message, flush=True)
|
||||||
log.write(f'[{datetime.now().strftime("%H:%M:%S | %d.%m.%Y")}] {message}\n')
|
log.write(f'[{datetime.now().strftime("%H:%M:%S | %d.%m.%Y")}] {message}\n')
|
||||||
|
|
||||||
log.close()
|
log.close()
|
||||||
|
@ -1,7 +1,9 @@
|
|||||||
#-*- coding: utf-8 -*-
|
#-*- coding: utf-8 -*-
|
||||||
|
|
||||||
from subprocess import run
|
from os import makedirs, path
|
||||||
|
from subprocess import check_output
|
||||||
from traceback import format_exc
|
from traceback import format_exc
|
||||||
|
from uuid import uuid4
|
||||||
from functions import *
|
from functions import *
|
||||||
from bs4 import BeautifulSoup
|
from bs4 import BeautifulSoup
|
||||||
|
|
||||||
@ -16,16 +18,19 @@ async def getWaterLeft(cardid, filename, app=None):
|
|||||||
|
|
||||||
try:
|
try:
|
||||||
|
|
||||||
run(["touch", f"data/pages/{str(filename)}.html"])
|
# if path.exists(f"data/pages/{str(filename)}.html") is False:
|
||||||
|
# run(["touch", f"data/pages/{str(filename)}.html"])
|
||||||
|
|
||||||
if config["use_compiled_page_saver"]:
|
if config["use_compiled_page_saver"] is True:
|
||||||
run(["PageSaver/pageSaver", f"\"https://bwtaqua.com.ua/card-topup/?id={cardid}\"", ">", f"data/pages/{str(filename)}.html"])
|
proc = check_output(["PageSaver/pageSaver", f"https://bwtaqua.com.ua/card-topup/?id={cardid}"]) #, ">", f"data/pages/{str(filename)}.html"])
|
||||||
|
html_file = proc.decode("utf-8")
|
||||||
else:
|
else:
|
||||||
run(["node", "PageSaver/pageSaver.js", f"\"https://bwtaqua.com.ua/card-topup/?id={cardid}\"", ">", f"data/pages/{str(filename)}.html"])
|
proc = check_output(["node", "./PageSaver/pageSaver.js", f"https://bwtaqua.com.ua/card-topup/?id={cardid}"]) #, ">", f"data/pages/{str(filename)}.html"])
|
||||||
|
html_file = proc.decode("utf-8")
|
||||||
|
|
||||||
with open(f'data/pages/{str(filename)}.html') as f:
|
# with open(f'data/pages/{str(filename)}.html') as f:
|
||||||
html_file = f.read()
|
# html_file = f.read()
|
||||||
f.close()
|
# f.close()
|
||||||
|
|
||||||
soup = BeautifulSoup(html_file, 'html.parser')
|
soup = BeautifulSoup(html_file, 'html.parser')
|
||||||
|
|
||||||
@ -37,10 +42,19 @@ async def getWaterLeft(cardid, filename, app=None):
|
|||||||
|
|
||||||
appendLog(f"Exception occured: {exp} (user: {str(filename)}, cardid: {cardid})")
|
appendLog(f"Exception occured: {exp} (user: {str(filename)}, cardid: {cardid})")
|
||||||
|
|
||||||
|
try:
|
||||||
|
tmp_name = str(uuid4())
|
||||||
|
makedirs("tmp", exist_ok=True)
|
||||||
|
with open(path.join("tmp", tmp_name), "w", encoding="utf-8") as f:
|
||||||
|
f.write(html_file)
|
||||||
|
except NameError:
|
||||||
|
tmp_name = "N/A"
|
||||||
|
appendLog(f"'html_file' is not defined so I won't gather any tmp data")
|
||||||
|
|
||||||
if app != None:
|
if app != None:
|
||||||
await app.send_message(config["owner_id"], f"**Exception occured:**\n • User: `{str(filename)}`\n • Card: [{cardid}]({url})\n • Exception: `{exp}`\n • Traceback: `{format_exc()}`", disable_web_page_preview=True)
|
await app.send_message(config["owner_id"], f"**Exception occured:**\n • User: `{str(filename)}`\n • Card: [{cardid}]({url})\n • Exception: `{exp}`\n • TMP UUID: `{tmp_name}`\n • Traceback: `{format_exc()}`", disable_web_page_preview=True)
|
||||||
else:
|
else:
|
||||||
print(f'Exception occured and could not send to user: {exp}')
|
appendLog(f'Exception occured and could not send to user: {exp}')
|
||||||
|
|
||||||
output = "Failure"
|
output = "Failure"
|
||||||
|
|
||||||
|
Reference in New Issue
Block a user