forked from LindseyB/starwars-dot-gif
/
star_wars_gif.py
107 lines (90 loc) · 3.68 KB
/
star_wars_gif.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
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
#!/usr/bin/python
import urwid
import pysrt
from makeGifs import makeGif
choices = [ { 'name': "Star Wars: A New Hope", 'num': 4 },
{ 'name': "Star Wars: The Empire Strikes Back", 'num': 5 },
{ 'name': "Star Wars: Return of the Jedi", 'num': 6 } ]
sub_files = { 4: 'subs/IV-A.New.Hope[1977]DvDrip-aXXo.srt',
5: 'subs/V-The.Empire.Strikes.Back[1980]DvDrip-aXXo.srt',
6: 'subs/VI-Return.Of.The.Jedi[1983]DvDrip-aXXo.srt' }
source = 0
index = 0
subtitle = ""
def menu(title, choices):
body = [urwid.Text(title), urwid.Divider()]
for c in choices:
button = urwid.Button(c['name'])
urwid.connect_signal(button, 'click', item_chosen, c)
body.append(urwid.AttrMap(button, None, focus_map='reversed'))
return urwid.ListBox(urwid.SimpleFocusListWalker(body))
def item_chosen(button, choice):
global source
response = urwid.Text([u'You chose ', choice['name'], u'\n'])
source = choice['num']
done = urwid.Button(u'Ok')
urwid.connect_signal(done, 'click', search)
main.original_widget = urwid.Filler(urwid.Pile([response,
urwid.AttrMap(done, None, focus_map='reversed')]))
def search(button):
edit = urwid.Edit("Search quotes: ")
done = urwid.Button(u'Search')
urwid.connect_signal(done, 'click', find_quotes, edit)
exit_button = urwid.Button(u'Exit')
urwid.connect_signal(exit_button, 'click', exit)
main.original_widget = urwid.Filler(urwid.Pile([edit,
urwid.AttrMap(done, None, focus_map='reversed'),
urwid.AttrMap(exit_button, None, focus_map='reversed')]))
def find_quotes(button, edit):
subs = pysrt.open(sub_files[source])
search_text = edit.edit_text.lower()
def seek(item, quote):
for i in item.split(' '):
if not i in quote:
return False
return True
matching = [s for s in subs if seek(search_text, s.text.lower())]
body_text = "Select quote" if len(matching) > 0 else "No quotes found"
body = [urwid.Text(body_text), urwid.Divider()]
for m in matching:
button = urwid.Button(m.text)
urwid.connect_signal(button, 'click', add_custom_subtitle, subs.index(m))
body.append(urwid.AttrMap(button, None, focus_map='reversed'))
back_button = urwid.Button('Go back')
urwid.connect_signal(back_button, 'click', search)
body.append(urwid.AttrMap(back_button, None, focus_map='reversed'))
main.original_widget = urwid.ListBox(urwid.SimpleFocusListWalker(body))
def add_custom_subtitle(button, i):
global index
index = i
body = [urwid.Text("Add a custom quote?"), urwid.Divider()]
yes_button = urwid.Button("Yes")
urwid.connect_signal(yes_button, 'click', enter_custom_subtitle)
body.append(urwid.AttrMap(yes_button, None, focus_map='reversed'))
no_button = urwid.Button("No")
urwid.connect_signal(no_button, 'click', generate_gif)
body.append(urwid.AttrMap(no_button, None, focus_map='reversed'))
main.original_widget = urwid.ListBox(urwid.SimpleFocusListWalker(body))
def enter_custom_subtitle(button):
subtitle = urwid.Edit("Enter custom subtitle: ")
done = urwid.Button(u'Submit')
urwid.connect_signal(done, 'click', generate_gif_with_subtitle, subtitle)
main.original_widget = urwid.Filler(urwid.Pile([subtitle,
urwid.AttrMap(done, None, focus_map='reversed')]))
def generate_gif(button):
raise urwid.ExitMainLoop()
def generate_gif_with_subtitle(button, edit):
global subtitle
subtitle = edit.edit_text
raise urwid.ExitMainLoop()
def exit(button):
index = None
raise urwid.ExitMainLoop()
main = urwid.Padding(menu(u'Movie choice', choices), left=2, right=2)
top = urwid.Overlay(main, urwid.SolidFill(u'\N{MEDIUM SHADE}'),
align='center', width=('relative', 60),
valign='middle', height=('relative', 60),
min_width=20, min_height=9)
urwid.MainLoop(top, palette=[('reversed', 'standout', '')]).run()
if index:
makeGif(source, index, custom_subtitle=subtitle)