30c7ffa287a66b09fc7154dd3a58197796198d6b
1 # GNU MediaGoblin -- federated, autonomous media hosting
2 # Copyright (C) 2013 MediaGoblin contributors. See AUTHORS.
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.
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.
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/>.
22 from werkzeug
.exceptions
import MethodNotAllowed
, BadRequest
, NotImplemented
23 from werkzeug
.wrappers
import BaseResponse
25 from mediagoblin
.tools
.translate
import pass_to_ugettext
as _
26 from mediagoblin
.meddleware
.csrf
import csrf_exempt
27 from mediagoblin
.auth
.tools
import check_login_simple
28 from mediagoblin
.submit
.lib
import \
29 submit_media
, check_file_field
, get_upload_file_limits
, \
30 FileUploadLimit
, UserUploadLimit
, UserPastUploadLimit
33 from mediagoblin
.user_pages
.lib
import add_media_to_collection
34 from mediagoblin
.db
.models
import Collection
36 from .tools
import CmdTable
, response_xml
, check_form
, \
37 PWGSession
, PwgNamedArray
, PwgError
38 from .forms
import AddSimpleForm
, AddForm
41 _log
= logging
.getLogger(__name__
)
44 @CmdTable("pwg.session.login", True)
45 def pwg_login(request
):
46 username
= request
.form
.get("username")
47 password
= request
.form
.get("password")
48 user
= check_login_simple(username
, password
)
50 return PwgError(999, 'Invalid username/password')
51 request
.session
["user_id"] = user
.id
52 request
.session
.save()
56 @CmdTable("pwg.session.logout")
57 def pwg_logout(request
):
59 request
.session
.delete()
63 @CmdTable("pwg.getVersion")
64 def pwg_getversion(request
):
65 return "2.5.0 (MediaGoblin)"
68 @CmdTable("pwg.session.getStatus")
69 def pwg_session_getStatus(request
):
71 username
= request
.user
.username
74 return {'username': username
}
77 @CmdTable("pwg.categories.getList")
78 def pwg_categories_getList(request
):
79 catlist
= [{'id': -29711,
80 'uppercats': "-29711",
81 'name': "All my images"}]
84 collections
= Collection
.query
.filter_by(
85 get_actor
=request
.user
).order_by(Collection
.title
)
88 catlist
.append({'id': c
.id,
89 'uppercats': str(c
.id),
91 'comment': c
.description
95 'categories': PwgNamedArray(
111 @CmdTable("pwg.images.exist")
112 def pwg_images_exist(request
):
116 @CmdTable("pwg.images.addSimple", True)
117 def pwg_images_addSimple(request
):
118 form
= AddSimpleForm(request
.form
)
119 if not form
.validate():
120 _log
.error("addSimple: form failed")
124 dump
.append("%s=%r" % (f
.name
, f
.data
))
125 _log
.info("addSimple: %r %s %r", request
.form
, " ".join(dump
),
128 if not check_file_field(request
, 'image'):
132 entry
= submit_media(
133 mg_app
=request
.app
, user
=request
.user
,
134 submitted_file
=request
.files
['image'],
135 filename
=request
.files
['image'].filename
,
136 title
=six
.text_type(form
.name
.data
),
137 description
=six
.text_type(form
.comment
.data
))
139 collection_id
= form
.category
.data
140 if collection_id
> 0:
141 collection
= Collection
.query
.get(collection_id
)
142 if collection
is not None and collection
.actor
== request
.user
.id:
143 add_media_to_collection(collection
, entry
, "")
146 'image_id': entry
.id,
147 'url': entry
.url_for_self(
151 # Handle upload limit issues
152 except FileUploadLimit
:
154 _(u
'Sorry, the file size is too big.'))
155 except UserUploadLimit
:
157 _('Sorry, uploading this file will put you over your'
159 except UserPastUploadLimit
:
161 _('Sorry, you have reached your upload limit.'))
164 md5sum_matcher
= re
.compile(r
"^[0-9a-fA-F]{32}$")
167 def fetch_md5(request
, parm_name
, optional_parm
=False):
168 val
= request
.form
.get(parm_name
)
169 if (val
is None) and (not optional_parm
):
170 _log
.error("Parameter %s missing", parm_name
)
171 raise BadRequest("Parameter %s missing" % parm_name
)
172 if not md5sum_matcher
.match(val
):
173 _log
.error("Parameter %s=%r has no valid md5 value", parm_name
, val
)
174 raise BadRequest("Parameter %s is not md5" % parm_name
)
178 @CmdTable("pwg.images.addChunk", True)
179 def pwg_images_addChunk(request
):
180 o_sum
= fetch_md5(request
, 'original_sum')
181 typ
= request
.form
.get('type')
182 pos
= request
.form
.get('position')
183 data
= request
.form
.get('data')
187 if not typ
in ("file", "thumb"):
188 _log
.error("type %r not allowed for now", typ
)
191 _log
.info("addChunk for %r, type %r, position %d, len: %d",
192 o_sum
, typ
, pos
, len(data
))
194 _log
.info("addChunk: Ignoring thumb, because we create our own")
200 @CmdTable("pwg.images.add", True)
201 def pwg_images_add(request
):
202 _log
.info("add: %r", request
.form
)
203 form
= AddForm(request
.form
)
206 return {'image_id': 123456, 'url': ''}
211 if request
.method
not in ("GET", "POST"):
212 _log
.error("Method %r not supported", request
.method
)
213 raise MethodNotAllowed()
215 func
= CmdTable
.find_func(request
)
217 _log
.warn("wsphp: Unhandled %s %r %r", request
.method
,
218 request
.args
, request
.form
)
219 raise NotImplemented()
221 with
PWGSession(request
) as session
:
222 result
= func(request
)
224 if isinstance(result
, BaseResponse
):
227 response
= response_xml(result
)
228 session
.save_to_cookie(response
)