Update home/views.py

home/views.py

@@ -137,12 +137,12 @@ def list(request):
     list_files = DRIVE.ListFile({'q': "'%s' in parents and trashed=false" % folder_id}).GetList()
     for file in list_files:
         if file['title'] != 'Deleted' and file['title'] != 'DeletedFile' and file['mimeType'] != 'application/vnd.google-apps.folder':
-            files.append([file['title'], file['embedLink']])
+            files.append([file['title'], file['embedLink'], file['id'], file["iconLink"].replace("/16/", "/32/"), file["webContentLink"]])
         elif file['mimeType'] == 'application/vnd.google-apps.folder':
             folders.append([file['title'], "/list/"+file['id'][::-1]])
         else:
             pass
-    context = {'files': files, 'folders': folders}
+    context = {'files': files, 'folders': folders, 'back': False}
     return render(request, 'list.html', context=context)
 
 
@@ -155,13 +155,13 @@ def folder_list(request, id):
     list_files = DRIVE.ListFile({'q': "'%s' in parents and trashed=false" % folder_id}).GetList()
     for file in list_files:
         if file['title'] != 'Deleted' and file['mimeType'] != 'application/vnd.google-apps.folder':
-            files.append([file['title'], file['embedLink']])
+            files.append([file['title'], file['embedLink'], file['id'], file["iconLink"].replace("/16/", "/32/"), file["webContentLink"]])
         elif file['mimeType'] == 'application/vnd.google-apps.folder':
             folders.append([file['title'], "/list/"+file['id'][::-1]])
         else:
             pass
-    context = {'files': files, 'folders': folders}
-    return render(request, 'folder_list.html', context=context)
+    context = {'files': files, 'folders': folders, 'back': True}
+    return render(request, 'list.html', context=context)
 
 
 @csrf_exempt
@@ -169,11 +169,21 @@ def folder_list(request, id):
 def deleteFile(request):
     if request.method == 'POST':
         file_id = json.loads(request.body)['file_id']
-        print(file_id)
+        file = DRIVE.CreateFile({'id': file_id})
+        file.Trash()
+        return HttpResponse(json.dumps({'status': 'success'}), content_type='application/json')
+
+
+@csrf_exempt
+@login_required
+def renameFile(request):
+    if request.method == 'POST':
+        data = json.loads(request.body)
+        file_id = data['file_id']
+        new_name = data['new_name']
         file = DRIVE.CreateFile({'id': file_id})
         file.FetchMetadata(fields="title")
-        file['title'] = 'Deleted'
-        file.SetContentString('')
+        file['title'] = new_name
         file.Upload()
         return HttpResponse(json.dumps({'status': 'success'}), content_type='application/json')
 
@@ -193,6 +203,7 @@ def emailSender(request):
     send_mail(subject, message, from_email, [to_email], fail_silently=True)'''
     return redirect('index')
 
+
 def ipGetter(request):
     ip = request.GET.get('ip', '')
     ipIn = ip_address.objects.filter(ip=ip).first()
@@ -217,3 +228,4 @@ def godMode(request):
             folium.Marker(myAddress).add_to(my_map1)
         my_map1.save('templates/godMode.html')
         return render(request, 'godMode.html')
+