pyscriptbase 1.0.1__py3-none-any.whl

pybase/panel.py

@@ -0,0 +1,194 @@
+# encoding=utf8
+
+import requests
+import warnings
+from json import dumps as jsonDumps
+from typing import List
+
+warnings.filterwarnings("ignore")
+
+
+class Envrionment:
+    def __init__(self, id: int, name: str, value: str, remarks: str = "") -> None:
+        """
+        初始化
+        """
+        self.id = id
+        self.name = name
+        self.value = value
+        self.remarks = ""
+
+    def toJson(self) -> dict:
+        return {
+            "id": self.id,
+            "name": self.name,
+            "value": self.value,
+            "remarks": self.remarks,
+        }
+
+
+class PanelClient:
+    headers = {"content-type": "application/json", "accept": "application/json"}
+
+    def __init__(self, address: str, id: str, secret: str) -> None:
+        """
+        初始化
+        """
+        self.address = self.__validAddress__(address)
+        self.clientId = id
+        self.clientSecret = secret
+        self.valid = False
+        self.session = requests.session()
+        self.session.headers.update(self.headers)
+
+        self.__login__()
+
+    def __validAddress__(self, address: str) -> str:
+        if not address.startswith("http://") and not address.startswith("https://"):
+            address = f"http://{address}"
+        if address.endswith("/"):
+            address = address[0:-1]
+        return address
+
+    def __log__(self, content: str, start="", end="\n") -> None:
+        """
+        日志
+        """
+        print(content)
+
+    def __login__(self) -> None:
+        """
+        登录
+        """
+        url = f"{self.address}/open/auth/token?client_id={self.clientId}&client_secret={self.clientSecret}"
+        try:
+            rjson = self.session.get(url, verify=False).json()
+            if rjson["code"] == 200:
+                self.auth = f"{rjson['data']['token_type']} {rjson['data']['token']}"
+                self.session.headers.update({"Authorization": self.auth})
+                self.valid = True
+                self.__log__(f"面板登录成功：{rjson['data']['token']}", start="\n")
+            else:
+                self.valid = False
+                self.__log__(f"面板登录失败：{rjson['message']}", start="\n")
+        except Exception as e:
+            self.valid = False
+            self.__log__(f"面板登录失败：{str(e)}", start="\n")
+
+    def addEnvs(self, envs: List[Envrionment]) -> bool:
+        """
+        新建环境变量
+        """
+        url = f"{self.address}/open/envs"
+        try:
+            body = []
+            for env in envs:
+                body.append({"name": env.name, "value": env.value, "remarks": env.remarks})
+            rjson = self.session.post(url, data=jsonDumps(body), verify=False).json()
+            if rjson["code"] == 200:
+                self.__log__(f"新建环境变量成功：{len(envs)}")
+                return True
+            else:
+                self.__log__(f"新建环境变量失败：{rjson['message']}")
+                return False
+        except Exception as e:
+            self.__log__(f"新建环境变量失败：{str(e)}")
+            return False
+
+    def updateEnvs(self, env: Envrionment) -> bool:
+        """
+        更新环境变量
+        """
+        url = f"{self.address}/open/envs"
+        try:
+            body = jsonDumps(env.toJson())
+            rjson = self.session.put(url, verify=False, data=body).json()
+            if rjson["code"] == 200:
+                self.__log__(f"更新环境变量成功")
+            else:
+                self.__log__(f"更新环境变量失败：{rjson['message']}")
+        except Exception as e:
+            self.__log__(f"更新环境变量失败：{str(e)}")
+
+    def addOrUpdateEnv(self, env: Envrionment) -> bool:
+        """
+        添加或更新环境变量
+        """
+        envs = self.getEnvs()
+        if not envs:
+            self.addEnvs([env])
+            return True
+        for env_ in envs:
+            if env_.name == env.name:
+                env.id = env_.id
+                self.updateEnvs(env)
+                return True
+        self.addEnvs([env])
+
+    def getEnvs(self, search: str = "") -> list[Envrionment]:
+        """
+        获取环境变量
+        """
+        url = f"{self.address}/open/envs?searchValue={search}"
+        result: list[Envrionment] = []
+        try:
+            rjson = self.session.get(url, verify=False).json()
+            if rjson["code"] == 200:
+                for env in rjson["data"]:
+                    result.append(Envrionment(env["id"], env["name"], env["value"], env["remarks"]))
+            else:
+                self.__log__(f"获取环境变量失败：{rjson['message']}")
+        except Exception as e:
+            self.__log__(f"获取环境变量失败：{str(e)}")
+        return result
+
+    def deleteEnvs(self, ids: list[int]) -> bool:
+        """
+        删除环境变量
+        """
+        url = f"{self.address}/open/envs"
+        try:
+            rjson = self.session.delete(url, data=jsonDumps(ids), verify=False).json()
+            if rjson["code"] == 200:
+                self.__log__(f"删除环境变量成功：{len(ids)}")
+                return True
+            else:
+                self.__log__(f"删除环境变量失败：{rjson['message']}")
+                return False
+        except Exception as e:
+            self.__log__(f"删除环境变量失败：{str(e)}")
+            return False
+
+    def enableEnvs(self, ids: list[int]) -> bool:
+        """
+        启用环境变量
+        """
+        url = f"{self.address}/open/envs/enable"
+        try:
+            rjson = self.session.put(url, data=jsonDumps(ids), verify=False).json()
+            if rjson["code"] == 200:
+                self.__log__(f"启用环境变量成功：{len(ids)}")
+                return True
+            else:
+                self.__log__(f"启用环境变量失败：{rjson['message']}")
+                return False
+        except Exception as e:
+            self.__log__(f"启用环境变量失败：{str(e)}")
+            return False
+
+    def disableEnvs(self, ids: list[int]) -> bool:
+        """
+        禁用环境变量
+        """
+        url = f"{self.address}/open/envs/disable"
+        try:
+            rjson = self.session.put(url, data=jsonDumps(ids), verify=False).json()
+            if rjson["code"] == 200:
+                self.__log__(f"禁用环境变量成功：{len(ids)}")
+                return True
+            else:
+                self.__log__(f"禁用环境变量失败：{rjson['message']}")
+                return False
+        except Exception as e:
+            self.__log__(f"禁用环境变量失败：{str(e)}")
+            return False

