|
10 | 10 | class UserFileManager(BaseManager):
|
11 | 11 | def __init__(self, *args, **kwargs):
|
12 | 12 | super().__init__(*args, **kwargs)
|
13 |
| - self.file_model = UserFile |
| 13 | + self.user_file_model = UserFile |
14 | 14 |
|
15 | 15 | def create_file(self, params: dict) -> UserFile:
|
16 | 16 | def _rollback(vo: UserFile) -> None:
|
17 | 17 | _LOGGER.info(f"[ROLLBACK] Delete file : {vo.name} ({vo.file_id})")
|
18 | 18 | vo.delete()
|
19 | 19 |
|
20 |
| - file_vo: UserFile = self.file_model.create(params) |
| 20 | + file_vo: UserFile = self.user_file_model.create(params) |
21 | 21 | self.transaction.add_rollback(_rollback, file_vo)
|
22 | 22 |
|
23 | 23 | return file_vo
|
@@ -51,13 +51,13 @@ def get_file(
|
51 | 51 | if user_id:
|
52 | 52 | conditions["user_id"] = user_id
|
53 | 53 |
|
54 |
| - return self.file_model.get(**conditions) |
| 54 | + return self.user_file_model.get(**conditions) |
55 | 55 |
|
56 | 56 | def filter_files(self, **conditions) -> QuerySet:
|
57 |
| - return self.file_model.filter(**conditions) |
| 57 | + return self.user_file_model.filter(**conditions) |
58 | 58 |
|
59 | 59 | def list_files(self, query: dict) -> dict:
|
60 |
| - return self.file_model.query(**query) |
| 60 | + return self.user_file_model.query(**query) |
61 | 61 |
|
62 | 62 | def stat_files(self, query: dict) -> dict:
|
63 |
| - return self.file_model.stat(**query) |
| 63 | + return self.user_file_model.stat(**query) |
0 commit comments