예제 #1
0
파일: scanner.py 프로젝트: chenca/omdb
def scan_dirs(location):

    for d in location:
        movies = [i for i in os.listdir(d) if os.path.isdir(os.path.join(d, i))]
        movies.sort()
        for m in movies:
            if os.path.exists(os.path.join(d,m,"nfo.pickle")):
                print "\n\nNFO already found: %s" % m

                nfo = open(os.path.join(d,m,"nfo.pickle"), 'rb')
                info = pickle.load(nfo)
                nfo.close()
                (imdb_url, imdb_title, imdb_cover, imdb_date, imdb_rating, imdb_plot) = imdb_info(info['imdb_url'])
            else:
                print "\n\nSearching imdb info for: %s" % m
                (imdb_url, imdb_title, imdb_cover, imdb_date, imdb_rating, imdb_plot) = movie_info(m)

            if imdb_url == "":
                print "IMDB info not found for: %s" % m
                continue

            if imdb_cover == "":
               imdb_cover = "nocover"

            print "Searching youtube trailers for: %s" % imdb_title

            trailers = search_trailers(unescape(imdb_title) + " Official Trailer ")
            store_info(imdb_url, imdb_title, imdb_cover, imdb_date, imdb_rating, imdb_plot,
                       trailers, os.path.join(d, m))
예제 #2
0
파일: scanner.py 프로젝트: chenca/omdb
                info = pickle.load(nfo)
                nfo.close()
                (imdb_url, imdb_title, imdb_cover, imdb_date, imdb_rating, imdb_plot) = imdb_info(info['imdb_url'])
            else:
                print "\n\nSearching imdb info for: %s" % m
                (imdb_url, imdb_title, imdb_cover, imdb_date, imdb_rating, imdb_plot) = movie_info(m)

            if imdb_url == "":
                print "IMDB info not found for: %s" % m
                continue

            if imdb_cover == "":
               imdb_cover = "nocover"

            print "Searching youtube trailers for: %s" % imdb_title

            trailers = search_trailers(unescape(imdb_title) + " Official Trailer ")
            store_info(imdb_url, imdb_title, imdb_cover, imdb_date, imdb_rating, imdb_plot,
                       trailers, os.path.join(d, m))


if len(sys.argv) == 3:
    folder = sys.argv[1]
    imdb_url = sys.argv[2]

    (imdb_url, imdb_title, imdb_cover, imdb_date, imdb_rating, imdb_plot) = imdb_info(imdb_url)
    trailers = search_trailers(unescape(imdb_title) + " Official Trailer ")
    store_info(imdb_url, imdb_title, imdb_cover, imdb_date, imdb_rating, imdb_plot, trailers, folder)
else:
    scan_dirs(dirs)