Subversion Repositories SmartDukaan

'''
Created on 22-Mar-2010

@author: ashish
'''
from elixir import metadata, setup_all
from elixir.entity import Entity
from elixir.fields import Field
from elixir.options import using_options, using_table_options
from elixir.relationships import OneToMany, ManyToOne
from sqlalchemy import create_engine
from sqlalchemy.types import Integer, String, DateTime, Float, Boolean, Text, Enum
import datetime
import elixir

class Vendor(Entity):
    id = Field(Integer, primary_key=True, autoincrement=True)
    name = Field(String(50))
    warehouses = OneToMany('Warehouse')
    using_options(shortnames=True)
    using_table_options(mysql_engine="InnoDB")

class Warehouse(Entity):
    id = Field(Integer, primary_key=True, autoincrement=True)
    displayName = Field(String(50))
    location = Field(String(200))
    addedOn = Field(DateTime)
    lastCheckedOn = Field(DateTime)
    tinNumber = Field(String(50))
    pincode = Field(String(10))
    vendorString = Field(String(50))
    badInventory = OneToMany('BadInventorySnapshot')
    inventory = OneToMany('CurrentInventorySnapshot')
    inventoryHistory = OneToMany('ItemInventoryHistory')
    logisticsLocation = Field(Integer)
    vendor = ManyToOne('Vendor')
    billingType = Field(Integer)
    inventoryType = Field(Enum('GOOD', 'BAD', 'VIRTUAL'))
    warehouseType = Field(Enum('OURS', 'THIRD_PARTY'))
    shippingWarehouseId = Field(Integer)
    billingWarehouseId = Field(Integer)
    transferDelayInHours = Field(Integer)
    isAvailabilityMonitored = Field(Boolean)
    using_options(shortnames=True)
    using_table_options(mysql_engine="InnoDB")
    
    def __repr__(self):
        return "<warehouse>%s</warehouse>" %(self.location)

    @property
    def all_items(self):
        items = []
        for ci in self.inventory:
            items.append(ci.item)
        return items 

class EntityIDGenerator(Entity):
    id=Field(Integer, primary_key=True)
    using_options(shortnames=True)
    using_table_options(mysql_engine="InnoDB")
    
class Item(Entity):
    id = Field(Integer, primary_key=True, autoincrement=True)
    product_group = Field(String(100))
    brand = Field(String(100), default='', server_default='')
    model_number = Field(String(50), default='', server_default='')
    model_name = Field(String(100), default='', server_default='')
    color = Field(String(20), default='', server_default='')
    category = Field(Integer, default=0, server_default="0")
    comments = Field(String(200))
    catalog_item_id = Field(Integer)
    feature_id = Field(Integer)
    feature_description = Field(String(200))
    mrp = Field(Float)
    sellingPrice = Field(Float)
    weight = Field(Float)
    addedOn = Field(DateTime)
    updatedOn = Field(DateTime)
    startDate = Field(DateTime)
    retireDate = Field(DateTime)
    comingSoonStartDate = Field(DateTime)
    expectedArrivalDate = Field(DateTime)
    status = Field(Integer)
    status_description = Field(String(100))
    bestDealText = Field(String(100))
    bestDealValue = Field(Float)
    bestSellingRank = Field(Integer)
    currentInventory = OneToMany('CurrentInventorySnapshot')
    inventoryHistory = OneToMany('ItemInventoryHistory')
    statusChangeLog = OneToMany("ItemChangeLog")
    preferredWarehouse = Field(Integer)
    defaultForEntity = Field(Boolean)
    risky = Field(Boolean)
    expectedDelay = Field(Integer)
    warranty_period = Field(Integer)
    defaultWarehouse = Field(Integer)
    isWarehousePreferenceSticky = Field(Boolean)
    preferredVendor = Field(Integer)
    type = Field(Enum('SERIALIZED', 'NON_SERIALIZED'), default = 'NON_SERIALIZED', server_default='NON_SERIALIZED')
    hasItemNo = Field(Boolean)
    using_options(shortnames=True)
    using_table_options(mysql_engine="InnoDB")
    
    def __repr__(self):
        return "<Item>%d</item>" % (self.id)
    
    @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):
        return [ci.warehouse for ci in self.currentInventory]

class VendorItemMapping(Entity):
    vendor = ManyToOne('Vendor', primary_key=True)
    item_key = Field(String(200), primary_key=True)
    item = ManyToOne('Item')
    using_options(shortnames=True)
    using_table_options(mysql_engine="InnoDB")
    
class VendorItemPricing(Entity):
    vendor = ManyToOne('Vendor', primary_key=True)
    item = ManyToOne('Item', primary_key=True)
    mop = Field(Float)
    dealerPrice = Field(Float)
    transfer_price = Field(Float)
    using_options(shortnames=True)
    using_table_options(mysql_engine="InnoDB")

