Esempio n. 1
0
    def stats(self):
        query_directors = select(['persons.id', func.count('roles.person_id').label('count')],
                       from_obj=['persons', 'roles'],
                       whereclause="roles.person_id = persons.id AND roles.role_type = 'director'",
                       group_by=['persons.id'], order_by='count desc', limit=10)
        query_actors = select(['persons.id', func.count('roles.person_id').label('count')],
                       from_obj=['persons', 'roles'],
                       whereclause="roles.person_id = persons.id AND roles.role_type = 'cast'",
                       group_by=['persons.id'], order_by='count desc', limit=10)                       
        
        top_directors = DBSession.query(Person, 'count').from_statement(query_directors).all()
        top_actors = DBSession.query(Person, 'count').from_statement(query_actors).all()        
    
        ia = IMDb()

        top250_ids = [x.movieID for x in ia.get_top250_movies()]
        bottom100_ids = [x.movieID for x in ia.get_bottom100_movies()]
        
        top250_count = DBSession.query(Movie).filter(Movie.id.in_(top250_ids)).count()
        bottom100_count = DBSession.query(Movie).filter(Movie.id.in_(bottom100_ids)).count()
        total_count = DBSession.query(Movie).count()
        
        total_runtime = 1
        
        return {'top250_count': top250_count,
                'bottom100_count': bottom100_count,
                'total_count': total_count,
                'total_runtime' : total_runtime,
                'top_directors': top_directors,
                'top_actors': top_actors}
Esempio n. 2
0
 def get_or_create(cls, id, name):
     from sqlalchemy.orm.exc import NoResultFound
 
     try:
         p = DBSession.query(cls).filter_by(id=id).one()
         return p
     except NoResultFound, e:
         p = cls(id=id, name=name)
         DBSession.add(p)
         return p
Esempio n. 3
0
    def movie_cover(self, id):
        movie = DBSession.query(Movie).filter_by(id=id).one()

        if movie.cover is not None:
            response.content_type = 'image/png'
            response.headers.add('Content-Disposition:', 'inline; filename=%s-cover.jpg' % movie.id)

        return movie.cover            
Esempio n. 4
0
    def summary(self, id):
        movie = DBSession.query(Movie).filter_by(id=id).first()
        
        if movie is None:
            override_template(self.summary, 'genshi:cotufa.templates.movie-imdb')
            # emulate 'Movie' object with basic info
            movie = Movie.lightweight_instance_for(IMDb(), id)

        return {'movie': movie}
Esempio n. 5
0
    def add(self, id):
        import urllib2
    
        movie = DBSession.query(Movie).filter_by(id=id).first()

        if movie is None:
            lw = Movie.lightweight_instance_for(IMDb(), id)
            
            # Construct Movie object
            movie = Movie(id=lw['id'], title=lw['title'], year=lw['year'])
            
            for at in lw['attributes']:
                movie.attributes.append(at)
                
            for d in lw['directors']:
                movie.people.append(MovieRole(person=Person.get_or_create(d['id'], d['name']),
                                              role_type=u'director'))
                                              
            for w in lw['writers']:
                movie.people.append(MovieRole(person=Person.get_or_create(w['id'], w['name']),
                                              role_type=u'writer'))
                                              
            for c in lw['cast']:
                movie.people.append(MovieRole(person=Person.get_or_create(c['id'], c['person']),
                                              role_type=u'cast',
                                              character=c['character']))
            
            # Fetch cover image
            if lw['cover_url'] is not None:
                cover = urllib2.urlopen(lw['cover_url'])
                movie.cover = cover.read()
                cover.close()
                
            DBSession.add(movie)

            flash(u'%s has been added to the list.' % movie.title)
            
        redirect(url('/movies'))
Esempio n. 6
0
 def setup(self):
     try:
         new_attrs = {}
         new_attrs.update(self.attrs)
         new_attrs.update(self.do_get_dependencies())
         self.obj = self.klass(**new_attrs)
         DBSession.add(self.obj)
         DBSession.flush()
         return self.obj
     except:
         DBSession.rollback()
         raise
Esempio n. 7
0
 def remove(self, id):
     movie = DBSession.query(Movie).filter_by(id=id).one()
     DBSession.delete(movie)
     flash(u'%s has been removed from the list.' % movie.title)
     redirect(url('/movies'))
Esempio n. 8
0
 def movie_details(self, id):
     movie = DBSession.query(Movie).filter_by(id=id).one()
     return {'movie': movie}
Esempio n. 9
0
 def movies(self, format=u'list'):
     movies = DBSession.query(Movie).order_by('created_on desc')
     return {'format': format, 'movies': movies}
Esempio n. 10
0
 def test_query_obj(self):
     obj = DBSession.query(self.klass).one()
     for key, value in self.attrs.iteritems():
         assert_equals(getattr(obj, key), value)
Esempio n. 11
0
 def tearDown(self):
     DBSession.rollback()