Publish dataease via gitea-publish skill

2026-08-09 00:36:02 +00:00
commit ac82f67e09
24 changed files with 7145 additions and 0 deletions
@@ -0,0 +1,82 @@
+import json
+import time
+import uuid
+import base64
+import os
+import requests
+import jwt
+from cryptography.hazmat.primitives.ciphers import Cipher, algorithms, modes
+from cryptography.hazmat.primitives import padding
+from cryptography.hazmat.backends import default_backend
+import urllib3
+
+urllib3.disable_warnings()
+
+class DataEaseClient:
+    def __init__(self, base_url, access_key, secret_key):
+        # Separate base URL and API prefix
+        self.base_url = base_url.rstrip('/')
+        self.api_prefix = os.environ.get('DATAEASE_API_PREFIX', '/de2api')
+        self.access_key = access_key
+        self.secret_key = secret_key
+
+    def _get_signature(self, uid, timestamp):
+        src_str = f"{self.access_key}|{uid}|{timestamp}"
+        padder = padding.PKCS7(128).padder()
+        padded_data = padder.update(src_str.encode('utf-8')) + padder.finalize()
+        cipher = Cipher(
+            algorithms.AES(self.secret_key.encode('utf-8')),
+            modes.CBC(self.access_key.encode('utf-8')),
+            backend=default_backend()
+        )
+        encryptor = cipher.encryptor()
+        ciphertext = encryptor.update(padded_data) + encryptor.finalize()
+        return base64.b64encode(ciphertext).decode('utf-8')
+
+    def _get_headers(self):
+        timestamp = str(int(time.time() * 1000))
+        uid = str(uuid.uuid4())
+        signature = self._get_signature(uid, timestamp)
+        claims = {
+            "accessKey": self.access_key,
+            "signature": signature
+        }
+        token = jwt.encode(claims, self.secret_key, algorithm="HS256")
+        if isinstance(token, bytes):
+            token = token.decode('utf-8')
+        
+        return {
+            "accessKey": self.access_key,
+            "signature": signature,
+            "x-de-ask-token": token,
+            "timestamp": timestamp,
+            "nonce": uid,
+            "Content-Type": "application/json"
+        }
+
+    def get(self, path, params=None):
+        url = f"{self.base_url}{self.api_prefix}{path}"
+        response = requests.get(url, headers=self._get_headers(), params=params, verify=False)
+        return response
+
+    def post(self, path, payload=None):
+        url = f"{self.base_url}{self.api_prefix}{path}"
+        response = requests.post(url, headers=self._get_headers(), json=payload, verify=False)
+        return response
+
+    def get_dataset_fields(self, dataset_id):
+        return self.post(f"/datasetField/listByDatasetGroup/{dataset_id}").json()
+
+    def update_publish_status(self, dashboard_id, name, status=1, type='dashboard', mobile_layout=False, active_view_ids=None):
+        url = f"{self.base_url}{self.api_prefix}/dataVisualization/updatePublishStatus"
+        payload = {
+            "id": dashboard_id,
+            "name": name,
+            "mobileLayout": mobile_layout,
+            "activeViewIds": active_view_ids or [],
+            "status": status,
+            "type": type
+        }
+        response = requests.post(url, headers=self._get_headers(), json=payload, verify=False)
+        response.raise_for_status()
+        return response.json()