✨ 构建 Docekr API 镜像

.github/workflows/docker-publish.yml

@@ -0,0 +1,83 @@
+name: Docker Publish
+
+permissions:
+  contents: read
+  packages: write
+
+on:
+  workflow_dispatch:
+  push:
+    branches:
+      - main
+    tags:
+      - "v*"
+
+env:
+  IMAGE_NAME: ${{ github.repository }}
+
+jobs:
+  build-and-push:
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout repository
+        uses: actions/checkout@v4
+
+      - name: Extract metadata (tags, labels) for Docker
+        id: meta
+        uses: docker/metadata-action@v5
+        with:
+          images: |
+            ${{ env.IMAGE_NAME }}
+            ghcr.io/${{ env.IMAGE_NAME }}
+          tags: |
+            type=ref,event=branch
+            type=ref,event=tag
+
+      - name: Get version
+        id: get_version
+        run: echo ::set-output name=VERSION::${GITHUB_REF/refs\/tags\//}
+
+      - name: Set up QEMU
+        uses: docker/setup-qemu-action@v3
+
+      - name: Set up Docker Buildx
+        uses: docker/setup-buildx-action@v3
+
+      - name: Login to GitHub Container Registry
+        uses: docker/login-action@v3
+        with:
+          registry: ghcr.io
+          username: ${{ github.repository_owner }}
+          password: ${{ secrets.GITHUB_TOKEN }}
+
+      - name: Login to Docker Hub
+        uses: docker/login-action@v3
+        with:
+          username: ${{ secrets.DOCKERHUB_USERNAME }}
+          password: ${{ secrets.DOCKERHUB_TOKEN }}
+
+      - name: Build and push Docker image
+        uses: docker/build-push-action@v5
+        with:
+          build-args: |
+            MAINTAINER=${{ github.repository_owner }}
+            BRANCH=${{ github.ref_name }}
+            BUILD_SHA=${{ github.sha }}
+            BUILD_TAG=${{ steps.get_version.outputs.VERSION }}
+          context: .
+          platforms: |
+            linux/amd64
+            linux/arm64
+          file: ./Dockerfile
+          push: true
+          tags: ${{ steps.meta.outputs.tags }}
+          labels: ${{ steps.meta.outputs.labels }}
+
+      - name: Update repo description
+        uses: peter-evans/dockerhub-description@v4
+        with:
+          username: ${{ secrets.DOCKERHUB_USERNAME }}
+          password: ${{ secrets.DOCKERHUB_TOKEN }}
+          repository: ${{ env.IMAGE_NAME }}
+          short-description: ${{ github.event.repository.description }}
+          enable-url-completion: true

Dockerfile

@@ -0,0 +1,29 @@
+# 使用官方 Python 镜像作为基础镜像
+FROM python:3.12.2-alpine
+
+# 设置工作目录
+WORKDIR /app
+
+# 将当前目录中的文件添加到工作目录中
+COPY . /app
+
+# 安装依赖
+RUN pip install --no-cache-dir -r requirements.txt \
+    pip install --no-cache-dir flask
+
+# 时区
+ENV TZ="Asia/Shanghai"
+
+# 构建版本
+ARG BUILD_SHA
+ARG BUILD_TAG
+ENV BUILD_SHA=$BUILD_SHA
+ENV BUILD_TAG=$BUILD_TAG
+
+ENV WHITELIST_NUM=
+
+# 端口
+EXPOSE 10000
+
+# 运行应用程序
+CMD ["python", "./app/api_server.py"]

app/api_server.py

@@ -0,0 +1,115 @@
+#!/usr/bin/env python3
+# _*_ coding:utf-8 _*_
+
+import os
+import sys
+import json
+from datetime import datetime
+from flask import Flask, request, jsonify
+
+# 导入父目录的依赖
+current_dir = os.path.dirname(os.path.abspath(__file__))
+parent_dir = os.path.dirname(current_dir)
+sys.path.append(parent_dir)
+from telecom_class import Telecom
+
+telecom = Telecom()
+
+app = Flask(__name__)
+app.json.ensure_ascii = False
+app.json.sort_keys = False
+
+# 登录信息存储文件
+LOGIN_INFO_FILE = os.environ.get("CONFIG_PATH", "./config/login_info.json")
+
+
+def load_login_info():
+    """加载本地登录信息"""
+    try:
+        with open(LOGIN_INFO_FILE, "r", encoding="utf-8") as f:
+            return json.load(f)
+    except FileNotFoundError:
+        return {}
+
+
+def save_login_info(login_info):
+    """保存登录信息到本地"""
+    with open(LOGIN_INFO_FILE, "w", encoding="utf-8") as f:
+        json.dump(login_info, f, ensure_ascii=False, indent=2)
+
+
+@app.route("/login", methods=["POST", "GET"])
+def login():
+    """登录接口"""
+    if request.method == "POST":
+        data = request.get_json() or {}
+    else:
+        data = request.args
+    phonenum = data.get("phonenum")
+    password = data.get("password")
+    if not phonenum or not password:
+        return jsonify({"message": "手机号和密码不能为空"}), 400
+    elif whitelist_num := os.environ.get("WHITELIST_NUM"):
+        if not phonenum in whitelist_num:
+            return jsonify({"message": "手机号不在白名单"}), 400
+
+    login_info = load_login_info()
+    data = telecom.do_login(phonenum, password)
+    if data.get("responseData").get("resultCode") == "0000":
+        login_info[phonenum] = data["responseData"]["data"]["loginSuccessResult"]
+        login_info[phonenum]["password"] = password
+        login_info[phonenum]["createTime"] = datetime.now().strftime(
+            "%Y-%m-%d %H:%M:%S"
+        )
+        save_login_info(login_info)
+        return jsonify(data), 200
+    else:
+        return jsonify({"message": "登录失败"}), 400
+
+
+def query_data(query_func):
+    """
+    查询数据，如果本地没有登录信息或密码不匹配，则尝试登录后再查询
+    """
+    if request.method == "POST":
+        data = request.get_json() or {}
+    else:
+        data = request.args
+    phonenum = data.get("phonenum")
+    password = data.get("password")
+
+    login_info = load_login_info()
+    if phonenum in login_info and login_info[phonenum]["password"] == password:
+        telecom.set_login_info(login_info[phonenum])
+        data = query_func()
+        if data:
+            return jsonify(data), 200
+    # 重新登录
+    login_data, status_code = login()
+    if status_code == 200:
+        telecom.set_login_info(
+            json.loads(login_data.data)["responseData"]["data"]["loginSuccessResult"]
+        )
+        data = query_func()
+        if data:
+            return jsonify(data), 200
+        else:
+            return jsonify({"message": "查询失败"}), 400
+    else:
+        return jsonify({"message": "手机号或密码错误"}), 400
+
+
+@app.route("/qryImportantData", methods=["POST", "GET"])
+def qry_important_data():
+    """查询重要数据接口"""
+    return query_data(telecom.qry_important_data)
+
+
+@app.route("/userFluxPackage", methods=["POST", "GET"])
+def user_flux_package():
+    """查询流量包接口"""
+    return query_data(telecom.user_flux_package)
+
+
+if __name__ == "__main__":
+    app.run(debug=os.environ.get("DEBUG", False), host="0.0.0.0", port=10000)