Fix deleting media with attachments.
[mediagoblin.git] / mediagoblin / tools / files.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 mediagoblin import mg_globals
18
19
20 def delete_media_files(media):
21 """
22 Delete all files associated with a MediaEntry
23
24 Arguments:
25 - media: A MediaEntry document
26 """
27 no_such_files = []
28 for listpath in media.media_files.itervalues():
29 try:
30 mg_globals.public_store.delete_file(
31 listpath)
32 except OSError:
33 no_such_files.append("/".join(listpath))
34
35 for attachment in media.attachment_files:
36 try:
37 mg_globals.public_store.delete_file(
38 attachment['filepath'])
39 except OSError:
40 no_such_files.append("/".join(attachment['filepath']))
41
42 if no_such_files:
43 raise OSError(", ".join(no_such_files))