class WebApp(object):
def __init__(self):
"""
Initiate the classes used in the web page.
:return:
"""
self.wikipageviews = WikiPageViews()
self.boxofficemojo = BoxOfficeMojo()
self.wikifunctions = WikiSimpleAPIFunctions()
self.youtube = YouTube('AIzaSyDQ6enre5eE7f_BIegK-2MOBbBAlMWaJgI')
self.classifier = Classifier()
self.calculator = Calculator()
@cherrypy.expose
def index(self):
"""
Expose the index html file on the server.
:return html file:
"""
return file(os.path.join(WEB_DIR, u'index.html'))
@cherrypy.expose
def searchwiki(self, movie):
"""Search Wikipedia for article by name"""
cherrypy.response.headers['Content-Type'] = 'application/json'
return dumps(self.wikifunctions.search_suggestions(movie))
@cherrypy.expose
def searchboxofficemojo(self, movie):
"""Search BoxOfficeMojo for a movie by title"""
cherrypy.response.headers['Content-Type'] = 'application/json'
return dumps(self.boxofficemojo.get_search_results(movie))
@cherrypy.expose
def searchyoutubetrailer(self, movie):
"""Search YouTube for a movie"""
title = movie.encode('utf-8')
result = self.youtube.youtube_search(title, results=10)
cherrypy.response.headers['Content-Type'] = 'application/json'
return dumps(result)
@cherrypy.expose
def getwikipageviews(self, wikiname, boxofficemojoname):
"""Return Wikipedia page views with the start of the Wikipedia page life, to BoxOfficeMojo release date.
If there is no release date on BoxOfficeMojo then to the current month"""
wiki_name = wikiname.encode('utf-8')
box_name = boxofficemojoname.encode('utf-8')
try:
created_date = self.wikifunctions.page_created_date(wiki_name)
release_date = self.boxofficemojo.get_release_date(box_name)
page_views = self.wikipageviews.get_page_views_from_to_by_monthly(wiki_name, created_date, release_date)
result = {str(key): value for key, value in sorted(page_views.iteritems())}
cherrypy.response.headers['Content-Type'] = 'application/json'
return dumps(result)
except BoxOfficeMojoException:
created_date = self.wikifunctions.page_created_date(wiki_name)
current_date = datetime.now()
page_views = self.wikipageviews.get_page_views_from_to_by_monthly(wiki_name, created_date, current_date)
result = {str(key): value for key, value in sorted(page_views.iteritems())}
cherrypy.response.headers['Content-Type'] = 'application/json'
return dumps(result)
@cherrypy.expose
def getfirstsevendaysearnings(self, boxofficemojoname):
"""Return the first 7 days of earnings from BoxOfficeMojo"""
try:
box_name = boxofficemojoname.encode('utf-8')
result = self.boxofficemojo.get_first_seven_days(box_name)
cherrypy.response.headers['Content-Type'] = 'application/json'
return dumps(result)
except BoxOfficeMojoException as e:
cherrypy.response.headers['Content-Type'] = 'application/json'
return dumps({"error": e.message})
@cherrypy.expose
def getreleaseday(self, boxofficemojoname):
"""Get the release date from BoxOfficeMojo given the boxofficemojo url name"""
box_name = boxofficemojoname.encode('utf-8')
release_date = self.boxofficemojo.get_release_date(box_name)
cherrypy.response.headers['Content-Type'] = 'application/json'
return dumps({"release_date": release_date.strftime("%Y-%m-%d %H:%M:%S")})
@cherrypy.expose
def sentimentanalysisytcomments(self, youtubeid, boxofficemojoname):
"""Analyse the comments of a youtube video until BoxOfficeMojo release date. If there is no
BoxOfficeMojo release date use current date"""
youtube_id = youtubeid.encode('utf-8')
box_name = boxofficemojoname.encode('utf-8')
try:
release_date = self.boxofficemojo.get_release_date(box_name)
comments = self.youtube.youtube_comments(youtube_id, release_date)
score = self.classifier.classify_text(comments)
cherrypy.response.headers['Content-Type'] = 'application/json'
return dumps(score)
except BoxOfficeMojoException:
current_date = datetime.now()
try:
comments = self.youtube.youtube_comments(youtube_id, current_date)
except RequestError as e:
cherrypy.response.status = e.message['status']
cherrypy.response.headers['Content-Type'] = 'application/json'
return dumps({"error": e.message['body']})
score = self.classifier.classify_text(comments)
cherrypy.response.headers['Content-Type'] = 'application/json'
return dumps(score)
except RequestError as e:
cherrypy.response.status = e.message['status']
cherrypy.response.headers['Content-Type'] = 'application/json'
return dumps({"error": e.message['body']})
@cherrypy.expose
def calculatescore(self, words, pos, pageviewsforyear):
"""Calculates the blockbuster score"""
cherrypy.response.headers['Content-Type'] = 'application/json'
return dumps(self.calculator.calculate_score(int(words), float(pos), int(pageviewsforyear)))
