Subversion Repositories SmartDukaan

'''

Created on 03-Dec-2012

@author: Amar

'''

from elixir.entity import Entity

from elixir.fields import Field

from elixir.options import using_options, using_table_options

from shop2020.thriftpy.purchase.ttypes import PurchaseReturn as TPurchaseReturn, PurchaseReturnInventoryType

from shop2020.utils.Utils import to_java_date

from sqlalchemy.types import Integer, Float, DateTime, Boolean, Enum

import datetime

class PurchaseReturn(Entity):

    '''

    classdocs

    '''

    id = Field(Integer, primary_key=True, autoincrement=True)

    vendorId = Field(Integer)

    amount = Field(Float)

    returnTimestamp = Field(DateTime)

    isSettled = Field(Boolean)

    type = Field(Enum('REAL', 'VIRTUAL'), default = 'REAL', server_default='REAL')

    returnInventoryType = Field(Enum('GOOD', 'BAD'), default = 'GOOD', server_default='GOOD')

    using_options(shortnames=True)

    using_table_options(mysql_engine="InnoDB")

    def __init__(self, vendorId, amount):

        '''

        Constructor

        '''

        self.vendorId = vendorId

        self.amount = amount

        self.returnTimestamp = datetime.datetime.now()

        self.isSettled = False

    def to_thrift_object(self):

        t_purchaseReturn = TPurchaseReturn()

        t_purchaseReturn.id = self.id

        t_purchaseReturn.vendorId = self.vendorId

        t_purchaseReturn.amount = self.amount

        t_purchaseReturn.returnTimestamp = to_java_date(self.returnTimestamp)

        t_purchaseReturn.isSettled = self.isSettled

        t_purchaseReturn.returnInventoryType = PurchaseReturnInventoryType._NAMES_TO_VALUES[self.returnInventoryType]

        return t_purchaseReturn