Upload RefundService.py

service/RefundService.py

@@ -0,0 +1,224 @@
+from pydantic import BaseModel
+import uuid, hmac, hashlib, requests
+
+# --- Thông tin định danh ---
+PARTNER_CODE = "MOMO"
+ACCESS_KEY = "F8BBA842ECF85"
+SECRET_KEY = "K951B6PE1waDMi640xX08PD3vg6EkVlz"
+
+# --- Model yêu cầu hoàn tiền ---
+class RefundRequest(BaseModel):
+    partnerCode: str
+    orderId: str
+    requestId: str
+    amount: int
+    transId: int
+    description: str
+    lang: str = "vi"
+
+# --- Tạo chữ ký ---
+def generate_refund_signature(data: dict) -> str:
+    raw_data = (
+        f"accessKey={ACCESS_KEY}&amount={data['amount']}&description={data['description']}"
+        f"&orderId={data['orderId']}&partnerCode={PARTNER_CODE}"
+        f"&requestId={data['requestId']}&transId={data['transId']}"
+    )
+    return hmac.new(SECRET_KEY.encode(), raw_data.encode(), hashlib.sha256).hexdigest()
+
+def call_momo_refund(amount: int, trans_id: int, description: str = "Hoàn tiền giao dịch"):
+    order_id = str(uuid.uuid4())
+    request_id = str(uuid.uuid4())
+
+    refund = RefundRequest(
+        partnerCode=PARTNER_CODE,
+        orderId=order_id,
+        requestId=request_id,
+        amount=amount,
+        transId=trans_id,
+        description=description,
+    )
+
+    payload = refund.dict()
+    payload["signature"] = generate_refund_signature(payload)
+
+    response = requests.post(
+        "https://test-payment.momo.vn/v2/gateway/api/refund",
+        json=payload,
+        headers={"Content-Type": "application/json"},
+        timeout=90,
+    )
+
+    return response.json()
+
+
+# result = call_momo_refund(amount=100000, trans_id=4506777858)
+# print(result)
+
+
+import hmac
+import hashlib
+import requests
+from datetime import datetime
+import uuid
+
+# ====== CẤU HÌNH MẶC ĐỊNH ======
+VNP_URL = "https://sandbox.vnpayment.vn/merchant_webapi/api/transaction"
+VNP_TMN_CODE = 'MCQ3SXQ1'
+VNP_SECRET_KEY = '2QS3MSRVQQAFN36I5GA02DWB353KO4FD'
+IP_ADDR = "127.0.0.1"
+
+# ====== HÀM TẠO CHECKSUM HMAC SHA512 ======
+def hmac_sha512(key: str, data: str) -> str:
+    return hmac.new(key.encode(), data.encode(), hashlib.sha512).hexdigest()
+
+# ====== HÀM HOÀN TIỀN VNPAY ======
+def call_vnpay_refund(
+    txn_ref: str,
+    amount: int,
+    transaction_date: str,
+    create_by: str,
+    order_info: str,
+    transaction_no: str = "",      # optional
+    transaction_type: str = "02",  # "02": full refund, "03": partial
+    vnp_url: str = VNP_URL,
+    vnp_tmn_code: str = VNP_TMN_CODE,
+    vnp_secret_key: str = VNP_SECRET_KEY,
+    ip_addr: str = IP_ADDR,
+):
+    # Lấy thời gian hiện tại (vnp_CreateDate)
+    now = datetime.now().strftime("%Y%m%d%H%M%S")
+
+    # Tạo request body
+    params = {
+        "vnp_RequestId": str(uuid.uuid4().hex[:32]),
+        "vnp_Version": "2.1.0",
+        "vnp_Command": "refund",
+        "vnp_TmnCode": vnp_tmn_code,
+        "vnp_TransactionType": transaction_type,
+        "vnp_TxnRef": txn_ref,
+        "vnp_Amount": str(amount),  # nhân 100 nếu VNP yêu cầu
+        "vnp_TransactionNo": transaction_no,
+        "vnp_TransactionDate": transaction_date,
+        "vnp_CreateBy": create_by,
+        "vnp_CreateDate": now,
+        "vnp_IpAddr": ip_addr,
+        "vnp_OrderInfo": order_info
+    }
+
+    sign_data = "|".join([
+        params["vnp_RequestId"],
+        params["vnp_Version"],
+        params["vnp_Command"],
+        params["vnp_TmnCode"],
+        params["vnp_TransactionType"],
+        params["vnp_TxnRef"],
+        params["vnp_Amount"],
+        params["vnp_TransactionNo"],
+        params["vnp_TransactionDate"],
+        params["vnp_CreateBy"],
+        params["vnp_CreateDate"],
+        params["vnp_IpAddr"],
+        params["vnp_OrderInfo"]
+    ])
+
+
+    params["vnp_SecureHash"] = hmac_sha512(vnp_secret_key, sign_data)
+    response = requests.post(vnp_url, json=params, headers={"Content-Type": "application/json"})
+    return {
+        "status_code": response.status_code,
+        "data": response.json()
+    }
+
+
+
+import hmac
+import hashlib
+import requests
+import time
+from datetime import datetime
+import random
+import string
+
+# Cấu hình tài khoản ZaloPay
+APP_ID = "2553"
+KEY1 = "PcY4iZIKFCIdgZvA6ueMcMHHUbRLYjPL"
+QUERY_ENDPOINT = "https://sb-openapi.zalopay.vn/v2/query"
+REFUND_ENDPOINT = "https://sb-openapi.zalopay.vn/v2/refund"
+
+def generate_mac(data: str, key: str) -> str:
+    return hmac.new(key.encode(), data.encode(), hashlib.sha256).hexdigest()
+
+def generate_m_refund_id():
+    date_part = datetime.now().strftime("%y%m%d")
+    random_part = ''.join(random.choices(string.digits, k=10))
+    return f"{date_part}_{APP_ID}_{random_part}"
+
+def call_zalopay_refund(app_trans_id: str, amount: int, description: str, refund_fee_amount: int = None):
+    """
+    Hàm thực hiện hoàn tiền giao dịch ZaloPay dựa trên app_trans_id.
+
+    Args:
+        app_trans_id (str): Mã giao dịch của ứng dụng.
+        amount (int): Số tiền hoàn lại (đơn vị: VND).
+        description (str): Mô tả lý do hoàn tiền.
+        refund_fee_amount (int, optional): Số tiền phí hoàn (nếu có).
+    
+    Returns:
+        dict: Kết quả phản hồi từ ZaloPay.
+    """
+    # Bước 1: Truy vấn để lấy zp_trans_id
+    mac_input = f"{APP_ID}|{app_trans_id}|{KEY1}"
+    mac = generate_mac(mac_input, KEY1)
+
+    query_payload = {
+        "app_id": APP_ID,
+        "app_trans_id": app_trans_id,
+        "mac": mac
+    }
+
+    try:
+        query_response = requests.post(QUERY_ENDPOINT, data=query_payload)
+        query_result = query_response.json()
+        zp_trans_id = query_result.get("zp_trans_id")
+
+        if not zp_trans_id:
+            print("❌ Không tìm thấy zp_trans_id từ app_trans_id.")
+            return {"code": -1, "message": "Không tìm thấy zp_trans_id"}
+
+    except Exception as e:
+        print("❌ Lỗi khi truy vấn giao dịch:", e)
+        return {"code": -2, "message": "Lỗi truy vấn giao dịch"}
+
+    timestamp = int(time.time() * 1000)
+    m_refund_id = generate_m_refund_id()
+
+    if refund_fee_amount is not None:
+        hmac_data = f"{APP_ID}|{zp_trans_id}|{amount}|{refund_fee_amount}|{description}|{timestamp}"
+    else:
+        hmac_data = f"{APP_ID}|{zp_trans_id}|{amount}|{description}|{timestamp}"
+
+    refund_mac = generate_mac(hmac_data, KEY1)
+
+    refund_data = {
+        "app_id": APP_ID,
+        "m_refund_id": m_refund_id,
+        "zp_trans_id": zp_trans_id,
+        "amount": amount,
+        "timestamp": timestamp,
+        "description": description,
+        "mac": refund_mac
+    }
+
+    if refund_fee_amount is not None:
+        refund_data["refund_fee_amount"] = refund_fee_amount
+
+    try:
+        headers = {
+            "Content-Type": "application/x-www-form-urlencoded"
+        }
+        refund_response = requests.post(REFUND_ENDPOINT, data=refund_data, headers=headers)
+        return refund_response.json()
+
+    except Exception as e:
+        print("❌ Lỗi khi gửi hoàn tiền:", e)
+        return {"code": -3, "message": "Lỗi hoàn tiền"}