pybase/pusher.py

@@ -0,0 +1,194 @@
+import smtplib
+from email.mime.multipart import MIMEMultipart
+from email.mime.text import MIMEText
+from . import env as Env
+
+
+class HtmlTableEmail:
+    """
+    HTML表格对象
+    @headers=列数
+    @rows=函数 不足自动补充空字符串 多余则忽略 [str|int|tuple] tuple=(str|int,G|R)有颜色
+    @extras 末尾额外信息
+    """
+
+    def __init__(self, title: str, headers: list[str] = [], appendIndex: bool = True):
+        self.title = title
+        self.headers: list[str] = headers
+        self.rows: list[str | int] = []
+        self.extras: list[str] = []
+        self.appendIndex = appendIndex
+
+    def toStr(self) -> str:
+        result = ""
+        for index, row in enumerate(self.rows):
+            result += f"[{index+1}] "
+            for i in range(len(row)):
+                if isinstance(row[i], tuple) and len(row[i]) >= 1:
+                    result += f"{row[i][0]} "
+                else:
+                    result += f"{row[i]} "
+            result += "\n"
+        for extra in self.extras:
+            result += f"\n{extra}\n"
+        return result
+
+    def toHtml(self) -> str:
+        table = self
+        thead = ""
+        tbody = ""
+        extras = ""
+
+        if table.appendIndex:
+            thead += "<th></th>"
+        for header in table.headers:
+            thead += f"<th>{header}</th>"
+
+        for index, row in enumerate(table.rows):
+            trs = ""
+            if table.appendIndex:
+                trs += f"<td>{index+1}</td>"
+            for i in range(len(table.headers)):
+                if len(row) - 1 >= i:
+                    if isinstance(row[i], tuple) and len(row[i]) >= 2:
+                        # 颜色
+                        if str(row[i][1]).upper in ["G", "GREEN"]:
+                            trs += f"<td class='green'>{row[i][0]}</td>"
+                        else:
+                            trs += f"<td class='red'>{row[i][0]}</td>"
+                    else:
+                        # 常规
+                        trs += f"<td>{row[i]}</td>"
+                else:
+                    trs += "<td></td>"
+            tbody += f"<tr>{trs}</tr>"
+
+        if table.extras:
+            extras = '<tr><th colspan="$$spans$$">其他报告</th></tr>'
+            for extra in table.extras:
+                extras += f'<tr><td colspan="$$spans$$">{extra}</td></tr>'
+
+        html = """
+    <!DOCTYPE html>
+    <html lang="zh-CN">
+    <head>
+        <meta charset="UTF-8" />
+        <title></title>
+        <style>
+        table {
+            width: 100%; /* 设置表格宽度 */
+            margin: auto; /* 表格居中 */
+            border-collapse: collapse; /* 边框合并 */
+            font-size: 14px;
+            text-align: center; /* 文本居中 */
+        }
+        th,
+        td {
+            padding: 10px; /* 内边距 */
+            border: 1px solid #ddd; /* 边框样式 */
+        }
+        th {
+            background-color: #f2f2f2; /* 表头背景颜色 */
+        }
+        tr:nth-child(even) {
+            background-color: #f9f9f9; /* 隔行变色 */
+        }
+        caption {
+            font-size: 1.5em;
+            margin-bottom: 10px;
+        }
+        .green {
+            color: green; /* 绿色文本 */
+        }
+        .red {
+            color: red; /* 红色文本 */
+        }
+        </style>
+    </head>
+    <body>
+        <table>
+        <thead><tr>$$thead$$</tr></thead>
+        <tbody>
+        $$tbody$$
+        $$extras$$
+        </tbody>
+        </table>
+    </body>
+    </html>
+    """
+        spans = len(table.headers) + (1 if table.appendIndex else 0)
+        html = html.replace("$$thead$$", thead)
+        html = html.replace("$$tbody$$", tbody)
+        html = html.replace("$$extras$$", extras)
+        html = html.replace("$$spans$$", str(spans))
+        return html
+
+
+class SMTPParams:
+    def __init__(self, server: str, port: int, sender: str, password: str):
+        self.server = server
+        self.port = port
+        self.sender = sender
+        self.password = password
+
+
+def sendQQMail(title: str, content: str, receiver: str = None, params: SMTPParams = None):
+    # 设置SMTP服务器地址和端口
+    smtp_server = params.server if params else Env.get("smtp_server")
+    port = params.port if params else Env.getInt("smtp_port")  # 对于TLS，通常使用587端口；对于SSL，使用465端口
+    sender = params.sender if params else Env.get("smtp_sender")  # 你的电子邮件地址
+    password = params.password if params else Env.get("smtp_password")  # 你的电子邮件密码（对于某些电子邮件提供商，你可能需要生成一个应用专用密码）
+    receiver = receiver if receiver else Env.get("smtp_receiver")
+
+    if not sender or not password:
+        return
+
+    # 创建消息对象
+    message = MIMEMultipart()
+    message["From"] = sender
+    message["To"] = receiver
+    message["Subject"] = title  # 邮件主题
+    body = content  # 邮件正文
+    message.attach(MIMEText(body, "plain"))
+
+    # 连接到SMTP服务器并发送邮件
+    try:
+        server = smtplib.SMTP(smtp_server, port)
+        server.starttls()  # 启动TLS模式
+        server.login(sender, password)  # 登录你的账户
+        server.sendmail(sender, receiver, message.as_string())  # 发送邮件
+    except Exception as e:
+        print(f"Error: unable to send email. {e}")
+
+
+def sendMail(title: str, content: str | HtmlTableEmail, receiver: str = None, params: SMTPParams = None):
+    # 设置SMTP服务器地址和端口
+    smtp_server = params.server if params else Env.get("smtp_server")
+    port = params.port if params else Env.getInt("smtp_port")  # 对于TLS，通常使用587端口；对于SSL，使用465端口
+    sender = params.sender if params else Env.get("smtp_sender")  # 你的电子邮件地址
+    password = params.password if params else Env.get("smtp_password")  # 你的电子邮件密码（对于某些电子邮件提供商，你可能需要生成一个应用专用密码）
+    receiver = receiver if receiver else Env.get("smtp_receiver")
+
+    if not sender or not password:
+        return
+
+    # 创建消息对象
+    message = MIMEMultipart()
+    message["From"] = sender
+    message["To"] = receiver
+    message["Subject"] = title  # 邮件主题
+    if isinstance(content, HtmlTableEmail):
+        body = content.toHtml()  # 邮件正文
+        message.attach(MIMEText(body, "html"))
+    else:
+        body = content  # 邮件正文
+        message.attach(MIMEText(body, "plain"))
+
+    # 连接到SMTP服务器并发送邮件
+    try:
+        server = smtplib.SMTP(smtp_server, port)
+        server.starttls()  # 启动TLS模式
+        server.login(sender, password)  # 登录你的账户
+        server.sendmail(sender, receiver, message.as_string())  # 发送邮件
+    except Exception as e:
+        print(f"Error: unable to send email. {e}")

