import pdb
import random
import time
import uuid
from datetime import timedelta, datetime

from sqlalchemy import func

from app.models import db
from app.models.ticket_lib import Ticket
from app.models.train_station_lib import TrainStation


class Order(db.Model):
    __tablename__ = 'order'

    id = db.Column(db.Integer, primary_key=True)
    order_no = db.Column(db.String(120), unique=True, nullable=False)
    price = db.Column(db.Numeric(8, 2))
    payment_time = db.Column(db.DateTime)
    state = db.Column(db.Integer, default=0)
    passenger_id = db.Column(db.Integer, db.ForeignKey('passenger.id'), nullable=False, index=True)
    created_at = db.Column(db.DateTime, default=func.now())
    updated_at = db.Column(db.DateTime, default=func.now())

    passenger = db.relationship('Passenger', backref=db.backref('orders'))

    def __repr__(self):
        return f'<Order {self.id}>'

    @classmethod
    def setOrderState(cls, order_no, passenger_id, **kwargs):
        # order = cls.query.filter_by(order_no=order_no, passenger_id=passenger_id).first()
        # if order:
        #     # 更新订单的属性
        #     for key, value in kwargs.items():
        #         setattr(order, key, value)
        #         if key == 'state' and value == 1:
        #             for ticket in order.tickets:
        #                 Ticket.updateState(ticket)  # Update state to 1
        #     # 提交更改到数据库
        #     db.session.commit()
        #     return order
        # else:
        #     # 处理订单未找到的情况
        #     return None
        try:
            # 尝试从数据库中查询特定的订单
            order = cls.query.filter_by(order_no=order_no, passenger_id=passenger_id).first()
            if order:
                # 如果找到了订单，则根据传入的参数更新订单的属性
                for key, value in kwargs.items():
                    setattr(order, key, value)
                    # 如果更新了订单状态为1，则更新该订单下所有车票的状态
                    if key == 'state' and value == 1:
                        for ticket in order.tickets:
                            Ticket.updateState(ticket)
                # 提交更改到数据库
                db.session.commit()
                return order
            else:
                # 如果没有找到订单，返回None
                return None
        except Exception as e:
            # 如果发生异常，回滚数据库更改并重新抛出异常
            db.session.rollback()
            raise e


def generate_unique_id(length=4):
    # 获取当前的秒级时间戳
    timestamp_seconds = int(time.time())

    # 生成 4 位随机数
    random_number = random.randint(0, 10 ** length - 1)

    # 确保随机数是 4 位数字
    random_number_str = str(random_number).zfill(length)

    # 生成唯一标识符
    unique_id = f"{timestamp_seconds}{random_number_str}"
    return unique_id


def generateOrder(params, passenger_id):
    # Create a new Order object
    order = Order(
        order_no=generate_unique_id(5),
        passenger_id=passenger_id
    )

    for e in params['tickets']:
        # 创建Ticket对象
        ticket = Ticket.generateTicket(e, passenger_id)
        # 计算起始站和终点站
        from_station = TrainStation.query.filter_by(train_no=ticket.train_no, station_name=ticket.from_station).one()
        to_station = TrainStation.query.filter_by(train_no=ticket.train_no, station_name=ticket.to_station).one()
        # 计算区间价格
        train_stations = TrainStation.query.filter(
            TrainStation.train_no == ticket.train_no,
            TrainStation.index > from_station.index,
            TrainStation.index <= to_station.index
        ).all()
        ticket.price = sum(e.price for e in train_stations)
        # 计算座位号
        seat_count = db.session.query(func.count(Ticket.id)).filter_by(
            train_no=ticket.train_no,
            date=ticket.date
        ).scalar()
        ticket.seat_no = seat_count + 1

        # 转换时间 计算出发时间
        interval_days = (
                datetime.strptime(ticket.date, "%Y/%m/%d") - datetime.combine(from_station.departure_time.date(),
                                                                              datetime.min.time())).days
        ticket.departure_time = from_station.departure_time + timedelta(days=interval_days)
        ticket.arrival_time = to_station.arrival_time + timedelta(days=interval_days)

        order.tickets.append(ticket)

    order.price = sum(ticket.price for ticket in order.tickets)

    return order