Compare commits
20 Commits
v1.0
...
d74826f0b4
Author | SHA1 | Date | |
---|---|---|---|
d74826f0b4 | |||
b23f0f2436 | |||
4f423257f3 | |||
9fb509fc2b | |||
82a659cce2 | |||
c3dd6f61d6 | |||
dcf82ab6f2 | |||
|
1dd8b13297 | ||
|
200f25e130 | ||
|
a09c3fb0d4 | ||
|
affb54155c | ||
|
b97b10975d | ||
|
27e204d3cc | ||
|
4fd4f0a6a4 | ||
|
19b83c0631 | ||
fa8bdc0e1f | |||
b9a7d85674 | |||
fe1c6984b2 | |||
e7ef1d4613 | |||
85a756dcab |
1
.gitignore
vendored
1
.gitignore
vendored
@@ -152,3 +152,4 @@ cython_debug/
|
||||
# option (not recommended) you can uncomment the following to ignore the entire idea folder.
|
||||
#.idea/
|
||||
|
||||
.vscode
|
@@ -1,5 +1,6 @@
|
||||
{
|
||||
"dependencies": {
|
||||
"puppeteer": "^14.4.0"
|
||||
"puppeteer": "~19.5.2",
|
||||
"user-agents": "~1.0.1260"
|
||||
}
|
||||
}
|
||||
|
@@ -1,13 +1,15 @@
|
||||
// npm install https://github.com/GoogleChrome/puppeteer/
|
||||
|
||||
const puppeteer = require('puppeteer');
|
||||
const userAgent = require('user-agents');
|
||||
|
||||
(async () => {
|
||||
|
||||
const url = process.argv[2];
|
||||
const browser = await puppeteer.launch();
|
||||
const page = await browser.newPage();
|
||||
|
||||
await page.setUserAgent(userAgent.random().toString());
|
||||
|
||||
await page.goto(url, {waitUntil: 'load'});
|
||||
|
||||
const html = await page.content();
|
||||
|
11
README.md
11
README.md
@@ -1,12 +1,17 @@
|
||||
# BWTAqua
|
||||
|
||||
[](https://www.gnu.org/licenses/gpl-3.0.html)
|
||||
|
||||
Simple yet helpful bot to check BWT Aqua's card balance
|
||||
|
||||
## Requirements
|
||||
|
||||
* nodejs & npm
|
||||
* python3
|
||||
* git
|
||||
|
||||
## Installation
|
||||
|
||||
1. Download package
|
||||
1. `git clone https://git.end-play.xyz/profitroll/BWTAqua.git`
|
||||
2. `cd BWTAqua`
|
||||
@@ -15,6 +20,7 @@ Simple yet helpful bot to check BWT Aqua's card balance
|
||||
3. Install PageSaver:
|
||||
1. `cd PageSaver`
|
||||
2. `npm install`
|
||||
3. `chmod +x pageSaver` (If you want to use compiled page saver)
|
||||
4. Configure the bot:
|
||||
1. `cd ..`
|
||||
2. `nano config.json` (You can use any other text editor actually, for example `vim`)
|
||||
@@ -22,5 +28,6 @@ Simple yet helpful bot to check BWT Aqua's card balance
|
||||
* `python3 bwtbot.py`
|
||||
|
||||
## Configuration
|
||||
You can edit with vim, nano, on Windows it's Notepad or Notepad++. Whatever.
|
||||
If you don't know where to find bot_token and your id - here you can find some hints: [get bot token](https://www.siteguarding.com/en/how-to-get-telegram-bot-api-token), [get your id](https://www.alphr.com/telegram-find-user-id/), [get api_hash and api_id](https://core.telegram.org/api/obtaining_api_id).
|
||||
|
||||
You can edit with vim, nano, whatever.
|
||||
If you don't know where to find bot_token and your id - here you can find some hints: [get bot token](https://www.siteguarding.com/en/how-to-get-telegram-bot-api-token), [get your id](https://www.alphr.com/telegram-find-user-id/), [get api_hash and api_id](https://core.telegram.org/api/obtaining_api_id).
|
||||
|
305
bwtbot.py
305
bwtbot.py
@@ -1,173 +1,214 @@
|
||||
#-*- coding: utf-8 -*-
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
import traceback
|
||||
from pyrogram import Client, filters, idle
|
||||
from pyrogram.types import ForceReply, BotCommand, BotCommandScopeChat
|
||||
from os import getpid, system
|
||||
from subprocess import call
|
||||
from pyrogram import filters
|
||||
from pyrogram.client import Client
|
||||
from pyrogram.sync import idle
|
||||
from pyrogram.types import (
|
||||
ForceReply,
|
||||
BotCommand,
|
||||
BotCommandScopeChat,
|
||||
Message,
|
||||
ReplyKeyboardRemove,
|
||||
)
|
||||
from pyrogram.enums.chat_action import ChatAction
|
||||
from convopyro import Conversation, listen_message
|
||||
from functions import *
|
||||
from modules.colors import *
|
||||
from modules.bwt import *
|
||||
import subprocess
|
||||
import os
|
||||
|
||||
config = jsonLoad("config.json")
|
||||
|
||||
owner_id = config["owner_id"]
|
||||
|
||||
app = Client(config["bot_name"], api_id=config["api_id"], api_hash=config["api_hash"], bot_token=config["bot_token"])
|
||||
app = Client(
|
||||
config["bot_name"],
|
||||
api_id=config["api_id"],
|
||||
api_hash=config["api_hash"],
|
||||
bot_token=config["bot_token"],
|
||||
)
|
||||
|
||||
Conversation(app)
|
||||
|
||||
|
||||
@app.on_message(~ filters.scheduled & filters.command(["setcard", "задать карту"], prefixes=["/", ""]))
|
||||
async def setcard(_, msg):
|
||||
if userGet(msg.from_user.id, "context") is None:
|
||||
userSet(msg.from_user.id, "context", "set")
|
||||
await msg.reply_text(string("send_number"), reply_markup=ForceReply(placeholder=string("enter_number")))
|
||||
@app.on_message(
|
||||
~filters.scheduled
|
||||
& filters.command(["setcard", "задать карту"], prefixes=["/", ""])
|
||||
)
|
||||
async def setcard(_: Client, msg: Message):
|
||||
await msg.reply_text(
|
||||
string("send_number"),
|
||||
reply_markup=ForceReply(placeholder=string("enter_number")),
|
||||
)
|
||||
answer = await listen_message(_, msg.chat.id, timeout=None)
|
||||
if answer is None:
|
||||
return
|
||||
elif answer.text.strip() in ["/cancel", "cancel", "/відміна", "відміна"]:
|
||||
await msg.reply_text(string("cancel"), reply_markup=ReplyKeyboardRemove())
|
||||
return
|
||||
userSet(answer.from_user.id, "card", answer.text)
|
||||
appendLog(f"User {str(msg.from_user.id)} set card id to {answer.text}")
|
||||
await msg.reply_text(
|
||||
string("card_linked").format(answer.text), reply_markup=ReplyKeyboardRemove()
|
||||
)
|
||||
|
||||
|
||||
@app.on_message(
|
||||
~filters.scheduled
|
||||
& filters.command(["resetcard", "забути картку"], prefixes=["/", ""])
|
||||
)
|
||||
async def resetcard(_: Client, msg: Message):
|
||||
if "card" in jsonLoad("data/database.json")[str(msg.from_user.id)]:
|
||||
userReset(msg.from_user.id, "card")
|
||||
await msg.reply_text(string("card_unlinked"))
|
||||
appendLog(f"User {str(msg.from_user.id)} reseted his card")
|
||||
else:
|
||||
await msg.reply_text(string("cancel_first"))
|
||||
await msg.reply_text(string("card_not_linked").format(string("get_number")))
|
||||
appendLog(f"User {str(msg.from_user.id)} tried to reset non-existent card")
|
||||
|
||||
|
||||
@app.on_message(~ filters.scheduled & filters.command(["cancel", "відміна"], prefixes=["/", ""]))
|
||||
async def cancel(_, msg):
|
||||
if userGet(msg.from_user.id, "context") is not None:
|
||||
userReset(msg.from_user.id, "context")
|
||||
await msg.reply_text(string("cancel"))
|
||||
else:
|
||||
await msg.reply_text(string("cancel_none"))
|
||||
|
||||
|
||||
@app.on_message(~ filters.scheduled & filters.command(["resetcard", "забути картку"], prefixes=["/", ""]))
|
||||
async def resetcard(_, msg):
|
||||
if userGet(msg.from_user.id, "context") is None:
|
||||
@app.on_message(
|
||||
~filters.scheduled & filters.command(["balance", "баланс"], prefixes=["/", ""])
|
||||
)
|
||||
async def balance(_: Client, msg: Message):
|
||||
try:
|
||||
if "card" in jsonLoad("data/database.json")[str(msg.from_user.id)]:
|
||||
userReset(msg.from_user.id, "card")
|
||||
await msg.reply_text(string("card_unlinked"))
|
||||
appendLog(f"User {str(msg.from_user.id)} reseted his card")
|
||||
await app.send_chat_action(chat_id=msg.chat.id, action=ChatAction.TYPING)
|
||||
water_left = await getWaterLeft(
|
||||
userGet(msg.from_user.id, "card"), msg.from_user.id, app
|
||||
)
|
||||
if water_left == "":
|
||||
await msg.reply_text(
|
||||
string("error_new").format(
|
||||
f'https://bwtaqua.com.ua/card-topup/?id={userGet(msg.from_user.id, "card")}'
|
||||
)
|
||||
)
|
||||
# raise EmptyCardException("Card information is empty")
|
||||
elif water_left == "Failure":
|
||||
await msg.reply_text(
|
||||
string("error_occured").format(string("get_number"))
|
||||
)
|
||||
appendLog(
|
||||
f"User {str(msg.from_user.id)} could not get left water amount"
|
||||
)
|
||||
else:
|
||||
await msg.reply_text(string("card_balance").format(water_left))
|
||||
appendLog(
|
||||
f"User {str(msg.from_user.id)} has {water_left} liters remaining"
|
||||
)
|
||||
else:
|
||||
await msg.reply_text(string("card_not_linked").format(string("get_number")))
|
||||
appendLog(f"User {str(msg.from_user.id)} tried to reset non-existent card")
|
||||
else:
|
||||
await msg.reply_text(string("cancel_first"))
|
||||
|
||||
|
||||
@app.on_message(~ filters.scheduled & filters.command(["balance", "баланс"], prefixes=["/", ""]))
|
||||
async def balance(_, msg):
|
||||
if userGet(msg.from_user.id, "context") is None:
|
||||
try:
|
||||
if "card" in jsonLoad("data/database.json")[str(msg.from_user.id)]:
|
||||
await app.send_chat_action(chat_id=msg.chat.id, action=ChatAction.TYPING)
|
||||
water_left = await getWaterLeft(userGet(msg.from_user.id, "card"), msg.from_user.id, app)
|
||||
if water_left == "":
|
||||
raise EmptyCardException("Card information is empty")
|
||||
elif water_left == "Failure":
|
||||
await msg.reply_text(string("error_occured").format(string("get_number")))
|
||||
appendLog(f"User {str(msg.from_user.id)} could not get left water amount")
|
||||
else:
|
||||
await msg.reply_text(string("card_balance").format(water_left))
|
||||
appendLog(f"User {str(msg.from_user.id)} has {water_left} liters remaining")
|
||||
else:
|
||||
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")
|
||||
except Exception as exp:
|
||||
appendLog(
|
||||
f"User {str(msg.from_user.id)} tried to get balance without card set"
|
||||
)
|
||||
except Exception as exp:
|
||||
if msg.from_user.id != config["owner_id"]:
|
||||
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: `{traceback.format_exc()}`")
|
||||
appendLog(f"User {str(msg.from_user.id)} could not get left water amount")
|
||||
else:
|
||||
await msg.reply_text(string("cancel_first"))
|
||||
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")
|
||||
|
||||
|
||||
@app.on_message(~ filters.scheduled & filters.command(["topup", "refill", "поповнити"], prefixes=["/", ""]))
|
||||
async def topup_cmd(_, msg):
|
||||
if userGet(msg.from_user.id, "context") is None:
|
||||
try:
|
||||
if "card" in jsonLoad("data/database.json")[str(msg.from_user.id)]:
|
||||
await app.send_chat_action(chat_id=msg.chat.id, action=ChatAction.TYPING)
|
||||
await msg.reply_text(string("top_up").format(str(userGet(msg.from_user.id, "card"))))
|
||||
appendLog(f"User {str(msg.from_user.id)} requested top up")
|
||||
else:
|
||||
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")
|
||||
except Exception as exp:
|
||||
await msg.reply_text(exp)
|
||||
else:
|
||||
await msg.reply_text(string("cancel_first"))
|
||||
|
||||
|
||||
@app.on_message(~ filters.scheduled & filters.command(["start", "help", "допомога"], prefixes=["/", ""]))
|
||||
async def help(_, msg):
|
||||
if userGet(msg.from_user.id, "context") is None:
|
||||
await msg.reply_text(string("welcome").format(string("get_number")))
|
||||
if msg.from_user.language_code in jsonLoad("strings.json"):
|
||||
userSet(msg.from_user.id, "locale", msg.from_user.language_code)
|
||||
@app.on_message(
|
||||
~filters.scheduled
|
||||
& filters.command(["topup", "refill", "поповнити"], prefixes=["/", ""])
|
||||
)
|
||||
async def topup_cmd(_: Client, msg: Message):
|
||||
try:
|
||||
if "card" in jsonLoad("data/database.json")[str(msg.from_user.id)]:
|
||||
await app.send_chat_action(chat_id=msg.chat.id, action=ChatAction.TYPING)
|
||||
await msg.reply_text(
|
||||
string("top_up").format(str(userGet(msg.from_user.id, "card")))
|
||||
)
|
||||
appendLog(f"User {str(msg.from_user.id)} requested top up")
|
||||
else:
|
||||
userSet(msg.from_user.id, "locale", "en")
|
||||
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"
|
||||
)
|
||||
except Exception as exp:
|
||||
await msg.reply_text(str(exp))
|
||||
|
||||
|
||||
@app.on_message(
|
||||
~filters.scheduled
|
||||
& filters.command(["start", "help", "допомога"], prefixes=["/", ""])
|
||||
)
|
||||
async def help(_: Client, msg: Message):
|
||||
await msg.reply_text(string("welcome").format(string("get_number")))
|
||||
if msg.from_user.language_code in jsonLoad("strings.json"):
|
||||
userSet(msg.from_user.id, "locale", msg.from_user.language_code)
|
||||
else:
|
||||
await msg.reply_text(string("cancel_first"))
|
||||
userSet(msg.from_user.id, "locale", "en")
|
||||
|
||||
pid = os.getpid()
|
||||
|
||||
@app.on_message(~ filters.scheduled & filters.command(["kill", "die", "shutdown"], prefixes="/"))
|
||||
async def kill(_, msg):
|
||||
pid = getpid()
|
||||
|
||||
|
||||
@app.on_message(
|
||||
~filters.scheduled & filters.command(["kill", "die", "shutdown"], prefixes="/")
|
||||
)
|
||||
async def kill(_: Client, msg: Message):
|
||||
if msg.from_user.id == owner_id:
|
||||
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)
|
||||
async def any_message_handler(app, msg):
|
||||
if userGet(msg.from_user.id, "context") == "set":
|
||||
userSet(msg.from_user.id, "card", msg.text)
|
||||
userReset(msg.from_user.id, "context")
|
||||
appendLog(f"User {str(msg.from_user.id)} set card id to {msg.text}")
|
||||
await msg.reply_text(string("card_linked").format(msg.text))
|
||||
print(f"{nowtime()} {WHITE}Starting with PID {YELLOW}{pid}{RESET}")
|
||||
|
||||
print(f'{nowtime()} {WHITE}Starting with PID {YELLOW}{pid}{RESET}')
|
||||
app.start() # type: ignore
|
||||
app.send_message(owner_id, f"Starting bot with pid **{pid}**") # type: ignore
|
||||
|
||||
app.start()
|
||||
app.send_message(owner_id, f"Starting bot with pid **{pid}**")
|
||||
|
||||
app.set_bot_commands([
|
||||
BotCommand("help", "Меню допомоги"),
|
||||
BotCommand("balance", "Баланс картки"),
|
||||
BotCommand("topup", "Поповнити картку"),
|
||||
BotCommand("setcard", "Прив'язати картку"),
|
||||
BotCommand("resetcard", "Відв'язати картку"),
|
||||
BotCommand("cancel", "Відмінити операцію"),
|
||||
app.set_bot_commands(
|
||||
[
|
||||
BotCommand("help", "Меню допомоги"),
|
||||
BotCommand("balance", "Баланс картки"),
|
||||
BotCommand("topup", "Поповнити картку"),
|
||||
BotCommand("setcard", "Прив'язати картку"),
|
||||
BotCommand("resetcard", "Відв'язати картку"),
|
||||
],
|
||||
language_code="uk")
|
||||
language_code="uk",
|
||||
) # type: ignore
|
||||
|
||||
app.set_bot_commands([
|
||||
BotCommand("help", "Меню допомоги"),
|
||||
BotCommand("balance", "Баланс картки"),
|
||||
BotCommand("topup", "Поповнити картку"),
|
||||
BotCommand("setcard", "Прив'язати картку"),
|
||||
BotCommand("resetcard", "Відв'язати картку"),
|
||||
BotCommand("cancel", "Відмінити операцію"),
|
||||
app.set_bot_commands(
|
||||
[
|
||||
BotCommand("help", "Меню допомоги"),
|
||||
BotCommand("balance", "Баланс картки"),
|
||||
BotCommand("topup", "Поповнити картку"),
|
||||
BotCommand("setcard", "Прив'язати картку"),
|
||||
BotCommand("resetcard", "Відв'язати картку"),
|
||||
],
|
||||
language_code="ru")
|
||||
language_code="ru",
|
||||
) # type: ignore
|
||||
|
||||
app.set_bot_commands([
|
||||
BotCommand("help", "Help menu"),
|
||||
BotCommand("balance", "Card's balance"),
|
||||
BotCommand("topup", "Refill card"),
|
||||
BotCommand("setcard", "Link card"),
|
||||
BotCommand("resetcard", "Unlink card"),
|
||||
BotCommand("cancel", "Cancel operation"),
|
||||
])
|
||||
app.set_bot_commands(
|
||||
[
|
||||
BotCommand("help", "Help menu"),
|
||||
BotCommand("balance", "Card's balance"),
|
||||
BotCommand("topup", "Refill card"),
|
||||
BotCommand("setcard", "Link card"),
|
||||
BotCommand("resetcard", "Unlink card"),
|
||||
]
|
||||
) # type: ignore
|
||||
|
||||
app.set_bot_commands([
|
||||
BotCommand("help", "Help menu"),
|
||||
BotCommand("balance", "Card's balance"),
|
||||
BotCommand("topup", "Refill card"),
|
||||
BotCommand("setcard", "Link card"),
|
||||
BotCommand("resetcard", "Unlink card"),
|
||||
BotCommand("shutdown", "Turn off the bot"),
|
||||
BotCommand("cancel", "Cancel operation"),
|
||||
app.set_bot_commands(
|
||||
[
|
||||
BotCommand("help", "Help menu"),
|
||||
BotCommand("balance", "Card's balance"),
|
||||
BotCommand("topup", "Refill card"),
|
||||
BotCommand("setcard", "Link card"),
|
||||
BotCommand("resetcard", "Unlink card"),
|
||||
BotCommand("shutdown", "Turn off the bot"),
|
||||
],
|
||||
scope=BotCommandScopeChat(chat_id=owner_id))
|
||||
scope=BotCommandScopeChat(chat_id=owner_id),
|
||||
) # type: ignore
|
||||
|
||||
idle()
|
||||
|
||||
app.send_message(owner_id, f"Shutting down bot with pid **{pid}**")
|
||||
print(f'\n{nowtime()} {WHITE}Shutting down with PID {YELLOW}{pid}{RESET}')
|
||||
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}")
|
||||
|
||||
subprocess.call(f'kill -9 {pid}', shell=True)
|
||||
call(f"kill -9 {pid}", shell=True)
|
||||
|
@@ -4,5 +4,6 @@
|
||||
"api_id": 0,
|
||||
"api_hash": "",
|
||||
"bot_token": "",
|
||||
"bot_name": ""
|
||||
"bot_name": "",
|
||||
"use_compiled_page_saver": false
|
||||
}
|
90
functions.py
90
functions.py
@@ -1,32 +1,29 @@
|
||||
import json
|
||||
import os
|
||||
import shutil
|
||||
import gzip
|
||||
import time
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
from os import makedirs, stat
|
||||
from gzip import open as gzipopen
|
||||
from shutil import copyfileobj
|
||||
from time import sleep
|
||||
from ujson import loads, dumps
|
||||
from modules.colors import *
|
||||
from datetime import datetime
|
||||
from pathlib import Path
|
||||
|
||||
path = Path(__file__).resolve().parent
|
||||
|
||||
days_path = str(path)+"/assets/days/"
|
||||
users_path = str(path)+"/users/"
|
||||
logs_folder = str(path)+"/logs/"
|
||||
users_path = "users/"
|
||||
logs_folder = "logs/"
|
||||
|
||||
|
||||
def jsonSave(filename, value):
|
||||
with open(filename, 'w', encoding="utf-8") as f:
|
||||
json.dump(value, f, indent=4, ensure_ascii=False)
|
||||
f.close()
|
||||
with open(filename, "w", encoding="utf-8") as f:
|
||||
f.write(dumps(value, indent=4, ensure_ascii=False))
|
||||
|
||||
|
||||
def jsonLoad(filename):
|
||||
with open(filename, 'r', encoding="utf-8") as f:
|
||||
value = json.load(f)
|
||||
f.close()
|
||||
with open(filename, "r", encoding="utf-8") as f:
|
||||
value = loads(f.read())
|
||||
return value
|
||||
|
||||
|
||||
config = jsonLoad(f"{path}/config.json")
|
||||
config = jsonLoad(f"config.json")
|
||||
|
||||
log_size = config["log_size"]
|
||||
owner_id = config["owner_id"]
|
||||
@@ -35,59 +32,64 @@ owner_id = config["owner_id"]
|
||||
def nowtime():
|
||||
return f'{BBLACK}[{CYAN}{datetime.now().strftime("%H:%M:%S")}{BBLACK}]{RESET}'
|
||||
|
||||
|
||||
def checkSize():
|
||||
global logs_folder, log_size
|
||||
|
||||
|
||||
i = 0
|
||||
|
||||
while i < 2:
|
||||
try:
|
||||
log = os.stat(logs_folder + 'latest.log')
|
||||
log = stat(logs_folder + "latest.log")
|
||||
|
||||
if (log.st_size / 1024) > log_size:
|
||||
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:
|
||||
shutil.copyfileobj(f_in, f_out)
|
||||
|
||||
open(logs_folder + 'latest.log', 'w').close()
|
||||
|
||||
with open(logs_folder + "latest.log", "rb") as f_in:
|
||||
with gzipopen(
|
||||
f'{logs_folder}{datetime.now().strftime("%d.%m.%Y_%H:%M:%S")}.zip',
|
||||
"wb",
|
||||
) as f_out:
|
||||
copyfileobj(f_in, f_out)
|
||||
|
||||
open(logs_folder + "latest.log", "w").close()
|
||||
|
||||
i = 2
|
||||
|
||||
except FileNotFoundError:
|
||||
|
||||
try:
|
||||
log = open(logs_folder + 'latest.log', 'a')
|
||||
open(logs_folder + 'latest.log', 'a').close()
|
||||
log = open(logs_folder + "latest.log", "a")
|
||||
open(logs_folder + "latest.log", "a").close()
|
||||
except:
|
||||
try:
|
||||
os.mkdir(logs_folder)
|
||||
log = open(logs_folder + 'latest.log', 'a')
|
||||
open(logs_folder + 'latest.log', 'a').close()
|
||||
makedirs(logs_folder, exist_ok=True)
|
||||
log = open(logs_folder + "latest.log", "a")
|
||||
open(logs_folder + "latest.log", "a").close()
|
||||
except:
|
||||
pass
|
||||
|
||||
|
||||
i += 1
|
||||
|
||||
|
||||
def appendLog(message):
|
||||
global logs_folder
|
||||
|
||||
checkSize()
|
||||
|
||||
try:
|
||||
log = open(logs_folder + 'latest.log', 'a')
|
||||
open(logs_folder + 'latest.log', 'a').close()
|
||||
log = open(logs_folder + "latest.log", "a")
|
||||
open(logs_folder + "latest.log", "a").close()
|
||||
except:
|
||||
try:
|
||||
os.mkdir(logs_folder)
|
||||
log = open(logs_folder + 'latest.log', 'a')
|
||||
open(logs_folder + 'latest.log', 'a').close()
|
||||
makedirs(logs_folder, exist_ok=True)
|
||||
log = open(logs_folder + "latest.log", "a")
|
||||
open(logs_folder + "latest.log", "a").close()
|
||||
except:
|
||||
time.sleep(2)
|
||||
print('Log file could not be created')
|
||||
sleep(2)
|
||||
print("Log file could not be created")
|
||||
return
|
||||
|
||||
|
||||
print(message, flush=True)
|
||||
log.write(f'[{datetime.now().strftime("%H:%M:%S | %d.%m.%Y")}] {message}\n')
|
||||
|
||||
|
||||
log.close()
|
||||
|
||||
|
||||
@@ -107,15 +109,17 @@ def userSet(userid, key: str, value):
|
||||
database[str(userid)][key] = value
|
||||
jsonSave("data/database.json", database)
|
||||
|
||||
|
||||
def userReset(userid, key: str):
|
||||
database = jsonLoad("data/database.json")
|
||||
del database[str(userid)][key]
|
||||
jsonSave("data/database.json", database)
|
||||
|
||||
|
||||
def userGet(userid, key: str):
|
||||
try:
|
||||
return jsonLoad("data/database.json")[str(userid)][key]
|
||||
except KeyError:
|
||||
return None
|
||||
except FileNotFoundError:
|
||||
return None
|
||||
return None
|
||||
|
103
modules/bwt.py
103
modules/bwt.py
@@ -1,50 +1,95 @@
|
||||
#-*- coding: utf-8 -*-
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
import os
|
||||
import traceback
|
||||
from os import makedirs, path
|
||||
from subprocess import check_output
|
||||
from traceback import format_exc
|
||||
from uuid import uuid4
|
||||
from functions import *
|
||||
from bs4 import BeautifulSoup
|
||||
|
||||
config = jsonLoad("config.json")
|
||||
|
||||
|
||||
class EmptyCardException(Exception):
|
||||
pass
|
||||
|
||||
|
||||
async def getWaterLeft(cardid, filename, app=None):
|
||||
|
||||
url = f"https://bwtaqua.com.ua/card-topup/?id={cardid}"
|
||||
|
||||
|
||||
try:
|
||||
# if path.exists(f"data/pages/{str(filename)}.html") is False:
|
||||
# run(["touch", f"data/pages/{str(filename)}.html"])
|
||||
|
||||
os.system(f'touch data/pages/{str(filename)}.html')
|
||||
os.system(f'PageSaver/pageSaver "https://bwtaqua.com.ua/card-topup/?id={cardid}" > data/pages/{str(filename)}.html')
|
||||
appendLog(f"Trying to get liters for url '{url}'")
|
||||
|
||||
with open(f'data/pages/{str(filename)}.html') as f:
|
||||
html_file = f.read()
|
||||
f.close()
|
||||
|
||||
soup = BeautifulSoup(html_file, 'html.parser')
|
||||
|
||||
output = (soup.find_all("h3", class_="headline headline_center headline_pink js-payment-balance")[0].getText()).replace("Твій баланс ", "").replace(" л", "")
|
||||
|
||||
appendLog(f"Parsed {output} liters of water remaining (user: {str(filename)}, cardid: {cardid})")
|
||||
|
||||
except Exception as exp:
|
||||
|
||||
appendLog(f"Exception occured: {exp} (user: {str(filename)}, cardid: {cardid})")
|
||||
|
||||
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: `{traceback.format_exc()}`", disable_web_page_preview=True)
|
||||
if config["use_compiled_page_saver"] is True:
|
||||
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:
|
||||
print(f'Exception occured and could not send to user: {exp}')
|
||||
|
||||
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:
|
||||
# html_file = f.read()
|
||||
# f.close()
|
||||
|
||||
soup = BeautifulSoup(html_file, "html.parser")
|
||||
|
||||
output = (
|
||||
(
|
||||
soup.find_all(
|
||||
"h3",
|
||||
class_="headline headline_center headline_pink js-payment-balance",
|
||||
)[0].getText()
|
||||
)
|
||||
.replace("Твій баланс ", "")
|
||||
.replace(" л", "")
|
||||
)
|
||||
|
||||
appendLog(
|
||||
f"Parsed {output} liters of water remaining (user: {str(filename)}, cardid: {cardid})"
|
||||
)
|
||||
|
||||
except Exception as exp:
|
||||
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:
|
||||
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:
|
||||
appendLog(f"Exception occured and could not send to user: {exp}")
|
||||
|
||||
output = "Failure"
|
||||
|
||||
|
||||
return output
|
||||
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
||||
cardid = input("Enter card number: ")
|
||||
userid = input("Enter Telegram ID (optional): ")
|
||||
|
||||
|
||||
print(f"Card has {str(getWaterLeft(cardid, userid, app=None))} l. left")
|
||||
|
@@ -1,22 +1,22 @@
|
||||
RESET = '\u001b[0m'
|
||||
RESET = "\u001b[0m"
|
||||
|
||||
BLACK = '\u001b[30m'
|
||||
RED = '\u001b[31m'
|
||||
GREEN = '\u001b[32m'
|
||||
YELLOW = '\u001b[33m'
|
||||
BLUE = '\u001b[34m'
|
||||
MAGENTA = '\u001b[35m'
|
||||
CYAN = '\u001b[36m'
|
||||
WHITE = '\u001b[37m'
|
||||
BLACK = "\u001b[30m"
|
||||
RED = "\u001b[31m"
|
||||
GREEN = "\u001b[32m"
|
||||
YELLOW = "\u001b[33m"
|
||||
BLUE = "\u001b[34m"
|
||||
MAGENTA = "\u001b[35m"
|
||||
CYAN = "\u001b[36m"
|
||||
WHITE = "\u001b[37m"
|
||||
|
||||
BBLACK = '\u001b[30;1m'
|
||||
BRED = '\u001b[31;1m'
|
||||
BGREEN = '\u001b[32;1m'
|
||||
BYELLOW = '\u001b[33;1m'
|
||||
BBLUE = '\u001b[34;1m'
|
||||
BMAGENTA = '\u001b[35;1m'
|
||||
BCYAN = '\u001b[36;1m'
|
||||
BWHITE = '\u001b[37;1m'
|
||||
BBLACK = "\u001b[30;1m"
|
||||
BRED = "\u001b[31;1m"
|
||||
BGREEN = "\u001b[32;1m"
|
||||
BYELLOW = "\u001b[33;1m"
|
||||
BBLUE = "\u001b[34;1m"
|
||||
BMAGENTA = "\u001b[35;1m"
|
||||
BCYAN = "\u001b[36;1m"
|
||||
BWHITE = "\u001b[37;1m"
|
||||
|
||||
ULINE = '\u001b[4m'
|
||||
REVERSE = '\u001b[7m'
|
||||
ULINE = "\u001b[4m"
|
||||
REVERSE = "\u001b[7m"
|
||||
|
6
renovate.json
Normal file
6
renovate.json
Normal file
@@ -0,0 +1,6 @@
|
||||
{
|
||||
"$schema": "https://docs.renovatebot.com/renovate-schema.json",
|
||||
"extends": [
|
||||
"config:base"
|
||||
]
|
||||
}
|
@@ -1 +1,6 @@
|
||||
beautifulsoup4
|
||||
beautifulsoup4~=4.12.2
|
||||
convopyro==0.5
|
||||
pyrogram~=2.0.103
|
||||
tgcrypto~=1.2.5
|
||||
pathlib~=1.0.1
|
||||
ujson~=5.7.0
|
14
strings.json
14
strings.json
@@ -4,15 +4,14 @@
|
||||
"get_number": "**Get card number (Var. 1):**\nOn the front bottom side of your card, number may be found\n\n**Get card number (Var. 2):**\n1. Scan QR on the card\n2. Open webpage from code\n3. Numer should be found in **Номер карти \"Здорова Вода\"** or **Номер карти BWT Aqua** fields",
|
||||
"card_linked": "Linked card: `{0}`\n\nPlease, make sure the number is correct before using the bot",
|
||||
"card_unlinked": "Card was unlinked from your Telegram",
|
||||
"card_not_linked": "You don't have any linked card.\n\nВы можете задать её с помощью команды /setcard\n\n{0}",
|
||||
"card_not_linked": "You don't have any linked card.\n\nВYou can set it using /setcard\n\n{0}",
|
||||
"error_occured": "An error occurred while getting the amount of remaining water on the card.\n\nPlease make sure the linked card number is correct. If you are sure that the bot is broken, please contact @profitroll.\n\nLink your card: /setcard\n\n{0}",
|
||||
"error_new": "An error occurred while getting the amount of remaining water on the card.\n\nLast a few weeks BWT seems to return empty string to balance request from our server. We assume that our server has been blacklisted.\n\nTo check your balance you can use official [BWT App](https://bwtaqua.com.ua/en/#app) or simply bookmark this page: {0}.",
|
||||
"card_balance": "Card's balance is {0} l. of water",
|
||||
"top_up": "[Click here to top up](https://bwtaqua.com.ua/card-topup/?id={0})",
|
||||
"cancel": "Operation cancelled",
|
||||
"cancel_none": "Nothing to cancel",
|
||||
"cancel_first": "Operation ongoing. Cancel the current one using /cancel to run this action",
|
||||
"enter_number": "Enter card number",
|
||||
"send_number": "Please, send your card number"
|
||||
"send_number": "Please, send your card number\nIf you want to abort this operation, use /cancel"
|
||||
},
|
||||
"uk": {
|
||||
"welcome": "Привіт-привіт!\n\nЦей бот дозволяє дізнатись скільки літрів залишилось на вашій карточці.\n\n**Команди:**\n • /balance – дізнатись баланс карти\n • /setcard – приав'язати карту\n • /resetcard – відв'язати карту\n\n{0}\n\nРозробник **не має жодного відношення до BWT Aqua**, а бот створений лише для особистого, некомерційного використання.",
|
||||
@@ -21,12 +20,11 @@
|
||||
"card_unlinked": "Картку відв'язано від вашого Telegram",
|
||||
"card_not_linked": "У вас немає прив'язаної картки.\n\nВи можете зробити це за допомогою команди /setcard\n\n{0}",
|
||||
"error_occured": "При отриманні води на карточці виникла помилка.\n\nБудь ласка, упевніться що номер карти правильний. Якщо ви впевнені, що номер картки правильний та бот зламався – зв'яжіться з @profitroll.\n\nПрив'язати карту: /setcard\n\n{0}",
|
||||
"error_new": "При отриманні води на карточці виникла помилка.\n\nОстанні тижні BWT повертає нашому серверу порожні строки замість балансу. Є підозри, що сервер потрапив у блеклист.\n\nДля перевірки балансу рекомендуємо користуватись офіційним [додатком BWT](https://bwtaqua.com.ua/#app) або просто додати цю сторінку у закладки: {0}.",
|
||||
"card_balance": "На карточці {0} л. води",
|
||||
"top_up": "[Натисніть для поповнення](https://bwtaqua.com.ua/card-topup/?id={0})",
|
||||
"cancel": "Операція відмінена",
|
||||
"cancel_none": "Нема що відміняти",
|
||||
"cancel_first": "Триває інша операція. Відмініть триваючу операцію командою /cancel щоб запустити іншу дію",
|
||||
"cancel": "Операцію скасовано",
|
||||
"enter_number": "Введіть номер картки",
|
||||
"send_number": "Будь ласка, надішліть номер вашої картки"
|
||||
"send_number": "Будь ласка, надішліть номер вашої картки\nЯкщо ви хочете скасувати цю операцію, використовуйте /cancel"
|
||||
}
|
||||
}
|
Reference in New Issue
Block a user