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)
    items = OneToMany('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))
    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))
    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))
    currentInventory = OneToMany('CurrentInventorySnapshot')
    inventoryHistory = OneToMany('ItemInventoryHistory')
    category = ManyToOne("Category")
    def __repr__(self):
        return "<Item>%s</item>" % (self.name)
    
    def get_total_inventory(self):
        if self.currentInventory:
            i = 0
            for entry in self.currentInventory:
                i += entry.availability
            return i
        
        else:
            return 0;
        
    def get_all_warehouses(self):
        warehouses = []
        for warehouse in self.currentInventory.warehouse:
            warehouses.append(warehouse)
        return warehouses
    

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 ItemAttribs(Entity):
    id = Field(Integer, primary_key=True)
    def __repr__(self):
        pass
    

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
    

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