##// END OF EJS Templates
file-store: save DB entry on upload, and track access times.
marcink -
r3457:4532f88d default
parent child Browse files
Show More

The requested changes are too big and content was truncated. Show full diff

@@ -1,95 +1,115 b''
1 1 # -*- coding: utf-8 -*-
2 2
3 3 # Copyright (C) 2016-2019 RhodeCode GmbH
4 4 #
5 5 # This program is free software: you can redistribute it and/or modify
6 6 # it under the terms of the GNU Affero General Public License, version 3
7 7 # (only), as published by the Free Software Foundation.
8 8 #
9 9 # This program is distributed in the hope that it will be useful,
10 10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
11 11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 12 # GNU General Public License for more details.
13 13 #
14 14 # You should have received a copy of the GNU Affero General Public License
15 15 # along with this program. If not, see <http://www.gnu.org/licenses/>.
16 16 #
17 17 # This program is dual-licensed. If you wish to learn more about the
18 18 # RhodeCode Enterprise Edition, including its added features, Support services,
19 19 # and proprietary license terms, please see https://rhodecode.com/licenses/
20 20 import logging
21 21
22 22 from pyramid.view import view_config
23 23 from pyramid.response import FileResponse
24 24 from pyramid.httpexceptions import HTTPFound, HTTPNotFound
25 25
26 26 from rhodecode.apps._base import BaseAppView
27 27 from rhodecode.apps.file_store import utils
28 28 from rhodecode.apps.file_store.exceptions import (
29 FileNotAllowedException,FileOverSizeException)
29 FileNotAllowedException, FileOverSizeException)
30 30
31 31 from rhodecode.lib import helpers as h
32 32 from rhodecode.lib import audit_logger
33 33 from rhodecode.lib.auth import (CSRFRequired, NotAnonymous)
34 from rhodecode.model.db import Session, FileStore
34 35
35 36 log = logging.getLogger(__name__)
36 37
37 38
38 39 class FileStoreView(BaseAppView):
39 40 upload_key = 'store_file'
40 41
41 42 def load_default_context(self):
42 43 c = self._get_local_tmpl_context()
43 44 self.storage = utils.get_file_storage(self.request.registry.settings)
44 45 return c
45 46
46 47 @NotAnonymous()
47 48 @CSRFRequired()
48 49 @view_config(route_name='upload_file', request_method='POST', renderer='json_ext')
49 50 def upload_file(self):
50 51 self.load_default_context()
51 52 file_obj = self.request.POST.get(self.upload_key)
52 53
53 54 if file_obj is None:
54 55 return {'store_fid': None,
55 56 'access_path': None,
56 57 'error': '{} data field is missing'.format(self.upload_key)}
57 58
58 59 if not hasattr(file_obj, 'filename'):
59 60 return {'store_fid': None,
60 61 'access_path': None,
61 62 'error': 'filename cannot be read from the data field'}
62 63
63 64 filename = file_obj.filename
64 65
65 66 metadata = {
66 67 'user_uploaded': {'username': self._rhodecode_user.username,
67 68 'user_id': self._rhodecode_user.user_id,
68 69 'ip': self._rhodecode_user.ip_addr}}
69 70 try:
70 71 store_fid, metadata = self.storage.save_file(
71 72 file_obj.file, filename, extra_metadata=metadata)
72 73 except FileNotAllowedException:
73 74 return {'store_fid': None,
74 75 'access_path': None,
75 76 'error': 'File {} is not allowed.'.format(filename)}
76 77
77 78 except FileOverSizeException:
78 79 return {'store_fid': None,
79 80 'access_path': None,
80 81 'error': 'File {} is exceeding allowed limit.'.format(filename)}
81 82
83 try:
84 entry = FileStore.create(
85 file_uid=store_fid, filename=metadata["filename"],
86 file_hash=metadata["sha256"], file_size=metadata["size"],
87 file_description='upload attachment',
88 check_acl=False, user_id=self._rhodecode_user.user_id
89 )
90 Session().add(entry)
91 Session().commit()
92 log.debug('Stored upload in DB as %s', entry)
93 except Exception:
94 log.exception('Failed to store file %s', filename)
95 return {'store_fid': None,
96 'access_path': None,
97 'error': 'File {} failed to store in DB.'.format(filename)}
98
82 99 return {'store_fid': store_fid,
83 100 'access_path': h.route_path('download_file', fid=store_fid)}
84 101
85 102 @view_config(route_name='download_file')
86 103 def download_file(self):
87 104 self.load_default_context()
88 105 file_uid = self.request.matchdict['fid']
89 106 log.debug('Requesting FID:%s from store %s', file_uid, self.storage)
107
90 108 if not self.storage.exists(file_uid):
91 109 log.debug('File with FID:%s not found in the store', file_uid)
92 110 raise HTTPNotFound()
93 111
112 FileStore.bump_access_counter(file_uid)
113
94 114 file_path = self.storage.store_path(file_uid)
95 115 return FileResponse(file_path)
1 NO CONTENT: modified file
The requested commit or file is too big and content was truncated. Show full diff
General Comments 0
You need to be logged in to leave comments. Login now