Subversion Repositories SmartDukaan

'''
Created on 12-May-2010

@author: gaurav
'''
from datastore.DataDefinition import * #Phones, init, PhoneItem

from elixir import *
from datastore.DataDefinition import infibeam_data
import datetime


class DataHelper:
    
    def __init__(self):
        init()
    
    def add_vendor(self,name,url):
        v = Vendor()
        v.v_name = name
        v.v_url = url
        session.commit() 
        
    def set_all_crawled(self,bval):
        for ph in self.get_all_phones():
            ph.is_crawled = bval
            
    def set_crawled(self,url,bval):
        for ph in self.get_all_phones():
            if ph.url == url:
                ph.is_crawled = bval
        session.commit()     
                
    def get_all_phones(self):
        phones = PhoneItem.query.all()
        return phones
    
    def get_all_vendors(self):
        vi = Vendor.query.all()
        return vi
        
        
    def add_new_phone(self, url, name, source):
        pi = PhoneItem()
        pi.name = name
        pi.url = url
        pi.source = source
        session.commit()
        
    def add_price(self,url,price):
        for ph in self.get_all_phones():
            if ph.url == url:
                ph.price = price
        session.commit()
        
    def add_infiphone(self,name,shown_price,final_price):
        ai = infibeam_data()
        ai.name = name
        ai.shown_price = shown_price
        ai.final_price = final_price
        session.commit()
    
    def add_univervendor(self,name,site):
        ai = univercell_data()
        ai.v_name = name
        ai.v_site = site
        session.commit()
    
    def get_all_univervendors(self):
        vi = univercell_data.query.all()
        return vi
        
    def add_new_univerphone(self,name,shown_price,final_price):
        ai = univercell_items()
        ai.p_title = name
        ai.p_shown_price = shown_price
        ai.p_final_price = final_price
        session.commit()
        
    def add_ipbasic(self,name,site):
        ai = indiaplaza_data()
        ai.v_name = name
        ai.v_site = site
        session.commit()
    
    def add_mobstoreurl(self,url):
        ai = themobilestoreurls()
        ai.url = url
        session.commit()
        
       
    def add_naaptolurl(self,url):
        ai = naaptolurls()
        ai.url = url
        session.commit()
      
    def add_morenaaptolurl(self,url):
        ai = morenaaptolurls()
        ai.url = url
        session.commit()        
        
    def get_allmobstoreurls(self):
        ai = themobilestoreurls.query.all()
        return ai
    
    def get_allnaaptolurls(self):
        ai = naaptolurls.query.all()
        return ai
     
    def get_allmorenaaptolurls(self):
        ai = morenaaptolurls.query.all()
        return ai
    
    def add_new_mobstorephone(self,name,shown_pr,final_pr):
        ai = themobilestorephones()
        ai.name = name
        ai.shown_price = shown_pr
        ai.final_price = final_pr
        session.commit()
        
    def add_new_naaptolphone(self,name,range):
        temp = name.lower()
        if temp.find("null") != -1:
            return
        for n in self.get_allnaaptolphones(): 
            if n.name == name:
                if n.range == range:
                        return                            
        ai = naaptolphones()
        ai.name = name
        ai.range = range
        session.commit()
    
    def add_new_ntonlinesp(self,nid,name,price):
        ai = ntonlinesp()
        ai.nid = nid
        ai.name = name
        ai.price = price
        session.commit()
    
    def add_new_ntofflinesp(self,nid,name,price):
        ai = ntofflinesp()
        ai.nid = nid
        ai.name = name
        ai.price = price
        session.commit()
        
    def get_naaptolphone(self, name, range):
        query = naaptolphones.query.filter_by(name=name)
        query = query.filter_by(range=range)
        return query.one()
        

    def get_allmobstorephones(self):
        ai = themobilestorephones.query.all()
        return ai
        
    def get_allnaaptolphones(self):
        ai = naaptolphones.query.all()
        return ai
    
    def get_allntonlinesp(self):
        ai = ntonlinesp.query.all()
        return ai
    
    def get_allntofflinesp(self):
        ai = ntofflinesp.query.all()
        return ai
    
    def get_ntonlinespbynid(self,nid):
        ai = ntonlinesp.query.filter_by(nid=nid)
        return ai
    
    def get_ntofflinespbynid(self,nid):
        ai = ntofflinesp.query.filter_by(nid=nid)
        return ai
    
            
    def get_all_ipbasic(self):
        vi = indiaplaza_data.query.all()
        return vi
        
    def add_ipextra(self,name,shown_price,final_price,guarantee,shipinfo):
        ai = indiaplaza_items()
        ai.p_name = name
        ai.p_shown_price = shown_price
        ai.p_final_price = final_price
        ai.p_guaranteeinfo = guarantee
        ai.p_shipinfo = shipinfo
        session.commit()
        
    def get_all_infibeam_data(self):
        phones = infibeam_data.query.all()
        return phones
    
    
    def get_all_indiaplaza_phones(self):
        phones = indiaplaza_items.query.all()
        return phones
                    
    def get_all_univercell_phones(self):
        phones = univercell_items.query.all()
        return phones
    
    def get_infibeam_csv(self):
        phones = self.get_all_infibeam_phones()
        print phones
    
    def get_all_suppliers(self):
        sup = suppliers.query.all()
        return sup
    
    def get_all_models(self):
        mod = models.query.all()
        return mod
    
    def get_all_prices(self):
        pr = prices.query.all()
        return pr
    
    def get_suppId(self,name):
        for s in self.get_all_suppliers():
            if s.name == name:
                return s.id
    
    def get_supp_byId(self,id):
        for s in self.get_all_suppliers():
            if s.id == id:
                return s
    
    def get_supp_byName(self,name):
        supps = []
        for s in self.get_all_suppliers():
            if s.name == name:
                supps.append(s)
        return supps        
    
    def get_supp_bySite(self,site):
        supps = []
        for s in self.get_all_suppliers():
            if s.site == site:
                supps.append(s)
        return supps
    
    
    
    def get_modId(self,brand,model):
        for m in self.get_all_models():
            if m.brand == brand:
                if m.model == model:
                    return m.id                     
    
    def get_modbyId(self,id):
        for m in self.get_all_models():
            if m.id == id:
                return m                     
        
    def get_modbyModel(self,model):
        for m in self.get_all_models():
            if m.model == model:
                return m
    def get_modbyBrand(self,brand):
        mods = []
        for m in self.get_all_models():
            if m.brand == brand:
                mods.append(m)
        return mods                     
    
    def get_prbyId(self,id):
        for p in self.get_all_prices():
            if p.id == id:
                return p                     
    
    def get_prbySid(self,supplier_id):
        prc = []
        for p in self.get_all_prices():
            if p.supplier_id == supplier_id:
                prc.append(p)
        return prc
                                 
    def get_prbyMid(self,mobile_id):
        prc = []
        for p in self.get_all_prices():
            if p.mobile_id == mobile_id:
                prc.append(p)
        return prc
    
    
    def add_supplier(self,name,site):
        for s in self.get_all_suppliers():
            if s.name == name:
                now = datetime.datetime.now()
                s.last_crawled = str(now)
                session.commit()
                return             
        ai = suppliers()
        ai.name = name
        ai.site = site
        now = datetime.datetime.now()
        ai.last_crawled = str(now)
        session.commit()
    
    def add_models(self,brand,model):
        for m in self.get_all_models():
            if m.brand == brand:
                if m.model == model:
                    return                     
        ai = models()
        ai.brand = brand
        ai.model = model
        session.commit()
        
        
    def add_prices(self,mobile_id,supplier_id,quoted_price,final_price,extra_info):
        for p in self.get_all_prices():
            if p.mobile_id == mobile_id:
                if p.supplier_id == supplier_id:
                    if p.extra_info == extra_info:
                        return                           
        ai = prices()
        ai.mobile_id = mobile_id
        ai.supplier_id = supplier_id
        ai.quoted_price = quoted_price
        ai.final_price = final_price
        ai.extra_info = extra_info
        session.commit()
        
    def get_price_by_model(self, model_id, supplier_id):
        query = prices.query.filter_by(mobile_id=model_id)
        query = query.filter_by(supplier_id=supplier_id)
        return query.one()
    
    def add_gs_info(self,mid,guaranteeinfo,shipinfo):
        gs = guarantee_info()
        gs.mid = mid
        gs.guaranteeinfo = guaranteeinfo 
        gs.shipinfo = shipinfo
        session.commit() 
    
    def get_all_gs_info(self):
        gsi = guarantee_info.query.all()
        return gsi
    
    def get_gs_bymid(self,mid):
        gsi = guarantee_info.query.filter_by(mid=mid).one()
        return gsi
        
if __name__ == "__main__":
    datastore = DataHelper()
    datastore.get_infibeam_csv()