Subversion Repositories SmartDukaan

#!/usr/bin/python

import optparse

import xlrd, xlwt

import random, string

from elixir import *

from shop2020.model.v1.order.impl import DataService

from shop2020.model.v1.order.impl.DataService import HotspotStore

from shop2020.utils import EmailAttachmentSender

from shop2020.utils.EmailAttachmentSender import get_attachment_part

if __name__ == '__main__' and __package__ is None:

    import sys

    import os

    sys.path.insert(0, os.getcwd())

def upload_store_details(filename, circle_id, limit, db_hostname):

    DataService.initialize(dbname='transaction', db_hostname=db_hostname)

    workbook = xlrd.open_workbook(filename)

    sheet = workbook.sheet_by_index(0)

    l = []

    num_rows = sheet.nrows

    for rownum in range(1, num_rows):

        name = sheet.row_values(rownum)[0]

        code = sheet.row_values(rownum)[1]

        email = sheet.row_values(rownum)[2]

        hs = HotspotStore()

        hs.hotspotId = str(code)

        hs.companyId = 1

        hs.name = name

        hs.email = email

        hs.city = email

        hs.circleId = circle_id

        hs.isActive = 1

        hs.collectedAmount = 0

        hs.availableLimit  = limit

        hs.creditLimit = limit

        hs.password = ''.join(random.sample(string.digits,4))

        char_set = string.ascii_uppercase + string.digits

        hs.salt = ''.join(random.sample(char_set,25))

        l.append(hs)

    session.commit()

    wbk = xlwt.Workbook()

    sheet = wbk.add_sheet('main')

    heading_xf = xlwt.easyxf('font: bold on; align: wrap on, vert centre, horiz center')

    excel_integer_format = '0'

    integer_style = xlwt.XFStyle()

    integer_style.num_format_str = excel_integer_format

    sheet.write(0, 0, "StoreCode", heading_xf)

    sheet.write(0, 1, "Salt", heading_xf)

    c = 1 

    for s in l:

        EmailAttachmentSender.mail("cnc.center@shop2020.in", "5h0p2o2o", [s.email], "Password for Recharge Dashboard", "Password of the recharge dashboard to be opened from OCR dashboard is " + s.password , [], [], ["rajveer.singh@shop2020.in"])

        print s.email, s.hotspotId, s.salt, str(s.password)

        sheet.write(c, 0, s.hotspotId)

        sheet.write(c, 1, s.salt)      

        c = c + 1

    fname = "/tmp/store-details.xls"

    wbk.save(fname)

    EmailAttachmentSender.mail("cnc.center@shop2020.in", "5h0p2o2o", ["deepak.chawla@spiceretail.co.in","rajveer.singh@shop2020.in"], "New Stores to be added", "Please find attached list of store along with salt.", [get_attachment_part(fname)], [], [])

def main():

    parser = optparse.OptionParser()

    parser.add_option("-f", "--file", dest="filename",

                   default="Store.xls", type="string",

                   help="Read the store date from FILE",

                   metavar="FILE")

    parser.add_option("-c", "--circle", dest="circle_id",

                      type="int", help="Load these entries for circle",

                      metavar="CIRCLE")

    parser.add_option("-l", "--limit", dest="limit",

                      type="int", help="The daily limit for a store",

                      metavar="TYPE")

    parser.add_option("-H", "--host", dest="hostname",

                      default="localhost",

                      type="string", help="The HOST where the DB server is running",

                      metavar="HOST")

    (options, args) = parser.parse_args()

    if len(args) != 0:

        parser.error("You've supplied extra arguments. Are you sure you want to run this program?")

    filename = options.filename

    circle_id = options.circle_id

    limit = options.limit

    if filename is None or circle_id is None or limit is None:

        parser.error("A circle's id and a filename must be provided. Use -h for usage details.")

    upload_store_details(filename, circle_id, type, options.hostname)

if __name__ == '__main__':

    main()