forked from hybridNeo/Local-Movie-Recommendation
/
start.py
50 lines (49 loc) · 1.04 KB
/
start.py
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
import sys
import os
import omdb
import glob
import operator
reserved = ['EXTENDED','REMASTERD','DIRECTORS','UNRATED','AlTERNATE']
def search(list):
dic = {}
err_cnt = 0
for i in list:
try:
obj = omdb.title(i)
if(obj != []):
dic[i] = obj.imdb_rating
except:
err_cnt+=1
return (dic)
def clean(raw_list):
l = []
for i in raw_list:
cl = i
if('(' in cl):
cl = i.split('(')
cl = cl[0]
for j in reserved:
if(j in cl):
cl.split(j)
cl = cl[0]
l.append(cl)
return l
def main():
omdb.set_default('tomatoes', True)
dirname = input("Enter directory: ")
if(len(sys.argv) == 2):
dirname = sys.argv[1]
raw_movies = os.listdir(dirname)
print('Cleaning.....')
l = clean(raw_movies)
print('Retrieving Info...')
info = search(l)
sorted_x = sorted(info.items(), key=operator.itemgetter(1))
sorted_x = sorted_x[::-1]
for i in sorted_x:
print(i[0]+ ' ---------------' + str(i[1]))
# search = omdb.title('The Matrix')
# print(search.imdb_rating)
# print(search.tomato_rating)
if __name__ == '__main__':
main()