Subversion Repositories SmartDukaan

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

from shop2020.clients.UserClient import UserClient

from elixir import *

from sqlalchemy.sql.expression import and_, or_, desc, not_, distinct, cast, \

    between

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

from shop2020.thriftpy.model.v1.order.ttypes import OrderSource

from shop2020.thriftpy.model.v1.user.ttypes import User, Sex

import time

def change_email_other_sources_orders():

    DataService.initialize('transaction','192.168.190.114')

    user_client = UserClient().get_client()

    other_source_orders = Order.query.filter(or_(Order.source == OrderSource.FLIPKART, Order.source == OrderSource.SNAPDEAL)).all()

    for order in other_source_orders:

        user = None

        if order.source == OrderSource.FLIPKART :

            try:

                user_to_add = User()

                user_to_add.email = 'FK.'+ str(order.id) + '@mailinator.com'

                user_to_add.password = 'gR9zF-Ish2im6tbYFNivgA'

                user_to_add.communicationEmail = user_to_add.email

                user_to_add.sourceStartTime = int(round(time.time() * 1000))

                user_to_add.sourceId = OrderSource.FLIPKART

                user_to_add.sex = Sex.WONT_SAY

                user_to_add = user_client.createUser(user_to_add)

                user = user_to_add

            except:

                pass

        if order.source == OrderSource.SNAPDEAL :

            user = None

            try:

                user_client = UserClient().get_client()

                user_to_add = User()

                user_to_add.email = 'SD.'+ str(order.id) + '@mailinator.com'

                user_to_add.password = 'hSp5gvCsrrhZcceE6mzzOQ'

                user_to_add.communicationEmail = user_to_add.email

                user_to_add.sourceStartTime = int(round(time.time() * 1000))

                user_to_add.sourceId = OrderSource.SNAPDEAL

                user_to_add.sex = Sex.WONT_SAY

                user_to_add = user_client.createUser(user_to_add)

                user = user_to_add

            except:

                pass

        if user:

            order.customer_id = user.userId

            order.customer_email = user.email

    session.commit()

def main():

    change_email_other_sources_orders()

if __name__ == '__main__':

    main()