-
Notifications
You must be signed in to change notification settings - Fork 8
/
main.py
80 lines (70 loc) · 2.41 KB
/
main.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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
from .utils import cmd_exists, peerflix
import sys
import colorful
from .argument_parser import Parser
from .ezflix import Ezflix
from prettytable import PrettyTable
parser = Parser()
args = parser.parse()
media_player = args.media_player
if not cmd_exists('peerflix'):
sys.exit('This program requires Peerflix. https://github.com/mafintosh/peerflix')
if not cmd_exists('mpv') and args.media_player == 'mpv':
media_player = 'vlc'
if not args.query:
sys.exit(colorful.red("Search query not valid."))
def main():
ezflix = Ezflix(query=args.query,
media_type=args.media_type,
limit=args.limit,
sort_by=args.sort_by,
sort_order=args.sort_order,
quality=args.quality,
minimum_rating=args.minimum_rating,
language=args.language
)
torrents = ezflix.get_torrents()
if torrents is None or len(torrents) == 0:
sys.exit(colorful.red('No results found.'))
if args.latest:
latest = torrents[0]
file_path = ''
if args.subtitles:
file_path = ezflix.find_subtitles(latest['title'])
print("Playing " + latest['title'])
peerflix(latest['magnet'], media_player, args.media_type, args.subtitles, args.remove, file_path)
sys.exit()
row = PrettyTable()
row.field_names = ["Id", "Torrent"]
row.align = 'l'
for result in torrents:
row.add_row([result['id'], result['title']])
print(row)
print(colorful.bold("Make selection: (Enter quit to close the program)"))
try:
input = raw_input
except NameError:
pass
while True:
read = input()
if read == 'quit':
sys.exit()
try:
int_val = int(read)
except (ValueError, TypeError) as error:
print(colorful.red('Invalid selection.'))
continue
magnet = ezflix.get_magnet(int_val)
if not magnet:
print(colorful.red('Invalid selection.'))
continue
print("Playing " + magnet['title'])
file_path = ''
if args.subtitles:
file_path = ezflix.find_subtitles(magnet['title'])
peerflix(magnet['magnet'], media_player, args.media_type, args.subtitles, args.remove, file_path)
if __name__ == '__main__':
try:
main()
except KeyboardInterrupt:
sys.exit()