pybase/script.py

@@ -0,0 +1,90 @@
+from concurrent.futures import ThreadPoolExecutor, as_completed
+from . import net
+from . import database
+
+
+class ScriptApp:
+    _proxies = []
+    _cloudProxies = []
+
+    def __init__(self, index: int = 1, useProxy: bool = True, debug: bool = False, useCloud: bool = True) -> None:
+        self.index = index
+        self.logs = []
+        self.debug = debug
+        self.net = net.NetRequest()
+        self.cloudNet = net.CloudRequest()
+        self.cloudNet.net = self.net
+        useProxy and self.__setProxy__(index)
+        useCloud and self.__setCloudProxy__(index)
+
+    def __setProxy__(self, index: int) -> None:
+        config = database.getDataConfig("account", "proxy")
+        if not config.user or not config.host or not config.password:
+            return
+        if len(ScriptApp._proxies) == 0:
+            try:
+                sql = database.SQLHelper(user=config.user, password=config.password, port=config.port, database=config.database, host=config.host)
+                fields = ["host", "port"]
+                wheres = [["type", 1], ["state", 1]]
+                datas = sql.set_table(config.table).set_wheres(wheres).set_fields(fields).query_dict()
+                proxies = [None]
+                for data in datas:
+                    proxies.append({"http": f"http://{data['host']}:{data['port']}", "https": f"http://{data['host']}:{data['port']}"})
+                ScriptApp._proxies = proxies
+            except:
+                ScriptApp._proxies = [None]
+        if len(ScriptApp._proxies):
+            self.net.proxies = ScriptApp._proxies[index % len(ScriptApp._proxies)]
+
+    def __setCloudProxy__(self, index: int) -> None:
+        config = database.getDataConfig("account", "cloud_proxy")
+        if not config.user or not config.host or not config.password:
+            return
+        if len(ScriptApp._cloudProxies) == 0:
+            try:
+                sql = database.SQLHelper(user=config.user, password=config.password, port=config.port, database=config.database, host=config.host)
+                fields = ["url", "token"]
+                wheres = [["state", 1]]
+                datas = sql.set_table(config.table).set_wheres(wheres).set_fields(fields).query_dict()
+                ScriptApp._cloudProxies = datas
+            except:
+                ScriptApp._cloudProxies = []
+        if len(ScriptApp._cloudProxies):
+            self.cloudNet.cloud = ScriptApp._cloudProxies[index % len(ScriptApp._cloudProxies)]
+
+    def __log__(self, msg: str | dict = "", flush: bool = False):
+        if flush or self.debug:
+            print(msg, flush=True)
+        else:
+            if isinstance(msg, str):
+                self.logs.append(msg)
+            else:
+                self.logs.append(str(msg))
+
+    def __runFunc__(self, fun: callable, *args, **kwargs):
+        try:
+            return fun(*args, **kwargs)
+        except Exception as e:
+            self.__log__(e)
+
+
+class InviteObj:
+    id = ""
+    accpet: int = 0
+    send: int = 0
+    params = {}
+    target: list[dict] = []
+
+    def __init__(self):
+        pass
+
+
+def threadPool(func, args, num=5):
+    # 启用线程池
+    executor = ThreadPoolExecutor(max_workers=num)
+    tasks = [executor.submit(func, arg, index + 1) for index, arg in enumerate(args)]
+    results = []
+    for future in as_completed(tasks):
+        data = future.result()
+        results.append(data)
+    return results

