Subversion Repositories SmartDukaan

'''

Created on Apr 15, 2015

@author: amit

'''

from dtr.reports.affiliatereco import SMTP_SERVER, SMTP_PORT, SENDER, PASSWORD, \

    inc

from dtr.reports.usersegmentation import TMP_FILE

from dtr.sources.spice import todict

from dtr.utils.utils import fetchResponseUsingProxy, sendNotification

from email import encoders

from email.mime.base import MIMEBase

from email.mime.multipart import MIMEMultipart

from email.mime.text import MIMEText

from pymongo.mongo_client import MongoClient

import json

import smtplib

import xlwt

PRODUCT_DETAIL_API = "https://catalog.paytm.com/v1/p/%s"

client = MongoClient('mongodb://localhost:27017/')

boldStyle = xlwt.XFStyle()

f = xlwt.Font()

f.bold = True

boldStyle.font = f

def sendmail(email, message, fileName, title):

    if email == "":

        return

    mailServer = smtplib.SMTP(SMTP_SERVER, SMTP_PORT)

    mailServer.ehlo()

    mailServer.starttls()

    mailServer.ehlo()

    # Create the container (outer) email message.

    msg = MIMEMultipart()

    msg['Subject'] = title

    msg.preamble = title

    html_msg = MIMEText(message, 'html')

    msg.attach(html_msg)

    fileMsg = MIMEBase('application', 'vnd.ms-excel')

    fileMsg.set_payload(file(TMP_FILE).read())

    encoders.encode_base64(fileMsg)

    fileMsg.add_header('Content-Disposition', 'attachment;filename=' + fileName)

    msg.attach(fileMsg)

    email.append('amit.gupta@shop2020.in')

    MAILTO = email 

    mailServer.login(SENDER, PASSWORD)

    mailServer.sendmail(PASSWORD, MAILTO, msg.as_string())

def scrapepaytm():

    db = client.Catalog

    workbook = xlwt.Workbook()

    worksheet = workbook.add_sheet("Sample")

    add_headers(worksheet)

    row = 0

    for sku in db.MasterData.find({"source":"paytm.com"}):

        try:

            row += 1

            prodUrl = sku.get("url")

            prodIdentifier = prodUrl.split("?")[0].split("/")[-1]

            apiUrl = PRODUCT_DETAIL_API%(prodIdentifier)

            response = fetchResponseUsingProxy(apiUrl)

            prod = json.loads(response)

            worksheet.write(row, inc(), prod["product_id"])

            worksheet.write(row, inc(), prod['parent_id'])

            worksheet.write(row, inc(), prod['instock'])

            worksheet.write(row, inc(), prod['pay_type_supported']['COD'])

            worksheet.write(row, inc(), prod['productName'])

            worksheet.write(row, inc(), prod['actual_price'])

            worksheet.write(row, inc(), prod['offer_price'])

            worksheet.write(row, inc(), prod.get('promo_code'))

            worksheet.write(row, inc(), prod['promo_text'])

            sellerValues = prod['other_sellers'].get("values")

            if sellerValues > 0 and len(sellerValues) > 0:

                worksheet.write(row, inc(), sellerValues[0]['offer_price'])

            else:

                worksheet.write(row, inc(), None)

        except:

            pass

    workbook.save("/home/amit/paytmcatalog")

def add_headers(worksheet):

    global i

    i=0

    worksheet.write(0, inc(), 'productid')

    worksheet.write(0, inc(), 'parentid')

    worksheet.write(0, inc(), 'instock')

    worksheet.write(0, inc(), 'cod supported')

    worksheet.write(0, inc(), 'title')

    worksheet.write(0, inc(), 'actual price')

    worksheet.write(0, inc(), 'box price')

    worksheet.write(0,inc(), 'lowest price')

    worksheet.write(0,inc(), 'promo code')

    worksheet.write(0,inc(), 'promo text')

    worksheet.write(0, inc(), 'image Url')

def main():

    db = client.Dtr

    refunds = db.refund.findOne({"batch":1428949802})

    #refunds  = [{"timestamp" : "2015-04-14 00:00:02", "userId" : 2, "batch" : 1428949802, "userAmount" : 847 }]

    for ref in refunds:

        sendNotification([ref.get("userId")], 'Batch Credit', 'Cashback Credited', 'Rs.%s was added to your wallet yesterday'%(ref.get("userAmount")), 'url', 'http://api.profittill.com/cashbacks/mine?user_id=%s'%(ref.get("userId")))

if __name__ == '__main__':

    main()