Creating plugins views.py

mediagoblin/plugins/custom_subtitles/views.py

@@ -0,0 +1,162 @@
+# GNU MediaGoblin -- federated, autonomous media hosting
+# Copyright (C) 2011, 2012 MediaGoblin contributors.  See AUTHORS.
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+import six
+
+from datetime import datetime
+
+from itsdangerous import BadSignature
+from werkzeug.exceptions import Forbidden
+from werkzeug.utils import secure_filename
+
+from mediagoblin import messages
+from mediagoblin import mg_globals
+
+from mediagoblin.plugins.custom_subtitles import forms
+from mediagoblin.decorators import (require_active_login, active_user_from_url,
+                            get_media_entry_by_id, path_subtitle, user_may_delete_media)
+from mediagoblin.tools.metadata import (compact_and_validate, DEFAULT_CHECKER,
+                                        DEFAULT_SCHEMA)
+from mediagoblin.tools.response import (render_to_response,
+                                        redirect, redirect_obj, render_404)
+
+import mimetypes
+
+from mediagoblin.plugins.custom_subtitles.tools import open_subtitle,save_subtitle
+
+UNSAFE_MIMETYPES = [
+        'text/html',
+        'text/svg+xml']
+
+@get_media_entry_by_id
+@user_may_delete_media
+@require_active_login
+def edit_subtitles(request, media):
+    form = forms.EditSubtitlesForm(request.form)
+
+    # Add any subtitles
+    if 'subtitle_file' in request.files \
+        and request.files['subtitle_file']:
+        if mimetypes.guess_type(
+                request.files['subtitle_file'].filename)[0] in \
+                UNSAFE_MIMETYPES:
+            public_filename = secure_filename('{0}.notsafe'.format(
+                request.files['subtitle_file'].filename))
+        else:
+            public_filename = secure_filename(
+                    request.files['subtitle_file'].filename)
+
+        subtitle_public_filepath \
+            = mg_globals.public_store.get_unique_filepath(
+            ['media_entries', six.text_type(media.id), 'subtitle',
+             public_filename])
+
+        subtitle_public_file = mg_globals.public_store.get_file(
+            subtitle_public_filepath, 'wb')
+
+        try:
+            subtitle_public_file.write(
+                request.files['subtitle_file'].stream.read())
+        finally:
+            request.files['subtitle_file'].stream.close()
+
+        media.subtitle_files.append(dict(
+                name=form.subtitle_language.data \
+                    or request.files['subtitle_file'].filename,
+                filepath=subtitle_public_filepath,
+                created=datetime.utcnow(),
+                ))
+
+        media.save()
+
+        messages.add_message(
+            request,
+            messages.SUCCESS,
+            ("You added the subttile %s!") %
+                (form.subtitle_language.data or
+                 request.files['subtitle_file'].filename))
+
+        return redirect(request,
+                        location=media.url_for_self(request.urlgen))
+    return render_to_response(
+        request,
+        'mediagoblin/plugins/custom_subtitles/subtitles.html',
+        {'media': media,
+         'form': form})
+
+
+@require_active_login
+@get_media_entry_by_id
+@user_may_delete_media
+@path_subtitle
+def custom_subtitles(request,media,path=None):
+    text=""
+#    text = open_subtitle(path)
+    form = forms.CustomizeSubtitlesForm(request.form,
+                                         subtitle=text)
+    if request.method == 'POST' and form.validate():
+        subtitle_data = form.subtitle.data
+#        save_subtitle(path,subtitle_data)
+        return render_to_response(
+        request,
+        "mediagoblin/plugins/custom_subtitles/custom_subtitles.html",
+        {"path": path,
+         "media": media,
+         "form": form })
+
+    """delete_container = None
+                index = 0
+                for subtitle in media.subtitle_files:
+                    if subtitle["name"] == "Two And A Half Men S02E02.srt":
+                        delete_container = index
+                    index += 1
+                print media.subtitle_files.pop(delete_container)
+                media.save()"""
+
+    return render_to_response(
+        request,
+        "mediagoblin/plugins/custom_subtitles/custom_subtitles.html",
+        {"path": path,
+         "media": media,
+         "form": form })
+
+@require_active_login
+@get_media_entry_by_id
+@user_may_delete_media
+@path_subtitle
+def delete_subtitles(request,media,path=None):
+    text = open_subtitle(path)
+    form = forms.CustomizeSubtitlesForm(request.form,
+                                         subtitle=text)
+    if request.method == 'POST' and form.validate():
+        subtitle_data = form.subtitle.data
+        save_subtitle(path,subtitle_data)
+
+    """delete_container = None
+                index = 0
+                for subtitle in media.subtitle_files:
+                    if subtitle["name"] == "Two And A Half Men S02E02.srt":
+                        delete_container = index
+                    index += 1
+                print media.subtitle_files.pop(delete_container)
+                media.save()"""
+ 
+    return render_to_response(
+        request,
+        "mediagoblin/plugins/custom_subtitles/custom_subtitles.html",
+        {"path": path,
+         "media": media,
+         "form": form })