Merge remote-tracking branch 'cweb/master'

1 files changed, 122 insertions, 0 deletions

diff --git a/mediagoblin/plugins/api/views.py b/mediagoblin/plugins/api/views.py
new file mode 100644
index 00000000..9159fe65
--- /dev/null
+++ b/mediagoblin/plugins/api/views.py
@@ -0,0 +1,122 @@
+# 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 json
+import logging
+
+from os.path import splitext
+from werkzeug.exceptions import BadRequest, Forbidden
+from werkzeug.wrappers import Response
+
+from mediagoblin.decorators import require_active_login
+from mediagoblin.meddleware.csrf import csrf_exempt
+from mediagoblin.media_types import sniff_media
+from mediagoblin.plugins.api.tools import api_auth, get_entry_serializable, \
+        json_response
+from mediagoblin.submit.lib import check_file_field, prepare_queue_task, \
+    run_process_media, new_upload_entry
+
+_log = logging.getLogger(__name__)
+
+
+@csrf_exempt
+@api_auth
+@require_active_login
+def post_entry(request):
+    _log.debug('Posting entry')
+
+    if request.method == 'OPTIONS':
+        return json_response({'status': 200})
+
+    if request.method != 'POST':
+        _log.debug('Must POST against post_entry')
+        raise BadRequest()
+
+    if not check_file_field(request, 'file'):
+        _log.debug('File field not found')
+        raise BadRequest()
+
+    media_file = request.files['file']
+
+    media_type, media_manager = sniff_media(media_file)
+
+    entry = new_upload_entry(request.user)
+    entry.media_type = unicode(media_type)
+    entry.title = unicode(request.form.get('title')
+            or splitext(media_file.filename)[0])
+
+    entry.description = unicode(request.form.get('description'))
+    entry.license = unicode(request.form.get('license', ''))
+
+    entry.generate_slug()
+
+    # queue appropriately
+    queue_file = prepare_queue_task(request.app, entry, media_file.filename)
+
+    with queue_file:
+        queue_file.write(request.files['file'].stream.read())
+
+    # Save now so we have this data before kicking off processing
+    entry.save()
+
+    if request.form.get('callback_url'):
+        metadata = request.db.ProcessingMetaData()
+        metadata.media_entry = entry
+        metadata.callback_url = unicode(request.form['callback_url'])
+        metadata.save()
+
+    # Pass off to processing
+    #
+    # (... don't change entry after this point to avoid race
+    # conditions with changes to the document via processing code)
+    feed_url = request.urlgen(
+        'mediagoblin.user_pages.atom_feed',
+        qualified=True, user=request.user.username)
+    run_process_media(entry, feed_url)
+
+    return json_response(get_entry_serializable(entry, request.urlgen))
+
+
+@api_auth
+@require_active_login
+def api_test(request):
+    user_data = {
+            'username': request.user.username,
+            'email': request.user.email}
+
+    # TODO: This is the *only* thing using Response() here, should that
+    # not simply use json_response()?
+    return Response(json.dumps(user_data))
+
+
+def get_entries(request):
+    entries = request.db.MediaEntry.query
+
+    # TODO: Make it possible to fetch unprocessed media, or media in-processing
+    entries = entries.filter_by(state=u'processed')
+
+    # TODO: Add sort order customization
+    entries = entries.order_by(request.db.MediaEntry.created.desc())
+
+    # TODO: Fetch default and upper limit from config
+    entries = entries.limit(int(request.GET.get('limit') or 10))
+
+    entries_serializable = []
+
+    for entry in entries:
+        entries_serializable.append(get_entry_serializable(entry, request.urlgen))
+
+    return json_response(entries_serializable)