Subversion Repositories SmartDukaan

'''
Created on 22-Mar-2010

@author: ashish
'''
from elixir import *
from elixir.entity import Entity
from elixir.fields import Field
from sqlalchemy.types import Integer
from elixir.relationships import OneToMany, ManyToOne
import datetime



class Category(Entity):
    '''
    This class represents the catalog. 
    '''
    id = Field(Integer, primary_key=True)
    name = Field(String(30))
    isRoot = Field(Integer)
    addedOn = Field(DateTime)
    status = Field(Integer)
    items = ManyToMany('Item')
    children = OneToMany('Category', inverse='parent')
    parent = ManyToOne('Category', inverse='children')

   
    def __repr__(self):
        return "<id>%id </id>"%(self.id)


class Vendor(Entity):
    id = Field(Integer, primary_key=True)
    name = Field(String(30))
    status = Field(Integer)
    addedOn = Field(datetime)
    warehouses = ManyToMany('Warehouse')
    def __repr__(self):
        return "<Vendor>%s</Vendor>" %(self.name)

    
class Warehouse(Entity):
    id = Field(Integer, primary_key=True)
    location = Field(String(20))
    status = Field(Integer)
    addedOn = Field(datetime)
    vendors = ManyToMany('Vendor')
    inventory = OneToMany('CurrentInventorySnapshot')
    inventoryHistory = OneToMany('ItemInventoryHistory')
    
    def __repr__(self):
        return "<warehouse>%s</warehouse>" %(self.location)


class Item(Entity):
    id = Field(Integer, primary_key=True)
    name = Field(String(20))
    price = Field(Float)
    addedOn = Field(datetime)
    status = Field(Integer)
    startDate = Field(datetime)
    retireDate = Field(datetime)
    currentInventory = OneToMany('CurrentInventorySnapshot')
    inventoryHistory = OneToMany('ItemInventoryHistory')
    category = ManyToMany("Category")
    iteminfo = OneToMany("ItemInfo")
    statusChangeLog = OneToMany("ItemChangeLog")
    
    def __repr__(self):
        return "<Item>%s</item>" % (self.name)
    
    @property
    def get_total_inventory(self):
        if self.currentInventory:
            i = 0
            for entry in self.currentInventory:
                i += entry.availability
            return i
        
        else:
            return 0;
    
    @property    
    def get_all_warehouses(self):
        warehouses = []
        for warehouse in self.currentInventory.warehouse:
            warehouses.append(warehouse)
        return warehouses
    
class ItemInfo(Entity):
    key = Field(String(30))
    value = Field(String(100))
    item = ManyToOne("Item")
    
    def __repr__(self):
        return "<ItemInfo><key>%s</key><value>%s</value></ItemInfo>" % (self.key, self.value)

class CurrentInventorySnapshot(Entity):
    id = Field(Integer, primary_key=True)
    checkedOn = Field(DateTime, default=datetime.datetime)
    availibility = Field(Integer)
    item = ManyToOne("Item")
    warehouse = ManyToOne("Warehouse")
    
    def __repr__(self):
        return "<current inventory><availability> %s </availability><time>%s</time><item>%s</item><warehouse>%s</warehouse></current inventory>" %(self.availibility, self.checkedOn, self.item, self.warehouse)

#===============================================================================
# This is just a placeholder, will be using naveen's code for this.
#===============================================================================
    
class ItemInventoryHistory(Entity):
    id = Field(Integer, primary_key=True)    
    timestamp = Field(DateTime, default=datetime.datetime.now())
    availibility = Field(Integer)
    item = ManyToOne("Item")
    warehouse = ManyToOne("Warehouse")
    
    def __repr__(self):
        pass

class ItemChangeLog(Entity):
    id = Field(Integer, primary_key=True)
    old_status = Field(Integer)
    new_status = Field(Integer)
    timestamp = Field(datetime)
    item = ManyToOne("Item")
    
    def __repr__(self):
        return "<Log><id>%d</id><item>%d</item><old_status>%d</old_status><new_status>%d</new_status></Log>" %(self.id,self.item.id, self.old_status, self.new_status)    


def initialize():
    metadata.bind = "sqlite:///catalog.sqlite"
    metadata.bind.echo = True
    setup_all(True)