Initial commit

modules/bwt.py

@@ -0,0 +1,50 @@
+#-*- coding: utf-8 -*-
+
+import os
+import traceback
+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:
+
+        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')
+
+        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)
+        else:
+            print(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")

modules/colors.py

@@ -0,0 +1,22 @@
+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'
+
+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'