Subversion Repositories SmartDukaan

'''

Created on 12-May-2010

@author: gaurav

'''

from datastore.DataDefinition import * #Phones, init, PhoneItem

from elixir import *

import datetime

class DataHelper:

    def __init__(self):

        #init()

        pass

    def initxy(self):

        init()

        #pass

    def add_vendor(self,name,url):

        try:

            v = Vendor.query.filter_by(name=name)

            v = v.filter_by(url=url).one()

            return

        except:

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

        try:

            pi = PhoneItem.query.filter_by(name=name)

            pi = pi.filter_by(url=url)

            pi = pi.filter_by(source=source).one()

            return

        except:

            pi = PhoneItem()

            pi.name = name

            pi.url = url

            pi.source = source

            session.commit()

    def add_price(self,url,price):

        try:

            p = prices.query.filter_by(price=price)

            p = p.filter_by(url=url).one()

            return

        except:

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

        try:

            ai = infibeam_data.query.filter_by(name=name).one()

            return

        except:

            ai = infibeam_data()

            ai.name = name

            ai.shown_price = shown_price

            ai.final_price = final_price

            session.commit()

    def add_univervendor(self,name,site):

        try:

            ai = univercell_data.query.filter_by(name=name)

            ai = ai.filter_by(site=site).one()

            return

        except:

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

        try:

            ai = univercell_items.query.filter_by(name=name)

            ai = ai.filter_by(shown_price=shown_price)

            ai = ai.filter_by(final_price=final_price).one()

            return

        except:

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

        try:

            ai = indiaplaza_data.query.filter_by(name=name)

            ai = ai.filter_by(site=site).one()

            return

        except:

            ai = indiaplaza_data()

            ai.v_name = name

            ai.v_site = site

            session.commit()

    def add_mobstoreurl(self,url):

        try:

            ai = themobilestoreurls.query.filter_by(url=url).one()

            return

        except:

            ai = themobilestoreurls()

            ai.url = url

            session.commit()

    def add_pricesbolourl(self,url):

        try:

            ai = pricesbolourls.query.filter_by(url=url).one()

            return

        except:

            ai = pricesbolourls()

            ai.url = url

            session.commit()    

    def get_allpricesbolourl(self):

        vi = pricesbolourls.query.all()

        return vi

    def add_naaptolurl(self,url):

        try:

            n = naaptolurls.query.filter_by(url=url).one()

            return

        except:

            ai = naaptolurls()

            ai.url = url

            session.commit()

    def add_morenaaptolurl(self,url):

        try:

            n = morenaaptolurls.query.filter_by(url=url).one()

            return

        except:

            try:

                n = naaptolurls.query.filter_by(url=url).one()

                return

            except:

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

        try:

            ai = themobilestorephones.query.filter_by(name=name)

            ai = ai.filter_by(shown_price=shown_pr)

            ai = ai.filter_by(final_price=final_pr).one()

            return

        except:    

            ai = themobilestorephones()

            ai.name = name

            ai.shown_price = shown_pr

            ai.final_price = final_pr

            session.commit()

    def add_new_mobstorephone_new(self,name,shown_pr,final_pr,extra_info):

        try:

            ai = themobilestorephones_new.query.filter_by(name=name).one()

            return

        except:    

            ai = themobilestorephones_new()

            ai.name = name

            ai.shown_price = shown_pr

            ai.final_price = final_pr

            ai.extra_info = extra_info

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

        try:

            n = ntonlinesp.query.filter_by(nid=nid)

            n = n.filter_by(name=name).one()

            return

        except:    

            ai = ntonlinesp()

            ai.nid = nid

            ai.name = name

            ai.price = price

            session.commit()

    def add_new_ntofflinesp(self,nid,name,price):

        try:

            n = ntofflinesp.query.filter_by(nid=nid)

            n = n.filter_by(name=name).one()

            return

        except:    

            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_allmobstorephones_new(self):

        ai = themobilestorephones_new.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 add_babuchakurl(self,url,no_pages):

        try:

            ai = babuchak_urls.query.filter_by(url=url).one()

            return

        except:    

            ai = babuchak_urls()

            ai.url = url

            ai.no_pages = no_pages 

            session.commit()

    def get_allbabuchakurls(self):

        ai = babuchak_urls.query.all()

        return ai

    def add_babuchakphoneurl(self,url):

        try:

            ai = babuchak_phoneurls.query.filter_by(url=url).one()

            return

        except:    

            ai = babuchak_phoneurls()

            ai.url = url 

            session.commit()

    def get_allbabuchakphoneurls(self):

        ai = babuchak_phoneurls.query.all()

        return ai

    def add_babuchakphone(self,name,shown_price,final_price):

        try:

            ai = babuchak_phones.query.filter_by(name=name).one()

            return

        except:

            ai = babuchak_phones()

            ai.name = name

            ai.shown_price = shown_price

            ai.final_price = final_price    

            session.commit()

    def get_allbabuchakphones(self):

        ai = babuchak_phones.query.all()

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

        try:

            ai = indiaplaza_items.query.filter_by(p_name=name)

            ai = ai.filter_by(p_shown_price=shown_price)

            ai = ai.filter_by(p_final_price=final_price)

            ai = ai.filter_by(p_guaranteeinfo=guarantee)

            ai = ai.filter_by(p_shipinfo=shipinfo).one()

            return

        except:

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

        try:

            gs = guarantee_info.query.filter_by(mid=mid)

            gs = gs.filter_by(guaranteeinfo=guaranteeinfo)

            gs = gs.filter_by(shipinfo=shipinfo).one()

            return

        except:

            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

    def set_extra_vars(self,var,val,desc):    

        try:

            tm = extra_vars.query.filter_by(var=var).one()

            tm.val = val

            tm.desc = desc

        except:

            tm = extra_vars()

            tm.var = var

            tm.val = val

            tm.desc = desc       

    def get_extra_vars(self,var):

        try:

            #print "in try"

            tm = extra_vars.query.filter_by(var=var).one()

        except:

            #print "in except"

            tm = "EMPTY"

        return tm.val

if __name__ == "__main__":

    datastore = DataHelper()

    datastore.get_infibeam_csv()