mock_coupon_server.py

#!/usr/bin/env python3
"""
Simple mock coupon service for demonstration purposes
"""
import json
import time
from datetime import datetime, timedelta
from http.server import HTTPServer, BaseHTTPRequestHandler
from urllib.parse import urlparse, parse_qs
import uuid

# Mock data storage
coupons = []

# Sample coupons
sample_coupons = [
    {
        "id": str(uuid.uuid4()),
        "code": "WELCOME20",
        "name": "Welcome Bonus",
        "description": "20% off for new customers",
        "type": "WELCOME",
        "discountType": "PERCENTAGE",
        "discountValue": 20.0,
        "minimumOrderAmount": 50.0,
        "maximumDiscountAmount": 100.0,
        "usageLimit": 1000,
        "usageLimitPerUser": 1,
        "usageCount": 45,
        "validFrom": "2024-01-01T00:00:00",
        "validUntil": "2024-12-31T23:59:59",
        "isPublic": True,
        "status": "ACTIVE",
        "createdAt": "2024-01-01T00:00:00",
        "updatedAt": "2024-01-01T00:00:00"
    },
    {
        "id": str(uuid.uuid4()),
        "code": "FLASH15",
        "name": "Flash Sale",
        "description": "Limited time 15% discount",
        "type": "FLASH_SALE",
        "discountType": "PERCENTAGE",
        "discountValue": 15.0,
        "minimumOrderAmount": 25.0,
        "maximumDiscountAmount": 75.0,
        "usageLimit": 500,
        "usageLimitPerUser": 1,
        "usageCount": 234,
        "validFrom": "2024-03-01T00:00:00",
        "validUntil": "2024-03-31T23:59:59",
        "isPublic": True,
        "status": "ACTIVE",
        "createdAt": "2024-03-01T00:00:00",
        "updatedAt": "2024-03-01T00:00:00"
    }
]

coupons.extend(sample_coupons)

class CouponHandler(BaseHTTPRequestHandler):
    def _set_headers(self, status=200):
        self.send_response(status)
        self.send_header('Content-type', 'application/json')
        self.send_header('Access-Control-Allow-Origin', '*')
        self.send_header('Access-Control-Allow-Methods', 'GET, POST, PUT, DELETE, OPTIONS')
        self.send_header('Access-Control-Allow-Headers', 'Content-Type, Authorization')
        self.end_headers()

    def do_OPTIONS(self):
        self._set_headers()

    def do_GET(self):
        parsed_path = urlparse(self.path)
        path = parsed_path.path
        
        if path == '/api/v1/coupons':
            self._set_headers()
            response = {
                "content": coupons,
                "totalElements": len(coupons),
                "totalPages": 1,
                "size": len(coupons),
                "number": 0
            }
            self.wfile.write(json.dumps(response).encode())
        elif path == '/api/v1/coupons/health':
            self._set_headers()
            self.wfile.write(json.dumps({"status": "UP"}).encode())
        else:
            self._set_headers(404)
            self.wfile.write(json.dumps({"error": "Not found"}).encode())

    def do_POST(self):
        parsed_path = urlparse(self.path)
        path = parsed_path.path
        
        if path == '/api/v1/coupons':
            content_length = int(self.headers['Content-Length'])
            post_data = self.rfile.read(content_length)
            
            try:
                coupon_data = json.loads(post_data.decode('utf-8'))
                
                # Create new coupon
                new_coupon = {
                    "id": str(uuid.uuid4()),
                    "code": coupon_data.get("code"),
                    "name": coupon_data.get("name"),
                    "description": coupon_data.get("description", ""),
                    "type": coupon_data.get("type"),
                    "discountType": coupon_data.get("discountType"),
                    "discountValue": coupon_data.get("discountValue"),
                    "minimumOrderAmount": coupon_data.get("minimumOrderAmount"),
                    "maximumDiscountAmount": coupon_data.get("maximumDiscountAmount"),
                    "usageLimit": coupon_data.get("usageLimit"),
                    "usageLimitPerUser": coupon_data.get("usageLimitPerUser", 1),
                    "usageCount": 0,
                    "validFrom": coupon_data.get("validFrom"),
                    "validUntil": coupon_data.get("validUntil"),
                    "isPublic": coupon_data.get("isPublic", True),
                    "allowedUserIds": coupon_data.get("allowedUserIds"),
                    "applicableCategories": coupon_data.get("applicableCategories"),
                    "applicableProductIds": coupon_data.get("applicableProductIds"),
                    "status": "ACTIVE",
                    "createdAt": datetime.now().isoformat(),
                    "updatedAt": datetime.now().isoformat()
                }
                
                coupons.append(new_coupon)
                
                self._set_headers(201)
                self.wfile.write(json.dumps(new_coupon).encode())
                
            except json.JSONDecodeError:
                self._set_headers(400)
                self.wfile.write(json.dumps({"error": "Invalid JSON"}).encode())
        else:
            self._set_headers(404)
            self.wfile.write(json.dumps({"error": "Not found"}).encode())

    def log_message(self, format, *args):
        print(f"[{datetime.now()}] {format % args}")

if __name__ == '__main__':
    server_address = ('', 8086)
    httpd = HTTPServer(server_address, CouponHandler)
    print(f"Mock Coupon Service running on port 8086...")
    print(f"Access at: http://localhost:8086/api/v1/coupons")
    httpd.serve_forever()