Subversion Repositories SmartDukaan

Rev

Rev 3350 | Rev 4198 | Go to most recent revision | Show entire file | Ignore whitespace | Details | Blame | Last modification | View Log | RSS feed

Rev 3350 Rev 3440
Line 7... Line 7...
7 
import tornado.ioloop
 7 
import tornado.ioloop
8 
import tornado.web
 8 
import tornado.web
9 
import json, os, ConfigParser, sys
 9 
import json, os, ConfigParser, sys
10 
from PyLucene.Retriever import Retriever
 10 
from PyLucene.Retriever import Retriever
11 
from Utils import getItemsWithTopScore, isPriceSame, getProductClusters, getFilteredClustersWithTopScores, \
 11 
from Utils import getItemsWithTopScore, isPriceSame, getProductClusters, getFilteredClustersWithTopScores, \
12 
    getDisplayInfo
 12 
    getDisplayInfo, getSynonyms
13 
 
 13 
 
14 
cmd_folder = os.path.dirname(os.path.abspath(os.environ["HOME"] + "/code/trunk/PyProj/src/shop2020/"))
 14 
cmd_folder = os.path.dirname(os.path.abspath(os.environ["HOME"] + "/code/trunk/PyProj/src/shop2020/"))
15 
if cmd_folder not in sys.path:
 15 
if cmd_folder not in sys.path:
16 
    sys.path.insert(0, cmd_folder)
 16 
    sys.path.insert(0, cmd_folder)
17 
 
 17 
 
18 
from shop2020.clients.CatalogClient import CatalogClient
 18 
from shop2020.clients.CatalogClient import CatalogClient
19 
from shop2020.thriftpy.model.v1.catalog.ttypes import status
 19 
from shop2020.thriftpy.model.v1.catalog.ttypes import status
20
20 
 
21 
class BaseHandler(tornado.web.RequestHandler):
 21 
class BaseHandler(tornado.web.RequestHandler):
22 
    def get_current_user(self):
 22 
    def get_current_user(self):
23 
        return self.get_secure_cookie('userauth')
 23 
        return self.get_secure_cookie('userauth')
24
24
25 
class LoginHandler(BaseHandler):
 25 
class LoginHandler(BaseHandler):
Line 39... Line 39...
39 
            self.set_secure_cookie("userauth", username + '_' + password)
 39 
            self.set_secure_cookie("userauth", username + '_' + password)
40 
            self.redirect('/')
 40 
            self.redirect('/')
41 
        else:
 41 
        else:
42 
            self.redirect('/login')
 42 
            self.redirect('/login')
43 
 
 43 
 
- 
 
 44 
class FeedbackHandler(BaseHandler):
- 
 
 45
- 
 
 46 
    def save(self, entity, source, feedback_type, selected_item = None):
- 
 
 47 
        self.feedback_file = '/tmp/feedback.json'
- 
 
 48 
        file_to_read = open(self.feedback_file, 'r')
- 
 
 49
- 
 
 50 
        feedbacks_json = file_to_read.read()
- 
 
 51 
        file_to_read.close()
- 
 
 52
- 
 
 53 
        feedbacks = json.loads(feedbacks_json) if len(feedbacks_json) > 1 else {}
- 
 
 54
- 
 
 55 
        if entity not in feedbacks: feedbacks[entity] = {}
- 
 
 56
- 
 
 57 
        feedbacks[entity][source] = {'type': feedback_type}
- 
 
 58
- 
 
 59 
        if selected_item is not None:   feedbacks[entity][source]['selected_item'] = selected_item
- 
 
 60
- 
 
 61 
        file_to_write = open(self.feedback_file, 'w')
- 
 
 62 
        file_to_write.write(json.dumps(feedbacks))
- 
 
 63 
        file_to_write.close()
- 
 
 64
- 
 
 65 
    def post(self):
- 
 
 66 
        feedback_type = self.get_argument('type')
- 
 
 67 
        entity_id = self.get_argument('entityId')
- 
 
 68 
        price_data_source = self.get_argument('source')
- 
 
 69
- 
 
 70 
        print feedback_type, entity_id, price_data_source
- 
 
 71
- 
 
 72 
        if feedback_type == 'select':
- 
 
 73 
            selected_item = self.get_argument('selected')
- 
 
 74 
            print selected_item
- 
 
 75 
            self.save(entity_id, price_data_source, feedback_type, selected_item)
- 
 
 76 
        else:
- 
 
 77 
            self.save(entity_id, price_data_source, feedback_type)
- 
 
 78
- 
 
 79 
    def get(self):
- 
 
 80 
        print 'GET: Feedback data'
- 
 
 81 
        self.feedback_file = '/tmp/feedback.json'
- 
 
 82 
        file_to_read = open(self.feedback_file, 'r')
- 
 
 83
- 
 
 84 
        feedbacks_json = file_to_read.read()
- 
 
 85 
        file_to_read.close()
- 
 
 86
