MagoRevisited/database.py

from datetime import datetime, timedelta
from models import (
    Customer, Supplier, Product, OrderLine, SalesOrder, PurchaseOrder, OrderStatus
)
from typing import List, Optional


class FakeDatabase:
    def __init__(self):
        self.customers = self._init_customers()
        self.suppliers = self._init_suppliers()
        self.products = self._init_products()
        self.sales_orders = self._init_sales_orders()
        self.purchase_orders = self._init_purchase_orders()
        self.next_customer_id = max([c.id for c in self.customers]) + 1
        self.next_supplier_id = max([s.id for s in self.suppliers]) + 1
        self.next_product_id = max([p.id for p in self.products]) + 1
        self.next_so_id = max([o.id for o in self.sales_orders]) + 1
        self.next_po_id = max([o.id for o in self.purchase_orders]) + 1

    def _init_customers(self) -> List[Customer]:
        return [
            Customer(
                id=1, code="CUST001", name="Acme Corporation",
                email="info@acme.com", phone="+39 02 1234567",
                address="Via Roma 10", city="Milano", country="IT",
                credit_limit=50000, balance=15000
            ),
            Customer(
                id=2, code="CUST002", name="TechFlow Industries",
                email="contact@techflow.de", phone="+49 30 555888",
                address="Hauptstrasse 45", city="Berlin", country="DE",
                credit_limit=75000, balance=32000
            ),
            Customer(
                id=3, code="CUST003", name="European Logistics",
                email="sales@eulog.fr", phone="+33 1 4520 3580",
                address="Avenue des Champs 20", city="Paris", country="FR",
                credit_limit=100000, balance=8500
            ),
            Customer(
                id=4, code="CUST004", name="Alpine Manufacturing",
                email="orders@alpmfg.ch", phone="+41 44 5678900",
                address="Industriestrasse 12", city="Zurich", country="CH",
                credit_limit=60000, balance=22000
            ),
        ]

    def _init_suppliers(self) -> List[Supplier]:
        return [
            Supplier(
                id=1, code="SUPP001", name="Global Components Ltd",
                email="sales@globalcomp.com", phone="+44 20 7946 0958",
                address="123 Oxford Street", city="London", country="UK",
                payment_terms_days=30, balance=5000
            ),
            Supplier(
                id=2, code="SUPP002", name="Asia Electronics Co.",
                email="export@asiaelec.com", phone="+886 2 8797 3000",
                address="Taipei Industrial Park", city="Taipei", country="TW",
                payment_terms_days=45, balance=18000
            ),
            Supplier(
                id=3, code="SUPP003", name="Nordic Materials AB",
                email="info@nordicmat.se", phone="+46 8 123 4567",
                address="Industrígatan 5", city="Stockholm", country="SE",
                payment_terms_days=60, balance=3500
            ),
        ]

    def _init_products(self) -> List[Product]:
        return [
            Product(
                id=1, code="PROD001", description="Industrial Widget A",
                category="Electronics", unit_price=125.50,
                quantity_in_stock=250, reorder_level=50, supplier_id=1
            ),
            Product(
                id=2, code="PROD002", description="Premium Connector Set",
                category="Hardware", unit_price=45.00,
                quantity_in_stock=180, reorder_level=30, supplier_id=2
            ),
            Product(
                id=3, code="PROD003", description="Steel Bracket Type B",
                category="Structural", unit_price=22.75,
                quantity_in_stock=500, reorder_level=100, supplier_id=3
            ),
            Product(
                id=4, code="PROD004", description="Precision Bearing Set",
                category="Mechanical", unit_price=89.99,
                quantity_in_stock=120, reorder_level=40, supplier_id=1
            ),
            Product(
                id=5, code="PROD005", description="Control Module MCU-32",
                category="Electronics", unit_price=234.50,
                quantity_in_stock=75, reorder_level=20, supplier_id=2
            ),
        ]

    def _init_sales_orders(self) -> List[SalesOrder]:
        now = datetime.now()
        return [
            SalesOrder(
                id=1, order_number="SO-2024-0001",
                order_date=now - timedelta(days=10),
                customer_id=1, customer_name="Acme Corporation",
                status=OrderStatus.CONFIRMED,
                lines=[
                    OrderLine(id=1, product_id=1, product_code="PROD001",
                             product_description="Industrial Widget A",
                             quantity=10, unit_price=125.50, total_amount=1255.00),
                    OrderLine(id=2, product_id=4, product_code="PROD004",
                             product_description="Precision Bearing Set",
                             quantity=5, unit_price=89.99, total_amount=449.95),
                ],
                total_amount=1704.95,
                delivery_date=now + timedelta(days=15)
            ),
            SalesOrder(
                id=2, order_number="SO-2024-0002",
                order_date=now - timedelta(days=5),
                customer_id=2, customer_name="TechFlow Industries",
                status=OrderStatus.CONFIRMED,
                lines=[
                    OrderLine(id=3, product_id=5, product_code="PROD005",
                             product_description="Control Module MCU-32",
                             quantity=8, unit_price=234.50, total_amount=1876.00),
                ],
                total_amount=1876.00,
                delivery_date=now + timedelta(days=10)
            ),
            SalesOrder(
                id=3, order_number="SO-2024-0003",
                order_date=now - timedelta(days=2),
                customer_id=3, customer_name="European Logistics",
                status=OrderStatus.DRAFT,
                lines=[
                    OrderLine(id=4, product_id=2, product_code="PROD002",
                             product_description="Premium Connector Set",
                             quantity=20, unit_price=45.00, total_amount=900.00),
                    OrderLine(id=5, product_id=3, product_code="PROD003",
                             product_description="Steel Bracket Type B",
                             quantity=15, unit_price=22.75, total_amount=341.25),
                ],
                total_amount=1241.25,
                delivery_date=now + timedelta(days=20)
            ),
        ]

    def _init_purchase_orders(self) -> List[PurchaseOrder]:
        now = datetime.now()
        return [
            PurchaseOrder(
                id=1, order_number="PO-2024-0001",
                order_date=now - timedelta(days=15),
                supplier_id=1, supplier_name="Global Components Ltd",
                status=OrderStatus.PARTIALLY_RECEIVED,
                lines=[
                    OrderLine(id=1, product_id=1, product_code="PROD001",
                             product_description="Industrial Widget A",
                             quantity=100, unit_price=110.00, total_amount=11000.00),
                ],
                total_amount=11000.00,
                expected_delivery_date=now + timedelta(days=5)
            ),
            PurchaseOrder(
                id=2, order_number="PO-2024-0002",
                order_date=now - timedelta(days=8),
                supplier_id=2, supplier_name="Asia Electronics Co.",
                status=OrderStatus.CONFIRMED,
                lines=[
                    OrderLine(id=2, product_id=5, product_code="PROD005",
                             product_description="Control Module MCU-32",
                             quantity=50, unit_price=195.00, total_amount=9750.00),
                ],
                total_amount=9750.00,
                expected_delivery_date=now + timedelta(days=25)
            ),
            PurchaseOrder(
                id=3, order_number="PO-2024-0003",
                order_date=now - timedelta(days=1),
                supplier_id=3, supplier_name="Nordic Materials AB",
                status=OrderStatus.DRAFT,
                lines=[
                    OrderLine(id=3, product_id=3, product_code="PROD003",
                             product_description="Steel Bracket Type B",
                             quantity=200, unit_price=20.00, total_amount=4000.00),
                ],
                total_amount=4000.00,
                expected_delivery_date=now + timedelta(days=30)
            ),
        ]

    # CRUD Operations
    def get_customers(self) -> List[Customer]:
        return self.customers

    def get_customer(self, customer_id: int) -> Optional[Customer]:
        return next((c for c in self.customers if c.id == customer_id), None)

    def create_customer(self, customer: Customer) -> Customer:
        customer.id = self.next_customer_id
        self.next_customer_id += 1
        self.customers.append(customer)
        return customer

    def update_customer(self, customer_id: int, customer_data: dict) -> Optional[Customer]:
        customer = self.get_customer(customer_id)
        if customer:
            for key, value in customer_data.items():
                if hasattr(customer, key) and value is not None:
                    setattr(customer, key, value)
        return customer

    def get_suppliers(self) -> List[Supplier]:
        return self.suppliers

    def get_supplier(self, supplier_id: int) -> Optional[Supplier]:
        return next((s for s in self.suppliers if s.id == supplier_id), None)

    def create_supplier(self, supplier: Supplier) -> Supplier:
        supplier.id = self.next_supplier_id
        self.next_supplier_id += 1
        self.suppliers.append(supplier)
        return supplier

    def get_products(self) -> List[Product]:
        return self.products

    def get_product(self, product_id: int) -> Optional[Product]:
        return next((p for p in self.products if p.id == product_id), None)

    def create_product(self, product: Product) -> Product:
        product.id = self.next_product_id
        self.next_product_id += 1
        self.products.append(product)
        return product

    def get_sales_orders(self) -> List[SalesOrder]:
        return self.sales_orders

    def get_sales_order(self, order_id: int) -> Optional[SalesOrder]:
        return next((o for o in self.sales_orders if o.id == order_id), None)

    def create_sales_order(self, order: SalesOrder) -> SalesOrder:
        order.id = self.next_so_id
        self.next_so_id += 1
        self.sales_orders.append(order)
        return order

    def get_purchase_orders(self) -> List[PurchaseOrder]:
        return self.purchase_orders

    def get_purchase_order(self, order_id: int) -> Optional[PurchaseOrder]:
        return next((o for o in self.purchase_orders if o.id == order_id), None)

    def create_purchase_order(self, order: PurchaseOrder) -> PurchaseOrder:
        order.id = self.next_po_id
        self.next_po_id += 1
        self.purchase_orders.append(order)
        return order

    # Dashboard statistics
    def get_dashboard_stats(self):
        sales_to_fulfill = sum(1 for o in self.sales_orders if o.status in [OrderStatus.CONFIRMED, OrderStatus.DRAFT])
        purchase_to_receive = sum(1 for o in self.purchase_orders if o.status in [OrderStatus.CONFIRMED, OrderStatus.DRAFT, OrderStatus.PARTIALLY_RECEIVED])

        total_receivable = sum(c.balance for c in self.customers)
        total_payable = sum(s.balance for s in self.suppliers)

        low_stock_items = [p for p in self.products if p.quantity_in_stock <= p.reorder_level]

        return {
            "sales_to_fulfill": sales_to_fulfill,
            "purchase_to_receive": purchase_to_receive,
            "total_receivable": total_receivable,
            "total_payable": total_payable,
            "low_stock_count": len(low_stock_items),
            "low_stock_items": low_stock_items,
            "total_customers": len(self.customers),
            "total_suppliers": len(self.suppliers),
            "total_products": len(self.products),
        }


# Global database instance
db = FakeDatabase()