123 lines
4.0 KiB
Python
123 lines
4.0 KiB
Python
# 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.tools.response import json_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
|
|
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)
|