- 
 
 87 
        self.write(feedbacks_json)
- 
 
 88
44 
class MainHandler(BaseHandler):
 89 
class MainHandler(BaseHandler):
45 
 
 90 
 
46 
    @tornado.web.authenticated
 91 
    @tornado.web.authenticated
47 
    def get(self):
 92 
    def get(self):
48 
        self.loader = tornado.template.Loader('HTMLTemplates')
 93 
        self.loader = tornado.template.Loader('HTMLTemplates')
49 
        catalog_client = CatalogClient().get_client()
 94 
        catalog_client = CatalogClient().get_client()
50 
        items = catalog_client.getAllItemsByStatus(status.ACTIVE)
 95 
        items = catalog_client.getAllItemsByStatus(status.ACTIVE)
51 
        items.extend(catalog_client.getAllItemsByStatus(status.PAUSED))
 96 
        items.extend(catalog_client.getAllItemsByStatus(status.PAUSED))
52 
        items.extend(catalog_client.getAllItemsByStatus(status.PAUSED_BY_RISK))
 97 
        items.extend(catalog_client.getAllItemsByStatus(status.PAUSED_BY_RISK))
- 
 
 98 
        synonyms = getSynonyms()
53
99 
        print synonyms
54 
        retriever = Retriever()
 100 
        retriever = Retriever()
55 
        products = {}
 101 
        products = {}
56
102
57 
        for item in items:
 103 
        for item in items:
58 
            if item.category in (10002, 10003, 10004, 10005, 10010):  products[item.catalogItemId] = item
 104 
            if item.category in (10002, 10003, 10004, 10005, 10010):  products[item.catalogItemId] = item
Line 62... Line 108...
62 
        for item in sorted(products.itervalues(), key = lambda item: item.brand):
 108 
        for item in sorted(products.itervalues(), key = lambda item: item.brand):
63 
            try:
 109 
            try:
64 
                model_name = item.modelName.strip() if len(item.modelName.strip()) > 0 else None
 110 
                model_name = item.modelName.strip() if len(item.modelName.strip()) > 0 else None
65 
                model_number = item.modelNumber.strip() if len(item.modelNumber.strip()) > 0 else None
 111 
                model_number = item.modelNumber.strip() if len(item.modelNumber.strip()) > 0 else None
66
112
- 
 
 113 
                synonyms_for_this_model = synonyms[item.catalogItemId] if item.catalogItemId in synonyms else None
- 
 
 114
67 
                search_results = retriever.retrieve(model_number = model_number, model_name = model_name, brand = item.brand)
 115 
                search_results = retriever.retrieve(model_number = model_number, model_name = model_name, brand = item.brand, synonyms = synonyms_for_this_model)
68
116
69 
                clusters = getProductClusters(search_results)
 117 
                clusters = getProductClusters(search_results)
70 
                filtered_clusters = getFilteredClustersWithTopScores(clusters)
 118 
                filtered_clusters = getFilteredClustersWithTopScores(clusters)
71
119
72 
                display_info = getDisplayInfo(filtered_clusters)
 120 
                display_info = getDisplayInfo(filtered_clusters)
73
121
74 
                product_name = "%s " % item.brand
 122 
                product_name = "%s " % item.brand
75 
                product_name += "%s " % model_name if model_name is not None else ''
 123 
                product_name += "%s " % model_name if model_name is not None else ''
76 
                product_name += model_number if model_number is not None else ''
 124 
                product_name += model_number if model_number is not None else ''
77
125
- 
 
 126 
                display_info['entity_id'] = item.catalogItemId
78 
                display_info['product_name'] = product_name
 127 
                display_info['product_name'] = product_name
79 
                display_info['saholic'] = {'price': item.sellingPrice}
 128 
                display_info['saholic'] = {'price': item.sellingPrice}
80 
                comparative_prices.append(display_info)
 129 
                comparative_prices.append(display_info)
81 
            except Exception as e:
 130 
            except Exception as e:
82 
                print e
 131 
                print e
Line 91... Line 140...
91 
}
 140 
}
92 
 
 141 
 
93 
application = tornado.web.Application([
 142 
application = tornado.web.Application([
94 
                (r"/", MainHandler),
 143 
                (r"/", MainHandler),
95 
                (r"/login", LoginHandler),
 144 
                (r"/login", LoginHandler),
- 
 
 145 
                (r"/feedback", FeedbackHandler),
96 
                (r"/(jquery-1.6.2.min\.js)", tornado.web.StaticFileHandler, dict(path=settings['static_path']))
 146 
                (r"/(jquery-1.6.2.min\.js)", tornado.web.StaticFileHandler, dict(path=settings['static_path']))
97 
            ], **settings)
 147 
            ], **settings)
98 
 
 148 
 
99 
if __name__ == '__main__':
 149 
if __name__ == '__main__':
100 
    http_server = tornado.httpserver.HTTPServer(application)
 150 
    http_server = tornado.httpserver.HTTPServer(application)