mediagoblin/gmg_commands/addmedia.py

   1 # GNU MediaGoblin -- federated, autonomous media hosting
   2 # Copyright (C) 2011, 2012 MediaGoblin contributors.  See AUTHORS.
   3 #
   4 # This program is free software: you can redistribute it and/or modify
   5 # it under the terms of the GNU Affero General Public License as published by
   6 # the Free Software Foundation, either version 3 of the License, or
   7 # (at your option) any later version.
   8 #
   9 # This program is distributed in the hope that it will be useful,
  10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 # GNU Affero General Public License for more details.
  13 #
  14 # You should have received a copy of the GNU Affero General Public License
  15 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
  16
  17 from __future__ import print_function
  18
  19 import os
  20
  21 import six
  22
  23 from mediagoblin.db.models import LocalUser
  24 from mediagoblin.gmg_commands import util as commands_util
  25 from mediagoblin.submit.lib import (
  26     submit_media, get_upload_file_limits,
  27     FileUploadLimit, UserUploadLimit, UserPastUploadLimit)
  28
  29 from mediagoblin import mg_globals
  30
  31
  32 def parser_setup(subparser):
  33     subparser.add_argument(
  34         'username',
  35         help="Name of user this media entry belongs to")
  36     subparser.add_argument(
  37         'filename',
  38         help="Local file on filesystem")
  39     subparser.add_argument(
  40         "-d", "--description",
  41         help="Description for this media entry")
  42     subparser.add_argument(
  43         "-t", "--title",
  44         help="Title for this media entry")
  45     subparser.add_argument(
  46         "-l", "--license",
  47         help=(
  48             "License this media entry will be released under. "
  49             "Should be a URL."))
  50     subparser.add_argument(
  51         "-T", "--tags",
  52         help=(
  53             "Comma separated list of tags for this media entry."))
  54     subparser.add_argument(
  55         "-s", "--slug",
  56         help=(
  57             "Slug for this media entry. "
  58             "Will be autogenerated if unspecified."))
  59
  60     subparser.add_argument(
  61         '--celery',
  62         action='store_true',
  63         help="Don't process eagerly, pass off to celery")
  64
  65
  66 def addmedia(args):
  67     # Run eagerly unless explicetly set not to
  68     if not args.celery:
  69         os.environ['CELERY_ALWAYS_EAGER'] = 'true'
  70
  71     app = commands_util.setup_app(args)
  72
  73     # get the user
  74     user = app.db.LocalUser.query.filter(
  75         LocalUser.username==args.username.lower()
  76     ).first()
  77     if user is None:
  78         print("Sorry, no user by username '%s'" % args.username)
  79         return
  80
  81     # check for the file, if it exists...
  82     filename = os.path.split(args.filename)[-1]
  83     abs_filename = os.path.abspath(args.filename)
  84     if not os.path.exists(abs_filename):
  85         print("Can't find a file with filename '%s'" % args.filename)
  86         return
  87
  88     upload_limit, max_file_size = get_upload_file_limits(user)
  89
  90     def maybe_unicodeify(some_string):
  91         # this is kinda terrible
  92         if some_string is None:
  93             return None
  94         if six.PY2:
  95             return six.text_type(some_string, 'utf-8')
  96         return some_string
  97
  98     try:
  99         submit_media(
 100             mg_app=app,
 101             user=user,
 102             submitted_file=open(abs_filename, 'rb'), filename=filename,
 103             title=maybe_unicodeify(args.title),
 104             description=maybe_unicodeify(args.description),
 105             license=maybe_unicodeify(args.license),
 106             tags_string=maybe_unicodeify(args.tags) or u"",
 107             upload_limit=upload_limit, max_file_size=max_file_size)
 108     except FileUploadLimit:
 109         print("This file is larger than the upload limits for this site.")
 110     except UserUploadLimit:
 111         print("This file will put this user past their upload limits.")
 112     except UserPastUploadLimit:
 113         print("This user is already past their upload limits.")