Exemplo n.º 1
0
 def getMovieInfoFromCache(self, res):
     '''
     return movieinfo object created from database result
     '''
     
     print "return movieinfo object"
     
     result = []
     for m in res:
         result.append(m)
     
     
     myMovieInfo = movieinfo()
     
     myMovieInfo.title = result[1].encode('utf-8')
     myMovieInfo.imdbid = result[2]
     myMovieInfo.genres = result[3]
     myMovieInfo.rating = result[4]
     myMovieInfo.runtime = result[5]
     myMovieInfo.tagline = result[6]
     myMovieInfo.totalVote = result[7]
     myMovieInfo.releaseDate = result[8]
     myMovieInfo. overview = result[9]
     myMovieInfo.posterImage = result[10]
     myMovieInfo.backdropImage = result[11]
     myMovieInfo.year = result[12]
     myMovieInfo.writer = result[13]
     myMovieInfo.director = result[14]
     myMovieInfo.castandrole = eval(result[15])
     myMovieInfo.trailer = result[16]
     
     
     return myMovieInfo
Exemplo n.º 2
0
    def getMovieInfo(self, name, imdbid='', videoUrl=''):
        # check if the info is available in cache
        print "checking if imdbid is provided..."
        print "imdbid = " + imdbid
        if imdbid:
            print "got imdbid"
            t = (imdbid,)
            searchQuery = "SELECT * FROM movie_info WHERE imdbid = ?"
            self.dbcur.execute(searchQuery, t )
            result =self.dbcur.fetchone()
            
            if result:
                return self.getMovieInfoFromCache(result)
        
        print "no imdbid provided."
        print "check if videoUrl is provided"
        if videoUrl:
            print "got videoUrl: " + videoUrl
            t = (videoUrl, )
            searchQuery = "SELECT * FROM movie_info WHERE url = ?"
            self.dbcur.execute(searchQuery, t)
            result =self.dbcur.fetchone()
            print "check if match found"
            if result:
                print "match found with the name of movie : " + name
                return self.getMovieInfoFromCache(result)
        
        # create movieinfo object to return
        myMovieInfo = movieinfo()
        
        # set file name to movieinfo object
        myMovieInfo.name = name
        
        # delete first bracket ( from name
        name = name.replace("(", "")
        name = name.replace(")", "")
        name = name.replace("[", "")
        name = name.replace("]", "")
        name = name.replace(".", " ")
        name = name.replace("_", " ")
        name = name.replace("{", "")
        name = name.replace("}", "")
        name = name.replace("720p", "")
        name = name.replace("1080p", "")
        
        mgroup = re.split('[0-9]{4}', name)
    
        if mgroup :
            searchName =  mgroup[0].strip()
        else:
            searchName = name
            
        m = re.search('[0-9]{4}', name)
        if m:
            searchYear = m.group(0)
        else:
            searchYear = ""
            
        # check if any movie info is avaiable in name, year pair
        # if year is available for better recognition
        if searchYear != "":
            print "searchYear: " + searchYear
            t = (searchName, searchYear)
            searchQuery = "SELECT * FROM movie_info WHERE title=? AND year=?"
            self.dbcur.execute(searchQuery, t)
            result =self.dbcur.fetchone()
            if result:
                return self.getMovieInfoFromCache(result)
            
            
        # invoke search
        movieId = self.searchMovie(searchName, searchYear)
        
        if movieId == 0 :
            return myMovieInfo
        
        DETAIL_URL = "http://api.themoviedb.org/3/movie/"+ str(movieId) +"?api_key="+self.API
        
        CREDIT_URL = "http://api.themoviedb.org/3/movie/"+ str(movieId) + "/credits?api_key="+ self.API
        
        TRAILER_URL = "http://api.themoviedb.org/3/movie/%s/videos?api_key=%s" %(str(movieId), self.API)
        
        
        
        result = urlopen(DETAIL_URL)
        mJson = json.load(result)
        
        creditResult = urlopen(CREDIT_URL)
        cJson = json.load(creditResult) 
        
        trailerResult = urlopen(TRAILER_URL)
        tJson = json.load(trailerResult)    
        
        # encode strings to "utf-8" becuase kodi give error while showing unicode character
        myMovieInfo.title =  mJson['title'].encode('utf-8')
        myMovieInfo.imdbid = mJson['imdb_id'].encode('utf-8')
        myMovieInfo.rating =  mJson['vote_average'] #rating
        myMovieInfo.totalVote = mJson['vote_count']
        myMovieInfo.overview =  mJson['overview'].encode('utf-8')
        myMovieInfo.runtime = mJson['runtime'] * 60
        myMovieInfo.tagline = mJson['tagline'].encode('utf-8') #short description
        myMovieInfo.releaseDate =  mJson['release_date'].encode('utf-8')
        tYear = mJson['release_date'].encode('utf-8')
        myMovieInfo.year = tYear[0:4]

        # get imdb rating 
        try:
            OMDB_URL = "http://www.omdbapi.com/?i=%s&plot=short&r=json" % myMovieInfo.imdbid
            omdbResult = urlopen(OMDB_URL)
            oJson = json.load(omdbResult) 
            
            if oJson['imdbVotes'] != 'N/A' and myMovieInfo.totalVote < oJson['imdbVotes']:
                myMovieInfo.rating =  oJson['imdbRating'] #rating
                myMovieInfo.totalVote = oJson['imdbVotes']
        except:
            pass
        gList = []
        for genre in mJson['genres'] :
            gList.append(genre['name'].encode('utf-8'))
            
        genres = ' / '.join(gList)
        
        myMovieInfo.genres =  genres
        
        try:
            myMovieInfo.posterImage = mJson['poster_path']
        except:
            pass
        try:
            myMovieInfo.backdropImage = mJson['backdrop_path']
        except: 
            pass
        
        try: 
            myMovieInfo.director = cJson['crew'][0]['name']
            myMovieInfo.writer = cJson['crew'][1]['name']
        except:
            pass
        
        try:
            myMovieInfo.trailer = str(tJson['results'][0]['key'])
        except:
            pass
        
        if cJson.has_key('cast'):
            listCast = cJson['cast']
            
            castName = []
            castChar = []
            for cast in listCast:
                name = cast['name']
                castName.append(name)
                character = cast['character']
                castChar.append(character)
            
            myMovieInfo.castandrole = zip(castName, castChar)
        
        self.saveMovieCache(myMovieInfo, videoUrl)
        
        return myMovieInfo