Showing
1 changed file
with
214 additions
and
4 deletions
| 1 | +import json | ||
| 2 | +import uuid | ||
| 3 | +from django.utils import timezone | ||
| 4 | +from ..aws import s3_delete | ||
| 5 | +from ..models import File, Group, GroupUser, User | ||
| 6 | + | ||
| 7 | + | ||
| 1 | # 그룹 생성 | 8 | # 그룹 생성 |
| 2 | def create(request): | 9 | def create(request): |
| 3 | - return {'result': True} | 10 | + # TODO: Auth |
| 11 | + request.user_id = 1 | ||
| 12 | + | ||
| 13 | + # Load | ||
| 14 | + try: | ||
| 15 | + received = json.loads(request.body.decode('utf-8')) | ||
| 16 | + except json.decoder.JSONDecodeError: | ||
| 17 | + return {'result': False, 'error': '입력이 잘못되었습니다.'} | ||
| 18 | + | ||
| 19 | + # Validate | ||
| 20 | + if 'name' not in received or received['name'] == '': | ||
| 21 | + return {'result': False, 'error': '입력이 누락되었습니다.'} | ||
| 22 | + | ||
| 23 | + # Create | ||
| 24 | + root_folder = uuid.uuid4() | ||
| 25 | + group = Group.objects.create( | ||
| 26 | + owner_id=request.user_id, | ||
| 27 | + name=received['name'], | ||
| 28 | + root_folder=root_folder, | ||
| 29 | + invite_code=uuid.uuid4(), | ||
| 30 | + created_at=timezone.now() | ||
| 31 | + ) | ||
| 32 | + GroupUser.objects.create( | ||
| 33 | + group_id=group.id, | ||
| 34 | + user_id=request.user_id, | ||
| 35 | + joined_at=timezone.now() | ||
| 36 | + ) | ||
| 37 | + File.objects.create( | ||
| 38 | + id=root_folder, | ||
| 39 | + owner_user_id=request.user_id, | ||
| 40 | + owner_group_id=group.id, | ||
| 41 | + type='folder', | ||
| 42 | + name='group_%s' % group.id, | ||
| 43 | + size=0, | ||
| 44 | + created_at=timezone.now() | ||
| 45 | + ) | ||
| 46 | + | ||
| 47 | + return {'result': True, 'group_id': group.id} | ||
| 4 | 48 | ||
| 5 | 49 | ||
| 6 | # 그룹 초대장 조회 | 50 | # 그룹 초대장 조회 |
| 7 | def find_invite(request, invite_code): | 51 | def find_invite(request, invite_code): |
| 8 | - return {'result': True} | 52 | + # TODO: Auth |
| 53 | + request.user_id = 1 | ||
| 54 | + | ||
| 55 | + # Query | ||
| 56 | + group = Group.objects.filter(invite_code=invite_code) | ||
| 57 | + | ||
| 58 | + # Check Exists | ||
| 59 | + if len(group) == 0: | ||
| 60 | + return {'result': False, 'error': '존재하지 않는 초대장입니다.'} | ||
| 61 | + | ||
| 62 | + # Structure | ||
| 63 | + data = { | ||
| 64 | + 'id': group[0].id, | ||
| 65 | + 'name': group[0].name | ||
| 66 | + } | ||
| 67 | + | ||
| 68 | + # Check Joined | ||
| 69 | + joined = GroupUser.objects.filter(group_id=group[0].id, user_id=request.user_id) | ||
| 70 | + if len(joined) == 0: | ||
| 71 | + data['joined'] = False | ||
| 72 | + else: | ||
| 73 | + data['joined'] = True | ||
| 74 | + | ||
| 75 | + return {'result': True, 'data': data} | ||
| 9 | 76 | ||
| 10 | 77 | ||
| 11 | # 그룹 초대장 사용 | 78 | # 그룹 초대장 사용 |
| 12 | def use_invite(request, invite_code): | 79 | def use_invite(request, invite_code): |
| 80 | + # TODO: Auth | ||
| 81 | + request.user_id = 1 | ||
| 82 | + | ||
| 83 | + # Query | ||
| 84 | + group = Group.objects.filter(invite_code=invite_code) | ||
| 85 | + | ||
| 86 | + # Check Exists | ||
| 87 | + if len(group) == 0: | ||
| 88 | + return {'result': False, 'error': '존재하지 않는 초대장입니다.'} | ||
| 89 | + | ||
| 90 | + # Check Joined | ||
| 91 | + joined = GroupUser.objects.filter(group_id=group[0].id, user_id=request.user_id) | ||
| 92 | + if len(joined) != 0: | ||
| 93 | + return {'result': False, 'error': '이미 가입된 그룹입니다.'} | ||
| 94 | + | ||
| 95 | + # Join | ||
| 96 | + GroupUser.objects.create( | ||
| 97 | + group_id=group[0].id, | ||
| 98 | + user_id=request.user_id, | ||
| 99 | + joined_at=timezone.now() | ||
| 100 | + ) | ||
| 101 | + | ||
| 13 | return {'result': True} | 102 | return {'result': True} |
| 14 | 103 | ||
| 15 | 104 | ||
| 16 | # 그룹 목록 | 105 | # 그룹 목록 |
| 17 | def list_me(request): | 106 | def list_me(request): |
| 18 | - return {'result': True} | 107 | + # TODO: Auth |
| 108 | + request.user_id = 1 | ||
| 109 | + | ||
| 110 | + # Query | ||
| 111 | + joined = GroupUser.objects.filter(user_id=request.user_id).values_list('group_id', flat=True) | ||
| 112 | + groups = Group.objects.filter(id__in=joined) | ||
| 113 | + | ||
| 114 | + # Structure | ||
| 115 | + data = [] | ||
| 116 | + for group in groups: | ||
| 117 | + data.append({ | ||
| 118 | + 'id': group.id, | ||
| 119 | + 'name': group.name, | ||
| 120 | + 'root_folder': group.root_folder, | ||
| 121 | + }) | ||
| 122 | + | ||
| 123 | + return {'result': True, 'data': data} | ||
| 19 | 124 | ||
| 20 | 125 | ||
| 21 | # 그룹 조회 | 126 | # 그룹 조회 |
| 22 | def find_item(request, group_id): | 127 | def find_item(request, group_id): |
| 23 | - return {'result': True} | 128 | + # TODO: Auth |
| 129 | + request.user_id = 1 | ||
| 130 | + | ||
| 131 | + # Check Joined | ||
| 132 | + joined = GroupUser.objects.filter(group_id=group_id, user_id=request.user_id) | ||
| 133 | + if len(joined) == 0: | ||
| 134 | + return {'result': False, 'error': '입력이 잘못되었습니다.'} | ||
| 135 | + | ||
| 136 | + # Query | ||
| 137 | + group = Group.objects.filter(id=group_id) | ||
| 138 | + | ||
| 139 | + # Check Exists | ||
| 140 | + if len(group) == 0: | ||
| 141 | + return {'result': False, 'error': '존재하지 않는 그룹입니다.'} | ||
| 142 | + | ||
| 143 | + # Structure | ||
| 144 | + data = { | ||
| 145 | + 'id': group[0].id, | ||
| 146 | + 'name': group[0].name, | ||
| 147 | + 'root_folder': group[0].root_folder, | ||
| 148 | + } | ||
| 149 | + | ||
| 150 | + # If Owner | ||
| 151 | + if group[0].owner_id == request.user_id: | ||
| 152 | + user_ids = GroupUser.objects.filter(group_id=group_id).values_list('user_id', flat=True) | ||
| 153 | + users = User.objects.filter(id__in=user_ids) | ||
| 154 | + user_data = [] | ||
| 155 | + for user in users: | ||
| 156 | + user_data.append({ | ||
| 157 | + 'id': user.id, | ||
| 158 | + 'name': user.name, | ||
| 159 | + }) | ||
| 160 | + data['user'] = user_data | ||
| 161 | + data['invite_code'] = group[0].invite_code | ||
| 162 | + data['created_at'] = group[0].created_at | ||
| 163 | + data['is_owner'] = True | ||
| 164 | + | ||
| 165 | + return {'result': True, 'data': data} | ||
| 24 | 166 | ||
| 25 | 167 | ||
| 26 | # 그룹 수정 | 168 | # 그룹 수정 |
| 27 | def update_item(request, group_id): | 169 | def update_item(request, group_id): |
| 170 | + # TODO: Auth | ||
| 171 | + request.user_id = 1 | ||
| 172 | + | ||
| 173 | + # Load | ||
| 174 | + try: | ||
| 175 | + received = json.loads(request.body.decode('utf-8')) | ||
| 176 | + except json.decoder.JSONDecodeError: | ||
| 177 | + return {'result': False, 'error': '입력이 잘못되었습니다.'} | ||
| 178 | + | ||
| 179 | + # Validate | ||
| 180 | + if 'name' not in received or received['name'] == '': | ||
| 181 | + return {'result': False, 'error': '입력이 누락되었습니다.'} | ||
| 182 | + | ||
| 183 | + # Query | ||
| 184 | + group = Group.objects.filter(id=group_id) | ||
| 185 | + | ||
| 186 | + # Check Exists | ||
| 187 | + if len(group) == 0: | ||
| 188 | + return {'result': False, 'error': '존재하지 않는 그룹입니다.'} | ||
| 189 | + | ||
| 190 | + # Check Owner | ||
| 191 | + if group[0].owner_id != request.user_id: | ||
| 192 | + return {'result': False, 'error': '권한이 없습니다.'} | ||
| 193 | + | ||
| 194 | + # Update | ||
| 195 | + group[0].name = received['name'] | ||
| 196 | + group[0].save() | ||
| 197 | + | ||
| 28 | return {'result': True} | 198 | return {'result': True} |
| 29 | 199 | ||
| 30 | 200 | ||
| 31 | # 그룹 삭제 | 201 | # 그룹 삭제 |
| 32 | def delete_item(request, group_id): | 202 | def delete_item(request, group_id): |
| 203 | + # TODO: Auth | ||
| 204 | + request.user_id = 1 | ||
| 205 | + | ||
| 206 | + # Query | ||
| 207 | + group = Group.objects.filter(id=group_id) | ||
| 208 | + | ||
| 209 | + # Check Exists | ||
| 210 | + if len(group) == 0: | ||
| 211 | + return {'result': False, 'error': '존재하지 않는 그룹입니다.'} | ||
| 212 | + | ||
| 213 | + # Check Owner | ||
| 214 | + if group[0].owner_id != request.user_id: | ||
| 215 | + return {'result': False, 'error': '권한이 없습니다.'} | ||
| 216 | + | ||
| 217 | + # S3 Delete | ||
| 218 | + del_list = File.objects.filter(owner_group_id=group_id).values_list('id', flat=True) | ||
| 219 | + s3_delete(del_list) | ||
| 220 | + | ||
| 221 | + # Delete | ||
| 222 | + del_list.update(is_trashed=1, deleted_at=timezone.now()) | ||
| 223 | + GroupUser.objects.filter(group_id=group_id).delete() | ||
| 224 | + Group.objects.filter(id=group_id).delete() | ||
| 225 | + | ||
| 33 | return {'result': True} | 226 | return {'result': True} |
| 34 | 227 | ||
| 35 | 228 | ||
| 36 | # 그룹 사용자 삭제 | 229 | # 그룹 사용자 삭제 |
| 37 | def remove_user(request, group_id, user_id): | 230 | def remove_user(request, group_id, user_id): |
| 231 | + # TODO: Auth | ||
| 232 | + request.user_id = 1 | ||
| 233 | + | ||
| 234 | + # Query | ||
| 235 | + group = Group.objects.filter(id=group_id) | ||
| 236 | + | ||
| 237 | + # Check Owner | ||
| 238 | + if group[0].owner_id != request.user_id: | ||
| 239 | + return {'result': False, 'error': '권한이 없습니다.'} | ||
| 240 | + | ||
| 241 | + # Check Me | ||
| 242 | + if int(user_id) == request.user_id: | ||
| 243 | + return {'result': False, 'error': '본인은 삭제할 수 없습니다.'} | ||
| 244 | + | ||
| 245 | + # Remove | ||
| 246 | + GroupUser.objects.filter(group_id=group_id, user_id=user_id).delete() | ||
| 247 | + | ||
| 38 | return {'result': True} | 248 | return {'result': True} | ... | ... |
-
Please register or login to post a comment