Subversion Repositories SmartDukaan

import urllib2
import requests
import ast
import time
from elixir import *
from elixir import metadata, setup_all
from sqlalchemy.engine import create_engine
from shop2020.model.v1.order.impl.model.RechargeDenomination import RechargeDenomination
    

engine = create_engine('mysql://root:shop2020@' + 'localhost' + '/' + 'transaction', pool_recycle=7200)
metadata.bind = engine
metadata.bind.echo = True
setup_all(True)

listCircle = [
['Andhra Pradesh','andhra_pradesh',1],
['Assam','assam',2],
['Bihar','bihar_&_jharkhand',3],
['Chennai','chennai',4],
['Delhi NCR','delhi_ncr',5],
['Gujarat','gujarat',6],
['Haryana','haryana',9],
['Himachal Pradesh','himachal_pradesh',7],
['Jammu','jammu_&_kashmir',8],
['Karnataka','karnataka',10],
['Kerela','kerala',11],
['Kolkata','kolkata',12],
['MP','madhya_pradesh_chhattisgarh',14],
['Maharashtra','maharashtra',13],
['Mumbai','mumbai',15],
['North East','north_east',16],
['Orissa','orissa',17],
['Punjab','punjab',18],
['Rajasthan','rajasthan',19],
['Tamil Nadu','tamil_nadu',20],
['UP East','up_east',21],
['UP(West)','up_west',22],
['West Bengal','west_bengal',23]
]

listOperator =[
               ['Aircel','aircel',6],
               ['Airtel','airtel',7],
               ['Idea','idea',10],
               ['Mtnl Delhi','mtnl',11],
               ['Mtnl Mumbai','mtnl',12],
               ['Reliance GSM','reliance_gsm',13],
               ['Reliance CDMA','reliance_cdma',14],
               ['Uninor','uninor',17],
               ['Vodafone','vodafone',19]
               ]

listRechargeType = ['top_up','special_recharge','2g_data','3g_data']

def fetch_plans(operator,circle,rechargeType):
    url='https://hub.paytm.com/api/plans/list?type=mobile&operator=%s&circle=%s&category=%s&auth=nosession&callback=jQuery15203827297973902207_1374842060208'%(operator[1],circle[1],rechargeType)
    time.sleep(2)
    req = urllib2.Request(url)
    response = urllib2.urlopen(req)
    json_input = response.read()
    start = json_input.rindex("\"plans\":") + len( "\"plans\":" )
    end = json_input.rindex( ",\"count\":", start )
    parse= json_input[start:end]
    plans = ast.literal_eval(parse)
    commit_details(operator,circle,rechargeType,plans)
    
    
def commit_details(operator,circle,rechargeType,plans):
    for plan in plans:
        value = int(plan['value'])
        talktime = str(plan['talktime'])
        validity = str(plan['validity'])
        description = str(plan['full_desc'].replace("\u2022",""))
        description = description.replace("\/", "/")
        description = description.replace("\n",".")
        denomination = RechargeDenomination()
        denomination.operatorId = operator[2]
        denomination.circleId = circle[2]
        if rechargeType == 'top_up':
            denomination.denominationType = 1
        else:
            denomination.denominationType = 2
        denomination.validity = validity
        denomination.amount = value
        denomination.description = "Talktime (Rs)"+talktime+"."+description
        session.commit()




def main():
    for operator in listOperator:
        for circle in listCircle:
            for rechargeType in listRechargeType:
                print "Trying to fetch recharge plans for Operator = %s , Circle = %s, Recharge Type = %s" %(operator[0],circle[0],rechargeType)
                print "\n"
                fetch_plans(operator,circle,rechargeType)
    
if __name__ == "__main__":
    main()