Exceptional handling for customizing subitles

This commit is contained in:
saksham1115 2016-08-13 17:40:28 +00:00
parent 464d7972b1
commit 19b4f43166
2 changed files with 60 additions and 24 deletions

View File

@ -18,14 +18,25 @@ from mediagoblin import mg_globals
import os import os
def open_subtitle(path): def open_subtitle(path):
status = True
subtitle_public_filepath = path subtitle_public_filepath = path
with mg_globals.public_store.get_file( try:
subtitle_public_filepath, 'rb') as subtitle_public_file: with mg_globals.public_store.get_file(
text = subtitle_public_file.read().decode('utf-8','ignore') subtitle_public_filepath, 'rb') as subtitle_public_file:
return text text = subtitle_public_file.read().decode('utf-8','ignore')
return (text,status)
except:
status = False
return ('',status)
def save_subtitle(path,text): def save_subtitle(path,text):
status = True
subtitle_public_filepath = path subtitle_public_filepath = path
with mg_globals.public_store.get_file( try:
subtitle_public_filepath, 'wb') as subtitle_public_file: with mg_globals.public_store.get_file(
subtitle_public_file.write(text.encode('utf-8','ignore')) subtitle_public_filepath, 'wb') as subtitle_public_file:
subtitle_public_file.write(text.encode('utf-8','ignore'))
return status
except:
status = False
return (status)

View File

@ -122,27 +122,52 @@ def custom_subtitles(request,media,id=None):
for subtitle in media.subtitle_files: for subtitle in media.subtitle_files:
if subtitle["id"] == id: if subtitle["id"] == id:
path = subtitle["filepath"] path = subtitle["filepath"]
text="" text = ""
text = open_subtitle(path) value = open_subtitle(path)
form = forms.CustomizeSubtitlesForm(request.form, text, status = value[0], value[1]
subtitle=text) if status == True :
if request.method == 'POST' and form.validate(): form = forms.CustomizeSubtitlesForm(request.form,
subtitle_data = form.subtitle.data subtitle=text)
save_subtitle(path,subtitle_data) if request.method == 'POST' and form.validate():
subtitle_data = form.subtitle.data
status = save_subtitle(path,subtitle_data)
if status == True:
messages.add_message(
request,
messages.SUCCESS,
("Subtitle file changed!!!"))
return redirect(request,
location=media.url_for_self(request.urlgen))
else :
messages.add_message(
request,
messages.ERROR,
("Couldn't edit the subtitles!!!"))
return redirect(request,
location=media.url_for_self(request.urlgen))
return render_to_response(
request,
"mediagoblin/plugins/subtitles/custom_subtitles.html",
{"id": id,
"media": media,
"form": form })
else:
index = 0
for subtitle in media.subtitle_files:
if subtitle["id"] == id:
delete_container = index
media.subtitle_files.pop(delete_container)
media.save()
break
index += 1
messages.add_message( messages.add_message(
request, request,
messages.SUCCESS, messages.ERROR,
("Subtitle file changed!!!")) ("File link broken! Upload the subtitle again"))
return redirect(request, return redirect(request,
location=media.url_for_self(request.urlgen)) location=media.url_for_self(request.urlgen))
return render_to_response(
request,
"mediagoblin/plugins/subtitles/custom_subtitles.html",
{"id": id,
"media": media,
"form": form })
@require_active_login @require_active_login
@get_media_entry_by_id @get_media_entry_by_id
@ -167,4 +192,4 @@ def delete_subtitles(request,media):
("Subtitle file deleted!!!")) ("Subtitle file deleted!!!"))
return redirect(request, return redirect(request,
location=media.url_for_self(request.urlgen)) location=media.url_for_self(request.urlgen))