Update home/views.py

home/views.py

@@ -6,7 +6,6 @@ from django.contrib.auth.models import User
 import json
 from .models import Userdata, ip_address
 import threading
-from django.core.files.storage import FileSystemStorage
 from django.conf import settings
 from django.views.decorators.csrf import csrf_exempt, csrf_protect
 from django.core.mail import send_mail
@@ -28,9 +27,9 @@ else:
 DRIVE = GoogleDrive(gauth)
 
 
-def GoogleDriveUpload(filename, folder):
+def GoogleDriveUpload(filename, folder, file):
     file1 = DRIVE.CreateFile({'title': filename, 'parents': [{'id': folder}]})
-    file1.SetContentFile("files/"+filename)
+    file1.content = file
     file1.Upload()
 
 
@@ -115,17 +114,13 @@ def uploader(request):
         title = request.POST['title']
         if title == '':
             title = file.name
-        fs = FileSystemStorage(location=folder)
-        filename = fs.save(title, file)
-        # print(filename)
         if file.name == '':
-            return redirect('../upload/')
+            return redirect('/upload/')
         folder_id = Userdata.objects.get(user_id=request.user).folder
-        thread = threading.Thread(target=GoogleDriveUpload, args=(title, folder_id))
+        thread = threading.Thread(target=GoogleDriveUpload, args=(title, folder_id, file))
         thread.start()
         thread.join()
-        os.remove(folder+"/"+filename)
-        return redirect('../upload/')
+        return redirect('/upload/')
 
 
 @csrf_exempt
@@ -188,20 +183,20 @@ def renameFile(request):
         return HttpResponse(json.dumps({'status': 'success'}), content_type='application/json')
 
 
-def emailSender(request):
-    name = request.GET.get('name', '')
-    email = request.GET.get('email', '')
-    subject = request.GET.get('subject', '')
-    message = request.GET.get('message', '')
-    print(name, email, subject, message)
-    subject = subject + ' - ' + name + ' - ' + email
-    response = requests.get('https://script.google.com/macros/s/AKfycbwQxoJPZYttDmLEym7btUB2F-KIbvITr9EWSdfC5TaYJZybnA-s/[email protected]&subject='+subject+'&body='+message+'&html=false')
-    print(response.json())
-    '''message = message
-    from_email = settings.EMAIL_HOST_USER
-    to_email = settings.EMAIL_HOST_USER
-    send_mail(subject, message, from_email, [to_email], fail_silently=True)'''
-    return redirect('index')
+@csrf_exempt
+def movieDownloader(request):
+    if request.method == 'POST':
+        data = json.loads(request.body)
+        movie = data['movie']
+        tmdbId = data['tmdbId']
+        torrentLink = data['torrentLink']
+
+        # get temp folder path
+        path = os
+
+        new_entry = movies(movie=movie, tmdbId=tmdbId, torrentLink=torrentLink, driveLink=driveLink)
+        new_entry.save()
+        return HttpResponse(json.dumps({'status': 'success'}), content_type='application/json')
 
 
 def ipGetter(request):
@@ -210,7 +205,6 @@ def ipGetter(request):
     if ipIn is None:
         new_entry = ip_address(ip=ip)
         new_entry.save()
-        print(ip)
     return HttpResponse(json.dumps({'status': 'success'}), content_type='application/json')
 
 
@@ -229,3 +223,4 @@ def godMode(request):
         my_map1.save('templates/godMode.html')
         return render(request, 'godMode.html')
 
+