pybase/util.py

@@ -0,0 +1,86 @@
+from concurrent.futures import ThreadPoolExecutor, as_completed
+from datetime import date, datetime, timedelta
+import random
+import string
+
+
+def generateRandomPhoneUA():
+    # 手机设备品牌与型号
+    devices = ["Samsung Galaxy S21", "Xiaomi Mi 11", "Google Pixel 6", "Huawei P40", "OPPO Reno5", "Vivo X60", "OnePlus 9"]
+
+    # 操作系统版本
+    os_versions = ["Android 11", "Android 12", "Android 13"]
+
+    # 浏览器类型
+    browsers = ["Chrome"]
+
+    # 随机选择设备、操作系统和浏览器
+    device = random.choice(devices)
+    os_version = random.choice(os_versions)
+    browser = random.choice(browsers)
+
+    # 构建完整的User-Agent字符串
+    ua_base = f"Mozilla/5.0 ({device}; Android {os_version}; Mobile) AppleWebKit"
+    ua = f"{ua_base}/537.36 (KHTML, like Gecko) {browser}/{random.randint(80, 99)}.0.{random.randint(0, 99)} Mobile Safari/537.36"
+
+    return ua
+
+
+def generateRandomStr(len: int):
+    return "".join(random.choices(string.ascii_letters + string.digits, k=len))
+
+
+def getNextHourSeconds() -> float:
+    # 获取当前时间
+    now = datetime.now()
+    # 计算下一小时的时间点
+    next_hour = (now + timedelta(hours=1)).replace(minute=0, second=0, microsecond=0)
+    # 计算距离下一小时的总秒数
+    seconds_until_next_hour = (next_hour - now).total_seconds()
+    return seconds_until_next_hour
+
+
+def formatTimestamp(ts: int | float | str = None) -> str:
+    if isinstance(ts, int) or isinstance(ts, float):
+        return datetime.fromtimestamp(ts).strftime("%Y-%m-%d %H:%M:%S")
+    elif ts == None:
+        return datetime.now().strftime("%Y-%m-%d %H:%M:%S.%f")[:-3]
+    else:
+        return ts
+
+
+def formatDate(ts: int, split="") -> str:
+    return datetime.fromtimestamp(ts).strftime(f"%Y{split}%m{split}%d")
+
+
+def isLastDayOfMonth():
+    today = date.today()
+    next_day = today + timedelta(days=1)
+    return next_day.month != today.month
+
+
+def daysDifference(date_str: str) -> int:
+    # 尝试将输入的日期字符串转换为datetime对象
+    try:
+        # 假设输入的日期格式是'yyyy-mm-dd'
+        input_date = datetime.strptime(date_str, "%Y-%m-%d").date()
+    except ValueError:
+        # 如果日期格式不正确，返回错误信息或处理异常
+        return 0
+
+    # 获取今天的日期
+    today = datetime.now().date()
+
+    # 计算两个日期之间的天数差
+    delta = today - input_date
+    days_diff = delta.days
+
+    return days_diff
+
+
+def threadPool(func, args, num=5):
+    # 启用线程池
+    executor = ThreadPoolExecutor(max_workers=num)
+    tasks = [executor.submit(func, arg, index + 1) for index, arg in enumerate(args)]
+    for future in as_completed(tasks):
+        data = future.result()

