Subversion Repositories SmartDukaan

'''
Created on 29-Jul-2011

@author: Chandranshu
'''
from elixir.entity import Entity
from elixir.fields import Field
from elixir.options import using_options, using_table_options
from elixir.relationships import OneToMany
from shop2020.purchase.main.model.LineItem import LineItem
from shop2020.thriftpy.purchase.ttypes import PurchaseOrder as TPurchaseOrder, \
    POStatus
from shop2020.utils.Utils import to_java_date
from sqlalchemy.types import Integer, String, Float, DateTime, Enum
import datetime
import time




class PurchaseOrder(Entity):
    '''
    classdocs
    '''
    id = Field(Integer, primary_key=True, autoincrement=True)
    poNumber = Field(String(20))
    supplierId = Field(Integer)
    buyerId = Field(Integer)
    warehouseId = Field(Integer)
    warehouseAddressId = Field(Integer)
    shippingWarehouseId = Field(Integer) 
    shippingWarehouseAddressId = Field(Integer)
    status = Field(Integer)
    createdAt = Field(DateTime)
    updatedAt = Field(DateTime)
    totalCost = Field(Float)
    freightCharges = Field(Float)
    realizedCost = Field(Float)
    realizedFreightCharges = Field(Float)
    type = Field(Enum('REAL', 'VIRTUAL'), default = 'REAL', server_default='REAL')
    taxType = Field(Integer)
    lineitems = OneToMany("LineItem")
    purchases = OneToMany("Purchase")
    using_options(shortnames=True)
    using_table_options(mysql_engine="InnoDB")
    
    def __init__(self, t_purchase_order):
        '''
        Constructor
        '''
        #Generate the poNumber
        self.supplierId = t_purchase_order.supplierId
        self.warehouseId = t_purchase_order.warehouseId
        if t_purchase_order.shippingWarehouseId:
            self.shippingWarehouseId = t_purchase_order.shippingWarehouseId
        else:
            self.shippingWarehouseId = t_purchase_order.warehouseId
        
        #TODO: This should be INIT. Category manager should review and mark it as ready.
        self.status = POStatus.READY
        self.createdAt = datetime.datetime.now()
        self.totalCost = t_purchase_order.totalCost
        self.freightCharges = t_purchase_order.freightCharges or 0
        self.realizedCost = 0
        self.realizedFreightCharges = 0
        if t_purchase_order.lineitems:
            for t_lineitem in t_purchase_order.lineitems:
                if t_lineitem.quantity == 0:            
                    t_purchase_order.lineitems.pop(t_purchase_order.lineitems.index(t_lineitem))
            self.lineitems = [LineItem(self, t_lineitem) for t_lineitem in t_purchase_order.lineitems]
            self.totalCost = sum([t_lineitem.quantity * t_lineitem.unitPrice for t_lineitem in t_purchase_order.lineitems])
        self.type = t_purchase_order.type
        self.poNumber = t_purchase_order.poNumber
        self.taxType = t_purchase_order.taxType
        self.buyerId = t_purchase_order.buyerId
        self.warehouseAddressId = t_purchase_order.warehouseAddressId
        self.shippingWarehouseAddressId = t_purchase_order.shippingWarehouseAddressId
                
        
    def to_thrift_object(self):
        t_purchase_order = TPurchaseOrder()
        t_purchase_order.id = self.id
        t_purchase_order.poNumber = self.poNumber
        t_purchase_order.supplierId = self.supplierId
        t_purchase_order.warehouseId = self.warehouseId
        t_purchase_order.warehouseAddressId = self.warehouseAddressId   
        t_purchase_order.shippingWarehouseId = self.shippingWarehouseId
        t_purchase_order.shippingWarehouseAddressId = self.shippingWarehouseAddressId
        t_purchase_order.status = self.status
        t_purchase_order.createdAt = to_java_date(self.createdAt)
        if self.updatedAt:
            t_purchase_order.updatedAt = to_java_date(self.updatedAt)
        t_purchase_order.totalCost = self.totalCost
        t_purchase_order.freightCharges = self.freightCharges
        t_purchase_order.realizedCost = self.realizedCost
        t_purchase_order.realizedFreightCharges = self.realizedFreightCharges
        
        t_purchase_order.lineitems = [lineitem.to_thrift_object() for lineitem in self.lineitems]
        t_purchase_order.taxType = self.taxType
        t_purchase_order.buyerId = self.buyerId
        return t_purchase_order

    def set_po_number(self):
        #Changed from SORL to PO after discussion from Rajneesh
        self.poNumber = "PO/" + time.strftime("%m-%y") + "/" + str(self.id)