server.py
4.51 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
#!/usr/bin/env python
import os
import web
from web import form
import tempfile
import sys
import simplejson as json
import apt
import re
import socket
sys.path.insert(0,"/var/www/AppRecommender/src/")
from config import Config
from recommender import *
from user import *
from data import DebianPackage
import urllib
# avoid "RuntimeError: maximum recursion depth exceeded"
sys.setrecursionlimit(50000)
class Fake:
def GET(self):
return render_plain.fake()
class Index:
def GET(self):
return render.index()
class About:
def GET(self):
return render.about()
class Package:
def GET(self, pkg_name):
cfg = Config()
pkg = DebianPackage(pkg_name)
pkg.load_details()
return render_plain.package(pkg)
class AppRecommender:
def __init__(self):
logging.info("Setting up AppRecommender...")
self.cfg = Config()
self.rec = Recommender(self.cfg)
self.requests_dir = "/var/www/AppRecommender/src/web/requests/"
if not os.path.exists(self.requests_dir):
os.makedirs(self.requests_dir)
def POST(self):
web_input = web.input(pkgs_file={})
user_dir = tempfile.mkdtemp(prefix='',dir=self.requests_dir)
user_id = user_dir.split("/")[-1]
uploaded_file = os.path.join(user_dir,"uploaded_file")
if web_input['pkgs_file'].value:
lines = web_input['pkgs_file'].file.readlines()
with open(uploaded_file, "w") as uploaded:
uploaded.writelines(lines)
with open(uploaded_file) as uploaded:
if uploaded.readline().startswith('POPULARITY-CONTEST'):
user = PopconSystem(uploaded_file,user_id)
else:
user = PkgsListSystem(uploaded_file,user_id)
if len(user.pkg_profile) < 10:
return render.error(["Could not extract profile from uploaded file. It must have at least 10 applications."],
"/","RECOMMENDATION")
else:
self.rec.set_strategy("knn_eset")
user.maximal_pkg_profile()
prediction = self.rec.get_recommendation(user,12).get_prediction()
logging.info("Prediction for user %s" % user.user_id)
logging.info(str(prediction))
recommendation = [result[0] for result in prediction]
pkgs_details = []
for pkg_name in recommendation:
logging.info("Getting details of package %s" % pkg_name)
pkg = DebianPackage(pkg_name)
pkg.load_summary()
pkgs_details.append(pkg)
if pkgs_details:
logging.info("Rendering recommendation...")
return render.apprec(pkgs_details)
else:
return render.error(["No recommendation produced for the uploaded file."],"/","RECOMMENDATION")
# parsing json from screenshots - can be usefull in the future...
# def _packages_attrs(self, recommends): #recommends is result of _recommends()
# all_recommended_packages = []
# recommended_pkgs_attrs = {}
# json_file_path = 'static/json/screenshots.json' #FIXME: go to config file
# json_file = open(json_file_path)
# json_data = json.load(json_file)
# for strategy, result in recommends.items():
# all_recommended_packages.extend(result)
# for pkg_attrs_dict in json_data['screenshots']:
# if pkg_attrs_dict['name'] in all_recommended_packages:
# recommended_pkgs_attrs[pkg_attrs_dict['name']] = pkg_attrs_dict
# return recommended_pkgs_attrs
# to be used if it is not under apache
#def add_global_hook():
# g = web.storage({"counter": "1"})
# def _wrapper(handler):
# web.ctx.globals = g
# return handler()
# return _wrapper
render = web.template.render('/var/www/AppRecommender/src/web/templates/', base='layout', globals={'hasattr':hasattr})
render_plain = web.template.render('/var/www/AppRecommender/src/web/templates/', globals={'hasattr':hasattr})
urls = ('/', 'Index',
'/index', 'Index',
'/apprec', 'AppRecommender',
'/support', 'Support',
'/about', 'About',
'/package/(.*)', 'Package'
)
web.webapi.internalerror = web.debugerror
#if __name__ == "__main__":
cfg = Config()
app = web.application(urls, globals(), autoreload=False)
application = app.wsgifunc()
# apprec = web.application(urls, globals())
# apprec.add_processor(add_global_hook())
# apprec.run()