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, ManyToMany

from sqlalchemy import create_engine

from sqlalchemy.types import Integer, String, DateTime, Float, Boolean, Text,\

    Date, 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')

    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")

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()