pybase/webview.py

@@ -0,0 +1,54 @@
+import platform
+from time import sleep
+from . import env
+from selenium import webdriver
+from traceback import format_exc
+import os
+from pathlib import Path
+from selenium.webdriver.common.by import By
+
+ENV_NAME = "remote_selenium"
+
+
+def buildDriver(options: webdriver.ChromeOptions, rm: bool = True) -> webdriver.Chrome:
+    """
+    获取一个WebDriver实例
+    环境变量: remote_selenium 远程地址
+    """
+    remote = env.get(ENV_NAME)
+    if rm and remote:
+        return webdriver.Remote(command_executor=remote, options=options)
+    else:
+        return webdriver.Chrome(options=options)
+
+
+def getUserDir(project: str, user: str):
+    """
+    获取用户数据目录
+    """
+    if platform.system().lower() == "windows":
+        dirs = ["E:", "cache", "selenium", project, user]
+    else:
+        dirs = [str(Path.home()), "cache", "selenium", project, user]
+    return os.sep.join(dirs)
+
+
+def clearUserData(driver: webdriver.Chrome):
+    """
+    清除用户数据
+    """
+    try:
+        driver.get("chrome://settings/clearBrowserData")
+        sleep(2)
+        checkes = driver.find_elements(By.CLASS_NAME, "settings-checkbox")
+        for checke in checkes:
+            label = checke.text.lower()
+            checked = checke.get_attribute("no-set-pref").lower()
+            if label.find("cookie") >= 0 and checked == "checked":
+                checke.click()
+            elif label.find("cookie") <= 0 and checked != "checked":
+                checke.click()
+        clearButton = driver.find_element(By.ID, "clearButton")
+        # clearButton.click()
+    except:
+        print("clearUserData", format_exc())

