Subversion Repositories SmartDukaan

'''

Created on Jul 8, 2015

@author: amit

'''

from dtr.storage import Mysql

from dtr.storage.DataService import MissingAmazonOrderUsers, \

    PendingToTrackAmazonOrderUsers, initialize, RetailerAddresses, Users, \

    User_addresses, All_user_addresses, RetailerLinks, Retailers

from elixir import *

from pymongo.mongo_client import MongoClient

from sqlalchemy.sql.expression import func

def main():

    #Mysql.deleteTableData(tableName)

    try:

        initialize()

        #get all users

        maxUserId = session.query(func.max(All_user_addresses.user_id)).first()

        if maxUserId is None:

            maxUserId = 0

        else:

            (maxUserId,) = maxUserId

        query = session.query(Users)

        if maxUserId>0:

            query = query.filter_by(id=maxUserId)

        for user in query.all():

            userAddress = All_user_addresses()

            userAddress.user_id = user.id

            address = None 

            if user.address_line_1:

                address = user.address_line_1

            if user.address_line_2:

                if address:

                    address += " " +  user.address_line_2

                else:

                    address = user.address_line_2

            userAddress.address = address

            userAddress.city = user.city

            userAddress.pincode = user.pincode

            userAddress.state = user.state

            userAddress.source = 'user_profile'

            ra2 = session.query(Retailers).filter(RetailerLinks.user_id==user.id).filter(Retailers.id==RetailerLinks.retailer_id).first()

            if ra2:

                ua2 = All_user_addresses()

                ua2.user_id = user.id

                ua2.store_name = ra2.title

                ua2.address = ra2.address

                ua2.city = ra2.city

                ua2.pincode = ra2.pin

                ua2.state = ra2.state

                ua2.source = 'retailer_master'

            ra1 = session.query(RetailerAddresses).filter(RetailerLinks.user_id==user.id).filter(RetailerAddresses.retailer_id==RetailerLinks.retailer_id).first()

            if ra1:

                ua1 = All_user_addresses()

                ua1.user_id= user.id

                ua1.store_name = ra2.title

                ua1.address = ra1.address

                ua1.city = ra1.city

                ua1.pincode = ra1.pin

                ua1.state = ra1.state

                ua1.source = 'retailer_crm_edited'

            session.commit()

        if maxUserId ==0:

            for address in session.query(User_addresses).all():

                ua3 = All_user_addresses()

                ua3.user_id = address.user_id

                ua3.store_name = address.store_name

                ua3.address = address.address

                ua3.city = address.city

                ua3.state = address.state

                ua3.pincode = address.pincode

                ua3.source = 'fos_crm'

            session.commit()

    finally:

        session.close()

if __name__ == '__main__':

    main()