class VendorItemProcurementDelay(Entity):
    vendor = ManyToOne('Vendor', primary_key=True)
    item = ManyToOne('Item', primary_key=True)
    procurementDelay = Field(Integer)
    using_options(shortnames=True)
    using_table_options(mysql_engine="InnoDB")

class VendorHolidays(Entity):
    vendor = ManyToOne('Vendor', primary_key=True)
    holidayType = Field(Integer, primary_key=True, autoincrement=False)
    holidayValue = Field(Integer, primary_key=True, autoincrement=False)
    occasion = Field(String(100))
    using_options(shortnames=True)
    using_table_options(mysql_engine="InnoDB")
        
class CurrentInventorySnapshot(Entity):
    item = ManyToOne("Item", primary_key=True)
    warehouse = ManyToOne("Warehouse", primary_key=True)
    availibility = Field(Integer)
    reserved = Field(Integer)
    using_options(shortnames=True)
    using_table_options(mysql_engine="InnoDB")
    
    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)

class BadInventorySnapshot(Entity):
    item = ManyToOne("Item", primary_key=True)
    warehouse = ManyToOne("Warehouse", primary_key=True)
    availability = Field(Integer)
    using_options(shortnames=True)
    using_table_options(mysql_engine="InnoDB")

class ItemInventoryHistory(Entity):
    id = Field(Integer, primary_key=True, autoincrement = True)    
    timestamp = Field(DateTime, default=datetime.datetime.now())
    availibility = Field(Integer)
    item = ManyToOne("Item")
    warehouse = ManyToOne("Warehouse")
    using_options(shortnames=True)
    using_table_options(mysql_engine="InnoDB")
    
    def __repr__(self):
        pass

class ItemChangeLog(Entity):
    id = Field(Integer, primary_key=True, autoincrement=True)
    old_status = Field(Integer)
    new_status = Field(Integer)
    timestamp = Field(DateTime)
    item = ManyToOne("Item")
    using_options(shortnames=True)
    using_table_options(mysql_engine="InnoDB")
    
    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)    

class Category(Entity):
    id = Field(Integer, primary_key=True, autoincrement=False)
    label = Field(String(50))
    description = Field(String(200))
    parent_category_id = Field(Integer)
    display_name = Field(String(50))
    using_options(shortnames=True)
    using_table_options(mysql_engine="InnoDB")

class SimilarItems(Entity):
    item = ManyToOne('Item', primary_key=True)
    catalog_item_id = Field(Integer, primary_key=True, autoincrement=False)
    using_options(shortnames=True)
    using_table_options(mysql_engine="InnoDB")

class ProductNotification(Entity):
    item = ManyToOne('Item', primary_key=True)
    email = Field(String(100), primary_key=True, autoincrement=False)
    addedOn = Field(DateTime, primary_key=True, autoincrement=False) 
    using_options(shortnames=True)
    using_table_options(mysql_engine="InnoDB")

class Source(Entity):
    id = Field(Integer, primary_key=True, autoincrement=True)
    name = Field(String(50))
    identifier = Field(String(100))
    using_options(shortnames=True)
    using_table_options(mysql_engine="InnoDB")

class SourceItemPricing(Entity):
    source = ManyToOne('Source', primary_key=True)
    item = ManyToOne('Item', primary_key=True)
    mrp = Field(Float)
    sellingPrice = Field(Float)
    using_options(shortnames=True)
    using_table_options(mysql_engine="InnoDB")
    
class AuthorizationLog(Entity):
    id = Field(Integer, primary_key=True, autoincrement = True)    
    timestamp = Field(DateTime, default=datetime.datetime.now())
    item = ManyToOne('Item')
    username = Field(String(30))
    reason = Field(Text)
    using_options(shortnames=True)
    using_table_options(mysql_engine="InnoDB")

class MissedInventoryUpdate(Entity):
    id = Field(Integer, primary_key=True, autoincrement=True)
    itemKey = Field(String(200))
    quantity = Field(Integer)
    warehouseId = Field(Integer)
    isIgnored = Field(Boolean)
    timestamp = Field(DateTime)
    using_options(shortnames=True)
    using_table_options(mysql_engine="InnoDB")

class ItemAvailabilityCache(Entity):
    itemId = Field(Integer, primary_key=True, autoincrement=False)
    warehouseId = Field(Integer)
    expectedDelay = Field(Integer)
    billingWarehouseId = Field(Integer)
    sellingPrice = Field(Float) 
    using_options(shortnames=True)
    using_table_options(mysql_engine="InnoDB")

def initialize(dbname='catalog', db_hostname="localhost"):
    #metadata.bind = "sqlite:///inventory-new.sqlite" #need to read it from configserver.
    #metadata.bind = 'mysql://root:shop2020@localhost/catalog'
    engine = create_engine('mysql://root:shop2020@' + db_hostname + '/' + dbname, pool_recycle=7200)
    metadata.bind = engine
    metadata.bind.echo = True
    setup_all(True)

if __name__=="__main__":
    initialize()