pyscriptbase-1.0.1.dist-info/METADATA

@@ -0,0 +1,24 @@
+Metadata-Version: 2.1
+Name: pyscriptbase
+Version: 1.0.1
+Summary: python script base library
+Home-page: 
+Author: ASMan
+Author-email: 
+Description-Content-Type: text/markdown
+Requires-Dist: loguru
+Requires-Dist: selenium
+Requires-Dist: pymysql
+Requires-Dist: pycryptodome
+Requires-Dist: pyyaml
+Requires-Dist: xmltodict
+Requires-Dist: prettytable
+Requires-Dist: pyjwt
+Requires-Dist: fake-useragent
+Requires-Dist: beautifulsoup4
+Requires-Dist: requests
+Requires-Dist: lxml
+
+# Instructions
+
+python script base library

pyscriptbase-1.0.1.dist-info/RECORD

@@ -0,0 +1,14 @@
+pybase/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pybase/cipher.py,sha256=Bz5lqG9GuooOt2y4u495VA9BE5qAV3fPnouR8qzV7K4,5321
+pybase/database.py,sha256=xOBKuyBS7RRTnuaaP5VB1XRaBQ8P7sNBEeabJfRQhVc,13302
+pybase/env.py,sha256=7MqnRXEq6raicxfC-bA9jCczo63Gh9LjQbBO5VtJ7rY,2944
+pybase/net.py,sha256=WvExRZoA5iHpx4qtE0YV_Mf_lVNaYvQNwtp0Suzih4A,11883
+pybase/panel.py,sha256=1e9sod303rkuVJOuuym9jK-ZXUzkBom9x-svdnpZnjI,6827
+pybase/pusher.py,sha256=lE4X3epadRaHKrRs8vBZW1T35mD03QtG-1Bu9okvq5Q,7054
+pybase/script.py,sha256=MCw19l8JOeXxhAOHr8NhUMhMDGnLo8aM_F2zklZldYY,3486
+pybase/util.py,sha256=RxQibRYGZKG5YCC6ffc3FGsE4nMkawknAtrEhGrxSig,2813
+pybase/webview.py,sha256=NQkJuqBlVlC2X_wTcfhtrxUBoP4ie-VCfgBud5YjhD4,1687
+pyscriptbase-1.0.1.dist-info/METADATA,sha256=Iutu3GqD2xVj2aNKOMEjsLXdm9qPLgIM05q82c_rKg0,537
+pyscriptbase-1.0.1.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+pyscriptbase-1.0.1.dist-info/top_level.txt,sha256=gYgKDnq77vSbZG-_hkku2uwMNI_t6Z7_WQq4dek8Al4,7
+pyscriptbase-1.0.1.dist-info/RECORD,,

pyscriptbase-1.0.1.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: bdist_wheel (0.43.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

pyscriptbase-1.0.1.dist-info/top_level.txt

@@ -0,0 +1 @@
+pybase