Subversion Repositories SmartDukaan

Rev

Blame | Last modification | View Log | RSS feed

'''
Created on 20-Jul-2010

@author: ashish
'''
from shop2020.thriftpy.model.v1.user.ttypes import Widget as TWidget
from shop2020.thriftpy.model.v1.user.ttypes import WidgetItem as TWidgetItem
from shop2020.thriftpy.model.v1.user.ttypes import WidgetType as WType
from shop2020.model.v1.user.impl.Dataservice import Widget, WidgetItem
from shop2020.thriftpy.model.v1.user.ttypes import RatingsWidget as TRatings
from shop2020.thriftpy.model.v1.user.ttypes import RatingType as RType
from shop2020.utils.Utils import to_java_date


def to_t_widget(widget):
    if not widget:
        return
    t_widget = TWidget()
    t_widget.id = widget.id
    t_widget.customer_id = widget.customer_id
    t_widget.name = widget.name
    t_widget.type = widget.type
    t_widget_items = []
    
    for widgetitem in widget.items:
        
        t_widget_item = TWidgetItem()
        t_widget_item.id = widgetitem.item.id
        t_widget_item.enabled = widgetitem.item.enabled
        t_widget_item.item_id = widgetitem.item.item_id
        t_widget_item.snippet = widgetitem.item.snippet
        t_widget_item.timestamp = to_java_date(widgetitem.added_on)
        t_widget_items.append(t_widget_item)
    t_widget.items = t_widget_items
    return t_widget

def to_t_ratings(user_ratings, all_ratings, item_id, user_id):
    t_ratings = TRatings()
    t_ratings.catalog_item_id = item_id
    t_ratings.user_id= user_id
    t_ratings.ratings = {}
    if user_ratings:
        t_ratings.ratings[RType.USER_CURRENT] = user_ratings.rating
    if all_ratings:
        for rating in all_ratings:
            t_ratings.ratings[rating.type] = rating.rating
    return t_ratings