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()