add workflow partner1site,dev
This commit is contained in:
root
parent
684465c71f
commit
19f9a062cc
|
|
@ -1,177 +1,178 @@
|
||||||
#!/usr/bin/env python3
|
#!/usr/bin/env python3
|
||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
"""
|
"""
|
||||||
partner1site 多数据来源 API 请求模块
|
partner1site 多数据来源 API 请求模块
|
||||||
支持自动分页，并存储到 PostgreSQL
|
支持自动分页，并存储到 PostgreSQL
|
||||||
"""
|
"""
|
||||||
|
|
||||||
import random
|
import random
|
||||||
import hmac
|
import hmac
|
||||||
import hashlib
|
import hashlib
|
||||||
import base64
|
import base64
|
||||||
import requests
|
import requests
|
||||||
import json
|
import json
|
||||||
import uuid
|
import uuid
|
||||||
from datetime import datetime, timezone, timedelta
|
from datetime import datetime, timezone, timedelta
|
||||||
from typing import List, Dict, Any
|
from typing import List, Dict, Any
|
||||||
import psycopg2
|
import psycopg2
|
||||||
|
|
||||||
# ======= 配置区 =======
|
# ======= 配置区 =======
|
||||||
ACCESS_KEY = "75c4ab4d-6a67-4aed-8b1d-5bb64fd36afc"
|
ACCESS_KEY = "75c4ab4d-6a67-4aed-8b1d-5bb64fd36afc"
|
||||||
SECRET_KEY = "117347a7dd066a50a4d2973c5f3d5ba9101094c5"
|
SECRET_KEY = "117347a7dd066a50a4d2973c5f3d5ba9101094c5"
|
||||||
|
|
||||||
# URL 用占位符，外部替换
|
# URL 用占位符，外部替换
|
||||||
BASE_URLS = {
|
BASE_URLS = {
|
||||||
"visits": "http://onesite.tek.cn/api/summary/visits", # 客户拜访数据
|
"visits": "http://onesite.tek.cn/api/summary/visits", # 客户拜访数据
|
||||||
"reports": "http://onesite.tek.cn/api/summary/visits", # 报备数据
|
"reports": "http://onesite.tek.cn/api/summary/visits", # 报备数据
|
||||||
"pos_datas": "http://onesite.tek.cn/api/summary/visits", # POS数据
|
"pos_datas": "http://onesite.tek.cn/api/summary/visits", # POS数据
|
||||||
"customer_and_contact_datas": "http://onesite.tek.cn/api/summary/visits" # 客户及联系人数据
|
"customer_and_contact_datas": "http://onesite.tek.cn/api/summary/visits" # 客户及联系人数据
|
||||||
}
|
}
|
||||||
|
|
||||||
PG_DSN = dict(
|
PG_DSN = dict(
|
||||||
database="dataops_db",
|
database="dataops_db",
|
||||||
user="dbuser_dba",
|
user="dbuser_dba",
|
||||||
password="EmBRxnmmjnE3",
|
password="EmBRxnmmjnE3",
|
||||||
host="124.221.232.219",
|
host="124.221.232.219",
|
||||||
port="5432"
|
port="5432"
|
||||||
)
|
)
|
||||||
|
|
||||||
API_ID = "2460976d-00c1-47d9-84b2-33e66d68" # 外部传入 api_id，占位符
|
API_ID = "2460976d-00c1-47d9-84b2-33e66d68" # 外部传入 api_id，占位符
|
||||||
# ======================
|
# ======================
|
||||||
|
|
||||||
|
|
||||||
class Partner1SiteClient:
|
class Partner1SiteClient:
|
||||||
"""
|
"""
|
||||||
partner1site API 客户端
|
partner1site API 客户端
|
||||||
"""
|
"""
|
||||||
|
|
||||||
def __init__(self, access_key: str, secret_key: str):
|
def __init__(self, access_key: str, secret_key: str):
|
||||||
self.ak = access_key
|
self.ak = access_key
|
||||||
self.sk = secret_key
|
self.sk = secret_key
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def urlsafe_b64encode(data: bytes) -> str:
|
def urlsafe_b64encode(data: bytes) -> str:
|
||||||
return base64.urlsafe_b64encode(data).decode()
|
return base64.urlsafe_b64encode(data).decode()
|
||||||
|
|
||||||
def gen_token(self, expire_sec: int = 600) -> str:
|
def gen_token(self, expire_sec: int = 600) -> str:
|
||||||
"""生成 token —— 按调通算法"""
|
"""生成 token —— 按调通算法"""
|
||||||
random_num = str(random.randint(100000, 999999))
|
random_num = str(random.randint(100000, 999999))
|
||||||
deadline = int(datetime.now(timezone.utc).timestamp()) + expire_sec
|
deadline = int(datetime.now(timezone.utc).timestamp()) + expire_sec
|
||||||
parm_str = f"{random_num}:{deadline}"
|
parm_str = f"{random_num}:{deadline}"
|
||||||
enc_parm = self.urlsafe_b64encode(parm_str.encode())
|
enc_parm = self.urlsafe_b64encode(parm_str.encode())
|
||||||
sign = hmac.new(self.sk.encode(), enc_parm.encode(), hashlib.sha1).digest()
|
sign = hmac.new(self.sk.encode(), enc_parm.encode(), hashlib.sha1).digest()
|
||||||
enc_sign = self.urlsafe_b64encode(sign.hex().encode())
|
enc_sign = self.urlsafe_b64encode(sign.hex().encode())
|
||||||
return f"{self.ak}:{enc_sign}:{enc_parm}"
|
return f"{self.ak}:{enc_sign}:{enc_parm}"
|
||||||
|
|
||||||
def fetch_all_pages(self, api_name: str, params: Dict[str, Any]):
|
def fetch_all_pages(self, api_name: str, params: Dict[str, Any]):
|
||||||
"""通用分页请求"""
|
"""通用分页请求"""
|
||||||
if api_name not in BASE_URLS:
|
if api_name not in BASE_URLS:
|
||||||
raise ValueError(f"未知 API 数据来源: {api_name}")
|
raise ValueError(f"未知 API 数据来源: {api_name}")
|
||||||
|
|
||||||
base_url = BASE_URLS[api_name]
|
base_url = BASE_URLS[api_name]
|
||||||
all_data = []
|
all_data = []
|
||||||
page_num = 0
|
page_num = 0
|
||||||
page_size = 50 # 固定页容量
|
page_size = 50 # 固定页容量
|
||||||
|
|
||||||
while True:
|
while True:
|
||||||
token = self.gen_token()
|
token = self.gen_token()
|
||||||
params_with_paging = dict(params)
|
params_with_paging = dict(params)
|
||||||
params_with_paging.update({
|
params_with_paging.update({
|
||||||
"token": token,
|
"token": token,
|
||||||
"size": page_size,
|
"size": page_size,
|
||||||
"page": page_num
|
"page": page_num
|
||||||
})
|
})
|
||||||
|
|
||||||
resp = requests.get(base_url, params=params_with_paging, timeout=15)
|
resp = requests.get(base_url, params=params_with_paging, timeout=15)
|
||||||
resp.raise_for_status()
|
resp.raise_for_status()
|
||||||
data_json = resp.json()
|
data_json = resp.json()
|
||||||
|
|
||||||
if data_json.get("code") != 100 or not data_json.get("success", False):
|
if data_json.get("code") != 100 or not data_json.get("success", False):
|
||||||
raise RuntimeError(f"{api_name} API 错误: {data_json.get('message')}")
|
raise RuntimeError(f"{api_name} API 错误: {data_json.get('message')}")
|
||||||
|
|
||||||
content = data_json.get("data", {}).get("content", [])
|
content = data_json.get("data", {}).get("content", [])
|
||||||
all_data.extend(content)
|
all_data.extend(content)
|
||||||
print(f"[{api_name}] 页码 {page_num} -> 本页 {len(content)} 条")
|
print(f"[{api_name}] 页码 {page_num} -> 本页 {len(content)} 条")
|
||||||
|
|
||||||
if data_json.get("data", {}).get("last", True):
|
if data_json.get("data", {}).get("last", True):
|
||||||
break
|
break
|
||||||
|
|
||||||
page_num += 1
|
page_num += 1
|
||||||
|
|
||||||
return all_data
|
return all_data
|
||||||
|
|
||||||
|
|
||||||
def save_json_to_pg(data: list, api_id: str) -> None:
|
def save_json_to_pg(data: list, api_id: str) -> None:
|
||||||
"""写入 PostgreSQL：软删历史 + 插入新数据"""
|
"""写入 PostgreSQL：软删历史 + 插入新数据"""
|
||||||
print(f"[save_to_pg] API={api_id} 写入 PG，记录数={len(data)}")
|
print(f"[save_to_pg] API={api_id} 写入 PG，记录数={len(data)}")
|
||||||
sql = """
|
sql = """
|
||||||
UPDATE data_api.api_data
|
UPDATE data_api.api_data
|
||||||
SET is_loaded = '1'
|
SET is_loaded = '1'
|
||||||
WHERE api_id = %s;
|
WHERE api_id = %s;
|
||||||
|
|
||||||
INSERT INTO data_api.api_data
|
INSERT INTO data_api.api_data
|
||||||
(id, api_id, data, total_num, is_loaded, status,
|
(id, api_id, data, total_num, is_loaded, status,
|
||||||
request_tm, execute_tm, remark)
|
request_tm, execute_tm, remark)
|
||||||
VALUES (%s, %s, %s, %s, '0', '0',
|
VALUES (%s, %s, %s, %s, '0', '0',
|
||||||
current_timestamp(0), current_timestamp(0), '');
|
current_timestamp(0), current_timestamp(0), '');
|
||||||
"""
|
"""
|
||||||
try:
|
try:
|
||||||
with psycopg2.connect(**PG_DSN) as conn:
|
with psycopg2.connect(**PG_DSN) as conn:
|
||||||
with conn.cursor() as cur:
|
with conn.cursor() as cur:
|
||||||
cur.execute(sql,
|
cur.execute(sql,
|
||||||
(api_id,
|
(api_id,
|
||||||
str(uuid.uuid4()),
|
str(uuid.uuid4()),
|
||||||
api_id,
|
api_id,
|
||||||
json.dumps(data, ensure_ascii=False),
|
json.dumps(data, ensure_ascii=False),
|
||||||
len(data)))
|
len(data)))
|
||||||
conn.commit()
|
conn.commit()
|
||||||
print(f"[save_to_pg] API={api_id} 写入完成")
|
print(f"[save_to_pg] API={api_id} 写入完成")
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
raise RuntimeError(f"PG写入错误: {e}")
|
raise RuntimeError(f"PG写入错误: {e}")
|
||||||
|
|
||||||
|
|
||||||
def get_previous_date(days: int = 0) -> str:
|
def get_previous_date(days: int = 0) -> str:
|
||||||
"""获取 N 天前的日期字符串"""
|
"""获取 N 天前的日期字符串"""
|
||||||
return (datetime.now() - timedelta(days=days)).strftime("%Y-%m-%d")
|
return (datetime.now() - timedelta(days=days)).strftime("%Y-%m-%d")
|
||||||
|
|
||||||
|
|
||||||
def main():
|
def main():
|
||||||
client = Partner1SiteClient(ACCESS_KEY, SECRET_KEY)
|
client = Partner1SiteClient(ACCESS_KEY, SECRET_KEY)
|
||||||
|
|
||||||
# ✅ 1️⃣ 客户拜访数据
|
# ✅ 1️⃣ 客户拜访数据
|
||||||
visits_data = client.fetch_all_pages(
|
visits_data = client.fetch_all_pages(
|
||||||
api_name="visits",
|
api_name="visits",
|
||||||
params={"startInsertDate": get_previous_date(5), "endInsertDate": get_previous_date(0)}
|
# params={"startInsertDate": get_previous_date(5), "endInsertDate": get_previous_date(0)}
|
||||||
)
|
params={}
|
||||||
save_json_to_pg(visits_data, API_ID)
|
)
|
||||||
|
save_json_to_pg(visits_data, API_ID)
|
||||||
# ❌ 2️⃣ 报备数据（注释掉，后续可启用）
|
|
||||||
"""
|
# ❌ 2️⃣ 报备数据（注释掉，后续可启用）
|
||||||
reports_data = client.fetch_all_pages(
|
"""
|
||||||
api_name="reports",
|
reports_data = client.fetch_all_pages(
|
||||||
params={"startApplyDate": get_previous_date(5), "endApplyDate": get_previous_date(0)}
|
api_name="reports",
|
||||||
)
|
params={"startApplyDate": get_previous_date(5), "endApplyDate": get_previous_date(0)}
|
||||||
save_json_to_pg(reports_data, API_ID)
|
)
|
||||||
"""
|
save_json_to_pg(reports_data, API_ID)
|
||||||
|
"""
|
||||||
# ❌ 3️⃣ POS 数据（注释掉，后续可启用）
|
|
||||||
"""
|
# ❌ 3️⃣ POS 数据（注释掉，后续可启用）
|
||||||
pos_data = client.fetch_all_pages(
|
"""
|
||||||
api_name="pos_datas",
|
pos_data = client.fetch_all_pages(
|
||||||
params={"startPosInsertDate": get_previous_date(10), "endPosInsertDate": get_previous_date(0)}
|
api_name="pos_datas",
|
||||||
)
|
params={"startPosInsertDate": get_previous_date(10), "endPosInsertDate": get_previous_date(0)}
|
||||||
save_json_to_pg(pos_data, API_ID)
|
)
|
||||||
"""
|
save_json_to_pg(pos_data, API_ID)
|
||||||
|
"""
|
||||||
# ❌ 4️⃣ 客户及联系人数据（注释掉，后续可启用）
|
|
||||||
"""
|
# ❌ 4️⃣ 客户及联系人数据（注释掉，后续可启用）
|
||||||
cust_contact_data = client.fetch_all_pages(
|
"""
|
||||||
api_name="customer_and_contact_datas",
|
cust_contact_data = client.fetch_all_pages(
|
||||||
params={"customerId": 0, "customerContactId": 0}
|
api_name="customer_and_contact_datas",
|
||||||
)
|
params={"customerId": 0, "customerContactId": 0}
|
||||||
save_json_to_pg(cust_contact_data, API_ID)
|
)
|
||||||
"""
|
save_json_to_pg(cust_contact_data, API_ID)
|
||||||
|
"""
|
||||||
|
|
||||||
if __name__ == "__main__":
|
|
||||||
main()
|
if __name__ == "__main__":
|
||||||
|
main()
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue