Subversion Repositories SmartDukaan

from shop2020.thriftpy.model.v1.user.ttypes import Address as TAddress, UserCommunication as TUserCommunication, \
    User as TUser, Cart as TCart, Line as TLine, MasterAffiliate as TMasterAffiliate, \
    Affiliate as TAffiliate, Tracker as TTracker, TrackLog as TTrackLog, Promotion as TPromotion, \
    Coupon as TCoupon, Discount as TDiscount, ItemCouponDiscount as TItemCouponDiscount,\
    Voucher
    
from shop2020.utils.Utils import to_java_date

def to_t_address(address):
    t_address = TAddress()
    t_address.id = address.id
    t_address.line1 = address.line_1
    t_address.line2 = address.line_2
    t_address.landmark = address.landmark
    t_address.city = address.city
    t_address.state = address.state
    t_address.pin = address.pin
    t_address.country = address.country
    t_address.enabled = address.enabled
    t_address.type = address.type
    t_address.addedOn = to_java_date(address.added_on)
    t_address.name = address.name
    t_address.phone = address.phone
    return t_address    

def to_t_user(user):
    t_user = TUser()
    if user is None:
        t_user.userId = -1
        return t_user
    
    t_user.userId = user.id
    t_user.name = user.name
    t_user.email = user.email
    t_user.password = user.password
    t_user.sex = user.sex

    t_user.communicationEmail = user.communication_email
    t_user.defaultAddressId = user.default_address_id
    t_user.isAnonymous = user.is_anonymous
    t_user.activeCartId = user.active_cart_id
    t_user.dateOfBirth = user.date_of_birth
    t_user.mobileNumber = user.mobile_number
    t_user.source = user.source
    t_user.sourceStartTime = to_java_date(user.source_start_time)
    t_user.trustLevel = user.trust_level
    t_user.lastLogin = to_java_date(user.last_login)
    t_user.lastLogout = to_java_date(user.last_logout)
    t_user.activeSince = to_java_date(user.active_since)
    t_user.isFacebookUser = False
    if user.fbusers and user.fbusers[0]:
        t_user.isFacebookUser = True
        t_user.facebookAccessToken = user.fbusers[0].facebook_access_token
        t_user.facebookId = user.fbusers[0].facebook_id
    
    t_user.sourceId = 1
    #put source
    if user.sources and user.sources[0]:
        t_user.sourceId = user.sources[0].source_id
        
    #put all addresses
    if user.addresses:
        t_user.addresses = list()
        for address in user.addresses:
            t_user.addresses.append(to_t_address(address))
    return t_user

def to_t_cart(cart):
    t_cart = TCart()
    if cart:
        t_cart.id = cart.id
        t_cart.status = cart.cart_status
        lines = []
        if cart.lines:
            for line in cart.lines:
                lines.append(to_t_line(line))
        t_cart.lines = lines
        t_cart.totalPrice = cart.total_price
        t_cart.discountedPrice = cart.discounted_price
        t_cart.couponCode = cart.coupon_code
        t_cart.createdOn = to_java_date(cart.created_on)
        t_cart.updatedOn = to_java_date(cart.updated_on)
        t_cart.addressId = cart.address_id
        t_cart.checkedOutOn = to_java_date(cart.checked_out_on)
        t_cart.pickupStoreId = cart.pickupStoreId
    return t_cart
    
def to_t_line(line):
    t_line = TLine()
    if line:
        t_line.cartId = line.cart_id
        t_line.itemId = line.item_id
        t_line.quantity = line.quantity
        t_line.estimate = line.estimate
        t_line.actualPrice = line.actual_price
        t_line.discountedPrice = line.discounted_price
        t_line.discounts = [to_t_discount(discount) for discount in line.discounts]
        t_line.createdOn = to_java_date(line.created_on)
        t_line.updatedOn = to_java_date(line.updated_on)
        t_line.lineStatus = line.line_status
        t_line.insurer = line.insurer
        t_line.insuranceAmount = line.insuranceAmount
    return t_line

def to_t_discount(discount):
    t_discount = TDiscount()
    
    if discount:
        t_discount.cart_id = discount.line_cart_id
        t_discount.item_id = discount.line_item_id
        t_discount.discount = discount.discount
        t_discount.quantity = discount.quantity
    
    return t_discount

def to_t_user_communication(user_communication):
    t_user_communication = TUserCommunication()
    
    if user_communication:
        t_user_communication.id = user_communication.id
        t_user_communication.userId = user_communication.user_id
        t_user_communication.communicationType = user_communication.communication_type
        t_user_communication.orderId = user_communication.order_id
        t_user_communication.airwaybillNo = user_communication.airwaybill_no
        t_user_communication.replyTo = user_communication.reply_to
        t_user_communication.productName = user_communication.product_name
        t_user_communication.subject = user_communication.subject
        t_user_communication.message = user_communication.message
        t_user_communication.communication_timestamp = to_java_date(user_communication.communication_timestamp)
        
    return t_user_communication

def to_t_master_affiliate(master_affiliate):
    t_master_affiliate = TMasterAffiliate()
    
    if master_affiliate:
        t_master_affiliate.id = master_affiliate.id
        t_master_affiliate.name = master_affiliate.name
        t_master_affiliate.addedOn = to_java_date(master_affiliate.added_on)

    return t_master_affiliate

def to_t_affiliate(affiliate):
    t_affiliate = TAffiliate()
    
    if affiliate:
        t_affiliate.id = affiliate.id
        t_affiliate.name = affiliate.name
        t_affiliate.url = affiliate.url
        t_affiliate.masterAffiliateId = affiliate.master_affiliate_id
        t_affiliate.addedOn = to_java_date(affiliate.added_on)
        
    return t_affiliate

def to_t_tracker(tracker):
    t_tracker = TTracker()
    
    if tracker:
        t_tracker.id = tracker.id
        t_tracker.affiliateId = tracker.affiliate_id
        t_tracker.addedOn = to_java_date(tracker.added_on)
    
    return t_tracker

def to_t_track_log(track_log):
    t_track_log = TTrackLog()
    
    if track_log:
        t_track_log.id = track_log.id
        t_track_log.affiliateId = track_log.affiliate_id
        t_track_log.userId = track_log.user_id
        t_track_log.eventType = track_log.event_id
        t_track_log.url = track_log.url
        t_track_log.data = track_log.data
        t_track_log.addedOn = to_java_date(track_log.added_on)

    return t_track_log

def to_t_promotion(promotion):
    t_promotion = TPromotion()
    
    if promotion:
        t_promotion.id = promotion.id
        t_promotion.name = promotion.name
        t_promotion.ruleExecutionSrc = promotion.rule_execution_src
        t_promotion.startOn = to_java_date(promotion.start_on)
        t_promotion.type = promotion.type
        t_promotion.endOn = to_java_date(promotion.end_on)
        
        #if promotion.coupons:
        #    t_promotion.coupons = [to_t_coupon(coupon) for coupon in promotion.coupons]
        
        t_promotion.createdOn = to_java_date(promotion.created_on)
    return t_promotion

def to_t_coupon(coupon):
    t_coupon = TCoupon()

    if coupon:
        t_coupon.couponCode = coupon.coupon_code
        t_coupon.promotion = to_t_promotion(coupon.promotion)
        t_coupon.arguments = coupon.arguments
    
    return t_coupon

def to_t_item_coupon_discount(item_coupon_discount):
    t_item_coupon_discount = TItemCouponDiscount()
    
    if item_coupon_discount:
        t_item_coupon_discount.itemId = item_coupon_discount[0]
        t_item_coupon_discount.couponCode = item_coupon_discount[1]
        t_item_coupon_discount.discount = item_coupon_discount[2]
    
    return t_item_coupon_discount

def to_t_voucher(voucher):
    t_voucher = Voucher()
    
    if voucher:
        t_voucher.id = voucher.id
        t_voucher.voucherCode = voucher.voucherCode
        t_voucher.voucherType = voucher.voucherType
        t_voucher.issuedOn = to_java_date(voucher.issuedOn)
        t_voucher.expiredOn = to_java_date(voucher.expiredOn)
        t_voucher.amount = voucher.amount
        t_voucher.userEmail = voucher.email
    return t_voucher